import { Worker } from 'bullmq'

import { config } from './config'
import { judgeQueueName, type JudgeJobData } from './judge/job'
import { failAbandonedSubmission, judgeSubmission } from './judge/run'
import { flowchartQueueName, type FlowchartJobData } from './flowchart/job'
import { evaluateFlowchart } from './flowchart/run'
import { createBlockingRedis } from './redis'

const worker = new Worker<JudgeJobData>(judgeQueueName, async (job) => judgeSubmission(job.data), {
  connection: createBlockingRedis(),
  concurrency: config.judgeConcurrency,
})

const flowchartWorker = new Worker<FlowchartJobData>(
  flowchartQueueName,
  // attemptsMade 是「此前已经失败过几次」，当前这次还没计进去，
  // 所以最后一次尝试的判据是 attemptsMade + 1 >= attempts
  async (job) =>
    evaluateFlowchart(job.data, {
      isFinalAttempt: job.attemptsMade + 1 >= (job.opts.attempts ?? 1),
    }),
  { connection: createBlockingRedis(), concurrency: 2 },
)

worker.on('ready', () => {
  console.log(`Judge worker ready (concurrency=${config.judgeConcurrency})`)
})
worker.on('failed', async (job, error) => {
  console.error(`Judge job ${job?.id ?? 'unknown'} failed`, error)
  // 队列没配 attempts，失败即终局；worker 被杀掉那种 BullMQ 走完 stalled 重试也会
  // 落到这里。不在这里写一个终态，提交就永远停在「等待评分」，没有任何人会再管它。
  const submissionId = job?.data.submissionId
  if (!submissionId) return
  try {
    await failAbandonedSubmission(submissionId, error)
  } catch (markError) {
    console.error(`Failed to mark submission ${submissionId} as system error`, markError)
  }
})
worker.on('error', (error) => {
  console.error('Judge worker error', error)
})
flowchartWorker.on('ready', () => console.log('Flowchart worker ready (concurrency=2)'))
flowchartWorker.on('failed', (job, error) => console.error(`Flowchart job ${job?.id ?? 'unknown'} failed`, error))
flowchartWorker.on('error', (error) => console.error('Flowchart worker error', error))

async function shutdown() {
  await worker.close()
  await flowchartWorker.close()
  process.exit(0)
}

process.on('SIGINT', shutdown)
process.on('SIGTERM', shutdown)