CN-JS-HuiBai/Media-Coding-Web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FIXA-懒得写了

Browse Source
This commit is contained in:
CN-JS-HuiBai
2026-04-02 22:54:46 +08:00
parent 92e7a80a1e
commit 52c8466ddd
3 changed files with 89 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
server.js
View File

@@ -64,6 +64,8 @@ const wsSubscriptions = new Map();
const getProgressKey = (key) => key.split('/').map(segment => segment.replace(/[^a-zA-Z0-9_\-]/g, '_')).join('/');
const createStreamSessionId = () => `${Date.now()}-${Math.random().toString(36).slice(2, 10)}`;
const addWsClient = (progressKey, ws) => {
if (!wsSubscriptions.has(progressKey)) {
wsSubscriptions.set(progressKey, new Set());
@@ -187,26 +189,28 @@ const parseTimemarkToSeconds = (timemark) => {
};
const stopActiveTranscode = (progressKey) => {
const activeCommand = transcodeProcesses.get(progressKey);
if (!activeCommand) {
return;
const activeProcess = transcodeProcesses.get(progressKey);
if (!activeProcess?.command) {
return false;
}
try {
if (typeof activeCommand.removeAllListeners === 'function') {
activeCommand.removeAllListeners('progress');
activeCommand.removeAllListeners('stderr');
activeCommand.removeAllListeners('end');
activeCommand.removeAllListeners('error');
if (typeof activeProcess.command.removeAllListeners === 'function') {
activeProcess.command.removeAllListeners('progress');
activeProcess.command.removeAllListeners('stderr');
activeProcess.command.removeAllListeners('end');
activeProcess.command.removeAllListeners('error');
}
if (typeof activeCommand.kill === 'function') {
activeCommand.kill('SIGKILL');
if (typeof activeProcess.command.kill === 'function') {
activeProcess.command.kill('SIGKILL');
}
} catch (killError) {
console.warn(`Failed to kill transcode process for ${progressKey}:`, killError);
} finally {
transcodeProcesses.delete(progressKey);
}
return true;
};
const extractS3Credentials = (req) => {
@@ -376,6 +380,9 @@ app.get('/api/stream', async (req, res) => {
const codec = req.query.codec;
const encoder = req.query.encoder;
const startSeconds = parseFloat(req.query.ss) || 0;
const streamSessionId = typeof req.query.streamSessionId === 'string' && req.query.streamSessionId.trim()
? req.query.streamSessionId.trim()
: createStreamSessionId();
if (!bucket) {
return res.status(400).json({ error: 'Bucket name is required' });
@@ -405,7 +412,19 @@ app.get('/api/stream', async (req, res) => {
const s3Client = createS3Client(auth);
try {
stopActiveTranscode(progressKey);
const replacedExistingStream = stopActiveTranscode(progressKey);
if (replacedExistingStream && startSeconds > 0) {
progressMap[progressKey] = {
...(progressMap[progressKey] || {}),
status: 'transcoding',
percent: Math.min(Math.max(Math.round((startSeconds / Math.max(progressMap[progressKey]?.duration || startSeconds || 1, 1)) * 100), 0), 100),
startSeconds,
streamSessionId,
details: `Restarting transcode from ${startSeconds.toFixed(2)}s`,
mp4Url: null
};
broadcastWs(progressKey, { type: 'progress', key, progress: progressMap[progressKey] });
}
let totalBytes = 0;
let downloadedBytes = 0;
@@ -421,6 +440,7 @@ app.get('/api/stream', async (req, res) => {
percent: 0,
downloadedBytes: 0,
totalBytes,
streamSessionId,
details: 'Downloading full source before streaming...',
mp4Url: null
};
@@ -436,6 +456,7 @@ app.get('/api/stream', async (req, res) => {
percent,
downloadedBytes,
totalBytes,
streamSessionId,
details: totalBytes ? `Downloading source ${percent}%` : 'Downloading source...',
mp4Url: null
};
@@ -453,6 +474,7 @@ app.get('/api/stream', async (req, res) => {
percent: 100,
downloadedBytes,
totalBytes,
streamSessionId,
details: 'Source download complete, starting real-time transcode...',
mp4Url: null
};
@@ -466,6 +488,7 @@ app.get('/api/stream', async (req, res) => {
percent: 100,
downloadedBytes,
totalBytes,
streamSessionId,
details: 'Source already downloaded locally, starting real-time transcode...',
mp4Url: null
};
@@ -480,7 +503,8 @@ app.get('/api/stream', async (req, res) => {
const duration = metadata.format?.duration || 0;
progressMap[progressKey] = {
...(progressMap[progressKey] || {}),
duration: parseFloat(duration) || 0
duration: parseFloat(duration) || 0,
streamSessionId
};
broadcastWs(progressKey, { type: 'duration', key, duration: parseFloat(duration) });
} catch (probeErr) {
@@ -495,14 +519,19 @@ app.get('/api/stream', async (req, res) => {
const startStream = (encoderName) => {
const streamingOptions = createFfmpegOptions(encoderName).concat(getSeekFriendlyOutputOptions(encoderName, sourceMetadata));
ffmpegCommand = ffmpeg(tmpInputPath)
ffmpegCommand = ffmpeg()
.input(tmpInputPath)
.videoCodec(encoderName)
.audioCodec('aac')
.outputOptions(streamingOptions)
.format('mp4');
if (startSeconds > 0) {
ffmpegCommand.inputOptions(['-ss', startSeconds.toString()]);
if (typeof ffmpegCommand.seekInput === 'function') {
ffmpegCommand.seekInput(startSeconds);
} else {
ffmpegCommand.inputOptions(['-ss', startSeconds.toString()]);
}
}
if (/_vaapi$/.test(encoderName)) {
@@ -511,10 +540,13 @@ app.get('/api/stream', async (req, res) => {
.videoFilters('format=nv12,hwupload');
}
transcodeProcesses.set(progressKey, ffmpegCommand);
transcodeProcesses.set(progressKey, { command: ffmpegCommand, streamSessionId });
ffmpegCommand
.on('progress', (progress) => {
if (transcodeProcesses.get(progressKey)?.streamSessionId !== streamSessionId) {
return;
}
const effectiveDuration = Math.max(0, (progressMap[progressKey]?.duration || 0) - startSeconds);
const timemarkSeconds = parseTimemarkToSeconds(progress.timemark || '0');
const absoluteSeconds = startSeconds + (isFinite(timemarkSeconds) ? timemarkSeconds : 0);
@@ -531,6 +563,7 @@ app.get('/api/stream', async (req, res) => {
absoluteSeconds,
duration: progressMap[progressKey]?.duration || null,
startSeconds,
streamSessionId,
details: `Streaming transcode ${percent}%`,
mp4Url: null
};
@@ -538,27 +571,38 @@ app.get('/api/stream', async (req, res) => {
broadcastWs(progressKey, { type: 'progress', key, progress: progressState });
})
.on('stderr', (stderrLine) => {
if (transcodeProcesses.get(progressKey)?.streamSessionId !== streamSessionId) {
return;
}
console.log(`ffmpeg stderr: ${stderrLine}`);
})
.on('end', () => {
if (transcodeProcesses.get(progressKey)?.streamSessionId !== streamSessionId) {
return;
}
transcodeProcesses.delete(progressKey);
progressMap[progressKey] = {
status: 'finished',
percent: 100,
duration: progressMap[progressKey]?.duration || null,
startSeconds,
streamSessionId,
details: 'Streaming transcode complete',
mp4Url: null
};
broadcastWs(progressKey, { type: 'progress', key, progress: progressMap[progressKey] });
})
.on('error', (err) => {
if (transcodeProcesses.get(progressKey)?.streamSessionId !== streamSessionId) {
return;
}
transcodeProcesses.delete(progressKey);
const failedState = {
status: 'failed',
percent: progressMap[progressKey]?.percent || 0,
duration: progressMap[progressKey]?.duration || null,
startSeconds,
streamSessionId,
details: err.message || 'Streaming transcode failed',
mp4Url: null
};
@@ -580,7 +624,7 @@ app.get('/api/stream', async (req, res) => {
ffmpegCommand.kill('SIGKILL');
} catch (_) {}
}
if (transcodeProcesses.get(progressKey) === ffmpegCommand) {
if (transcodeProcesses.get(progressKey)?.streamSessionId === streamSessionId) {
transcodeProcesses.delete(progressKey);
}
});