@laitszkin/apollo-toolkit 4.1.4 → 5.0.1

package/packages/tools/generate-storyboard-images/dist/index.js

@@ -3,91 +3,7 @@ import path from 'node:path';
 import https from 'node:https';
 import http from 'node:http';
 import { fileURLToPath } from 'node:url';
-function parseArgs(args) {
-    const parsed = {
-        contentName: null,
-        projectDir: '.',
-        envFile: null,
-        apiUrl: null,
-        apiKey: null,
-        promptsFile: null,
-        prompts: [],
-        imageModel: null,
-        aspectRatio: null,
-        imageSize: null,
-        quality: null,
-        style: null,
-        help: false,
-    };
-    for (let i = 0; i < args.length; i++) {
-        const arg = args[i];
-        if (arg === '--help' || arg === '-h') {
-            parsed.help = true;
-            continue;
-        }
-        if (arg.startsWith('--')) {
-            const eqIndex = arg.indexOf('=');
-            let key;
-            let value;
-            if (eqIndex !== -1) {
-                key = arg.slice(2, eqIndex);
-                value = arg.slice(eqIndex + 1);
-            }
-            else {
-                key = arg.slice(2);
-                if (key === 'prompt') {
-                    // --prompt can be multiple
-                    value = args[++i] || '';
-                    parsed.prompts.push(value);
-                    continue;
-                }
-                value = args[++i] || '';
-            }
-            switch (key) {
-                case 'input':
-                case 'content-name':
-                    parsed.contentName = value;
-                    break;
-                case 'project-dir':
-                    parsed.projectDir = value;
-                    break;
-                case 'env-file':
-                    parsed.envFile = value;
-                    break;
-                case 'api-url':
-                    parsed.apiUrl = value;
-                    break;
-                case 'api-key':
-                    parsed.apiKey = value;
-                    break;
-                case 'prompts-file':
-                    parsed.promptsFile = value;
-                    break;
-                case 'image-model':
-                    parsed.imageModel = value;
-                    break;
-                case 'aspect-ratio':
-                    parsed.aspectRatio = value;
-                    break;
-                case 'image-size':
-                case 'size':
-                    parsed.imageSize = value;
-                    break;
-                case 'quality':
-                    parsed.quality = value;
-                    break;
-                case 'style':
-                    parsed.style = value;
-                    break;
-            }
-        }
-        else if (!parsed.contentName && !arg.startsWith('-')) {
-            // positional: content name
-            parsed.contentName = arg;
-        }
-    }
-    return parsed;
-}
+import { UserInputError, createToolRunner } from '../../../tool-utils/dist/index.js';
 function sanitizeComponent(name, fallback) {
     return name
         .replace(/[\\/:*?"<>|]+/g, '_')
@@ -163,7 +79,7 @@ function parsePromptEntries(raw) {
         if (typeof item === 'string') {
             const prompt = item.trim();
             if (!prompt)
-                throw new Error(`Empty prompt at index ${i}`);
+                throw new UserInputError(`Empty prompt at index ${i}`);
             items.push({ title: `scene-${i + 1}`, prompt });
         }
         else if (item && typeof item === 'object') {
@@ -171,15 +87,15 @@ function parsePromptEntries(raw) {
             const prompt = String(obj.prompt || '').trim();
             const title = String(obj.title || `scene-${i + 1}`).trim() || `scene-${i + 1}`;
             if (!prompt)
-                throw new Error(`Empty prompt in object at index ${i}`);
+                throw new UserInputError(`Empty prompt in object at index ${i}`);
             items.push({ title, prompt });
         }
         else {
-            throw new Error(`Invalid item type at index ${i}: expected string or object`);
+            throw new UserInputError(`Invalid item type at index ${i}: expected string or object`);
         }
     }
     if (items.length === 0)
-        throw new Error('No prompts found.');
+        throw new UserInputError('No prompts found.');
     return items;
 }
 function parsePromptsFile(filePath) {
@@ -190,7 +106,7 @@ function parsePromptsFile(filePath) {
     if (raw && typeof raw === 'object') {
         const scenes = raw.scenes;
         if (!Array.isArray(scenes) || scenes.length === 0) {
-            throw new Error('Object mode requires a top-level "scenes" array.');
+            throw new UserInputError('Object mode requires a top-level "scenes" array.');
         }
         const characters = {};
         if (Array.isArray(raw.characters)) {
@@ -213,12 +129,12 @@ function parsePromptsFile(filePath) {
         for (let si = 0; si < scenes.length; si++) {
             const scene = scenes[si];
             if (!scene || typeof scene !== 'object') {
-                throw new Error(`Invalid scene at index ${si}: expected object.`);
+                throw new UserInputError(`Invalid scene at index ${si}: expected object.`);
             }
             const title = String(scene.title || `scene-${si + 1}`).trim() || `scene-${si + 1}`;
             const description = String(scene.description || '').trim();
             if (!description)
-                throw new Error(`Scene ${si}: 'description' is required.`);
+                throw new UserInputError(`Scene ${si}: 'description' is required.`);
             let promptPayload = {
                 scene_title: title,
                 description,
@@ -248,44 +164,56 @@ function parsePromptsFile(filePath) {
         }
         return items;
     }
-    throw new Error('Top-level JSON must be an array or an object.');
+    throw new UserInputError('Top-level JSON must be an array or an object.');
 }
-export async function generateStoryboardImagesHandler(args, context) {
-    const stdout = context.stdout || process.stdout;
-    const stderr = context.stderr || process.stderr;
-    try {
-        const opts = parseArgs(args);
-        if (opts.help || (!opts.contentName)) {
-            stdout.write(`Usage: apltk generate-storyboard-images --input <name> [options]
-
-Generate storyboard images from prompts via OpenAI-compatible API.
-
-Options:
-  --input, --content-name <name>  Output subfolder name under pictures/
-  --project-dir <path>            Project root (default: .)
-  --env-file <path>               Path to .env file
-  --api-url <url>                 API base URL for /images/generations
-  --api-key <key>                 API key
-  --prompts-file <path>           JSON file with prompt entries
-  --prompt <text>                 Image prompt (repeatable)
-  --image-model <model>           Image model (default: gpt-image-1)
-  --aspect-ratio <ratio>          Aspect ratio, e.g. 16:9
-  --image-size <size>             Image size, e.g. 1024x768
-  --quality <q>                   Image quality
-  --style <style>                 Image style
-
-Either --prompts-file or at least one --prompt is required.
-`);
-            return opts.contentName ? 1 : 0;
+const schema = {
+    options: {
+        'input': { type: 'string' },
+        'content-name': { type: 'string' },
+        'project-dir': { type: 'string', default: '.' },
+        'env-file': { type: 'string' },
+        'api-url': { type: 'string' },
+        'api-key': { type: 'string' },
+        'prompts-file': { type: 'string' },
+        'prompt': { type: 'string', multiple: true },
+        'image-model': { type: 'string' },
+        'aspect-ratio': { type: 'string' },
+        'image-size': { type: 'string' },
+        'size': { type: 'string' },
+        'quality': { type: 'string' },
+        'style': { type: 'string' },
+    },
+    allowPositionals: true,
+    usage: 'apltk generate-storyboard-images --input <name> [options]',
+    description: 'Generate storyboard images from prompts via OpenAI-compatible API.',
+    handler: async (values, positionals, context) => {
+        const stdout = context.stdout || process.stdout;
+        const stderr = context.stderr || process.stderr;
+        const contentName = values['input'] ||
+            values['content-name'] ||
+            positionals[0] ||
+            null;
+        const projectDir = values['project-dir'] || '.';
+        const envFile = values['env-file'] ?? null;
+        const apiUrl = values['api-url'] ?? null;
+        const apiKey = values['api-key'] ?? null;
+        const promptsFile = values['prompts-file'] ?? null;
+        const prompts = values['prompt'] || [];
+        const imageModel = values['image-model'] ?? null;
+        const aspectRatio = values['aspect-ratio'] ?? null;
+        const imageSize = values['image-size'] || values['size'] || null;
+        const quality = values['quality'] ?? null;
+        const style = values['style'] ?? null;
+        if (!contentName) {
+            throw new UserInputError('--input or --content-name is required.');
         }
-        const projectDir = path.resolve(opts.projectDir);
-        const contentName = opts.contentName;
+        const resolvedProjectDir = path.resolve(projectDir);
         // Resolve API config
         const __dirname = path.dirname(fileURLToPath(import.meta.url));
         const sourceRoot = context.sourceRoot || path.resolve(__dirname, '..', '..');
         // Try loading env file
-        const envFilePath = opts.envFile
-            ? path.resolve(opts.envFile)
+        const envFilePath = envFile
+            ? path.resolve(envFile)
             : path.join(sourceRoot, 'openai-text-to-image-storyboard', '.env');
         if (fs.existsSync(envFilePath)) {
             const envContent = fs.readFileSync(envFilePath, 'utf-8');
@@ -309,60 +237,58 @@ Either --prompts-file or at least one --prompt is required.
                 }
             }
         }
-        const apiUrl = opts.apiUrl || process.env.OPENAI_API_URL || '';
-        const apiKey = opts.apiKey || process.env.OPENAI_API_KEY || '';
-        const imageModel = opts.imageModel || process.env.OPENAI_IMAGE_MODEL || 'gpt-image-1';
-        const aspectRatio = opts.aspectRatio || process.env.OPENAI_IMAGE_RATIO || process.env.OPENAI_IMAGE_ASPECT_RATIO || null;
-        const imageSize = opts.imageSize || process.env.OPENAI_IMAGE_SIZE || null;
-        const quality = opts.quality || process.env.OPENAI_IMAGE_QUALITY || null;
-        const style = opts.style || process.env.OPENAI_IMAGE_STYLE || null;
-        if (!apiUrl) {
-            stderr.write('Error: Missing API URL. Set --api-url or OPENAI_API_URL.\n');
-            return 1;
+        const resolvedApiUrl = apiUrl || process.env.OPENAI_API_URL || '';
+        const resolvedApiKey = apiKey || process.env.OPENAI_API_KEY || '';
+        const resolvedImageModel = imageModel || process.env.OPENAI_IMAGE_MODEL || 'gpt-image-1';
+        const resolvedAspectRatio = aspectRatio || process.env.OPENAI_IMAGE_RATIO || process.env.OPENAI_IMAGE_ASPECT_RATIO || null;
+        const resolvedImageSize = imageSize || process.env.OPENAI_IMAGE_SIZE || null;
+        const resolvedQuality = quality || process.env.OPENAI_IMAGE_QUALITY || null;
+        const resolvedStyle = style || process.env.OPENAI_IMAGE_STYLE || null;
+        if (!resolvedApiUrl) {
+            throw new UserInputError('Missing API URL. Set --api-url or OPENAI_API_URL.');
         }
-        if (!apiKey) {
-            stderr.write('Error: Missing API key. Set --api-key or OPENAI_API_KEY.\n');
-            return 1;
+        if (!resolvedApiKey) {
+            throw new UserInputError('Missing API key. Set --api-key or OPENAI_API_KEY.');
         }
         // Build prompt items
         let promptItems;
-        if (opts.promptsFile) {
-            promptItems = parsePromptsFile(path.resolve(opts.promptsFile));
+        if (promptsFile) {
+            promptItems = parsePromptsFile(path.resolve(promptsFile));
         }
-        else if (opts.prompts.length > 0) {
-            promptItems = opts.prompts.map((p, i) => ({ title: `scene-${i + 1}`, prompt: p.trim() }));
+        else if (prompts.length > 0) {
+            promptItems = prompts.map((p, i) => ({ title: `scene-${i + 1}`, prompt: p.trim() }));
         }
         else {
-            stderr.write('Error: Either --prompts-file or at least one --prompt is required.\n');
-            return 1;
+            throw new UserInputError('Either --prompts-file or at least one --prompt is required.');
         }
         if (promptItems.length === 0) {
-            stderr.write('Error: No prompts provided.\n');
-            return 1;
+            throw new UserInputError('No prompts provided.');
         }
-        const outputDir = path.join(projectDir, 'pictures', sanitizeComponent(contentName, 'untitled-content'));
+        const outputDir = path.join(resolvedProjectDir, 'pictures', sanitizeComponent(contentName, 'untitled-content'));
         fs.mkdirSync(outputDir, { recursive: true });
         const records = [];
+        let failures = 0;
         for (let i = 0; i < promptItems.length; i++) {
             const item = promptItems[i];
             const titleSlug = sanitizeComponent(item.title, `scene-${i + 1}`);
             const imagePath = uniquePath(path.join(outputDir, `${String(i + 1).padStart(2, '0')}_${titleSlug}.png`));
             const payload = {
-                model: imageModel,
+                model: resolvedImageModel,
                 prompt: item.prompt,
             };
-            if (aspectRatio)
-                payload.aspect_ratio = aspectRatio;
-            if (imageSize)
-                payload.size = imageSize;
-            if (quality)
-                payload.quality = quality;
-            if (style)
-                payload.style = style;
-            const response = await postJson(apiUrl, '/images/generations', apiKey, payload);
+            if (resolvedAspectRatio)
+                payload.aspect_ratio = resolvedAspectRatio;
+            if (resolvedImageSize)
+                payload.size = resolvedImageSize;
+            if (resolvedQuality)
+                payload.quality = resolvedQuality;
+            if (resolvedStyle)
+                payload.style = resolvedStyle;
+            const response = await postJson(resolvedApiUrl, '/images/generations', resolvedApiKey, payload);
             const data = response.data;
             if (!Array.isArray(data) || data.length === 0) {
-                stderr.write(`Error: No image data returned for prompt ${i + 1}.\n`);
+                stderr.write(`No image data returned for prompt ${i + 1}.\n`);
+                failures++;
                 continue;
             }
             const first = data[0];
@@ -374,7 +300,8 @@ Either --prompts-file or at least one --prompt is required.
                 imageBytes = await fetchBinary(first.url);
             }
             else {
-                stderr.write(`Error: Image payload missing b64_json/url for prompt ${i + 1}.\n`);
+                stderr.write(`Image payload missing b64_json/url for prompt ${i + 1}.\n`);
+                failures++;
                 continue;
             }
             fs.writeFileSync(imagePath, imageBytes);
@@ -393,29 +320,27 @@ Either --prompts-file or at least one --prompt is required.
         // Write summary
         const summary = {
             content_name: contentName,
-            project_dir: projectDir,
+            project_dir: resolvedProjectDir,
             output_dir: outputDir,
-            image_model: imageModel,
+            image_model: resolvedImageModel,
             images: records,
         };
-        if (aspectRatio)
-            summary.aspect_ratio = aspectRatio;
-        if (imageSize)
-            summary.image_size = imageSize;
+        if (resolvedAspectRatio)
+            summary.aspect_ratio = resolvedAspectRatio;
+        if (resolvedImageSize)
+            summary.image_size = resolvedImageSize;
         const summaryPath = path.join(outputDir, 'storyboard.json');
         fs.writeFileSync(summaryPath, JSON.stringify(summary, null, 2), 'utf-8');
         stdout.write(`[OK] Wrote plan to ${summaryPath}\n`);
-        return 0;
-    }
-    catch (err) {
-        const msg = err instanceof Error ? err.message : 'Unknown error';
-        stderr.write(`Error: ${msg}\n`);
-        return 1;
-    }
-}
+        if (failures > 0) {
+            stderr.write(`Warning: ${failures} out of ${promptItems.length} prompts failed to generate images.\n`);
+        }
+        return failures > 0 ? 1 : 0;
+    },
+};
 export const tool = {
     name: 'generate-storyboard-images',
     category: 'media',
     description: 'Generate storyboard images from prompts via OpenAI-compatible API.',
-    handler: generateStoryboardImagesHandler,
+    handler: createToolRunner(schema),
 };