ai-control-center 1.15.2

package/lib/actions/reject.js

@@ -0,0 +1,307 @@
+import chalk from 'chalk';
+import { copyFileSync, existsSync, mkdirSync, readdirSync, readFileSync, writeFileSync } from 'fs';
+import { resolve } from 'path';
+import { logActivity, printActivityFooter, printActivityHeader } from '../utils/activity-log.js';
+import { printError, printSuccess } from '../utils/display.js';
+import { getStatus, getWorkflowDir, updateStatus } from '../utils/pipeline.js';
+import { runCopilot } from '../utils/runner.js';
+import { spinCycle, spinner } from '../utils/spinner.js';
+import { statusBar } from '../utils/status-bar.js';
+
+// ─── Multi-line reason input (paste-safe) ─────────────────────────────────────
+// Behaves like Claude Code: paste freely including newlines.
+// Submit: press Enter on a blank line, or Ctrl+D.
+// Cancel: Ctrl+C exits the process.
+//
+// Raw-mode approach — reads stdin char-by-char so pasted \n chars are buffered
+// rather than immediately submitted (inquirer 'input' submits on first \n).
+
+function readReasonInput() {
+  return new Promise((resolve, _reject) => {
+    const stdin = process.stdin;
+    const isTTY = !!stdin.isTTY;
+    const savedRaw = isTTY ? stdin.isRaw : false;
+
+    if (isTTY) {
+      stdin.setRawMode(true);
+      stdin.resume();
+    }
+    stdin.setEncoding('utf8');
+
+    statusBar.pause();
+
+    process.stdout.write('\n');
+    process.stdout.write(
+      `  ${chalk.cyan('◆')}  ${chalk.bold('Reason for rejection')} ` +
+      chalk.dim('(be specific so Copilot can fix it)') + '\n'
+    );
+    process.stdout.write(
+      chalk.dim('     Paste freely · press Enter on a blank line or Ctrl+D to submit\n\n')
+    );
+    process.stdout.write('  ');
+
+    let buffer      = '';   // full accumulated text
+    let currentLine = '';   // chars typed on the current visual line
+    let lastWasNL   = false; // was the previous processed char a newline?
+
+    const cleanup = () => {
+      stdin.off('data', onData);
+      if (isTTY) stdin.setRawMode(savedRaw || false);
+      statusBar.resume();
+    };
+
+    const submit = () => {
+      cleanup();
+      process.stdout.write('\n');
+      resolve((buffer + currentLine).trim());
+    };
+
+    const onData = (chunk) => {
+      // ── Normalise line endings ──────────────────────────────────────────────
+      // Replace \r\n (Windows) and bare \r (old Mac) with \n so the loop only
+      // ever sees a single newline per line break.
+      const normalised = chunk.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
+
+      // ── Paste vs. typed detection (same approach as Claude Code CLI) ────────
+      // A manually-typed Enter arrives as a 1-char chunk ("\n").
+      // Pasted text arrives as one large multi-char chunk with embedded newlines.
+      //
+      // Rule: only ARM the double-Enter-to-submit trigger when the newline came
+      // from a single-character chunk — i.e. the user physically pressed Enter.
+      // Newlines embedded inside a paste chunk are mid-paste line breaks and
+      // must NOT set lastWasNL, so a multi-paragraph paste never auto-submits.
+      const isTyped = normalised.length === 1;
+
+      for (const char of normalised) {
+
+        // ── Ctrl+C ─────────────────────────────────────────────────────────
+        if (char === '\u0003') {
+          cleanup();
+          process.stdout.write('\n');
+          process.exit(0);
+        }
+
+        // ── Ctrl+D (EOF) ────────────────────────────────────────────────────
+        if (char === '\u0004') {
+          if ((buffer + currentLine).trim()) submit();
+          return;
+        }
+
+        // ── Backspace ───────────────────────────────────────────────────────
+        if (char === '\u007f' || char === '\b') {
+          if (currentLine.length > 0) {
+            currentLine = currentLine.slice(0, -1);
+            process.stdout.write('\b \b');
+          }
+          lastWasNL = false;
+          continue;
+        }
+
+        // ── Newline ─────────────────────────────────────────────────────────
+        if (char === '\n') {
+          // Double-Enter on blank line → submit, BUT only when both presses
+          // were typed (not pasted). lastWasNL is only true after a typed Enter.
+          if (currentLine === '' && lastWasNL && buffer.trim()) {
+            submit();
+            return;
+          }
+          buffer += currentLine + '\n';
+          currentLine = '';
+          // Arm double-Enter only for typed newlines; pasted newlines do NOT
+          // change lastWasNL so a fast multi-paragraph paste never fires submit.
+          lastWasNL = isTyped;
+          process.stdout.write('\r\n  ');
+          continue;
+        }
+
+        // ── Printable character ─────────────────────────────────────────────
+        lastWasNL = false;
+        currentLine += char;
+        process.stdout.write(char);
+      }
+    };
+
+    stdin.on('data', onData);
+  });
+}
+
+// ── Stages that can enter this flow ───────────────────────────────────────────
+const REJECT_STAGES    = ['review_complete', 'arch_complete', 'implementation_complete'];
+const REDISPATCH_STAGE = 'rejected'; // re-send to Copilot without asking for a new reason
+
+export async function rejectAction() {
+  const status    = getStatus();
+  const stage     = status.stage;
+  const featureId = status.current_feature;
+
+  if (!REJECT_STAGES.includes(stage) && stage !== REDISPATCH_STAGE) {
+    printError(`Cannot dispatch fixes — current stage is "${stage}".`);
+    return;
+  }
+
+  printActivityHeader('PIPELINE — Dispatch Fixes to Copilot');
+
+  const workflowDir = getWorkflowDir();
+
+  // ─── Stage: rejected ──────────────────────────────────────────────────────
+  // Feature already rejected. Re-dispatch the existing review to Copilot.
+  // No reason prompt, no reject.sh — just send the review comments to Copilot.
+  if (stage === REDISPATCH_STAGE) {
+    logActivity('PIPELINE', 'Feature already rejected. Re-dispatching review comments to Copilot...');
+    await dispatchReviewToCopilot(workflowDir, featureId, status.rejection_reason || 'Fix all blockers and warnings from the review.');
+    return true;
+  }
+
+  // ─── Stage: review_complete / arch_complete / implementation_complete ─────
+  // First rejection: ask reason, update state, then dispatch to Copilot.
+  let reason = '';
+  while (reason.length < 10) {
+    reason = await readReasonInput();
+    if (reason.length < 10) {
+      process.stdout.write(chalk.red('  Please provide at least 10 characters.\n'));
+    }
+  }
+
+  // Create rejection record inline (no external .sh dependency)
+  const reasonOneLiner = reason.replace(/\s+/g, ' ').trim();
+  const _wfDir       = getWorkflowDir();
+  const _rejectedDir = resolve(_wfDir, 'rejected');
+  if (!existsSync(_rejectedDir)) mkdirSync(_rejectedDir, { recursive: true });
+  const _rejReviewsDir = resolve(_wfDir, 'reviews');
+  if (existsSync(_rejReviewsDir)) {
+    const _rr = readdirSync(_rejReviewsDir).filter(f => f.endsWith('.md')).sort().reverse();
+    if (_rr.length) copyFileSync(resolve(_rejReviewsDir, _rr[0]), resolve(_rejectedDir, _rr[0]));
+  }
+  const _rts   = new Date();
+  const _rpad  = n => String(n).padStart(2, '0');
+  const _rstamp = `${_rts.getFullYear()}${_rpad(_rts.getMonth()+1)}${_rpad(_rts.getDate())}-${_rpad(_rts.getHours())}${_rpad(_rts.getMinutes())}${_rpad(_rts.getSeconds())}`;
+  writeFileSync(resolve(_rejectedDir, `REJECTION-${_rstamp}.md`), [
+    `# Rejection Record`,
+    `**Feature:** ${featureId}`,
+    `**Rejected at:** ${_rts.toLocaleString()}`,
+    `**Reason:** ${reasonOneLiner}`,
+    `**Status:** REJECTED ❌`,
+    ``,
+    `## Required Fixes`,
+    reason.trim(),
+    ``,
+  ].join('\n'));
+  const _rejectHistory = [...(status.history || []), { id: featureId, stage: 'rejected', time: _rts.toISOString() }];
+  updateStatus({ stage: 'rejected', rejection_reason: reasonOneLiner, next: 'copilot_fix', history: _rejectHistory });
+  printSuccess('Feature rejected — dispatching fixes to Copilot.');
+
+  await dispatchReviewToCopilot(workflowDir, featureId, reason.trim());
+  return true;
+}
+
+// ─── Core: find review → build prompt → send to Copilot ─────────────────────
+
+async function dispatchReviewToCopilot(workflowDir, featureId, fallbackReason) {
+  // ── After Gemini review → pass review blockers to Copilot ────────────────
+  const reviewsDir = resolve(workflowDir, 'reviews');
+  if (existsSync(reviewsDir)) {
+    const reviewFiles = readdirSync(reviewsDir)
+      .filter(f => f.endsWith('.md') && readFileSync(resolve(reviewsDir, f), 'utf8').trim().length > 50)
+      .sort()
+      .reverse();
+
+    if (reviewFiles.length) {
+      const reviewRelPath = `.ai-workflow/reviews/${reviewFiles[0]}`;
+      const reviewAbsPath = resolve(reviewsDir, reviewFiles[0]);
+      const actionItems   = parseActionItems(reviewAbsPath);
+
+      const copilotPrompt = buildFixPrompt(reviewRelPath, fallbackReason, actionItems);
+
+      logActivity('COPILOT', `Fixing review blockers from ${reviewRelPath}...`);
+
+      const stopCycle = spinCycle('Copilot', [
+        'Reading review blockers and warnings...',
+        'Fixing architecture issues...',
+        'Addressing code quality findings...',
+        'Updating test classes...',
+        'Finalizing fixes...',
+      ], 5000, 'blue');
+
+      await runCopilot(
+        `${copilotPrompt}\n\nFix all the listed issues in the codebase. ` +
+        `Make targeted changes only — do not refactor anything beyond what is required to address the blockers.`,
+        { featureId }
+      );
+
+      stopCycle();
+      spinner.succeed('Fixes dispatched to Copilot — run Review Code to verify');
+
+      updateStatus({ stage: 'implementation_complete', next: 'review' });
+      logActivity('COPILOT', 'Fixes applied — stage reset to implementation_complete', 'success');
+      logActivity('PIPELINE', 'Run "Review Code" next to verify the fixes.', 'info');
+      printActivityFooter();
+      return;
+    }
+  }
+
+  // ── Fallback: no review file → fix from tasks context ─────────────────────
+  const tasksDir = resolve(workflowDir, 'tasks');
+  if (existsSync(tasksDir)) {
+    const files = readdirSync(tasksDir).filter(f => f.endsWith('.md')).sort().reverse();
+    if (files.length) {
+      const tasksRelPath = `.ai-workflow/tasks/${files[0]}`;
+
+      logActivity('COPILOT', `No review file found. Using tasks context: ${tasksRelPath}...`);
+
+      const stopCycle = spinCycle('Copilot', [
+        'Reading tasks and issue description...',
+        'Implementing the fix...',
+        'Updating affected files...',
+      ], 5000, 'blue');
+
+      await runCopilot(
+        `Read ${tasksRelPath} for context and fix the following issue:\n\n${fallbackReason}\n\n` +
+        `Make targeted changes only — do not refactor anything unrelated to the issue.`,
+        { featureId }
+      );
+
+      stopCycle();
+      spinner.succeed('Fix dispatched to Copilot — run Review Code to verify');
+
+      updateStatus({ stage: 'implementation_complete', next: 'review' });
+      logActivity('COPILOT', 'Fix applied — stage reset to implementation_complete', 'success');
+      logActivity('PIPELINE', 'Run "Review Code" next to verify the fix.', 'info');
+    } else {
+      printError('No review file or tasks file found. Cannot dispatch to Copilot.');
+    }
+  }
+
+  printActivityFooter();
+}
+
+// ── Helpers ────────────────────────────────────────────────────────────────────
+
+function buildFixPrompt(reviewFile, reason, actionItems) {
+  const lines = [
+    `Read ${reviewFile}`,
+    `Fix all ✗ Blockers and ~ Warnings listed in the review.`,
+    '',
+    `Rejection reason: "${reason}"`,
+  ];
+  if (actionItems.length) {
+    lines.push('');
+    lines.push('Action Items:');
+    actionItems.forEach((item, i) => lines.push(`${i + 1}. ${item}`));
+  }
+  return lines.join('\n');
+}
+
+function parseActionItems(reviewFilePath) {
+  try {
+    const content       = readFileSync(reviewFilePath, 'utf8');
+    const actionSection = content.split(/## Action Items for Copilot/i)[1];
+    if (!actionSection) return [];
+    return actionSection
+      .split('\n')
+      .filter(l => /^\d+\./.test(l.trim()))
+      .map(l => l.replace(/^\d+\.\s*\*?\*?/, '').replace(/\*?\*?$/, '').trim())
+      .filter(Boolean);
+  } catch {
+    return [];
+  }
+}