@lovelybunch/cli 1.0.75-alpha.0 → 1.0.75-alpha.10

package/dist/commands/resource.js

@@ -3,14 +3,80 @@ import chalk from 'chalk';
 import ora from 'ora';
 import fetch, { FormData, fileFromSync } from 'node-fetch';
 import { promises as fs } from 'fs';
-import { writeFileSync, unlinkSync, mkdtempSync } from 'fs';
+import { writeFileSync, unlinkSync, mkdtempSync, existsSync } from 'fs';
 import path from 'path';
 import { tmpdir } from 'os';
+import inquirer from 'inquirer';
 // ============================================================================
 // Shared Utilities
 // ============================================================================
 const DEFAULT_API_BASE = 'http://localhost';
 const GENERATION_TIMEOUT = 300_000; // 5 minutes
+/**
+ * Determine MIME type from file extension
+ */
+function getMimeType(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    const mimeTypes = {
+        // Images
+        '.png': 'image/png',
+        '.jpg': 'image/jpeg',
+        '.jpeg': 'image/jpeg',
+        '.gif': 'image/gif',
+        '.webp': 'image/webp',
+        '.svg': 'image/svg+xml',
+        '.bmp': 'image/bmp',
+        '.ico': 'image/x-icon',
+        '.tiff': 'image/tiff',
+        '.tif': 'image/tiff',
+        '.avif': 'image/avif',
+        // Audio
+        '.mp3': 'audio/mpeg',
+        '.wav': 'audio/wav',
+        '.ogg': 'audio/ogg',
+        '.flac': 'audio/flac',
+        '.aac': 'audio/aac',
+        '.m4a': 'audio/mp4',
+        '.wma': 'audio/x-ms-wma',
+        '.opus': 'audio/opus',
+        // Video
+        '.mp4': 'video/mp4',
+        '.webm': 'video/webm',
+        '.mov': 'video/quicktime',
+        '.avi': 'video/x-msvideo',
+        '.mkv': 'video/x-matroska',
+        '.flv': 'video/x-flv',
+        '.wmv': 'video/x-ms-wmv',
+        '.m4v': 'video/mp4',
+        // Documents
+        '.pdf': 'application/pdf',
+        '.json': 'application/json',
+        '.xml': 'application/xml',
+        '.csv': 'text/csv',
+        '.txt': 'text/plain',
+        '.md': 'text/markdown',
+        '.html': 'text/html',
+        '.css': 'text/css',
+        '.js': 'text/javascript',
+        '.ts': 'text/typescript',
+        // Archives
+        '.zip': 'application/zip',
+        '.gz': 'application/gzip',
+        '.tar': 'application/x-tar',
+    };
+    return mimeTypes[ext] || 'application/octet-stream';
+}
+/**
+ * Format file size in bytes to human-readable string
+ */
+function formatFileSize(bytes) {
+    if (bytes === 0)
+        return '0 B';
+    const k = 1024;
+    const sizes = ['B', 'KB', 'MB', 'GB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
+}
 /**
  * Check if the Coconut server is running
  */
@@ -557,12 +623,515 @@ configured via "nut config set replicate <token>".
     }
 });
 // ============================================================================
+// List Resources Command
+// ============================================================================
+const listResourcesCommand = new Command('list')
+    .description('List all resources')
+    .alias('ls')
+    .option('-s, --search <query>', 'Filter resources by name, description, or tags')
+    .option('--type <type>', 'Filter by media type prefix (e.g. image, audio, video)')
+    .option('--json', 'Output as JSON')
+    .option('--api <url>', 'API server URL', DEFAULT_API_BASE)
+    .addHelpText('after', `
+Examples:
+  $ nut resource list
+  $ nut resource ls --search "logo"
+  $ nut resource list --type image --json
+`)
+    .action(async (options) => {
+    const isHealthy = await checkServerHealth(options.api);
+    if (!isHealthy) {
+        console.error(chalk.red('Error: Coconut server is not running.'));
+        console.error(chalk.yellow('Start the server with "nut serve" or specify a different URL with --api'));
+        process.exit(1);
+    }
+    const spinner = ora('Loading resources...').start();
+    try {
+        const response = await fetch(`${options.api}/api/v1/resources`);
+        const result = await response.json();
+        if (!response.ok || !result.success || !result.data) {
+            spinner.fail('Failed to load resources');
+            process.exit(1);
+        }
+        let resources = result.data;
+        // Apply type filter
+        if (options.type) {
+            const typePrefix = options.type.toLowerCase();
+            resources = resources.filter(r => r.type.toLowerCase().startsWith(typePrefix));
+        }
+        // Apply search filter (client-side, matching frontend behavior)
+        if (options.search) {
+            const query = options.search.toLowerCase();
+            resources = resources.filter(r => r.name.toLowerCase().includes(query) ||
+                r.metadata.description?.toLowerCase().includes(query) ||
+                r.metadata.tags?.some(tag => tag.toLowerCase().includes(query)));
+        }
+        spinner.stop();
+        if (options.json) {
+            console.log(JSON.stringify(resources, null, 2));
+            return;
+        }
+        if (resources.length === 0) {
+            if (options.search || options.type) {
+                console.log(chalk.yellow('No resources found matching your filters.'));
+            }
+            else {
+                console.log(chalk.yellow('No resources found.'));
+                console.log(chalk.gray('Use "nut resource add <file>" to upload one.'));
+            }
+            return;
+        }
+        console.log('\n' + chalk.bold.underline(`Resources (${resources.length})`));
+        console.log();
+        for (const r of resources) {
+            console.log(chalk.cyan('●') + ' ' + chalk.bold(r.name) + chalk.gray(` (${r.id})`));
+            console.log('  ' + chalk.gray(`Type: ${r.type}  Size: ${formatFileSize(r.size)}  Uploaded: ${new Date(r.uploadedAt).toLocaleDateString()}`));
+            if (r.metadata.description) {
+                const desc = r.metadata.description.length > 80
+                    ? r.metadata.description.slice(0, 80).trimEnd() + '...'
+                    : r.metadata.description;
+                console.log('  ' + chalk.gray(`Description: ${desc}`));
+            }
+            if (r.metadata.tags && r.metadata.tags.length > 0) {
+                console.log('  ' + chalk.gray(`Tags: ${r.metadata.tags.join(', ')}`));
+            }
+            console.log();
+        }
+    }
+    catch (error) {
+        spinner.fail('Failed to load resources');
+        console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+        process.exit(1);
+    }
+});
+// ============================================================================
+// Get Resource Command
+// ============================================================================
+const getResourceCommand = new Command('get')
+    .description('Get details for a specific resource')
+    .alias('show')
+    .argument('<id>', 'Resource ID')
+    .option('--json', 'Output as JSON')
+    .option('-o, --output <path>', 'Download resource file to local path')
+    .option('--api <url>', 'API server URL', DEFAULT_API_BASE)
+    .addHelpText('after', `
+Examples:
+  $ nut resource get res-1234567890-abc123def
+  $ nut resource show res-1234567890-abc123def --json
+  $ nut resource get res-1234567890-abc123def -o ./downloaded-file.png
+`)
+    .action(async (id, options) => {
+    const isHealthy = await checkServerHealth(options.api);
+    if (!isHealthy) {
+        console.error(chalk.red('Error: Coconut server is not running.'));
+        console.error(chalk.yellow('Start the server with "nut serve" or specify a different URL with --api'));
+        process.exit(1);
+    }
+    const spinner = ora('Loading resource...').start();
+    try {
+        // Fetch all resources and find by ID (the list endpoint returns full metadata)
+        const response = await fetch(`${options.api}/api/v1/resources`);
+        const result = await response.json();
+        if (!response.ok || !result.success || !result.data) {
+            spinner.fail('Failed to load resources');
+            process.exit(1);
+        }
+        const resource = result.data.find(r => r.id === id);
+        if (!resource) {
+            spinner.fail(`Resource '${id}' not found`);
+            process.exit(1);
+        }
+        spinner.stop();
+        // Handle download
+        if (options.output) {
+            const downloadSpinner = ora(`Downloading to ${options.output}...`).start();
+            try {
+                await downloadToFile(`${options.api}/api/v1/resources/${id}?download=true`, options.output);
+                downloadSpinner.succeed(`Downloaded to ${chalk.cyan(path.resolve(options.output))}`);
+            }
+            catch (error) {
+                downloadSpinner.fail('Failed to download');
+                console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+                process.exit(1);
+            }
+            return;
+        }
+        if (options.json) {
+            console.log(JSON.stringify(resource, null, 2));
+            return;
+        }
+        // Display formatted details
+        console.log();
+        console.log(chalk.bold.cyan(resource.name));
+        console.log();
+        console.log(`  ${chalk.gray('ID:')}         ${resource.id}`);
+        console.log(`  ${chalk.gray('Type:')}       ${resource.type}`);
+        console.log(`  ${chalk.gray('Size:')}       ${formatFileSize(resource.size)}`);
+        console.log(`  ${chalk.gray('Uploaded:')}   ${new Date(resource.uploadedAt).toLocaleString()}`);
+        if (resource.metadata.tags && resource.metadata.tags.length > 0) {
+            console.log(`  ${chalk.gray('Tags:')}       ${resource.metadata.tags.join(', ')}`);
+        }
+        if (resource.metadata.description) {
+            console.log(`  ${chalk.gray('Description:')} ${resource.metadata.description}`);
+        }
+        if (resource.metadata.generationPrompt) {
+            console.log(`  ${chalk.gray('Prompt:')}     ${resource.metadata.generationPrompt}`);
+        }
+        if (resource.metadata.generationModel) {
+            console.log(`  ${chalk.gray('Model:')}      ${resource.metadata.generationModel}`);
+        }
+        console.log();
+        console.log(chalk.bold('Paths:'));
+        console.log(`  ${chalk.gray('File:')}       .nut/resources/files/${resource.path}`);
+        if (resource.thumbnailPath) {
+            console.log(`  ${chalk.gray('Thumbnail:')} .nut/resources/thumbnails/${resource.thumbnailPath}`);
+        }
+        console.log(`  ${chalk.gray('Metadata:')}  .nut/resources/metadata/${resource.id}.json`);
+        console.log();
+    }
+    catch (error) {
+        spinner.fail('Failed to load resource');
+        console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+        process.exit(1);
+    }
+});
+// ============================================================================
+// Add Resource Command
+// ============================================================================
+/**
+ * Check if a string looks like a URL
+ */
+function isUrl(input) {
+    return /^https?:\/\//i.test(input);
+}
+/**
+ * Download a URL to a temp file and return the temp path, filename, and MIME type.
+ * Caller is responsible for cleaning up the temp file/directory.
+ */
+async function downloadUrlToTemp(url) {
+    const response = await fetch(url);
+    if (!response.ok) {
+        throw new Error(`Failed to download ${url}: ${response.statusText}`);
+    }
+    const buffer = Buffer.from(await response.arrayBuffer());
+    const contentType = response.headers.get('content-type') || '';
+    // Derive a filename from the URL path, falling back to a timestamp-based name
+    const urlPath = new URL(url).pathname;
+    let fileName = path.basename(urlPath);
+    if (!fileName || fileName === '/' || !path.extname(fileName)) {
+        // Try to infer extension from content-type
+        const extMap = {
+            'image/png': '.png',
+            'image/jpeg': '.jpg',
+            'image/webp': '.webp',
+            'image/gif': '.gif',
+            'image/svg+xml': '.svg',
+            'audio/mpeg': '.mp3',
+            'audio/wav': '.wav',
+            'audio/ogg': '.ogg',
+            'video/mp4': '.mp4',
+            'video/webm': '.webm',
+            'application/pdf': '.pdf',
+            'application/json': '.json',
+        };
+        const ext = Object.entries(extMap).find(([ct]) => contentType.includes(ct))?.[1] || '.bin';
+        fileName = `downloaded-${Date.now()}${ext}`;
+    }
+    const mimeType = contentType.split(';')[0].trim() || getMimeType(fileName);
+    const tempDir = mkdtempSync(path.join(tmpdir(), 'coconut-add-'));
+    const tempFilePath = path.join(tempDir, fileName);
+    writeFileSync(tempFilePath, buffer);
+    return { tempDir, tempFilePath, fileName, mimeType };
+}
+const addResourceCommand = new Command('add')
+    .description('Upload a file or URL to resources')
+    .alias('upload')
+    .argument('<file...>', 'File path(s) or URL(s) to upload')
+    .option('-t, --tags <tags>', 'Comma-separated tags')
+    .option('-d, --description <text>', 'Description of the resource')
+    .option('--json', 'Output as JSON')
+    .option('--api <url>', 'API server URL', DEFAULT_API_BASE)
+    .addHelpText('after', `
+Examples:
+  $ nut resource add ./screenshot.png
+  $ nut resource upload ./logo.svg -t "branding, logo" -d "Company logo"
+  $ nut resource add ./image1.png ./image2.png -t "batch"
+  $ nut resource add https://example.com/photo.png
+  $ nut resource add ./data.json --json
+`)
+    .action(async (files, options) => {
+    const isHealthy = await checkServerHealth(options.api);
+    if (!isHealthy) {
+        console.error(chalk.red('Error: Coconut server is not running.'));
+        console.error(chalk.yellow('Start the server with "nut serve" or specify a different URL with --api'));
+        process.exit(1);
+    }
+    const results = [];
+    for (const filePath of files) {
+        let localPath;
+        let fileName;
+        let mimeType;
+        let tempDir = null;
+        if (isUrl(filePath)) {
+            // ── URL: download to temp file first ──
+            const spinner = ora(`Downloading ${filePath}...`).start();
+            try {
+                const downloaded = await downloadUrlToTemp(filePath);
+                tempDir = downloaded.tempDir;
+                localPath = downloaded.tempFilePath;
+                fileName = downloaded.fileName;
+                mimeType = downloaded.mimeType;
+                spinner.succeed(`Downloaded ${chalk.cyan(fileName)}`);
+            }
+            catch (error) {
+                spinner.fail('Failed to download URL');
+                console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+                process.exit(1);
+            }
+        }
+        else {
+            // ── Local file ──
+            localPath = path.resolve(filePath);
+            if (!existsSync(localPath)) {
+                console.error(chalk.red(`File not found: ${localPath}`));
+                process.exit(1);
+            }
+            fileName = path.basename(filePath);
+            mimeType = getMimeType(localPath);
+        }
+        const uploadSpinner = ora(`Uploading ${fileName}...`).start();
+        try {
+            const formData = new FormData();
+            const file = fileFromSync(localPath, mimeType);
+            formData.set('file', file, fileName);
+            if (options.tags) {
+                formData.set('tags', options.tags);
+            }
+            if (options.description) {
+                formData.set('description', options.description);
+            }
+            const response = await fetch(`${options.api}/api/v1/resources`, {
+                method: 'POST',
+                body: formData,
+            });
+            const result = await response.json();
+            if (!response.ok || !result.success || !result.data) {
+                const errorMessage = result.error?.message || 'Upload failed';
+                uploadSpinner.fail(`Failed to upload ${fileName}`);
+                console.error(chalk.red(errorMessage));
+                process.exit(1);
+            }
+            const resource = result.data;
+            results.push(resource);
+            uploadSpinner.succeed(`Uploaded ${chalk.cyan(resource.name)}`);
+            if (!options.json) {
+                console.log();
+                console.log(`  ${chalk.gray('ID:')}        ${resource.id}`);
+                console.log(`  ${chalk.gray('Name:')}      ${resource.name}`);
+                console.log(`  ${chalk.gray('Type:')}      ${resource.type}`);
+                console.log(`  ${chalk.gray('Size:')}      ${formatFileSize(resource.size)}`);
+                if (resource.metadata.tags && resource.metadata.tags.length > 0) {
+                    console.log(`  ${chalk.gray('Tags:')}      ${resource.metadata.tags.join(', ')}`);
+                }
+                if (resource.metadata.description) {
+                    console.log(`  ${chalk.gray('Desc:')}      ${resource.metadata.description}`);
+                }
+                console.log(`  ${chalk.gray('File:')}      .nut/resources/files/${resource.path}`);
+                if (resource.thumbnailPath) {
+                    console.log(`  ${chalk.gray('Thumbnail:')} .nut/resources/thumbnails/${resource.thumbnailPath}`);
+                }
+                console.log(`  ${chalk.gray('Metadata:')}  .nut/resources/metadata/${resource.id}.json`);
+                console.log();
+            }
+        }
+        catch (error) {
+            uploadSpinner.fail(`Failed to upload ${fileName}`);
+            console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+            process.exit(1);
+        }
+        finally {
+            // Clean up temp directory if we downloaded from a URL
+            if (tempDir) {
+                try {
+                    unlinkSync(localPath);
+                    fs.rmdir(tempDir).catch(() => { });
+                }
+                catch {
+                    // Ignore cleanup errors
+                }
+            }
+        }
+    }
+    if (options.json) {
+        console.log(JSON.stringify(results.length === 1 ? results[0] : results, null, 2));
+    }
+});
+// ============================================================================
+// Update Resource Command
+// ============================================================================
+const updateResourceCommand = new Command('update')
+    .description('Update resource metadata (tags, description)')
+    .argument('<id>', 'Resource ID to update')
+    .option('-t, --tags <tags>', 'Comma-separated tags (replaces existing)')
+    .option('-d, --description <text>', 'Description (replaces existing)')
+    .option('--json', 'Output as JSON')
+    .option('--api <url>', 'API server URL', DEFAULT_API_BASE)
+    .addHelpText('after', `
+Examples:
+  $ nut resource update res-123 -t "logo, branding"
+  $ nut resource update res-123 -d "Updated description"
+  $ nut resource update res-123 -t "new-tag" -d "New desc" --json
+`)
+    .action(async (id, options) => {
+    if (!options.tags && !options.description) {
+        console.error(chalk.red('Error: At least one of --tags or --description must be provided.'));
+        process.exit(1);
+    }
+    const isHealthy = await checkServerHealth(options.api);
+    if (!isHealthy) {
+        console.error(chalk.red('Error: Coconut server is not running.'));
+        console.error(chalk.yellow('Start the server with "nut serve" or specify a different URL with --api'));
+        process.exit(1);
+    }
+    const spinner = ora('Updating resource...').start();
+    try {
+        const metadata = {};
+        if (options.tags !== undefined) {
+            metadata.tags = options.tags.split(',').map(t => t.trim());
+        }
+        if (options.description !== undefined) {
+            metadata.description = options.description;
+        }
+        const response = await fetch(`${options.api}/api/v1/resources/${id}`, {
+            method: 'PUT',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ metadata }),
+        });
+        const result = await response.json();
+        if (!response.ok || !result.success || !result.data) {
+            const errorMessage = result.error?.message || 'Update failed';
+            spinner.fail('Failed to update resource');
+            console.error(chalk.red(errorMessage));
+            process.exit(1);
+        }
+        const resource = result.data;
+        spinner.succeed(`Updated ${chalk.cyan(resource.name)}`);
+        if (options.json) {
+            console.log(JSON.stringify(resource, null, 2));
+        }
+        else {
+            console.log();
+            console.log(`  ${chalk.gray('ID:')}          ${resource.id}`);
+            console.log(`  ${chalk.gray('Name:')}        ${resource.name}`);
+            if (resource.metadata.tags && resource.metadata.tags.length > 0) {
+                console.log(`  ${chalk.gray('Tags:')}        ${resource.metadata.tags.join(', ')}`);
+            }
+            if (resource.metadata.description) {
+                console.log(`  ${chalk.gray('Description:')} ${resource.metadata.description}`);
+            }
+            console.log();
+        }
+    }
+    catch (error) {
+        spinner.fail('Failed to update resource');
+        console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+        process.exit(1);
+    }
+});
+// ============================================================================
+// Delete Resource Command
+// ============================================================================
+const deleteResourceCommand = new Command('delete')
+    .description('Delete a resource')
+    .alias('rm')
+    .argument('<id>', 'Resource ID to delete')
+    .option('-f, --force', 'Skip confirmation prompt')
+    .option('--json', 'Output as JSON')
+    .option('--api <url>', 'API server URL', DEFAULT_API_BASE)
+    .addHelpText('after', `
+Examples:
+  $ nut resource delete res-1234567890-abc123def
+  $ nut resource rm res-1234567890-abc123def --force
+  $ nut resource delete res-123 --json
+`)
+    .action(async (id, options) => {
+    const isHealthy = await checkServerHealth(options.api);
+    if (!isHealthy) {
+        console.error(chalk.red('Error: Coconut server is not running.'));
+        console.error(chalk.yellow('Start the server with "nut serve" or specify a different URL with --api'));
+        process.exit(1);
+    }
+    // Fetch resource details first to show name in confirmation
+    const listSpinner = ora('Loading resource...').start();
+    let resourceName = id;
+    try {
+        const listResponse = await fetch(`${options.api}/api/v1/resources`);
+        const listResult = await listResponse.json();
+        if (listResult.success && listResult.data) {
+            const found = listResult.data.find(r => r.id === id);
+            if (found) {
+                resourceName = found.name;
+            }
+            else {
+                listSpinner.fail(`Resource '${id}' not found`);
+                process.exit(1);
+            }
+        }
+        listSpinner.stop();
+    }
+    catch {
+        listSpinner.stop();
+        // Proceed anyway; the delete endpoint will 404 if not found
+    }
+    // Confirm deletion
+    if (!options.force) {
+        const { confirm } = await inquirer.prompt([{
+                type: 'confirm',
+                name: 'confirm',
+                message: `Delete resource "${resourceName}" (${id})?`,
+                default: false,
+            }]);
+        if (!confirm) {
+            console.log(chalk.yellow('Operation cancelled.'));
+            return;
+        }
+    }
+    const spinner = ora('Deleting resource...').start();
+    try {
+        const response = await fetch(`${options.api}/api/v1/resources/${id}`, {
+            method: 'DELETE',
+        });
+        const result = await response.json();
+        if (!response.ok || !result.success) {
+            const errorMessage = result.error?.message || 'Delete failed';
+            spinner.fail('Failed to delete resource');
+            console.error(chalk.red(errorMessage));
+            process.exit(1);
+        }
+        spinner.succeed(`Deleted ${chalk.cyan(resourceName)}`);
+        if (options.json) {
+            console.log(JSON.stringify({ success: true, id, name: resourceName }, null, 2));
+        }
+    }
+    catch (error) {
+        spinner.fail('Failed to delete resource');
+        console.error(chalk.red(error instanceof Error ? error.message : 'Unknown error'));
+        process.exit(1);
+    }
+});
+// ============================================================================
 // Parent Resource Command
 // ============================================================================
 const resource = new Command('resource')
     .description('Manage and generate resources (images, audio, video)')
     .alias('resources');
-// Add subcommands
+// Add CRUD subcommands
+resource.addCommand(listResourcesCommand);
+resource.addCommand(getResourceCommand);
+resource.addCommand(addResourceCommand);
+resource.addCommand(updateResourceCommand);
+resource.addCommand(deleteResourceCommand);
+// Add generation subcommands
 resource.addCommand(generateImageCommand);
 resource.addCommand(generateAudioCommand);
 resource.addCommand(generateVideoCommand);