70 lines
2 KiB
JavaScript
70 lines
2 KiB
JavaScript
|
import { join } from 'path';
|
||
|
|
import { unlink } from 'fs/promises';
|
||
|
|
import { tmpdir } from 'os';
|
||
|
|
import { query } from '../db/client.js';
|
||
|
|
import { downloadFromS3, uploadToS3 } from '../s3/client.js';
|
||
|
|
import { transcodeVideo } from '../ffmpeg/executor.js';
|
||
|
|
|
||
|
|
const S3_BUCKET = process.env.S3_BUCKET || 'wild-dragon';
|
||
|
|
|
||
|
|
export const proxyWorker = async (job) => {
|
||
|
|
const { assetId, inputKey, outputKey } = job.data;
|
||
|
|
|
||
|
|
const tmpDir = tmpdir();
|
||
|
|
const inputPath = join(tmpDir, `proxy-input-${job.id}.mov`);
|
||
|
|
const outputPath = join(tmpDir, `proxy-output-${job.id}.mp4`);
|
||
|
|
|
||
|
|
try {
|
||
|
|
// Download from S3
|
||
|
|
job.updateProgress(10);
|
||
|
|
console.log(`[proxy] Downloading ${inputKey} for asset ${assetId}`);
|
||
|
|
await downloadFromS3(S3_BUCKET, inputKey, inputPath);
|
||
|
|
|
||
|
|
// Transcode
|
||
|
|
job.updateProgress(30);
|
||
|
|
console.log(`[proxy] Transcoding asset ${assetId}`);
|
||
|
|
await transcodeVideo(inputPath, outputPath, {
|
||
|
|
videoCodec: 'libx264',
|
||
|
|
videoPreset: 'fast',
|
||
|
|
videoBitrate: '10M',
|
||
|
|
audioCodec: 'aac',
|
||
|
|
audioBitrate: '192k',
|
||
|
|
});
|
||
|
|
|
||
|
|
// Upload to S3
|
||
|
|
job.updateProgress(70);
|
||
|
|
console.log(`[proxy] Uploading to ${outputKey}`);
|
||
|
|
await uploadToS3(S3_BUCKET, outputKey, outputPath);
|
||
|
|
|
||
|
|
// Update database
|
||
|
|
job.updateProgress(90);
|
||
|
|
console.log(`[proxy] Updating asset record for ${assetId}`);
|
||
|
|
await query(
|
||
|
|
'UPDATE assets SET proxy_s3_key = $1, status = $2 WHERE id = $3',
|
||
|
|
[outputKey, 'ready', assetId]
|
||
|
|
);
|
||
|
|
|
||
|
|
job.updateProgress(100);
|
||
|
|
console.log(`[proxy] Asset ${assetId} proxy complete`);
|
||
|
|
|
||
|
|
return { assetId, outputKey };
|
||
|
|
} catch (error) {
|
||
|
|
console.error(`[proxy] Error processing asset ${assetId}:`, error);
|
||
|
|
await query(
|
||
|
|
'UPDATE assets SET status = $1 WHERE id = $2',
|
||
|
|
['error', assetId]
|
||
|
|
);
|
||
|
|
throw error;
|
||
|
|
} finally {
|
||
|
|
// Cleanup
|
||
|
|
try {
|
||
|
|
await Promise.all([
|
||
|
|
unlink(inputPath).catch(() => {}),
|
||
|
|
unlink(outputPath).catch(() => {}),
|
||
|
|
]);
|
||
|
|
} catch (err) {
|
||
|
|
console.error(`[proxy] Cleanup error for job ${job.id}:`, err);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|