337 lines
9.7 KiB
TypeScript
337 lines
9.7 KiB
TypeScript
import { join, basename, extname } from 'node:path';
|
||
import { randomUUID } from 'node:crypto';
|
||
import { rm } from 'node:fs/promises';
|
||
import type {
|
||
DashConvertOptions,
|
||
DashConvertResult,
|
||
VideoProfile,
|
||
ThumbnailConfig,
|
||
ConversionProgress,
|
||
CodecType
|
||
} from '../types';
|
||
import {
|
||
checkFFmpeg,
|
||
checkMP4Box,
|
||
checkNvenc,
|
||
checkAV1Support,
|
||
getVideoMetadata,
|
||
ensureDir
|
||
} from '../utils';
|
||
import { selectProfiles, createProfilesFromStrings } from '../config/profiles';
|
||
import { generateThumbnailSprite, generatePoster } from './thumbnails';
|
||
import { encodeProfilesToMP4 } from './encoding';
|
||
import { packageToDash } from './packaging';
|
||
|
||
/**
|
||
* Convert video to DASH format with NVENC acceleration
|
||
* Two-stage approach: FFmpeg encoding → MP4Box packaging
|
||
*/
|
||
export async function convertToDash(
|
||
options: DashConvertOptions
|
||
): Promise<DashConvertResult> {
|
||
const {
|
||
input,
|
||
outputDir,
|
||
segmentDuration = 2,
|
||
profiles: userProfiles,
|
||
customProfiles,
|
||
codec = 'dual',
|
||
useNvenc,
|
||
generateThumbnails = true,
|
||
thumbnailConfig = {},
|
||
generatePoster: shouldGeneratePoster = true,
|
||
posterTimecode = '00:00:01',
|
||
parallel = true,
|
||
onProgress
|
||
} = options;
|
||
|
||
// Create unique temp directory
|
||
const tempDir = join('/tmp', `dash-converter-${randomUUID()}`);
|
||
await ensureDir(tempDir);
|
||
|
||
try {
|
||
return await convertToDashInternal(
|
||
input,
|
||
outputDir,
|
||
tempDir,
|
||
segmentDuration,
|
||
userProfiles,
|
||
customProfiles,
|
||
codec,
|
||
useNvenc,
|
||
generateThumbnails,
|
||
thumbnailConfig,
|
||
shouldGeneratePoster,
|
||
posterTimecode,
|
||
parallel,
|
||
onProgress
|
||
);
|
||
} finally {
|
||
// Cleanup temp directory
|
||
try {
|
||
await rm(tempDir, { recursive: true, force: true });
|
||
} catch (err) {
|
||
console.warn(`Warning: Failed to cleanup temp directory: ${tempDir}`);
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* Internal conversion logic
|
||
*/
|
||
async function convertToDashInternal(
|
||
input: string,
|
||
outputDir: string,
|
||
tempDir: string,
|
||
segmentDuration: number,
|
||
userProfiles: VideoProfile[] | undefined,
|
||
customProfiles: string[] | undefined,
|
||
codec: CodecType,
|
||
useNvenc: boolean | undefined,
|
||
generateThumbnails: boolean,
|
||
thumbnailConfig: ThumbnailConfig,
|
||
generatePosterFlag: boolean,
|
||
posterTimecode: string,
|
||
parallel: boolean,
|
||
onProgress?: (progress: ConversionProgress) => void
|
||
): Promise<DashConvertResult> {
|
||
|
||
// Validate dependencies
|
||
if (!await checkFFmpeg()) {
|
||
throw new Error('FFmpeg is not installed or not in PATH');
|
||
}
|
||
|
||
if (!await checkMP4Box()) {
|
||
throw new Error('MP4Box is not installed or not in PATH. Install gpac package.');
|
||
}
|
||
|
||
// Report progress
|
||
const reportProgress = (stage: ConversionProgress['stage'], percent: number, message?: string, currentProfile?: string) => {
|
||
if (onProgress) {
|
||
onProgress({ stage, percent, message, currentProfile });
|
||
}
|
||
};
|
||
|
||
reportProgress('analyzing', 0, 'Analyzing input video...');
|
||
|
||
// Get video metadata
|
||
const metadata = await getVideoMetadata(input);
|
||
|
||
// Check NVENC availability
|
||
const nvencAvailable = useNvenc !== false ? await checkNvenc() : false;
|
||
const willUseNvenc = useNvenc === true ? true : (useNvenc === false ? false : nvencAvailable);
|
||
|
||
if (useNvenc === true && !nvencAvailable) {
|
||
throw new Error('NVENC requested but not available. Check NVIDIA drivers and GPU support.');
|
||
}
|
||
|
||
// Select profiles
|
||
let profiles: VideoProfile[];
|
||
|
||
if (customProfiles && customProfiles.length > 0) {
|
||
// User specified custom profiles via CLI
|
||
const result = createProfilesFromStrings(
|
||
customProfiles,
|
||
metadata.width,
|
||
metadata.height,
|
||
metadata.fps,
|
||
metadata.videoBitrate
|
||
);
|
||
|
||
// Show errors if any
|
||
if (result.errors.length > 0) {
|
||
console.warn('\n⚠️ Profile warnings:');
|
||
for (const error of result.errors) {
|
||
console.warn(` - ${error}`);
|
||
}
|
||
console.warn('');
|
||
}
|
||
|
||
profiles = result.profiles;
|
||
|
||
if (profiles.length === 0) {
|
||
throw new Error('No valid profiles found in custom list. Check warnings above.');
|
||
}
|
||
} else if (userProfiles) {
|
||
// Programmatic API usage
|
||
profiles = userProfiles;
|
||
} else {
|
||
// Default: auto-select based on source
|
||
profiles = selectProfiles(
|
||
metadata.width,
|
||
metadata.height,
|
||
metadata.fps,
|
||
metadata.videoBitrate
|
||
);
|
||
}
|
||
|
||
if (profiles.length === 0) {
|
||
throw new Error('No suitable profiles found for input video resolution');
|
||
}
|
||
|
||
// Create video name directory
|
||
const inputBasename = basename(input, extname(input));
|
||
const videoOutputDir = join(outputDir, inputBasename);
|
||
|
||
// Clean up previous conversion if exists
|
||
try {
|
||
await rm(videoOutputDir, { recursive: true, force: true });
|
||
} catch (err) {
|
||
// Directory might not exist, that's fine
|
||
}
|
||
|
||
await ensureDir(videoOutputDir);
|
||
|
||
// Determine which codecs to use based on codec parameter
|
||
const codecs: Array<{ type: 'h264' | 'av1'; codec: string; preset: string }> = [];
|
||
|
||
if (codec === 'h264' || codec === 'dual') {
|
||
const h264Codec = willUseNvenc ? 'h264_nvenc' : 'libx264';
|
||
const h264Preset = willUseNvenc ? 'p4' : 'medium';
|
||
codecs.push({ type: 'h264', codec: h264Codec, preset: h264Preset });
|
||
}
|
||
|
||
if (codec === 'av1' || codec === 'dual') {
|
||
// Check for AV1 hardware encoder
|
||
const av1Support = await checkAV1Support();
|
||
const av1Codec = av1Support.available ? av1Support.encoder! : 'libsvtav1';
|
||
const av1Preset = av1Support.available ? (av1Codec === 'av1_nvenc' ? 'p4' : 'medium') : '8';
|
||
codecs.push({ type: 'av1', codec: av1Codec, preset: av1Preset });
|
||
}
|
||
|
||
const codecNames = codecs.map(c => c.type.toUpperCase()).join(' + ');
|
||
reportProgress('analyzing', 20, `Using ${codecNames} encoding (${willUseNvenc ? 'GPU' : 'CPU'})`, undefined);
|
||
|
||
const maxConcurrent = willUseNvenc ? 3 : 2;
|
||
|
||
// STAGE 1: Encode profiles to MP4 for each codec (parallel - heavy work)
|
||
const codecMP4Paths = new Map<'h264' | 'av1', Map<string, string>>();
|
||
|
||
for (let codecIndex = 0; codecIndex < codecs.length; codecIndex++) {
|
||
const { type, codec: videoCodec, preset: codecPreset } = codecs[codecIndex];
|
||
const codecProgress = codecIndex / codecs.length;
|
||
const codecProgressRange = 1 / codecs.length;
|
||
|
||
reportProgress('encoding', 25 + codecProgress * 40, `Stage 1: Encoding ${type.toUpperCase()} (${profiles.length} profiles)...`);
|
||
|
||
const tempMP4Paths = await encodeProfilesToMP4(
|
||
input,
|
||
tempDir,
|
||
profiles,
|
||
videoCodec,
|
||
codecPreset,
|
||
metadata.duration,
|
||
segmentDuration,
|
||
metadata.fps || 25,
|
||
metadata.audioBitrate,
|
||
parallel,
|
||
maxConcurrent,
|
||
type, // Pass codec type to differentiate output files
|
||
undefined, // optimizations - for future use
|
||
(profileName, percent) => {
|
||
const profileIndex = profiles.findIndex(p => p.name === profileName);
|
||
const baseProgress = 25 + codecProgress * 40;
|
||
const profileProgress = (percent / 100) * (40 * codecProgressRange / profiles.length);
|
||
reportProgress('encoding', baseProgress + profileProgress, `Encoding ${type.toUpperCase()} ${profileName}...`, `${type}-${profileName}`);
|
||
|
||
// Also report individual profile progress
|
||
if (onProgress) {
|
||
onProgress({
|
||
stage: 'encoding',
|
||
percent: baseProgress + profileProgress,
|
||
currentProfile: `${type}-${profileName}`,
|
||
profilePercent: percent,
|
||
message: `Encoding ${type.toUpperCase()} ${profileName}...`
|
||
});
|
||
}
|
||
}
|
||
);
|
||
|
||
codecMP4Paths.set(type, tempMP4Paths);
|
||
}
|
||
|
||
reportProgress('encoding', 65, 'Stage 1 complete: All codecs and profiles encoded');
|
||
|
||
// STAGE 2: Package to DASH using MP4Box (light work, fast)
|
||
reportProgress('encoding', 70, `Stage 2: Creating DASH with MP4Box...`);
|
||
|
||
const manifestPath = await packageToDash(
|
||
codecMP4Paths,
|
||
videoOutputDir,
|
||
profiles,
|
||
segmentDuration,
|
||
codec
|
||
);
|
||
|
||
// Collect all video paths from all codecs
|
||
const videoPaths: string[] = [];
|
||
for (const mp4Paths of codecMP4Paths.values()) {
|
||
videoPaths.push(...Array.from(mp4Paths.values()));
|
||
}
|
||
|
||
reportProgress('encoding', 80, 'Stage 2 complete: DASH created');
|
||
|
||
// Generate thumbnails
|
||
let thumbnailSpritePath: string | undefined;
|
||
let thumbnailVttPath: string | undefined;
|
||
|
||
if (generateThumbnails) {
|
||
reportProgress('thumbnails', 80, 'Generating thumbnail sprites...');
|
||
|
||
const thumbConfig: Required<ThumbnailConfig> = {
|
||
width: thumbnailConfig.width || 160,
|
||
height: thumbnailConfig.height || 90,
|
||
interval: thumbnailConfig.interval || 1, // 1 секунда по умолчанию
|
||
columns: thumbnailConfig.columns || 10
|
||
};
|
||
|
||
const thumbResult = await generateThumbnailSprite(
|
||
input,
|
||
videoOutputDir,
|
||
metadata.duration,
|
||
thumbConfig
|
||
);
|
||
|
||
thumbnailSpritePath = thumbResult.spritePath;
|
||
thumbnailVttPath = thumbResult.vttPath;
|
||
|
||
reportProgress('thumbnails', 90, 'Thumbnails generated');
|
||
}
|
||
|
||
// Generate poster
|
||
let posterPath: string | undefined;
|
||
|
||
if (generatePosterFlag) {
|
||
reportProgress('thumbnails', 92, 'Generating poster image...');
|
||
|
||
posterPath = await generatePoster(
|
||
input,
|
||
videoOutputDir,
|
||
posterTimecode
|
||
);
|
||
|
||
reportProgress('thumbnails', 95, 'Poster generated');
|
||
}
|
||
|
||
// Generate MPD manifest
|
||
reportProgress('manifest', 95, 'Finalizing manifest...');
|
||
|
||
// Note: manifestPath is already created by MP4Box in packageToDash
|
||
// No need for separate generateManifest function
|
||
|
||
reportProgress('complete', 100, 'Conversion complete!');
|
||
|
||
return {
|
||
manifestPath,
|
||
videoPaths,
|
||
thumbnailSpritePath,
|
||
thumbnailVttPath,
|
||
posterPath,
|
||
duration: metadata.duration,
|
||
profiles,
|
||
usedNvenc: willUseNvenc,
|
||
codecType: codec
|
||
};
|
||
}
|
||
|