import { Worker, WorkerOptions, Processor } from 'bullmq';

import { QueueName } from '@app/lib/queue/QueueRegistry';

import { mailProcessor } from  '@app/lib/queue/processors/mail';

type ProcessorMap = Record<QueueName, Processor>;

const processorMap: ProcessorMap = {

  mail: mailProcessor

};

// Per-queue concurrency

// TODO: When we process images, we need to bump for that queue

const concurrencyMap: Record<QueueName, number> = {

  mail: 10

};

const workers: Worker[] = [];

export function bootWorkers(): void {

  for (const [name, processor] of Object.entries(processorMap)) {

    const queueName = name as QueueName;

    const worker = new Worker(queueName, processor, {

      connection: {

        host: process.env.VALKEY_HOST,

        port: Number(process.env.VALKEY_PORT)

      },

      concurrency: concurrencyMap[queueName],

    } satisfies WorkerOptions);

    worker.on('failed', (job, err) => {

      console.error(`[${queueName}] job ${job?.id} failed:`, err.message);

    });

    workers.push(worker);

  }

}

export async function closeAllWorkers(): Promise<void> {

  await Promise.all(workers.map(w => w.close()));

}