import { join } from 'path';
import { tmpdir } from 'os';
import { unlink } from 'fs/promises';
import { query } from '../db/client.js';
import { downloadFromS3, uploadToS3 } from '../s3/client.js';
import { trimSegment } from '../ffmpeg/executor.js';

const S3_BUCKET = process.env.S3_BUCKET || 'wild-dragon';

export const trimWorker = async (job) => {
  const { clipInstanceId, assetId, sourceInFrames, sourceOutFrames } = job.data;
  const jobId = job.id;

  const tmpDir = tmpdir();
  const downloadPath = join(tmpDir, `trim-${jobId}-src`);
  const outputPath = join(tmpDir, `trim-${jobId}.mov`);

  try {
    const assetRes = await query(
      'SELECT original_s3_key FROM assets WHERE id = $1 LIMIT 1',
      [assetId]
    );

    if (assetRes.rows.length === 0) {
      throw new Error(`Asset not found: ${assetId}`);
    }

    const { original_s3_key: sourceKey } = assetRes.rows[0];

    await job.updateProgress(10);
    console.log(`[trim] Downloading asset ${assetId} from S3 (${sourceKey})`);
    await downloadFromS3(S3_BUCKET, sourceKey, downloadPath);

    await job.updateProgress(40);
    console.log(`[trim] Trimming frames ${sourceInFrames} → ${sourceOutFrames}`);
    await trimSegment(downloadPath, outputPath, sourceInFrames, sourceOutFrames);

    await job.updateProgress(70);
    const s3Key = `temp-segments/${clipInstanceId}.mov`;
    console.log(`[trim] Uploading trimmed segment to ${s3Key}`);
    await uploadToS3(S3_BUCKET, s3Key, outputPath);

    await job.updateProgress(85);
    await query(
      `INSERT INTO temp_segments (clip_instance_id, s3_key, expires_at)
       VALUES ($1, $2, NOW() + INTERVAL '24 hours')`,
      [clipInstanceId, s3Key]
    );

    await job.updateProgress(100);
    console.log(`[trim] Job ${jobId} complete for clip ${clipInstanceId}`);

    return { clipInstanceId, s3Key };

  } catch (error) {
    console.error(`[trim] Error in job ${jobId}:`, error);
    throw error;
  } finally {
    await Promise.all([
      unlink(downloadPath).catch(() => {}),
      unlink(outputPath).catch(() => {}),
    ]);
  }
};