import { Queue, Worker, QueueScheduler } from "bullmq";
import { Redis } from "ioredis";
import { ScanService } from "@shieldai/darkwatch";
import { AlertPipeline } from "@shieldai/darkwatch";

const redisUrl = process.env.REDIS_URL || "redis://localhost:6379";
const connection = new Redis(redisUrl);

const scanQueue = new Queue("darkwatch-scans", { connection });
const alertQueue = new Queue("darkwatch-alerts", { connection });

const scanWorker = new Worker(
  "darkwatch-scans",
  async (job) => {
    const { userId, source } = job.data;
    const scanService = new ScanService();
    const resultCount = await scanService.runScan(userId, source);
    return { resultCount, completedAt: new Date().toISOString() };
  },
  { connection, concurrency: 3 }
);

const alertWorker = new Worker(
  "darkwatch-alerts",
  async () => {
    const pipeline = new AlertPipeline();
    const sent = await pipeline.sendPendingAlerts();
    return { sent, processedAt: new Date().toISOString() };
  },
  { connection, concurrency: 1 }
);

const scheduler = new QueueScheduler("darkwatch-alerts", { connection });

scanWorker.on("completed", (job) => {
  console.log(`[Scan] Job ${job.id} completed: ${JSON.stringify(job.returnvalue)}`);
});

scanWorker.on("failed", (job, err) => {
  console.error(`[Scan] Job ${job.id} failed: ${err.message}`);
});

alertWorker.on("completed", (job) => {
  console.log(`[Alert] Job ${job.id} completed: ${JSON.stringify(job.returnvalue)}`);
});

alertWorker.on("failed", (job, err) => {
  console.error(`[Alert] Job ${job.id} failed: ${err.message}`);
});

export async function addScanJob(userId: string, source?: string) {
  return scanQueue.add("scan", { userId, source }, {
    attempts: 3,
    backoff: { type: "exponential", delay: 5000 },
    jobId: `scan-${userId}-${Date.now()}`,
  });
}

export async function scheduleAlertProcessing() {
  return alertQueue.add("process-alerts", {}, {
    repeat: { pattern: "*/5 * * * *" },
    jobId: "alert-processor-recurring",
  });
}

scanWorker.on("waiting", () => console.log("[Worker] Scan worker ready"));
alertWorker.on("waiting", () => console.log("[Worker] Alert worker ready"));

console.log("Job workers started");