File size: 4,443 Bytes
d47b053 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 | import { spawn } from 'child_process'
import { createLogger } from './logger'
import {
registerManimProcess,
unregisterManimProcess,
wasManimProcessCancelled
} from './manim-process-registry'
import {
buildManimArgs,
buildResult,
createExecutionState,
elapsedSeconds,
handleStderrData,
handleStdoutData,
normalizeExecuteOptions,
startMemoryMonitor
} from './manim-executor-runtime'
const logger = createLogger('ManimExecutor')
export interface ManimExecutionResult {
success: boolean
stdout: string
stderr: string
peakMemoryMB: number
exitCode?: number
}
export interface ManimExecuteOptions {
jobId: string
quality: string
frameRate?: number
format?: 'mp4' | 'png'
sceneName?: string
tempDir: string
mediaDir: string
timeoutMs?: number
}
export function executeManimCommand(
codeFile: string,
options: ManimExecuteOptions
): Promise<ManimExecutionResult> {
const normalizedOptions = normalizeExecuteOptions(options)
const args = buildManimArgs(codeFile, normalizedOptions)
logger.info(`Job ${normalizedOptions.jobId}: starting manim process`, {
command: `manim ${args.join(' ')}`,
cwd: normalizedOptions.tempDir
})
return new Promise((resolve) => {
const startTime = Date.now()
const state = createExecutionState()
const proc = spawn('manim', args, { cwd: normalizedOptions.tempDir })
registerManimProcess(normalizedOptions.jobId, proc)
const memoryMonitor = startMemoryMonitor(proc, normalizedOptions, state)
let timeoutTimer: NodeJS.Timeout | null = null
let settled = false
const settle = (result: ManimExecutionResult): void => {
if (settled) {
return
}
settled = true
if (timeoutTimer) {
clearTimeout(timeoutTimer)
}
clearInterval(memoryMonitor)
unregisterManimProcess(normalizedOptions.jobId)
resolve(result)
}
proc.stdout.on('data', (data) => {
handleStdoutData(state, normalizedOptions.jobId, data.toString())
})
proc.stderr.on('data', (data) => {
handleStderrData(state, normalizedOptions.jobId, data.toString())
})
timeoutTimer = setTimeout(() => {
const elapsed = elapsedSeconds(startTime)
logger.warn(`Job ${normalizedOptions.jobId}: manim render timeout (${elapsed}s), killing process`, {
peakMemoryMB: state.peakMemoryMB
})
proc.kill('SIGKILL')
settle(
buildResult(
false,
state,
state.stderr || `Manim render timeout (${Math.round(normalizedOptions.timeoutMs / 1000)} seconds)`
)
)
}, normalizedOptions.timeoutMs)
proc.on('close', (code) => {
const elapsed = elapsedSeconds(startTime)
const cancelled = wasManimProcessCancelled(normalizedOptions.jobId)
if (cancelled) {
logger.warn(`Job ${normalizedOptions.jobId}: Manim cancelled`, { elapsed: `${elapsed}s` })
settle(buildResult(false, state, 'Job cancelled', code ?? undefined))
return
}
if (code === 0) {
logger.info(`Job ${normalizedOptions.jobId}: manim completed`, {
elapsed: `${elapsed}s`,
exitCode: code,
stdoutLength: state.stdout.length,
stderrLength: state.stderr.length,
peakMemoryMB: state.peakMemoryMB
})
settle(buildResult(true, state, undefined, code ?? undefined))
return
}
logger.error(`Job ${normalizedOptions.jobId}: manim exited with error`, {
elapsed: `${elapsed}s`,
exitCode: code,
stdoutLength: state.stdout.length,
stderrLength: state.stderr.length,
stderrPreview: state.stderr.slice(-500),
peakMemoryMB: state.peakMemoryMB
})
settle(buildResult(false, state, undefined, code ?? undefined))
})
proc.on('error', (error) => {
const elapsed = elapsedSeconds(startTime)
const cancelled = wasManimProcessCancelled(normalizedOptions.jobId)
if (cancelled) {
logger.warn(`Job ${normalizedOptions.jobId}: Manim cancelled`, { elapsed: `${elapsed}s` })
settle(buildResult(false, state, 'Job cancelled'))
return
}
logger.error(`Job ${normalizedOptions.jobId}: manim process start failed`, {
elapsed: `${elapsed}s`,
errorMessage: error.message,
errorStack: error.stack
})
settle(buildResult(false, state, error.message))
})
})
}
|