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)