s9n-devops-agent 2.0.10 → 2.0.11-dev.1

package/src/setup-cs-devops-agent.js

@@ -27,6 +27,7 @@ import path from 'path';
 import { execSync } from 'child_process';
 import { fileURLToPath } from 'url';
 import { dirname } from 'path';
+import { credentialsManager } from './credentials-manager.js';
 import {
   colors,
   status,
@@ -113,9 +114,116 @@ function backupFile(filePath) {
   return null;
 }
 
-// ============================================================================
+function setupFolderStructure(projectRoot) {
+  log.header();
+  log.title('📂 Checking Folder Structure');
+
+  // Folders defined in folders.md
+  const standardFolders = [
+    'src',
+    'test',
+    'docs',
+    'scripts',
+    'deploy_test',
+    'product_requirement_docs',
+    'infrastructure'
+  ];
+
+  const missingFolders = standardFolders.filter(folder => !fs.existsSync(path.join(projectRoot, folder)));
+
+  if (missingFolders.length === 0) {
+    log.info('Standard folder structure already exists.');
+    return;
+  }
+
+  log.info('Found missing standard folders:');
+  missingFolders.forEach(folder => console.log(`   - ${folder}/`));
+  console.log();
+
+  explain(`
+${colors.bright}Recommended Structure:${colors.reset}
+Standard folders help organize your code, tests, and documentation.
+This structure is compatible with the DevOps Agent's automation tools.
+  `);
+
+  // We can't use await here because this function is synchronous in the flow, 
+  // but the main flow is async. We should probably make this async or use prompt from ui-utils.
+  // However, looking at the code structure, helper functions are sync or async.
+  // Let's return the missing folders and handle the prompt in the main function or make this async.
+  return missingFolders;
+}
+
+function checkContractsExist(projectRoot) {
+  const contractsDir = path.join(projectRoot, 'House_Rules_Contracts');
+  if (!fs.existsSync(contractsDir)) return false;
+  
+  const requiredContracts = [
+    'FEATURES_CONTRACT.md',
+    'API_CONTRACT.md',
+    'DATABASE_SCHEMA_CONTRACT.md',
+    'SQL_CONTRACT.json',
+    'THIRD_PARTY_INTEGRATIONS.md',
+    'INFRA_CONTRACT.md'
+  ];
+  
+  return requiredContracts.every(file => fs.existsSync(path.join(contractsDir, file)));
+}
+
+async function generateContracts(projectRoot) {
+  log.header();
+  log.title('📜 Generating Contracts');
+  
+  const scriptsDir = path.join(projectRoot, 'scripts', 'contract-automation');
+  const generateScript = path.join(scriptsDir, 'generate-contracts.js');
+  const analyzeScript = path.join(scriptsDir, 'analyze-with-llm.js');
+  
+  if (!fs.existsSync(generateScript)) {
+    log.error('Contract generation scripts not found!');
+    return;
+  }
+  
+  try {
+    // 1. Run local scan
+    info('Running local codebase scan...');
+    execSync(`node "${generateScript}"`, { cwd: projectRoot, stdio: 'inherit' });
+    success('Scan complete.');
+    
+    // 2. Run LLM analysis if key exists
+    if (credentialsManager.hasGroqApiKey()) {
+      info('Enhancing contracts with AI analysis...');
+      try {
+        // Inject env for the child process
+        const env = { ...process.env };
+        credentialsManager.injectEnv(); // Ensure current process has it
+        // Note: execSync inherits env by default, but explicit is safer if we modified it
+        if (!env.GROQ_API_KEY && credentialsManager.getGroqApiKey()) {
+           env.GROQ_API_KEY = credentialsManager.getGroqApiKey();
+           env.OPENAI_API_KEY = credentialsManager.getGroqApiKey();
+        }
+
+        execSync(`node "${analyzeScript}" --scan-results=House_Rules_Contracts/contract-scan-results.json`, { 
+          cwd: projectRoot, 
+          stdio: 'inherit',
+          env
+        });
+        success('AI analysis complete.');
+      } catch (err) {
+        warn(`AI analysis failed: ${err.message}`);
+        warn('Contracts generated but without AI enhancements.');
+      }
+    } else {
+      warn('Skipping AI analysis (no Groq API Key).');
+      warn('Run "s9n-devops-agent creds set-groq-key <key>" later to enable this.');
+    }
+    
+  } catch (error) {
+    log.error(`Contract generation failed: ${error.message}`);
+  }
+}
+
+// ===========================================================================
 // SETUP FUNCTIONS
-// ============================================================================
+// ===========================================================================
 
 async function setupNpmPackages(projectRoot) {
   log.header();
@@ -865,6 +973,47 @@ function printInstructions(initials) {
   log.header();
 }
 
+async function setupEnvFile(projectRoot) {
+  log.header();
+  log.title('🔑 Setting up Environment Variables');
+  
+  const envPath = path.join(projectRoot, '.env');
+  let envContent = '';
+  
+  if (fs.existsSync(envPath)) {
+    envContent = fs.readFileSync(envPath, 'utf8');
+    log.info('.env file already exists');
+  } else {
+    log.info('Creating .env file');
+  }
+  
+  // Check for OPENAI_API_KEY
+  if (!envContent.includes('OPENAI_API_KEY=')) {
+    console.log();
+    explain(`
+${colors.bright}Groq API Key Setup${colors.reset}
+The contract automation features use Groq LLM (via OpenAI compatibility).
+You can enter your API key now, or set it later in the .env file.
+    `);
+    
+    const apiKey = await prompt('Enter Groq API Key (leave empty to skip)');
+    
+    if (apiKey) {
+      const newLine = envContent.endsWith('\n') || envContent === '' ? '' : '\n';
+      envContent += `${newLine}# Groq API Key for Contract Automation\nOPENAI_API_KEY=${apiKey}\n`;
+      fs.writeFileSync(envPath, envContent);
+      log.success('Added OPENAI_API_KEY to .env');
+    } else {
+      log.warn('Skipped Groq API Key. Contract automation features may not work.');
+      if (!fs.existsSync(envPath)) {
+        fs.writeFileSync(envPath, '# Environment Variables\n');
+      }
+    }
+  } else {
+    log.info('OPENAI_API_KEY is already configured in .env');
+  }
+}
+
 // ============================================================================
 // CLEANUP FUNCTIONS
 // ============================================================================
@@ -976,6 +1125,37 @@ ${colors.bright}How:${colors.reset} Creates branches like dev_abc_2025-10-31
   tip(`Your branches will be named: ${colors.cyan}dev_${initials}_YYYY-MM-DD${colors.reset}`);
   console.log();
   
+  // Groq API Key Setup
+  sectionTitle('Groq API Key (Contract Automation)');
+  explain(`
+${colors.bright}What:${colors.reset} API Key for Groq (llama-3.1-70b-versatile)
+${colors.bright}Why:${colors.reset} Required for AI-Optimized Contract Automation System
+${colors.bright}Security:${colors.reset} Stored locally in ${colors.yellow}local_deploy/credentials.json${colors.reset} (gitignored)
+  `);
+
+  const hasKey = credentialsManager.hasGroqApiKey();
+  let groqKey = null;
+
+  if (hasKey) {
+    info('Groq API Key is already configured.');
+    const update = await confirm('Do you want to update it?', false);
+    if (update) {
+      groqKey = await prompt('Enter your Groq API Key');
+    }
+  } else {
+    groqKey = await prompt('Enter your Groq API Key (leave empty to skip)');
+  }
+
+  if (groqKey && groqKey.trim()) {
+    credentialsManager.setGroqApiKey(groqKey.trim());
+    success('Groq API Key saved securely.');
+  } else if (!hasKey) {
+    warn('Skipping Groq API Key setup.');
+    warn('NOTE: Contract Automation features (analyze-with-llm.js) will NOT work without this key.');
+  }
+  
+  console.log();
+
   // Confirm before proceeding
   drawSection('Configuration Summary', [
     `${status.folder} Branch prefix: ${colors.cyan}dev_${initials}_${colors.reset}`,
@@ -983,7 +1163,9 @@ ${colors.bright}How:${colors.reset} Creates branches like dev_abc_2025-10-31
     `${status.checkmark} VS Code settings and tasks`,
     `${status.checkmark} NPM packages and scripts`,
     `${status.checkmark} Commit message files`,
-    `${status.checkmark} House rules for AI agents`
+    `${status.checkmark} House rules for AI agents`,
+    `${status.folder} Standard folder structure check`,
+    `${status.checkmark} Contract files check`
   ]);
   console.log();
   
@@ -999,6 +1181,37 @@ ${colors.bright}How:${colors.reset} Creates branches like dev_abc_2025-10-31
   console.log();
   
   try {
+    // Check and setup folder structure first
+    const missingFolders = setupFolderStructure(projectRoot);
+    if (missingFolders && missingFolders.length > 0) {
+      const createFolders = await confirm('Create missing standard folders?', true);
+      if (createFolders) {
+        missingFolders.forEach(folder => {
+          ensureDirectoryExists(path.join(projectRoot, folder));
+          log.success(`Created ${folder}/`);
+        });
+      } else {
+        log.warn('Skipping folder creation.');
+      }
+    }
+
+    // Check for contracts
+    if (!checkContractsExist(projectRoot)) {
+      log.header();
+      log.title('📜 Contract Files Missing');
+      explain(`
+${colors.bright}Contract System:${colors.reset}
+This project uses a Contract System to coordinate multiple AI agents.
+It seems like this is a fresh setup or contracts are missing.
+We can scan your codebase and generate them now.
+      `);
+      
+      const shouldGenerate = await confirm('Generate contract files now?', true);
+      if (shouldGenerate) {
+        await generateContracts(projectRoot);
+      }
+    }
+
     // Run setup steps
     const packageJson = await setupNpmPackages(projectRoot);
     setupVSCodeSettings(projectRoot, initials);
@@ -1006,6 +1219,9 @@ ${colors.bright}How:${colors.reset} Creates branches like dev_abc_2025-10-31
     setupCommitFiles(projectRoot, initials);
     createRunScripts(projectRoot, initials, packageJson);
     
+    // Setup .env with API keys
+    await setupEnvFile(projectRoot);
+    
     // Clean up DevOpsAgent files to avoid duplicates
     cleanupDevOpsAgentFiles(projectRoot);