imhcode 1.0.2 → 1.0.4

package/bin/imhcode.js

@@ -296,6 +296,7 @@ async function runExecuteCommand(restArgs) {
 
   console.log(`─`.repeat(60));
   try {
+    await markSprintStarted(cwd, sprintNum);
     execSync(`sh "${masterScript}"`, { stdio: 'inherit', cwd });
     console.log(`─`.repeat(60));
     console.log(`\n🏁 Sprint ${sprintNum} execution complete!\n`);
@@ -704,6 +705,11 @@ async function cmdRun(orc, args) {
   const outputIdx    = restArgs.indexOf('--output');
   const outputDir    = outputIdx >= 0 ? restArgs[outputIdx + 1] : undefined;
 
+  const sprintIdx    = restArgs.indexOf('--sprint');
+  const sprintVal    = sprintIdx >= 0 ? parseInt(restArgs[sprintIdx + 1], 10) : undefined;
+  const taskIdx      = restArgs.indexOf('--task');
+  const taskVal      = taskIdx >= 0 ? parseInt(restArgs[taskIdx + 1], 10) : undefined;
+
   const agentsDir = resolveAgentsDir();
   const cwd       = process.cwd();
   const config    = loadLocalConfig(cwd);
@@ -736,6 +742,10 @@ async function cmdRun(orc, args) {
 
   const agent  = orc.getAgent(agents, agentId);
 
+  if (live && sprintVal !== undefined) {
+    await markSprintStarted(cwd, sprintVal);
+  }
+
   // Pass the routed engine + model to runAgent (Fix 0b Part A)
   const result = await orc.runAgent(
     agent, task,
@@ -743,6 +753,12 @@ async function cmdRun(orc, args) {
     criteria
   );
 
+  if (!result.dryRun && result.errors.length === 0) {
+    if (sprintVal !== undefined && taskVal !== undefined) {
+      await markTaskCompleted(cwd, sprintVal, taskVal);
+    }
+  }
+
   console.log('\n' + '─'.repeat(72));
   if (result.errors.length > 0 && !result.dryRun) {
     console.error(`\n❌ [IMH-Code] Execution failed with errors:`);
@@ -1662,28 +1678,40 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
   const testingModeRaw = extractAnswer(brainstormContent, 'Q31', 'A').trim().toUpperCase().charAt(0);
   const detectedTesting = testingModeRaw === 'B' ? 'balanced' : testingModeRaw === 'C' ? 'strict' : 'fast';
 
-  // Detect scope from brainstorm content
-  const b = brainstormContent.toLowerCase();
-  const hasFrontend  = true; // Always assume frontend
-  const hasBackend   = (() => {
-    // Check Q31-equivalent backend answer or keyword detection
-    const backendAnswer = extractAnswer(brainstormContent, 'Q16', '').trim();
-    if (backendAnswer && backendAnswer !== '*(edit if needed)*') return true;
-    return /laravel|django|fastapi|express|spring\s*boot|node\.js.*api|postgresql|mysql|redis/i.test(brainstormContent);
-  })();
-  const hasMobile    = /flutter|react native|ios.*app|android.*app|expo/i.test(brainstormContent);
-  const needsWorktrees = /worktree.*yes|yes.*worktree/i.test(brainstormContent);
-  const hasPayments  = /stripe|paypal|payment|yes.*payment/i.test(brainstormContent);
-  const hasRealtime  = /websocket|real-time.*yes|yes.*real-time|socket\.io/i.test(brainstormContent);
-  const designStyle  = (() => {
-    const m = brainstormContent.match(/Your Answer:\s*\*(.*?(?:glassmorphism|neumorphism|brutalism|claymorphism|minimalist|modern saas).*?)\*/i);
-    return m ? m[1].trim() : 'Modern SaaS';
+  const ansQ8  = extractAnswer(brainstormContent, 'Q8', 'Dark Premium').trim();
+  const ansQ9  = extractAnswer(brainstormContent, 'Q9', 'Modern SaaS').trim();
+  const ansQ12 = extractAnswer(brainstormContent, 'Q12', 'system toggle').toLowerCase();
+  const ansQ14 = extractAnswer(brainstormContent, 'Q14', 'no').toLowerCase();
+  const ansQ16 = extractAnswer(brainstormContent, 'Q16', 'no').toLowerCase();
+  const ansQ21 = extractAnswer(brainstormContent, 'Q21', 'no').toLowerCase();
+  const ansQ22 = extractAnswer(brainstormContent, 'Q22', 'no').toLowerCase();
+  const ansQ25 = extractAnswer(brainstormContent, 'Q25', 'no').toLowerCase();
+
+  const hasFrontend = true; // Always assume frontend
+
+  // Robust scope detection: only true if user answer is not "no", "none", or standard default fallback placeholder
+  const hasBackend = (() => {
+    if (ansQ16.includes('no') || ansQ16.includes('none') || ansQ16.includes('edit if needed') || ansQ16.trim() === '') {
+      return false;
+    }
+    return true;
   })();
-  const colorStyle   = (() => {
-    const m = brainstormContent.match(/Your Answer:\s*\*(.*?(?:vibrant|dark premium|minimal neutral|pastel|corporate|bold).*?)\*/i);
-    return m ? m[1].trim() : 'Dark Premium';
+
+  const hasMobile = (() => {
+    if (ansQ25.includes('no') || ansQ25.includes('none') || ansQ25.includes('edit if needed') || ansQ25.trim() === '') {
+      return false;
+    }
+    return true;
   })();
 
+  const needsWorktrees = ansQ14.includes('yes') || ansQ14.includes('true');
+  const hasPayments    = !ansQ22.includes('no') && !ansQ22.includes('none') && !ansQ22.includes('edit if needed') && ansQ22.trim().length > 0;
+  const hasRealtime    = !ansQ21.includes('no') && !ansQ21.includes('none') && !ansQ21.includes('edit if needed') && ansQ21.trim().length > 0;
+
+  const designStyle = ansQ9.replace(/\*\(.*?\)\*/g, '').trim() || 'Modern SaaS';
+  const colorStyle  = ansQ8.replace(/\*\(.*?\)\*/g, '').trim() || 'Dark Premium';
+  const darkmode    = ansQ12.replace(/\*\(.*?\)\*/g, '').trim() || 'system toggle';
+
   // Save testing mode to config
   if (config) {
     config.testing_mode = detectedTesting;
@@ -1691,21 +1719,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
     try { fs.writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf8'); } catch {}
   }
 
-  // Fix 5: Create workspace directories lazily based on brainstorm answers
-  if (hasFrontend) {
-    const frontendDir = path.join(cwd, 'frontend');
-    if (!fs.existsSync(frontendDir)) {
-      fs.mkdirSync(frontendDir, { recursive: true });
-      console.log(`  📁 Created frontend/ (scope confirmed in brainstorming)`);
-    }
-  }
-  if (hasBackend) {
-    const backendDir = path.join(cwd, 'backend');
-    if (!fs.existsSync(backendDir)) {
-      fs.mkdirSync(backendDir, { recursive: true });
-      console.log(`  📁 Created backend/ (scope confirmed in brainstorming)`);
-    }
-  }
+  // Create workspace directories lazily (only worktrees since it's a layout helper, NOT frontend/backend which must remain empty for official CLI scaffolding)
   if (needsWorktrees) {
     const wtDir = path.join(cwd, '.worktrees');
     if (!fs.existsSync(wtDir)) {
@@ -1716,7 +1730,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
 
   // Generate PROJECT_BRIEF.md
   const briefContent = generateProjectBrief(userPrompt, brainstormContent, {
-    hasFrontend, hasBackend, hasMobile, designStyle, colorStyle, detectedTesting
+    hasFrontend, hasBackend, hasMobile, designStyle, colorStyle, darkmode, detectedTesting
   });
   fs.writeFileSync(path.join(cwd, BRIEF_MD), briefContent, 'utf8');
   console.log(`  ✅ Created PROJECT_BRIEF.md`);
@@ -1724,7 +1738,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
   // Try LLM-powered sprint generation first (Fix 0)
   const llmSprintResult = await tryLLMSprintGeneration(
     cwd, userPrompt, brainstormContent, config,
-    { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, detectedTesting, needsWorktrees }
+    { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, darkmode, detectedTesting, needsWorktrees }
   );
 
   let lastSprintNum;
@@ -1736,7 +1750,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
     // Fix 5: Fallback to smart static generation
     lastSprintNum = await generateStaticSprintPlans(
       cwd,
-      { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, detectedTesting },
+      { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, darkmode, detectedTesting },
       config
     );
   }
@@ -1750,7 +1764,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
 
   // Update compact context
   const contextContent = buildContextContent(userPrompt, {
-    hasFrontend, hasBackend, hasMobile, detectedTesting, lastSprintNum, designStyle, colorStyle
+    hasFrontend, hasBackend, hasMobile, detectedTesting, lastSprintNum, designStyle, colorStyle, darkmode
   });
   fs.mkdirSync(path.join(cwd, LOCAL_DIR_NAME), { recursive: true });
   fs.writeFileSync(path.join(cwd, CONTEXT_MD), contextContent, 'utf8');
@@ -1761,7 +1775,7 @@ async function generateSprintPlans(cwd, userPrompt, brainstormContent, config) {
  * Attempt LLM-generated sprint plans. Returns { sprintCount } on success, null on failure.
  */
 async function tryLLMSprintGeneration(cwd, userPrompt, brainstormContent, config, scope) {
-  const { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, detectedTesting } = scope;
+  const { hasFrontend, hasBackend, hasMobile, hasPayments, hasRealtime, designStyle, colorStyle, darkmode, detectedTesting } = scope;
 
   const agentList = [
     'planner (planning)', 'designer (frontend)', 'nextjs-executor (frontend)', 'react-executor (frontend)',
@@ -1788,7 +1802,7 @@ ${userPrompt}
 ${brainstormContent}
 
 ## Scope
-- Frontend: ${hasFrontend ? 'YES' : 'NO'} (Design style: ${designStyle}, Colors: ${colorStyle})
+- Frontend: ${hasFrontend ? 'YES' : 'NO'} (Design style: ${designStyle}, Colors: ${colorStyle}, Dark Mode Strategy: ${darkmode})
 - Backend:  ${hasBackend  ? 'YES' : 'NO'}
 - Mobile:   ${hasMobile   ? 'YES' : 'NO'}
 - Payments: ${hasPayments ? 'YES' : 'NO'}
@@ -1838,7 +1852,8 @@ Return ONLY this JSON structure (no markdown, no explanation):
 8. Max 6 tasks per sprint for clean execution
 9. If design style is Glassmorphism/Neumorphism — include a designer task for design tokens in Sprint 1
 10. If payments needed — include payment integration task in a sprint
-11. If realtime needed — include WebSocket/SSE task in a sprint`;
+11. If realtime needed — include WebSocket/SSE task in a sprint
+12. Under Frontend, STRICTLY respect the Dark Mode Strategy: if strategy is "light only", do NOT generate any dark mode setups, variables, toggles or dark styles. If strategy is "always dark", design the UI strictly for dark backgrounds.`;
 
   const llmOutput = await invokePlanningLLM(llmPrompt, config, cwd);
   if (!llmOutput) return null;
@@ -2021,7 +2036,16 @@ ${tasks.map((t, i) => `- [ ] Task ${i+1}: ${t.task} [\`${t.agent}\`]`).join('\n'
 # Model:  ${routedModel || 'default'} via ${routedEngine || 'default'}
 # Tier:   ${t.tier}
 CWD="$(cd "$(dirname "\${BASH_SOURCE[0]}")" && pwd)"
-cd "$CWD/../../../.."
+cd "$CWD/../../.."
+
+# Check if task is already completed
+PROGRESS_FILE="$CWD/../progress.md"
+if [ -f "$PROGRESS_FILE" ]; then
+  if grep -q -e "- \\[[xX]\\] Task ${taskNum}:" "$PROGRESS_FILE"; then
+    echo "✅ Task ${taskNum} is already completed. Skipping."
+    exit 0
+  fi
+fi
 
 TASK="${taskDesc.replace(/"/g, '\\"')}"
 
@@ -2031,9 +2055,9 @@ echo "   Model:  ${routedModel || 'default'} via ${routedEngine || 'default'}"
 echo "   Tier:   ${t.tier}"
 
 if command -v imhcode >/dev/null 2>&1; then
-  imhcode agent run ${t.agent} "$TASK" --live ${engineFlag} ${modelFlag}
+  imhcode agent run ${t.agent} "$TASK" --live ${engineFlag} ${modelFlag} --sprint ${sprintNum} --task ${taskNum}
 else
-  node "$(npm root -g)/imhcode/bin/imhcode.js" agent run ${t.agent} "$TASK" --live ${engineFlag} ${modelFlag}
+  node "$(npm root -g)/imhcode/bin/imhcode.js" agent run ${t.agent} "$TASK" --live ${engineFlag} ${modelFlag} --sprint ${sprintNum} --task ${taskNum}
 fi
 `;
     fs.writeFileSync(path.join(tasksDir, `task_${taskNum}.sh`), taskScript, { mode: 0o755 });
@@ -2251,14 +2275,182 @@ async function updateCompactContext(cwd, completedSprint) {
   if (!fs.existsSync(contextPath)) return;
   try {
     let content = fs.readFileSync(contextPath, 'utf8');
-    content = content.replace(
-      /Current Sprint\nSprint \d+ \(not started\)/,
-      `Current Sprint\nSprint ${completedSprint + 1}`
-    );
+    const pattern = /(## Current Sprint\r?\nSprint )\d+\s*(\([^)]*\))?/i;
+    if (pattern.test(content)) {
+      content = content.replace(pattern, `$1${completedSprint + 1} (not started)`);
+    } else {
+      content = content.replace(
+        /Current Sprint\r?\nSprint \d+ \(not started\)/,
+        `Current Sprint\nSprint ${completedSprint + 1} (not started)`
+      );
+    }
     fs.writeFileSync(contextPath, content, 'utf8');
+
+    // Update PROJECT_BRIEF.md Current Sprint
+    const briefPath = path.join(cwd, BRIEF_MD);
+    if (fs.existsSync(briefPath)) {
+      try {
+        let briefContent = fs.readFileSync(briefPath, 'utf8');
+        const currentSprintPattern = /(Current Sprint:\s*Sprint\s*)\d+/i;
+        if (currentSprintPattern.test(briefContent)) {
+          briefContent = briefContent.replace(currentSprintPattern, `$1${completedSprint + 1}`);
+          fs.writeFileSync(briefPath, briefContent, 'utf8');
+        }
+      } catch {}
+    }
   } catch { /* non-critical */ }
 }
 
+async function markSprintStarted(cwd, sprintNum) {
+  const progressPath = path.join(cwd, DOCS_DIR, `sprint-${sprintNum}`, 'progress.md');
+  const briefPath = path.join(cwd, BRIEF_MD);
+
+  // 1. Update progress.md
+  if (fs.existsSync(progressPath)) {
+    try {
+      let progressContent = fs.readFileSync(progressPath, 'utf8');
+      if (progressContent.includes('Status: 🟡 Not Started')) {
+        progressContent = progressContent.replace('Status: 🟡 Not Started', 'Status: 🟢 In Progress');
+        const dateStr = new Date().toLocaleString();
+        progressContent = progressContent.replace('Start: —', `Start: ${dateStr}`);
+        fs.writeFileSync(progressPath, progressContent, 'utf8');
+        console.log(`  🔄 Updated sprint progress to: 🟢 In Progress`);
+      }
+    } catch (err) {
+      console.warn(`  ⚠️ Could not update sprint progress.md: ${err.message}`);
+    }
+  }
+
+  // 2. Update PROJECT_BRIEF.md status
+  if (fs.existsSync(briefPath)) {
+    try {
+      let briefContent = fs.readFileSync(briefPath, 'utf8');
+      const rowPattern = new RegExp(`^\\|\\s*${sprintNum}\\s*\\|([^|]+)\\|([^|]+)\\|`, 'm');
+      const match = briefContent.match(rowPattern);
+      if (match) {
+        const currentStatus = match[2].trim();
+        if (currentStatus.includes('Not Started') || currentStatus.includes('Pending')) {
+          briefContent = updateSprintLogTable(briefContent, sprintNum, '🟢 In Progress');
+          
+          // Also check if Current Sprint needs updating
+          const currentSprintPattern = /(Current Sprint:\s*Sprint\s*)\d+/i;
+          if (currentSprintPattern.test(briefContent)) {
+            briefContent = briefContent.replace(currentSprintPattern, `$1${sprintNum}`);
+          }
+          
+          fs.writeFileSync(briefPath, briefContent, 'utf8');
+          console.log(`  🔄 Updated PROJECT_BRIEF.md sprint ${sprintNum} status to: 🟢 In Progress`);
+        }
+      }
+    } catch (err) {
+      console.warn(`  ⚠️ Could not update PROJECT_BRIEF.md: ${err.message}`);
+    }
+  }
+
+  // 3. Update context.md status
+  await updateContextSprintStatus(cwd, sprintNum, 'in progress');
+}
+
+async function markTaskCompleted(cwd, sprintNum, taskNum) {
+  const progressPath = path.join(cwd, DOCS_DIR, `sprint-${sprintNum}`, 'progress.md');
+  const briefPath = path.join(cwd, BRIEF_MD);
+
+  if (!fs.existsSync(progressPath)) return;
+
+  try {
+    let progressContent = fs.readFileSync(progressPath, 'utf8');
+    const lines = progressContent.split('\n');
+    let taskLineIndex = -1;
+    for (let i = 0; i < lines.length; i++) {
+      const taskPattern = new RegExp(`^-\\s*\\[\\s*\\]\\s*Task\\s*${taskNum}\\b`, 'i');
+      if (taskPattern.test(lines[i])) {
+        taskLineIndex = i;
+        lines[i] = lines[i].replace(/-\s*\[\s*\]/, '- [x]');
+        break;
+      }
+    }
+
+    if (taskLineIndex === -1) {
+      console.warn(`  ⚠️ Could not find Task ${taskNum} checkbox in progress.md`);
+      return;
+    }
+
+    progressContent = lines.join('\n');
+    console.log(`  ✅ Marked Task ${taskNum} as completed in progress.md`);
+
+    const incompleteTasksCount = (progressContent.match(/^-\s*\[\s*\]\s*Task\b/gim) || []).length;
+    
+    if (incompleteTasksCount === 0) {
+      progressContent = progressContent.replace(/Status:\s*🟢\s*In\s*Progress/i, 'Status: ✅ Complete');
+      progressContent = progressContent.replace(/Status:\s*🟡\s*Not\s*Started/i, 'Status: ✅ Complete');
+      
+      const dateStr = new Date().toLocaleString();
+      progressContent = progressContent.replace('End: —', `End: ${dateStr}`);
+      
+      console.log(`  🏁 All tasks in Sprint ${sprintNum} completed! Sprint is now ✅ Complete.`);
+
+      if (fs.existsSync(briefPath)) {
+        try {
+          let briefContent = fs.readFileSync(briefPath, 'utf8');
+          briefContent = updateSprintLogTable(briefContent, sprintNum, '✅ Complete');
+          fs.writeFileSync(briefPath, briefContent, 'utf8');
+          console.log(`  🔄 Updated PROJECT_BRIEF.md sprint ${sprintNum} status to: ✅ Complete`);
+        } catch (err) {
+          console.warn(`  ⚠️ Could not update PROJECT_BRIEF.md: ${err.message}`);
+        }
+      }
+    }
+    
+    fs.writeFileSync(progressPath, progressContent, 'utf8');
+  } catch (err) {
+    console.warn(`  ⚠️ Could not update task progress: ${err.message}`);
+  }
+}
+
+function updateSprintLogTable(content, sprintNum, newStatus) {
+  const lines = content.split('\n');
+  let inSprintLog = false;
+  for (let i = 0; i < lines.length; i++) {
+    const line = lines[i];
+    if (line.startsWith('## Sprint Log')) {
+      inSprintLog = true;
+      continue;
+    }
+    if (inSprintLog && line.startsWith('#')) {
+      inSprintLog = false;
+    }
+    if (inSprintLog) {
+      const match = line.match(new RegExp(`^\\|\\s*${sprintNum}\\s*\\|([^|]+)\\|([^|]+)\\|`));
+      if (match) {
+        const title = match[1].trim();
+        lines[i] = `| ${sprintNum} | ${title} | ${newStatus} |`;
+        break;
+      }
+    }
+  }
+  return lines.join('\n');
+}
+
+async function updateContextSprintStatus(cwd, sprintNum, status) {
+  const contextPath = path.join(cwd, CONTEXT_MD);
+  if (!fs.existsSync(contextPath)) return;
+  try {
+    let content = fs.readFileSync(contextPath, 'utf8');
+    const pattern = /(## Current Sprint\r?\nSprint \d+)\s*\([^)]*\)/i;
+    if (pattern.test(content)) {
+      content = content.replace(pattern, `$1 (${status})`);
+    } else {
+      const simplePattern = /(## Current Sprint\r?\nSprint \d+)/i;
+      if (simplePattern.test(content)) {
+        content = content.replace(simplePattern, `$1 (${status})`);
+      }
+    }
+    fs.writeFileSync(contextPath, content, 'utf8');
+  } catch (err) {
+    // non-critical
+  }
+}
+
 // ─── Utilities ────────────────────────────────────────────────────────────────
 
 function askQuestion(query) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "imhcode",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "IMH-Code — Imam Hussain Coding Harness Platform. A fast-first multi-agent AI coding framework with intelligent model routing. 19 generic role-based agents (planner, nextjs-executor, laravel-executor, etc.), configurable testing strategy, and 7 token-saving optimizations for rapid MVP development.",
   "main": "index.js",
   "bin": {