import { getRedisConnection } from '../src/index.ts';
import { Queue } from 'bullmq'

export async function clearAllJobs() {
  const connection = getRedisConnection();
  const queueNames = [
    'image-download',
    'image-generate',
    'perfect-prompt',
    'perfect-sentence-prompt'
  ];

  for (const queueName of queueNames) {
    const queue = new Queue(queueName, { connection });
    await queue.drain();
    await queue.clean(0, 1000, 'completed');
    await queue.clean(0, 1000, 'failed');
    console.log(`Cleared all jobs in queue: ${queueName}`);
    await queue.close();
  }

  await connection.quit();
}

clearAllJobs().then(() => {
  console.log('All jobs cleared.');
  process.exit(0);
}).catch((error) => {
  console.error('Error clearing jobs:', error);
  process.exit(1);
});