dank-ai 1.0.5 → 1.0.7

package/bin/dank

@@ -75,6 +75,21 @@ program
     await buildCommand(options);
   });
 
+// Production build command - build and optionally push production images
+program
+  .command('build:prod')
+  .description('Build production Docker images with custom naming and tagging')
+  .option('-c, --config <file>', 'Configuration file path', 'dank.config.js')
+  .option('--push', 'Push images to registry after building')
+  .option('--tag <tag>', 'Custom tag for images (default: latest)')
+  .option('--registry <registry>', 'Docker registry URL (e.g., docker.io, ghcr.io)')
+  .option('--namespace <namespace>', 'Docker namespace/organization')
+  .option('--force', 'Force rebuild without cache')
+  .action(async (options) => {
+    const { productionBuildCommand } = require('../lib/cli/production-build');
+    await productionBuildCommand(options);
+  });
+
 // Clean command - cleanup Docker resources
 program
   .command('clean')

package/lib/agent.js

@@ -500,6 +500,27 @@ class DankAgent {
     return this;
   }
 
+  /**
+   * Set agent image configuration for Docker builds
+   */
+  setAgentImageConfig(options = {}) {
+    const schema = Joi.object({
+      registry: Joi.string().optional(),
+      namespace: Joi.string().optional(),
+      tag: Joi.string().default('latest')
+    });
+
+    const { error, value } = schema.validate(options);
+    if (error) {
+      throw new Error(`Invalid agent image configuration: ${error.message}`);
+    }
+
+    this.config.agentImage = value;
+    return this;
+  }
+
+
+
   /**
    * Get the complete agent configuration for serialization
    */
@@ -589,6 +610,12 @@ class DankAgent {
       
       environment: Joi.object().pattern(Joi.string(), Joi.string()).default({}),
       
+      agentImage: Joi.object({
+        registry: Joi.string().optional(),
+        namespace: Joi.string().optional(),
+        tag: Joi.string().default('latest')
+      }).optional(),
+      
       custom: Joi.object().default({}),
       
       tools: Joi.object({

package/lib/cli/production-build.js

@@ -0,0 +1,110 @@
+/**
+ * CLI Production Build Command - Build and optionally push production Docker images
+ */
+
+const fs = require('fs-extra');
+const path = require('path');
+const chalk = require('chalk');
+const { DockerManager } = require('../docker/manager');
+
+async function productionBuildCommand(options) {
+  try {
+    console.log(chalk.yellow('🏗️  Building production Docker images...\n'));
+
+    // Load configuration
+    const configPath = path.resolve(options.config);
+    if (!await fs.pathExists(configPath)) {
+      throw new Error(`Configuration file not found: ${configPath}`);
+    }
+
+    const config = require(configPath);
+    if (!config.agents || !Array.isArray(config.agents)) {
+      throw new Error('No agents found in configuration');
+    }
+
+    // Initialize Docker manager
+    const dockerManager = new DockerManager();
+    await dockerManager.initialize();
+
+    // Build production images for each agent
+    const buildResults = [];
+    for (const agent of config.agents) {
+      try {
+        console.log(chalk.blue(`📦 Building production image for agent: ${agent.name}`));
+        
+        // Use agent's image config if available, otherwise use CLI options
+        const agentImageConfig = agent.config?.agentImage || {};
+        const buildOptions = {
+          tag: options.tag || agentImageConfig.tag || 'latest',
+          registry: options.registry || agentImageConfig.registry,
+          namespace: options.namespace || agentImageConfig.namespace,
+          force: options.force || false,
+          push: options.push || false
+        };
+        
+        const result = await dockerManager.buildProductionImage(agent, buildOptions);
+
+        buildResults.push({
+          agent: agent.name,
+          imageName: result.imageName,
+          success: true,
+          pushed: result.pushed || false
+        });
+
+        console.log(chalk.green(`✅ Successfully built: ${result.imageName}`));
+        if (result.pushed) {
+          console.log(chalk.green(`🚀 Successfully pushed: ${result.imageName}`));
+        }
+
+      } catch (error) {
+        console.error(chalk.red(`❌ Failed to build agent ${agent.name}:`), error.message);
+        buildResults.push({
+          agent: agent.name,
+          success: false,
+          error: error.message
+        });
+      }
+    }
+
+    // Print summary
+    console.log(chalk.yellow('\n📊 Build Summary:'));
+    console.log(chalk.gray('================'));
+    
+    const successful = buildResults.filter(r => r.success);
+    const failed = buildResults.filter(r => !r.success);
+    const pushed = buildResults.filter(r => r.pushed);
+
+    console.log(chalk.green(`✅ Successful builds: ${successful.length}`));
+    if (pushed.length > 0) {
+      console.log(chalk.blue(`🚀 Pushed to registry: ${pushed.length}`));
+    }
+    if (failed.length > 0) {
+      console.log(chalk.red(`❌ Failed builds: ${failed.length}`));
+    }
+
+    // List built images
+    if (successful.length > 0) {
+      console.log(chalk.cyan('\n📦 Built Images:'));
+      successful.forEach(result => {
+        console.log(chalk.gray(`  - ${result.imageName}`));
+      });
+    }
+
+    // List failed builds
+    if (failed.length > 0) {
+      console.log(chalk.red('\n❌ Failed Builds:'));
+      failed.forEach(result => {
+        console.log(chalk.gray(`  - ${result.agent}: ${result.error}`));
+      });
+      process.exit(1);
+    }
+
+    console.log(chalk.green('\n🎉 Production build completed successfully!'));
+
+  } catch (error) {
+    console.error(chalk.red('❌ Production build failed:'), error.message);
+    process.exit(1);
+  }
+}
+
+module.exports = { productionBuildCommand };

package/lib/docker/manager.js

@@ -413,6 +413,78 @@ class DockerManager {
     }
   }
 
+  /**
+   * Build production image with custom naming and tagging
+   */
+  async buildProductionImage(agent, options = {}) {
+    const {
+      tag = 'latest',
+      registry,
+      namespace,
+      force = false,
+      push = false
+    } = options;
+
+    // Construct production image name
+    let imageName = agent.name.toLowerCase();
+    
+    // Add namespace if provided
+    if (namespace) {
+      imageName = `${namespace}/${imageName}`;
+    }
+    
+    // Add registry if provided
+    if (registry) {
+      imageName = `${registry}/${imageName}`;
+    }
+    
+    // Add tag
+    imageName = `${imageName}:${tag}`;
+
+    this.logger.info(`Building production image for agent: ${agent.name} -> ${imageName}`);
+
+    try {
+      const buildContext = await this.createAgentBuildContext(agent);
+      
+      const stream = await this.docker.buildImage(buildContext, {
+        t: imageName,
+        dockerfile: 'Dockerfile',
+        nocache: force
+      });
+
+      await this.followBuildProgress(stream, `Production build for ${agent.name}`);
+      
+      this.logger.info(`Production image '${imageName}' built successfully`);
+      
+      // Clean up build context
+      await fs.remove(buildContext);
+      
+      let pushed = false;
+      
+      // Push to registry if requested
+      if (push) {
+        try {
+          this.logger.info(`Pushing image to registry: ${imageName}`);
+          const pushStream = await this.docker.getImage(imageName).push();
+          await this.followBuildProgress(pushStream, `Push ${imageName}`);
+          this.logger.info(`Successfully pushed image: ${imageName}`);
+          pushed = true;
+        } catch (pushError) {
+          this.logger.warn(`Failed to push image ${imageName}: ${pushError.message}`);
+          // Don't fail the build if push fails
+        }
+      }
+      
+      return {
+        imageName,
+        pushed
+      };
+      
+    } catch (error) {
+      throw new Error(`Failed to build production image: ${error.message}`);
+    }
+  }
+
   /**
    * Generate handlers code from agent configuration
    */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dank-ai",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "Dank Agent Service - Docker-based AI agent orchestration platform",
   "main": "lib/index.js",
   "exports": {