const assert = require('assert');
const http = require('http');
const { highlightRecordingService } = require('../services/highlightRecordingService');
const app = require('../app');

function request(server, method, route, body) {
    return new Promise((resolve, reject) => {
        const address = server.address();
        const payload = body == null ? '' : JSON.stringify(body);
        const req = http.request({
            host: '127.0.0.1',
            port: address.port,
            path: route,
            method,
            headers: payload ? {
                'Content-Type': 'application/json',
                'Content-Length': Buffer.byteLength(payload)
            } : {}
        }, response => {
            let data = '';
            response.setEncoding('utf8');
            response.on('data', chunk => { data += chunk; });
            response.on('end', () => {
                resolve({ status: response.statusCode, body: JSON.parse(data) });
            });
        });
        req.on('error', reject);
        if (payload) req.write(payload);
        req.end();
    });
}

async function run() {
    const server = http.createServer(app);
    await new Promise(resolve => server.listen(0, '127.0.0.1', resolve));
    try {
        const status = await request(server, 'GET', '/highlight/status');
        assert.strictEqual(status.status, 200);
        assert.deepStrictEqual(status.body.data, { queued: 0, recording: 0, knownTasks: 0 });

        const invalidFiles = await request(server, 'POST', '/highlight/files', { items: [] });
        assert.strictEqual(invalidFiles.status, 400);
        assert.strictEqual(invalidFiles.body.code, 10);

        const disabledFiles = await request(server, 'POST', '/highlight/files', {
            items: [{ highlightId: 5, classId: '1001', siteId: 'xdyui2', beginTime: 1785895298000 }]
        });
        assert.strictEqual(disabledFiles.status, 400);
        assert.strictEqual(disabledFiles.body.code, 10);
        assert.strictEqual(disabledFiles.body.message, '高光录制功能未启用');

        const invalidPreview = await request(server, 'POST', '/highlight/preview/by-class', {});
        assert.strictEqual(invalidPreview.status, 400);
        assert.strictEqual(invalidPreview.body.code, 10);

        const invalidClassFiles = await request(server, 'POST', '/highlight/fileExists', {});
        assert.strictEqual(invalidClassFiles.status, 400);
        assert.strictEqual(invalidClassFiles.body.code, 10);

        const disabledScheduled = await request(server, 'POST', '/highlight/recording/scheduled');
        assert.strictEqual(disabledScheduled.status, 400);
        assert.strictEqual(disabledScheduled.body.code, 10);
        assert.strictEqual(disabledScheduled.body.message, '高光录制功能未启用');

        const originalRunScheduledRecording = highlightRecordingService.runScheduledRecording;
        try {
            highlightRecordingService.runScheduledRecording = async () => ({
                mode: 'site',
                beginTime: 1785945600000,
                endTime: 1786031999999,
                received: 2,
                queued: 2,
                recording: 0,
                uploading: 0,
                generated: 0,
                invalid: 0,
                failed: 0
            });
            const scheduled = await request(server, 'POST', '/highlight/recording/scheduled');
            assert.strictEqual(scheduled.status, 200);
            assert.strictEqual(scheduled.body.code, 0);
            assert.strictEqual(scheduled.body.message, 'success');
            assert.strictEqual(scheduled.body.data.mode, 'site');
            assert.strictEqual(scheduled.body.data.received, 2);

            highlightRecordingService.runScheduledRecording = async () => ({
                mode: 'site',
                beginTime: 1785945600000,
                endTime: 1786031999999,
                received: 0,
                queued: 0,
                recording: 0,
                uploading: 0,
                generated: 0,
                invalid: 0,
                failed: 0
            });
            const scheduledEmpty = await request(server, 'POST', '/highlight/recording/scheduled');
            assert.strictEqual(scheduledEmpty.body.code, 0);
            assert.strictEqual(scheduledEmpty.body.message, '无高光数据');
        } finally {
            highlightRecordingService.runScheduledRecording = originalRunScheduledRecording;
        }

        const originalGetClassFileStatuses = highlightRecordingService.getClassFileStatuses;
        highlightRecordingService.getClassFileStatuses = async () => [
            { highlightId: 4, classId: '1001', siteId: 'xdyui2', generated: true, status: 'generated', url: 'https://example/4.mp4' },
            { highlightId: 5, classId: '1001', siteId: 'xdyui2', generated: true, status: 'generated', url: 'https://example/5.mp4' }
        ];
        const generatedFiles = await request(server, 'POST', '/highlight/fileExists', { siteId: 'xdyui2', classId: '1001' });
        assert.strictEqual(generatedFiles.body.code, 0);
        assert.strictEqual(generatedFiles.body.onlyHighlight, 1);
        assert.strictEqual(generatedFiles.body.classUrlList.length, 2);

        highlightRecordingService.getClassFileStatuses = async () => [
            { highlightId: 4, classId: '1001', siteId: 'xdyui2', generated: true, status: 'generated', url: 'https://example/4.mp4' },
            { highlightId: 5, classId: '1001', siteId: 'xdyui2', generated: false, status: 'uploading', url: null }
        ];
        const partialFiles = await request(server, 'POST', '/highlight/fileExists', { siteId: 'xdyui2', classId: '1001' });
        assert.strictEqual(partialFiles.body.code, 1);
        assert.strictEqual(partialFiles.body.message, '部分文件未生成');
        highlightRecordingService.getClassFileStatuses = originalGetClassFileStatuses;
    } finally {
        await new Promise(resolve => server.close(resolve));
    }
    console.log('highlight routes tests passed');
}

run().catch(error => {
    console.error(error);
    process.exitCode = 1;
});