@aws/ml-container-creator 0.2.1 → 0.2.2

package/src/lib/registry-command-handler.js

@@ -18,15 +18,15 @@
  *   search [--model, --architecture, --backend, --instance-type]
  */
 
-import os from 'node:os'
-import path from 'node:path'
-import { readFileSync } from 'node:fs'
-import { execSync } from 'node:child_process'
-import { fileURLToPath } from 'node:url'
-import DeploymentRegistry, { reconstructReplayFlags } from './deployment-registry.js'
+import os from 'node:os';
+import path from 'node:path';
+import { readFileSync } from 'node:fs';
+import { execSync } from 'node:child_process';
+import { fileURLToPath } from 'node:url';
+import DeploymentRegistry, { reconstructReplayFlags } from './deployment-registry.js';
 
-const PERSONAL_REGISTRY_PATH = path.join(os.homedir(), '.ml-container-creator', 'registry.json')
-const PROJECT_REGISTRY_PATH = path.join(process.cwd(), '.ml-container-creator', 'registry.json')
+const PERSONAL_REGISTRY_PATH = path.join(os.homedir(), '.ml-container-creator', 'registry.json');
+const PROJECT_REGISTRY_PATH = path.join(process.cwd(), '.ml-container-creator', 'registry.json');
 
 export default class RegistryCommandHandler {
     constructor() {
@@ -40,41 +40,41 @@ export default class RegistryCommandHandler {
      */
     async handle(args, options) {
         if (args.length === 0) {
-            this._showRegistryHelp()
-            return
+            this._showRegistryHelp();
+            return;
         }
 
-        const subcommand = args[0].toLowerCase()
+        const subcommand = args[0].toLowerCase();
 
         switch (subcommand) {
-            case 'log':
-                await this._handleLog(options)
-                break
-            case 'list':
-                this._handleList(options)
-                break
-            case 'get':
-                this._handleGet(args[1])
-                break
-            case 'remove':
-                this._handleRemove(args[1])
-                break
-            case 'replay':
-                await this._handleReplay(args[1], options)
-                break
-            case 'export':
-                this._handleExport(args[1], options)
-                break
-            case 'import':
-                await this._handleImport(args[1], options)
-                break
-            case 'search':
-                this._handleSearch(options)
-                break
-            default:
-                console.log(`Unknown registry subcommand: ${subcommand}`)
-                this._showRegistryHelp()
-                break
+        case 'log':
+            await this._handleLog(options);
+            break;
+        case 'list':
+            this._handleList(options);
+            break;
+        case 'get':
+            this._handleGet(args[1]);
+            break;
+        case 'remove':
+            this._handleRemove(args[1]);
+            break;
+        case 'replay':
+            await this._handleReplay(args[1], options);
+            break;
+        case 'export':
+            this._handleExport(args[1], options);
+            break;
+        case 'import':
+            await this._handleImport(args[1], options);
+            break;
+        case 'search':
+            this._handleSearch(options);
+            break;
+        default:
+            console.log(`Unknown registry subcommand: ${subcommand}`);
+            this._showRegistryHelp();
+            break;
         }
     }
 
@@ -87,12 +87,12 @@ export default class RegistryCommandHandler {
      * @param {object} options - Parsed CLI options from do/register
      */
     async _handleLog(options) {
-        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH
-        const registry = new DeploymentRegistry(registryPath)
+        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH;
+        const registry = new DeploymentRegistry(registryPath);
 
-        const deploymentConfig = options.deploymentConfig || options['deployment-config'] || ''
-        const architecture = options.architecture || ''
-        const backend = options.backend || ''
+        const deploymentConfig = options.deploymentConfig || options['deployment-config'] || '';
+        const architecture = options.architecture || '';
+        const backend = options.backend || '';
 
         const entry = {
             timestamp: new Date().toISOString(),
@@ -125,27 +125,27 @@ export default class RegistryCommandHandler {
                 source: 'local',
                 importedFrom: null
             }
-        }
+        };
 
         // Parse parameters from JSON string if provided
         if (options.parameters) {
             try {
                 entry.configuration.parameters = typeof options.parameters === 'string'
                     ? JSON.parse(options.parameters)
-                    : options.parameters
+                    : options.parameters;
             } catch (err) {
-                console.log(`Warning: Could not parse parameters JSON: ${err.message}`)
-                entry.configuration.parameters = {}
+                console.log(`Warning: Could not parse parameters JSON: ${err.message}`);
+                entry.configuration.parameters = {};
             }
         }
 
         try {
-            const id = registry.add(entry)
-            console.log(`✅ Deployment entry logged successfully.`)
-            console.log(`   Entry ID: ${id}`)
-            console.log(`   View details: ml-container-creator registry get ${id}`)
+            const id = registry.add(entry);
+            console.log('✅ Deployment entry logged successfully.');
+            console.log(`   Entry ID: ${id}`);
+            console.log(`   View details: ml-container-creator registry get ${id}`);
         } catch (err) {
-            console.log(`Error logging deployment entry: ${err.message}`)
+            console.log(`Error logging deployment entry: ${err.message}`);
         }
     }
 
@@ -158,34 +158,34 @@ export default class RegistryCommandHandler {
      * @param {object} options - Parsed CLI options
      */
     _handleList(options) {
-        const filters = this._extractFilters(options)
+        const filters = this._extractFilters(options);
 
-        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH)
-        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH)
+        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH);
+        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH);
 
-        const personalEntries = personalRegistry.list(filters).map(e => ({ ...e, _source: 'personal' }))
-        const projectEntries = projectRegistry.list(filters).map(e => ({ ...e, _source: 'project' }))
+        const personalEntries = personalRegistry.list(filters).map(e => ({ ...e, _source: 'personal' }));
+        const projectEntries = projectRegistry.list(filters).map(e => ({ ...e, _source: 'project' }));
 
-        const allEntries = [...personalEntries, ...projectEntries]
+        const allEntries = [...personalEntries, ...projectEntries];
 
         if (allEntries.length === 0) {
-            console.log('No deployment entries found.')
-            console.log('Use "./do/register" after a successful deployment to add an entry.')
-            return
+            console.log('No deployment entries found.');
+            console.log('Use "./do/register" after a successful deployment to add an entry.');
+            return;
         }
 
-        console.log('\nDeployment Registry Entries:\n')
+        console.log('\nDeployment Registry Entries:\n');
         for (const entry of allEntries) {
-            const id = entry.id || '(no id)'
-            const ts = entry.timestamp ? entry.timestamp.slice(0, 19) : '(no timestamp)'
-            const dc = entry.deployment?.deploymentConfig || '(none)'
-            const mn = entry.model?.modelName || '(none)'
-            const it = entry.infrastructure?.instanceType || '(none)'
-            const st = entry.status || '(none)'
-            const src = entry._source === 'project' ? ' [project]' : ''
-            console.log(`  ${id}  ${ts}  ${dc}  ${mn}  ${it}  ${st}${src}`)
+            const id = entry.id || '(no id)';
+            const ts = entry.timestamp ? entry.timestamp.slice(0, 19) : '(no timestamp)';
+            const dc = entry.deployment?.deploymentConfig || '(none)';
+            const mn = entry.model?.modelName || '(none)';
+            const it = entry.infrastructure?.instanceType || '(none)';
+            const st = entry.status || '(none)';
+            const src = entry._source === 'project' ? ' [project]' : '';
+            console.log(`  ${id}  ${ts}  ${dc}  ${mn}  ${it}  ${st}${src}`);
         }
-        console.log('')
+        console.log('');
     }
 
     /**
@@ -197,20 +197,20 @@ export default class RegistryCommandHandler {
      */
     _handleGet(id) {
         if (!id) {
-            console.log('Usage: ml-container-creator registry get <id>')
-            return
+            console.log('Usage: ml-container-creator registry get <id>');
+            return;
         }
 
-        const entry = this._findEntry(id)
+        const entry = this._findEntry(id);
 
         if (!entry) {
-            console.log(`Error: Entry "${id}" not found.`)
-            return
+            console.log(`Error: Entry "${id}" not found.`);
+            return;
         }
 
-        console.log(`\nDeployment Entry: ${id}\n`)
-        console.log(JSON.stringify(entry, null, 2))
-        console.log('')
+        console.log(`\nDeployment Entry: ${id}\n`);
+        console.log(JSON.stringify(entry, null, 2));
+        console.log('');
     }
 
     /**
@@ -222,23 +222,23 @@ export default class RegistryCommandHandler {
      */
     _handleRemove(id) {
         if (!id) {
-            console.log('Usage: ml-container-creator registry remove <id>')
-            return
+            console.log('Usage: ml-container-creator registry remove <id>');
+            return;
         }
 
-        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH)
+        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH);
         if (personalRegistry.remove(id)) {
-            console.log(`✅ Entry "${id}" removed from personal registry.`)
-            return
+            console.log(`✅ Entry "${id}" removed from personal registry.`);
+            return;
         }
 
-        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH)
+        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH);
         if (projectRegistry.remove(id)) {
-            console.log(`✅ Entry "${id}" removed from project registry.`)
-            return
+            console.log(`✅ Entry "${id}" removed from project registry.`);
+            return;
         }
 
-        console.log(`Error: Entry "${id}" not found.`)
+        console.log(`Error: Entry "${id}" not found.`);
     }
 
     /**
@@ -252,19 +252,19 @@ export default class RegistryCommandHandler {
      */
     async _handleReplay(id, options) {
         if (!id) {
-            console.log('Usage: ml-container-creator registry replay <id> [--model-name <name>] [--instance-type <type>] ...')
-            return
+            console.log('Usage: ml-container-creator registry replay <id> [--model-name <name>] [--instance-type <type>] ...');
+            return;
         }
 
-        const entry = this._findEntry(id)
+        const entry = this._findEntry(id);
 
         if (!entry) {
-            console.log(`Error: Entry "${id}" not found.`)
-            return
+            console.log(`Error: Entry "${id}" not found.`);
+            return;
         }
 
         // Build overrides from user-provided CLI options
-        const overrides = {}
+        const overrides = {};
         const overrideMap = {
             'deployment-config': '--deployment-config',
             'deploymentConfig': '--deployment-config',
@@ -275,37 +275,37 @@ export default class RegistryCommandHandler {
             'region': '--region',
             'model-format': '--model-format',
             'modelFormat': '--model-format'
-        }
+        };
 
         for (const [optKey, flagKey] of Object.entries(overrideMap)) {
-            if (options[optKey] != null) {
-                overrides[flagKey] = options[optKey]
+            if (options[optKey] !== null && options[optKey] !== undefined) {
+                overrides[flagKey] = options[optKey];
             }
         }
 
-        const flags = reconstructReplayFlags(entry, overrides)
+        const flags = reconstructReplayFlags(entry, overrides);
 
-        console.log(`\nReplaying deployment entry: ${id}`)
-        console.log(`  Deployment config: ${flags['--deployment-config'] || '(will prompt)'}`)
-        console.log(`  Model name: ${flags['--model-name'] || '(will prompt)'}`)
-        console.log(`  Instance type: ${flags['--instance-type'] || '(will prompt)'}`)
-        console.log(`  Region: ${flags['--region'] || '(will prompt)'}`)
-        console.log('')
+        console.log(`\nReplaying deployment entry: ${id}`);
+        console.log(`  Deployment config: ${flags['--deployment-config'] || '(will prompt)'}`);
+        console.log(`  Model name: ${flags['--model-name'] || '(will prompt)'}`);
+        console.log(`  Instance type: ${flags['--instance-type'] || '(will prompt)'}`);
+        console.log(`  Region: ${flags['--region'] || '(will prompt)'}`);
+        console.log('');
 
-        const flagArgs = []
+        const flagArgs = [];
         for (const [flag, value] of Object.entries(flags)) {
-            flagArgs.push(flag, value)
+            flagArgs.push(flag, value);
         }
 
         // Resolve the CLI script path relative to this module
-        const __filename = fileURLToPath(import.meta.url)
-        const cliPath = path.resolve(path.dirname(__filename), '../../bin/cli.js')
+        const __filename = fileURLToPath(import.meta.url);
+        const cliPath = path.resolve(path.dirname(__filename), '../../bin/cli.js');
 
         try {
-            execSync(`ml-container-creator ${flagArgs.join(' ')}`, { stdio: 'inherit' })
+            execSync(`ml-container-creator ${flagArgs.join(' ')}`, { stdio: 'inherit' });
         } catch {
             // Fallback: invoke via node + script path if binary is not on PATH
-            execSync(`${process.execPath} ${cliPath} ${flagArgs.join(' ')}`, { stdio: 'inherit' })
+            execSync(`${process.execPath} ${cliPath} ${flagArgs.join(' ')}`, { stdio: 'inherit' });
         }
     }
 
@@ -318,22 +318,22 @@ export default class RegistryCommandHandler {
      * @param {object} options - Parsed CLI options
      */
     _handleExport(id, options) {
-        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH
-        const registry = new DeploymentRegistry(registryPath)
+        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH;
+        const registry = new DeploymentRegistry(registryPath);
 
-        const exportOptions = {}
+        const exportOptions = {};
         if (options.status) {
-            exportOptions.status = options.status
+            exportOptions.status = options.status;
         }
 
-        const result = registry.exportEntries(id || null, exportOptions)
+        const result = registry.exportEntries(id || null, exportOptions);
 
         if (result.entries.length === 0) {
-            console.log('No entries to export.')
-            return
+            console.log('No entries to export.');
+            return;
         }
 
-        console.log(JSON.stringify(result, null, 2))
+        console.log(JSON.stringify(result, null, 2));
     }
 
     /**
@@ -346,50 +346,50 @@ export default class RegistryCommandHandler {
      */
     async _handleImport(filePath, options) {
         if (!filePath) {
-            console.log('Usage: ml-container-creator registry import <file> [--merge|--replace]')
-            return
+            console.log('Usage: ml-container-creator registry import <file> [--merge|--replace]');
+            return;
         }
 
-        let raw
+        let raw;
         try {
-            raw = readFileSync(filePath, 'utf8')
+            raw = readFileSync(filePath, 'utf8');
         } catch (err) {
-            console.log(`Error: File not found: ${filePath}`)
-            return
+            console.log(`Error: File not found: ${filePath}`);
+            return;
         }
 
-        let json
+        let json;
         try {
-            json = JSON.parse(raw)
+            json = JSON.parse(raw);
         } catch (err) {
-            console.log(`Error: Invalid JSON in ${filePath}: ${err.message}`)
-            return
+            console.log(`Error: Invalid JSON in ${filePath}: ${err.message}`);
+            return;
         }
 
         if (!json.version || !Array.isArray(json.entries)) {
-            console.log('Error: Invalid export format — missing required "version" or "entries" fields.')
-            return
+            console.log('Error: Invalid export format — missing required "version" or "entries" fields.');
+            return;
         }
 
-        let strategy = 'skip'
+        let strategy = 'skip';
         if (options.merge) {
-            strategy = 'merge'
+            strategy = 'merge';
         } else if (options.replace) {
-            strategy = 'replace'
+            strategy = 'replace';
         }
 
-        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH
-        const registry = new DeploymentRegistry(registryPath)
+        const registryPath = options.project ? PROJECT_REGISTRY_PATH : PERSONAL_REGISTRY_PATH;
+        const registry = new DeploymentRegistry(registryPath);
 
         try {
-            const result = registry.importEntries(json, strategy, path.basename(filePath))
-            console.log(`\nImport complete:`)
-            console.log(`  Added:     ${result.added}`)
-            console.log(`  Skipped:   ${result.skipped}`)
-            console.log(`  Conflicts: ${result.conflicts}`)
-            console.log('')
+            const result = registry.importEntries(json, strategy, path.basename(filePath));
+            console.log('\nImport complete:');
+            console.log(`  Added:     ${result.added}`);
+            console.log(`  Skipped:   ${result.skipped}`);
+            console.log(`  Conflicts: ${result.conflicts}`);
+            console.log('');
         } catch (err) {
-            console.log(`Error importing entries: ${err.message}`)
+            console.log(`Error importing entries: ${err.message}`);
         }
     }
 
@@ -402,33 +402,33 @@ export default class RegistryCommandHandler {
      * @param {object} options - Parsed CLI options
      */
     _handleSearch(options) {
-        const query = this._extractFilters(options)
+        const query = this._extractFilters(options);
 
-        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH)
-        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH)
+        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH);
+        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH);
 
-        const personalResults = personalRegistry.search(query).map(e => ({ ...e, _source: 'personal' }))
-        const projectResults = projectRegistry.search(query).map(e => ({ ...e, _source: 'project' }))
+        const personalResults = personalRegistry.search(query).map(e => ({ ...e, _source: 'personal' }));
+        const projectResults = projectRegistry.search(query).map(e => ({ ...e, _source: 'project' }));
 
-        const allResults = [...personalResults, ...projectResults]
+        const allResults = [...personalResults, ...projectResults];
 
         if (allResults.length === 0) {
-            console.log('No matching entries found.')
-            return
+            console.log('No matching entries found.');
+            return;
         }
 
-        console.log(`\nSearch Results (${allResults.length} match${allResults.length === 1 ? '' : 'es'}):\n`)
+        console.log(`\nSearch Results (${allResults.length} match${allResults.length === 1 ? '' : 'es'}):\n`);
         for (const entry of allResults) {
-            const id = entry.id || '(no id)'
-            const ts = entry.timestamp ? entry.timestamp.slice(0, 19) : '(no timestamp)'
-            const dc = entry.deployment?.deploymentConfig || '(none)'
-            const mn = entry.model?.modelName || '(none)'
-            const it = entry.infrastructure?.instanceType || '(none)'
-            const st = entry.status || '(none)'
-            const src = entry._source === 'project' ? ' [project]' : ''
-            console.log(`  ${id}  ${ts}  ${dc}  ${mn}  ${it}  ${st}${src}`)
+            const id = entry.id || '(no id)';
+            const ts = entry.timestamp ? entry.timestamp.slice(0, 19) : '(no timestamp)';
+            const dc = entry.deployment?.deploymentConfig || '(none)';
+            const mn = entry.model?.modelName || '(none)';
+            const it = entry.infrastructure?.instanceType || '(none)';
+            const st = entry.status || '(none)';
+            const src = entry._source === 'project' ? ' [project]' : '';
+            console.log(`  ${id}  ${ts}  ${dc}  ${mn}  ${it}  ${st}${src}`);
         }
-        console.log('')
+        console.log('');
     }
 
     /**
@@ -481,7 +481,7 @@ EXAMPLES:
   ml-container-creator registry export a1b2c3d4
   ml-container-creator registry import team-deployments.json --merge
   ml-container-creator registry search --model "meta-llama/*" --backend vllm
-`)
+`);
     }
 
     // ── Helper methods ──────────────────────────────────────────────
@@ -492,15 +492,15 @@ EXAMPLES:
      * @returns {object} Filter key-value pairs
      */
     _extractFilters(options) {
-        const filters = {}
-        if (options.backend) filters.backend = options.backend
-        if (options.architecture) filters.architecture = options.architecture
-        if (options.model) filters.model = options.model
+        const filters = {};
+        if (options.backend) filters.backend = options.backend;
+        if (options.architecture) filters.architecture = options.architecture;
+        if (options.model) filters.model = options.model;
         if (options['instance-type'] || options.instanceType) {
-            filters['instance-type'] = options['instance-type'] || options.instanceType
+            filters['instance-type'] = options['instance-type'] || options.instanceType;
         }
-        if (options.status) filters.status = options.status
-        return filters
+        if (options.status) filters.status = options.status;
+        return filters;
     }
 
     /**
@@ -509,11 +509,11 @@ EXAMPLES:
      * @returns {object|null} The matching entry, or null
      */
     _findEntry(id) {
-        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH)
-        const entry = personalRegistry.get(id)
-        if (entry) return entry
+        const personalRegistry = new DeploymentRegistry(PERSONAL_REGISTRY_PATH);
+        const entry = personalRegistry.get(id);
+        if (entry) return entry;
 
-        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH)
-        return projectRegistry.get(id)
+        const projectRegistry = new DeploymentRegistry(PROJECT_REGISTRY_PATH);
+        return projectRegistry.get(id);
     }
 }