cerevox 0.3.2 → 0.3.4

package/dist/mcp/servers/zerocut.js

@@ -17,6 +17,43 @@ const constants_1 = require("../../utils/constants");
 const videokit_1 = require("../../utils/videokit");
 const promises_1 = require("node:fs/promises");
 const node_path_1 = require("node:path");
+// 错误处理工具函数
+function createErrorResponse(error, operation) {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    console.error(`[${operation}] Error:`, error);
+    return {
+        content: [
+            {
+                type: 'text',
+                text: JSON.stringify({
+                    success: false,
+                    error: errorMessage,
+                    operation,
+                    timestamp: new Date().toISOString(),
+                }),
+            },
+        ],
+    };
+}
+// Session 状态检查
+function validateSession(operation) {
+    if (!session) {
+        throw new Error(`Session not initialized. Please call 'zerocut-project-open' first before using ${operation}.`);
+    }
+    return session;
+}
+// 文件名验证
+function validateFileName(fileName) {
+    if (!fileName || fileName.trim() === '') {
+        throw new Error('File name cannot be empty');
+    }
+    if (fileName.includes('..') ||
+        fileName.includes('/') ||
+        fileName.includes('\\')) {
+        throw new Error('Invalid file name: contains illegal characters');
+    }
+    return fileName.trim();
+}
 /* Configuration
 {
   "mcpServers": {
@@ -105,85 +142,266 @@ server.registerTool('zerocut-project-open', {
             .describe('The path of the file to upload.'),
     },
 }, async ({ localDir }, context) => {
-    session = await cerevox.launch({
-        browser: {
-            liveview: true,
-        },
-        keepAliveMS: 10000,
-        timeoutMS: 3600000,
-    });
-    const workDir = await initProject(session);
-    projectLocalDir = (0, node_path_1.resolve)(process.env.ZEROCUT_PROJECT_CWD || process.cwd(), localDir);
-    const syncDir = (0, node_path_1.resolve)(projectLocalDir, 'materials');
-    await (0, promises_1.mkdir)(syncDir, { recursive: true });
-    const materials = await listFiles(syncDir);
-    const files = session.files;
-    let progress = 0;
-    for (const material of materials) {
-        await files.upload(material, `${workDir}/materials/${(0, node_path_1.basename)(material)}`);
-        sendProgress(context, ++progress, materials.length, material);
-    }
-    return {
-        content: [
-            {
-                type: 'text',
-                text: JSON.stringify({
-                    sessionId: session.id,
-                    workDir,
-                    projectLocalDir,
-                    materials,
-                }),
+    try {
+        // 检查是否已有活跃session
+        if (session) {
+            console.warn('Session already exists, closing previous session');
+            try {
+                await session.close();
+            }
+            catch (closeError) {
+                console.warn('Failed to close previous session:', closeError);
+            }
+        }
+        // 验证API密钥
+        if (!process.env.CEREVOX_API_KEY) {
+            throw new Error('CEREVOX_API_KEY environment variable is required');
+        }
+        console.log('Launching new Cerevox session...');
+        session = await cerevox.launch({
+            browser: {
+                liveview: true,
             },
-        ],
-    };
+            keepAliveMS: 10000,
+            timeoutMS: 3600000,
+        });
+        if (!session) {
+            throw new Error('Failed to create Cerevox session');
+        }
+        console.log('Initializing project...');
+        const workDir = await initProject(session);
+        projectLocalDir = (0, node_path_1.resolve)(process.env.ZEROCUT_PROJECT_CWD || process.cwd(), localDir || '.');
+        const syncDir = (0, node_path_1.resolve)(projectLocalDir, 'materials');
+        try {
+            await (0, promises_1.mkdir)(syncDir, { recursive: true });
+        }
+        catch (mkdirError) {
+            console.warn('Failed to create materials directory:', mkdirError);
+            // 继续执行，可能目录已存在
+        }
+        let materials = [];
+        try {
+            materials = await listFiles(syncDir);
+        }
+        catch (listError) {
+            console.warn('Failed to list materials:', listError);
+            materials = [];
+        }
+        const files = session.files;
+        let progress = 0;
+        const uploadErrors = [];
+        for (const material of materials) {
+            try {
+                await files.upload(material, `${workDir}/materials/${(0, node_path_1.basename)(material)}`);
+                await sendProgress(context, ++progress, materials.length, material);
+            }
+            catch (uploadError) {
+                const errorMsg = `Failed to upload ${material}: ${uploadError}`;
+                console.error(errorMsg);
+                uploadErrors.push(errorMsg);
+            }
+        }
+        const result = {
+            success: true,
+            sessionId: session.id,
+            workDir,
+            projectLocalDir,
+            materials,
+            uploadErrors: uploadErrors.length > 0 ? uploadErrors : undefined,
+        };
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify(result),
+                },
+            ],
+        };
+    }
+    catch (error) {
+        // 不自动关闭session，让agent根据异常信息自行处理
+        return createErrorResponse(error, 'zerocut-project-open');
+    }
 });
 server.registerTool('zerocut-project-close', {
     title: 'Close Project',
     description: 'Close the current Cerevox session and release all resources.',
     inputSchema: {},
 }, async () => {
-    if (session) {
-        await session.close();
+    try {
+        if (session) {
+            console.log('Closing Cerevox session...');
+            await session.close();
+            session = null;
+            console.log('Session closed successfully');
+        }
+        else {
+            console.warn('No active session to close');
+        }
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify({
+                        success: true,
+                        message: 'Project closed successfully.',
+                        timestamp: new Date().toISOString(),
+                    }),
+                },
+            ],
+        };
+    }
+    catch (error) {
+        // 即使关闭失败，也要清理session引用
         session = null;
+        return createErrorResponse(error, 'zerocut-project-close');
     }
-    return {
-        content: [{ type: 'text', text: 'Cerevox session closed' }],
-    };
 });
 // 将完成的成品下载到本地
+// server.registerTool(
+//   'download-outputs',
+//   {
+//     title: 'Download Outputs',
+//     description: 'Download the output files from the server.',
+//     inputSchema: {},
+//   },
+//   async (_, context) => {
+//     const terminal = session!.terminal;
+//     const workDir = `/home/user/cerevox-zerocut/projects/${terminal.id}/output`;
+//     const outputDir = resolve(
+//       process.env.ZEROCUT_PROJECT_CWD || process.cwd(),
+//       projectLocalDir,
+//       'output'
+//     );
+//     await mkdir(outputDir, { recursive: true });
+//     let progress = 0;
+//     const outputs = (await session?.files.listFiles(workDir)) || [];
+//     const files = session!.files;
+//     for (const output of outputs) {
+//       await files.download(`${workDir}/${output}`, `${outputDir}/${output}`);
+//       sendProgress(context, ++progress, outputs.length, output);
+//     }
+//     return {
+//       content: [
+//         { type: 'text', text: JSON.stringify({ success: true, outputs }) },
+//       ],
+//     };
+//   }
+// );
 server.registerTool('download-outputs', {
-    title: 'Download Outputs',
+    title: 'Download Outputs from HTTP',
     description: 'Download the output files from the server.',
     inputSchema: {},
 }, async (_, context) => {
-    const terminal = session.terminal;
-    const workDir = `/home/user/cerevox-zerocut/projects/${terminal.id}/output`;
-    const outputDir = (0, node_path_1.resolve)(process.env.ZEROCUT_PROJECT_CWD || process.cwd(), projectLocalDir, 'output');
-    await (0, promises_1.mkdir)(outputDir, { recursive: true });
-    let progress = 0;
-    const outputs = (await session?.files.listFiles(workDir)) || [];
-    const files = session.files;
-    for (const output of outputs) {
-        await files.download(`${workDir}/${output}`, `${outputDir}/${output}`);
-        sendProgress(context, ++progress, outputs.length, output);
+    try {
+        // 验证session状态
+        const currentSession = validateSession('download-outputs');
+        console.log('Starting download outputs process...');
+        const terminal = currentSession.terminal;
+        if (!terminal) {
+            throw new Error('Terminal not available in current session');
+        }
+        const workDir = `/home/user/cerevox-zerocut/projects/${terminal.id}/output`;
+        const host = currentSession.getHost();
+        if (!host) {
+            throw new Error('Host information not available from session');
+        }
+        console.log(`Listing output files from: ${workDir}`);
+        let outputs = [];
+        try {
+            outputs = (await currentSession.files.listFiles(workDir)) || [];
+        }
+        catch (listError) {
+            console.warn('Failed to list output files:', listError);
+            outputs = [];
+        }
+        if (outputs.length === 0) {
+            console.log('No output files found');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: true,
+                            status: 'no_files',
+                            message: 'No output files found to download',
+                            sources: [],
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+        console.log(`Found ${outputs.length} output files`);
+        const publicDir = `/home/user/public/zerocut`;
+        try {
+            const copyCommand = `mkdir -p ${publicDir} && cp -R ${workDir} ${publicDir}`;
+            console.log(`Executing: ${copyCommand}`);
+            const copyResult = await terminal.run(copyCommand);
+            await copyResult.end();
+            console.log('Files copied to public directory successfully');
+        }
+        catch (copyError) {
+            console.error('Failed to copy files to public directory:', copyError);
+            throw new Error(`Failed to prepare files for download: ${copyError}`);
+        }
+        let progress = 0;
+        const downloadErrors = [];
+        const successfulDownloads = [];
+        const outputDir = (0, node_path_1.resolve)(process.env.ZEROCUT_PROJECT_CWD || process.cwd(), projectLocalDir, 'output');
+        await (0, promises_1.mkdir)(outputDir, { recursive: true });
+        const promises = outputs.map(async (output) => {
+            try {
+                const url = `https://${host}/public/zerocut/output/${output}`;
+                console.log(`Downloading: ${url}`);
+                const res = await fetch(url);
+                if (!res.ok) {
+                    throw new Error(`HTTP ${res.status}: ${res.statusText}`);
+                }
+                const content = await res.arrayBuffer();
+                const buffer = Buffer.from(content);
+                await (0, promises_1.writeFile)(`${outputDir}/${output}`, buffer);
+                try {
+                    await sendProgress(context, ++progress, outputs.length, output);
+                }
+                catch (progressError) {
+                    console.warn('Failed to send progress update:', progressError);
+                }
+                successfulDownloads.push(url);
+                return url;
+            }
+            catch (downloadError) {
+                const errorMsg = `Failed to download ${output}: ${downloadError}`;
+                console.error(errorMsg);
+                downloadErrors.push(errorMsg);
+                return null;
+            }
+        });
+        const results = await Promise.all(promises);
+        const sources = results.filter(url => url !== null);
+        const result = {
+            success: downloadErrors.length === 0,
+            status: 'downloaded',
+            sources,
+            totalFiles: outputs.length,
+            successfulDownloads: sources.length,
+            failedDownloads: downloadErrors.length,
+            downloadErrors: downloadErrors.length > 0 ? downloadErrors : undefined,
+            timestamp: new Date().toISOString(),
+        };
+        console.log(`Download completed: ${sources.length}/${outputs.length} files successful`);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify(result),
+                },
+            ],
+        };
+    }
+    catch (error) {
+        return createErrorResponse(error, 'download-outputs');
     }
-    // await session!.files.syncDownloadsDirectory(
-    //   outputDir,
-    //   workDir,
-    //   async metaData => {
-    //     await sendProgress(
-    //       context,
-    //       ++progress,
-    //       outputs.length,
-    //       JSON.stringify(metaData)
-    //     );
-    //   }
-    // );
-    return {
-        content: [
-            { type: 'text', text: JSON.stringify({ success: true, outputs }) },
-        ],
-    };
 });
 // 列出项目下的所有文件
 server.registerTool('list-project-files', {
@@ -191,67 +409,63 @@ server.registerTool('list-project-files', {
     description: 'List all files in the materials directory.',
     inputSchema: {},
 }, async () => {
-    const terminal = session.terminal;
-    if (!terminal) {
+    try {
+        // 验证session状态
+        const currentSession = validateSession('list-project-files');
+        console.log('Listing project files...');
+        const terminal = currentSession.terminal;
+        if (!terminal) {
+            throw new Error('Terminal not available in current session');
+        }
+        let cwd;
+        try {
+            cwd = await terminal.getCwd();
+        }
+        catch (cwdError) {
+            console.error('Failed to get current working directory:', cwdError);
+            throw new Error('Failed to get current working directory');
+        }
+        console.log(`Current working directory: ${cwd}`);
+        // 安全地列出各目录文件，失败时返回空数组
+        const listFilesWithFallback = async (path, dirName) => {
+            try {
+                const files = await currentSession.files.listFiles(path);
+                console.log(`Found ${files?.length || 0} files in ${dirName}`);
+                return files || [];
+            }
+            catch (error) {
+                console.warn(`Failed to list files in ${dirName} (${path}):`, error);
+                return [];
+            }
+        };
+        const [rootFiles, materialsFiles, outputFiles] = await Promise.all([
+            listFilesWithFallback(cwd, 'root'),
+            listFilesWithFallback(`${cwd}/materials`, 'materials'),
+            listFilesWithFallback(`${cwd}/output`, 'output'),
+        ]);
+        const result = {
+            success: true,
+            cwd,
+            root: rootFiles,
+            materials: materialsFiles,
+            output: outputFiles,
+            totalFiles: rootFiles.length + materialsFiles.length + outputFiles.length,
+            timestamp: new Date().toISOString(),
+        };
+        console.log(`Total files found: ${result.totalFiles}`);
         return {
-            content: [{ type: 'text', text: 'No session found' }],
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify(result),
+                },
+            ],
         };
     }
-    const cwd = await terminal.getCwd();
-    const rootFiles = await session?.files.listFiles(cwd);
-    const materialsFiles = await session?.files.listFiles(`${cwd}/materials`);
-    const outputFiles = await session?.files.listFiles(`${cwd}/output`);
-    return {
-        content: [
-            {
-                type: 'text',
-                text: JSON.stringify({
-                    root: rootFiles,
-                    materials: materialsFiles,
-                    output: outputFiles,
-                }),
-            },
-        ],
-    };
+    catch (error) {
+        return createErrorResponse(error, 'list-project-files');
+    }
 });
-// server.registerTool(
-//   'read-file',
-//   {
-//     title: 'Read File',
-//     description: 'Read the content of a file.',
-//     inputSchema: {
-//       path: z.string().describe('The path of the file to read.'),
-//     },
-//   },
-//   async ({ path }) => {
-//     const files = session!.files;
-//     const content: string = await files.read(path, { encoding: 'utf8' });
-//     return {
-//       content: [{ type: 'text', text: content }],
-//     };
-//   }
-// );
-// server.registerTool(
-//   'write-file',
-//   {
-//     title: 'Write File',
-//     description: 'Write the content to a file.',
-//     inputSchema: {
-//       path: z.string().describe('The path of the file to write.'),
-//       content: z.string().describe('The content to write.'),
-//     },
-//   },
-//   async ({ path, content }) => {
-//     const files = session!.files;
-//     await files.write(path, content);
-//     assetsCount++;
-//     return {
-//       content: [
-//         { type: 'text', text: JSON.stringify({ path, success: true }) },
-//       ],
-//     };
-//   }
-// );
 server.registerTool('search-context', {
     title: 'Search Context',
     description: 'Search the context.',
@@ -299,22 +513,59 @@ server.registerTool('generate-image', {
         saveToFileName: zod_1.z.string().describe('The filename to save.'),
     },
 }, async ({ prompt, size, saveToFileName }) => {
-    const ai = session.ai;
-    const res = await ai.generateImage({
-        prompt,
-        size,
-    });
-    if (res.url) {
-        const uri = await saveMertial(session, res.url, saveToFileName);
-        return {
-            content: [
-                { type: 'text', text: JSON.stringify({ source: res.url, uri }) },
-            ],
-        };
+    try {
+        // 验证session状态
+        const currentSession = validateSession('generate-image');
+        const validatedFileName = validateFileName(saveToFileName);
+        console.log(`Generating image with prompt: ${prompt.substring(0, 100)}...`);
+        const ai = currentSession.ai;
+        const res = await ai.generateImage({
+            prompt: prompt.trim(),
+            size,
+        });
+        if (!res) {
+            throw new Error('Failed to generate image: no response from AI service');
+        }
+        if (res.url) {
+            console.log('Image generated successfully, saving to materials...');
+            const uri = await saveMertial(currentSession, res.url, validatedFileName);
+            const result = {
+                success: true,
+                source: res.url,
+                uri,
+                prompt: prompt.substring(0, 100),
+                size,
+                timestamp: new Date().toISOString(),
+            };
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify(result),
+                    },
+                ],
+            };
+        }
+        else {
+            console.warn('Image generation completed but no URL returned');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: false,
+                            error: 'No image URL returned from AI service',
+                            response: res,
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+    }
+    catch (error) {
+        return createErrorResponse(error, 'generate-image');
     }
-    return {
-        content: [{ type: 'text', text: JSON.stringify(res) }],
-    };
 });
 server.registerTool('generate-video', {
     title: 'Generate Video',
@@ -335,38 +586,188 @@ server.registerTool('generate-video', {
             .describe('The end frame image URL.'),
     },
 }, async ({ prompt, saveToFileName, start_frame, end_frame, duration }, context) => {
-    const ai = session.ai;
-    let progress = 0;
-    const res = await ai.framesToVideo({
-        prompt,
-        start_frame,
-        end_frame,
-        duration,
-        resolution: '720p',
-        onProgress: async (metaData) => {
-            await sendProgress(context, ++progress, undefined, JSON.stringify(metaData));
-        },
-    });
-    if (res.url) {
-        const uri = await saveMertial(session, res.url, saveToFileName);
-        const { url, duration, ...opts } = res;
+    try {
+        // 验证session状态
+        const currentSession = validateSession('generate-video');
+        const validatedFileName = validateFileName(saveToFileName);
+        console.log(`Generating video with prompt: ${prompt.substring(0, 100)}...`);
+        const ai = currentSession.ai;
+        let progress = 0;
+        const res = await ai.framesToVideo({
+            prompt: prompt.trim(),
+            start_frame,
+            end_frame,
+            duration,
+            resolution: '720p',
+            onProgress: async (metaData) => {
+                try {
+                    await sendProgress(context, ++progress, undefined, JSON.stringify(metaData));
+                }
+                catch (progressError) {
+                    console.warn('Failed to send progress update:', progressError);
+                }
+            },
+        });
+        if (!res) {
+            throw new Error('Failed to generate video: no response from AI service');
+        }
+        if (res.url) {
+            console.log('Video generated successfully, saving to materials...');
+            const uri = await saveMertial(currentSession, res.url, validatedFileName);
+            const { url, duration: videoDuration, ...opts } = res;
+            const result = {
+                success: true,
+                source: url,
+                uri,
+                durationMs: Math.floor((videoDuration || duration) * 1000),
+                prompt: prompt.substring(0, 100),
+                timestamp: new Date().toISOString(),
+                ...opts,
+            };
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify(result),
+                    },
+                ],
+            };
+        }
+        else {
+            console.warn('Video generation completed but no URL returned');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: false,
+                            error: 'No video URL returned from AI service',
+                            response: res,
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+    }
+    catch (error) {
+        return createErrorResponse(error, 'generate-video');
+    }
+});
+server.registerTool('generate-video-fallback', {
+    title: 'Generate Ken Burns Motion as Video Fallback',
+    description: 'Generate the ken burns motion as video fallback.',
+    inputSchema: {
+        frame_index: zod_1.z.number().describe('The frame index.'),
+        image_path: zod_1.z.string().describe('The start frame image path（not URL).'),
+        duration: zod_1.z
+            .number()
+            .min(3)
+            .max(12)
+            .describe('The duration of the video.'),
+        size: zod_1.z
+            .enum([
+            '1024x1024',
+            '864x1152',
+            '1152x864',
+            '1280x720',
+            '720x1280',
+            '832x1248',
+            '1248x832',
+            '1512x648',
+        ])
+            .describe('The size of the image.'),
+        saveToFileName: zod_1.z.string().describe('The filename to save.'),
+    },
+}, async ({ frame_index, image_path, duration, size, saveToFileName }) => {
+    try {
+        // 验证session状态
+        const currentSession = validateSession('generate-video-fallback');
+        const validatedFileName = validateFileName(saveToFileName);
+        console.log(`Generating Ken Burns motion for frame ${frame_index}: ${image_path} (${duration}s, ${size})`);
+        const files = currentSession.files;
+        const terminal = currentSession.terminal;
+        if (!terminal) {
+            throw new Error('Terminal not available in current session');
+        }
+        const saveToPath = `/home/user/cerevox-zerocut/projects/${terminal.id}/materials/${validatedFileName}`;
+        const saveLocalPath = (0, node_path_1.resolve)(projectLocalDir, 'materials', validatedFileName);
+        // 解析尺寸参数
+        const sizeArray = size.split('x');
+        if (sizeArray.length !== 2) {
+            throw new Error(`Invalid size format: ${size}. Expected format: WIDTHxHEIGHT`);
+        }
+        const [widthStr, heightStr] = sizeArray;
+        const width = Number(widthStr);
+        const height = Number(heightStr);
+        if (isNaN(width) || isNaN(height) || width <= 0 || height <= 0) {
+            throw new Error(`Invalid dimensions: ${width}x${height}. Both width and height must be positive numbers`);
+        }
+        console.log(`Compiling Ken Burns motion command...`);
+        let command;
+        try {
+            command = await (0, videokit_1.compileKenBurnsMotion)(frame_index, image_path, duration, {
+                output: saveToPath,
+                width,
+                height,
+            });
+        }
+        catch (compileError) {
+            console.error('Failed to compile Ken Burns motion command:', compileError);
+            throw new Error(`Failed to compile Ken Burns motion: ${compileError}`);
+        }
+        console.log(`Executing FFmpeg command: ${command.substring(0, 100)}...`);
+        const res = await terminal.run(command);
+        const result = await res.json();
+        if (result.exitCode !== 0) {
+            console.error('FFmpeg command failed:', result);
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: false,
+                            error: 'Ken Burns motion generation failed',
+                            exitCode: result.exitCode,
+                            stderr: result.stderr,
+                            command: command.substring(0, 200),
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+        console.log('Ken Burns motion generated successfully, downloading file...');
+        try {
+            await files.download(saveToPath, saveLocalPath);
+        }
+        catch (downloadError) {
+            console.warn('Failed to download file to local:', downloadError);
+            // 继续执行，因为远程文件已生成成功
+        }
+        const resultData = {
+            success: true,
+            uri: saveToPath,
+            durationMs: Math.floor(duration * 1000),
+            frameIndex: frame_index,
+            imagePath: image_path,
+            size,
+            dimensions: { width, height },
+            timestamp: new Date().toISOString(),
+        };
+        console.log(`Ken Burns motion completed: ${saveToPath}`);
         return {
             content: [
                 {
                     type: 'text',
-                    text: JSON.stringify({
-                        source: url,
-                        uri,
-                        durationMs: Math.floor(duration * 1000),
-                        ...opts,
-                    }),
+                    text: JSON.stringify(resultData),
                 },
             ],
         };
     }
-    return {
-        content: [{ type: 'text', text: JSON.stringify(res) }],
-    };
+    catch (error) {
+        return createErrorResponse(error, 'generate-video-fallback');
+    }
 });
 server.registerTool('generate-bgm', {
     title: 'Generate BGM',
@@ -381,35 +782,71 @@ server.registerTool('generate-bgm', {
         saveToFileName: zod_1.z.string().describe('The filename to save.'),
     },
 }, async ({ prompt, duration, saveToFileName }, context) => {
-    const ai = session.ai;
-    let progress = 0;
-    const res = await ai.generateBGM({
-        prompt,
-        duration,
-        onProgress: async (metaData) => {
-            await sendProgress(context, metaData.Result?.Progress ?? ++progress, metaData.Result?.Progress ? 100 : undefined, JSON.stringify(metaData));
-        },
-    });
-    if (res.url) {
-        const uri = await saveMertial(session, res.url, saveToFileName);
-        const { url, duration, ...opts } = res;
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        source: url,
-                        uri,
-                        durationMs: Math.floor(duration * 1000),
-                        ...opts,
-                    }),
-                },
-            ],
-        };
+    try {
+        // 验证session状态
+        const currentSession = validateSession('generate-bgm');
+        const validatedFileName = validateFileName(saveToFileName);
+        console.log(`Generating BGM with prompt: ${prompt.substring(0, 100)}... (${duration}s)`);
+        const ai = currentSession.ai;
+        let progress = 0;
+        const res = await ai.generateBGM({
+            prompt: prompt.trim(),
+            duration,
+            onProgress: async (metaData) => {
+                try {
+                    await sendProgress(context, metaData.Result?.Progress ?? ++progress, metaData.Result?.Progress ? 100 : undefined, JSON.stringify(metaData));
+                }
+                catch (progressError) {
+                    console.warn('Failed to send progress update:', progressError);
+                }
+            },
+        });
+        if (!res) {
+            throw new Error('Failed to generate BGM: no response from AI service');
+        }
+        if (res.url) {
+            console.log('BGM generated successfully, saving to materials...');
+            const uri = await saveMertial(currentSession, res.url, validatedFileName);
+            const { url, duration: bgmDuration, ...opts } = res;
+            const result = {
+                success: true,
+                source: url,
+                uri,
+                durationMs: Math.floor((bgmDuration || duration) * 1000),
+                prompt: prompt.substring(0, 100),
+                requestedDuration: duration,
+                timestamp: new Date().toISOString(),
+                ...opts,
+            };
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify(result),
+                    },
+                ],
+            };
+        }
+        else {
+            console.warn('BGM generation completed but no URL returned');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: false,
+                            error: 'No BGM URL returned from AI service',
+                            response: res,
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+    }
+    catch (error) {
+        return createErrorResponse(error, 'generate-bgm');
     }
-    return {
-        content: [{ type: 'text', text: JSON.stringify(res) }],
-    };
 });
 server.registerTool('generate-scene-tts', {
     title: 'Generate Scene TTS',
@@ -417,6 +854,13 @@ server.registerTool('generate-scene-tts', {
     inputSchema: {
         text: zod_1.z.string().describe('The text to generate.'),
         saveToFileName: zod_1.z.string().describe('The filename to save.'),
+        speed: zod_1.z
+            .number()
+            .min(0.5)
+            .max(2)
+            .optional()
+            .default(1)
+            .describe('The speed of the tts.'),
         voiceName: zod_1.z
             .enum([
             'presenter_female',
@@ -437,32 +881,66 @@ server.registerTool('generate-scene-tts', {
         ])
             .describe('适合作为视频配音的中英文音色（推荐15个）。来源于系统音色清单，偏中性、耐听，覆盖新闻/主持、企业宣传、科普解说、纪录片与生活方式等场景。'),
     },
-}, async ({ text, voiceName, saveToFileName }) => {
-    const ai = session.ai;
-    const res = await ai.textToSpeech({
-        text,
-        voiceName,
-    });
-    if (res.url) {
-        const uri = await saveMertial(session, res.url, saveToFileName);
-        const { url, duration, ...opts } = res;
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        source: url,
-                        uri,
-                        durationMs: Math.floor(duration * 1000),
-                        ...opts,
-                    }),
-                },
-            ],
-        };
+}, async ({ text, voiceName, saveToFileName, speed }) => {
+    try {
+        // 验证session状态
+        const currentSession = validateSession('generate-scene-tts');
+        const validatedFileName = validateFileName(saveToFileName);
+        const finalSpeed = speed ?? 1;
+        console.log(`Generating TTS with voice: ${voiceName}, speed: ${finalSpeed}, text: ${text.substring(0, 100)}...`);
+        const ai = currentSession.ai;
+        const res = await ai.textToSpeech({
+            text: text.trim(),
+            voiceName,
+            speed: finalSpeed,
+        });
+        if (!res) {
+            throw new Error('Failed to generate TTS: no response from AI service');
+        }
+        if (res.url) {
+            console.log('TTS generated successfully, saving to materials...');
+            const uri = await saveMertial(currentSession, res.url, validatedFileName);
+            const { url, duration, ...opts } = res;
+            const result = {
+                success: true,
+                source: url,
+                uri,
+                durationMs: Math.floor((duration || 0) * 1000),
+                text: text.substring(0, 100),
+                voiceName,
+                speed: finalSpeed,
+                timestamp: new Date().toISOString(),
+                ...opts,
+            };
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify(result),
+                    },
+                ],
+            };
+        }
+        else {
+            console.warn('TTS generation completed but no URL returned');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: false,
+                            error: 'No TTS URL returned from AI service',
+                            response: res,
+                            timestamp: new Date().toISOString(),
+                        }),
+                    },
+                ],
+            };
+        }
+    }
+    catch (error) {
+        return createErrorResponse(error, 'generate-scene-tts');
     }
-    return {
-        content: [{ type: 'text', text: JSON.stringify(res) }],
-    };
 });
 server.registerTool('compile-and-run', {
     title: 'Compile And Run',
@@ -476,71 +954,89 @@ server.registerTool('compile-and-run', {
     },
 }, async ({ project, outputFileName }) => {
     try {
-        // Project is already validated by zVideoProject schema
-        const validated = project;
-        // Get working directory
-        const terminal = session.terminal;
+        // 验证session状态
+        const currentSession = validateSession('compile-and-run');
+        console.log('Starting video compilation and rendering...');
+        // 验证terminal可用性
+        const terminal = currentSession.terminal;
+        if (!terminal) {
+            throw new Error('Terminal not available in current session');
+        }
+        // 验证输出文件名安全性
+        const outFile = outputFileName || 'output.mp4';
+        const validatedFileName = validateFileName(outFile);
+        console.log(`Output file: ${validatedFileName}`);
+        // 构建工作目录路径
         const workDir = `/home/user/cerevox-zerocut/projects/${terminal.id}`;
         const outputDir = `${workDir}/output`;
-        // Set output file
-        const outFile = outputFileName || 'output.mp4';
-        const outputPath = `${outputDir}/${outFile}`;
-        // Update export configuration
-        validated.export.outFile = outputPath;
-        // Compile to FFmpeg command
-        const compiled = (0, videokit_1.compileToFfmpeg)(validated, {
-            workingDir: outputDir,
-            subtitleStrategy: 'auto',
-        });
-        // Log the FFmpeg command for debugging
-        console.log('FFmpeg Command:', compiled.cmd);
-        // Run FFmpeg
-        const result = await runFfmpeg(session, compiled);
+        const outputPath = `${outputDir}/${validatedFileName}`;
+        // Project已经通过zVideoProject schema验证
+        const validated = { ...project };
+        // 更新导出配置
+        validated.export = {
+            ...validated.export,
+            outFile: outputPath,
+        };
+        console.log('Compiling VideoProject to FFmpeg command...');
+        // 编译为FFmpeg命令
+        let compiled;
+        try {
+            compiled = (0, videokit_1.compileToFfmpeg)(validated, {
+                workingDir: outputDir,
+                subtitleStrategy: 'auto',
+            });
+        }
+        catch (compileError) {
+            console.error('Failed to compile VideoProject:', compileError);
+            throw new Error(`Failed to compile VideoProject: ${compileError}`);
+        }
+        console.log(`FFmpeg command generated (${compiled.cmd.length} chars)`);
+        console.log('FFmpeg Command:', compiled.cmd.substring(0, 200) + '...');
+        // 执行FFmpeg命令
+        console.log('Executing FFmpeg command...');
+        const result = await runFfmpeg(currentSession, compiled);
         if (result.exitCode === 0) {
+            console.log('Video compilation completed successfully');
+            const successResult = {
+                success: true,
+                outputPath,
+                outputFileName: validatedFileName,
+                command: compiled.cmd.substring(0, 500), // 限制命令长度
+                message: 'Video compilation completed successfully',
+                timestamp: new Date().toISOString(),
+            };
             return {
                 content: [
                     {
                         type: 'text',
-                        text: JSON.stringify({
-                            success: true,
-                            outputPath,
-                            command: compiled.cmd,
-                            message: 'Video compilation completed successfully',
-                        }),
+                        text: JSON.stringify(successResult),
                     },
                 ],
             };
         }
         else {
+            console.error(`FFmpeg failed with exit code: ${result.exitCode}`);
+            const failureResult = {
+                success: false,
+                exitCode: result.exitCode,
+                outputPath,
+                command: compiled.cmd.substring(0, 500),
+                stderr: result.stderr?.substring(0, 1000), // 限制错误输出长度
+                message: `FFmpeg exited with code ${result.exitCode}`,
+                timestamp: new Date().toISOString(),
+            };
             return {
                 content: [
                     {
                         type: 'text',
-                        text: JSON.stringify({
-                            success: false,
-                            exitCode: result.exitCode,
-                            command: compiled.cmd,
-                            message: `FFmpeg exited with code ${result.exitCode}`,
-                        }),
+                        text: JSON.stringify(failureResult),
                     },
                 ],
             };
         }
     }
     catch (error) {
-        const errorMessage = error instanceof Error ? error.message : String(error);
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        success: false,
-                        error: errorMessage,
-                        message: 'Failed to compile and run project',
-                    }),
-                },
-            ],
-        };
+        return createErrorResponse(error, 'compile-and-run');
     }
 });
 server.registerTool('get-video-project-schema', {