@relipa/ai-flow-kit 0.1.1-beta.0 → 0.1.2-beta.0

package/scripts/task.js

@@ -1,546 +1,546 @@
-const fs = require('fs-extra');
-const path = require('path');
-const chalk = require('chalk');
-const { confirm } = require('@inquirer/prompts');
-
-const PROJECT_DIR   = process.cwd();
-const AIFLOW_DIR    = path.join(PROJECT_DIR, '.aiflow');
-const CONTEXT_DIR   = path.join(AIFLOW_DIR, 'context');
-const CURRENT_FILE  = path.join(CONTEXT_DIR, 'current.json');
-const TASKS_DIR     = path.join(AIFLOW_DIR, 'tasks');
-
-// ──────────────────────────────────────────────────────────────
-// Entry point
-// ──────────────────────────────────────────────────────────────
-
-module.exports = async function task(action, options = {}) {
-    try {
-        if (action === 'list')                return await listTasks();
-        if (action === 'status')              return await showStatus();
-        if (action === 'pause')               return await pauseTask(options.note);
-        if (action === 'switch' && options.taskId) return await switchTask(options.taskId);
-        if (action === 'resume' && options.taskId) return await resumeTask(options.taskId);
-        if (action === 'reset'  && options.taskId) return await resetTask(options.taskId);
-        if (action === 'remove' && options.taskId) return await removeTask(options.taskId);
-        if (action === 'next')                return await nextGate(options.taskId);
-
-        console.log(chalk.yellow('Usage:'));
-        console.log(chalk.gray('  aiflow task status              — show active task and pending list'));
-        console.log(chalk.gray('  aiflow task list                — list all saved tasks'));
-        console.log(chalk.gray('  aiflow task pause               — pause current task'));
-        console.log(chalk.gray('  aiflow task pause --note "..."  — pause with a note'));
-        console.log(chalk.gray('  aiflow task switch <ticket-id>  — pause current + switch to another task'));
-        console.log(chalk.gray('  aiflow task resume <ticket-id>  — resume a paused task'));
-        console.log(chalk.gray('  aiflow task reset  <ticket-id>  — reset task to Gate 1 (keeps context)'));
-        console.log(chalk.gray('  aiflow task remove <ticket-id>  — permanently delete all task data'));
-        console.log(chalk.gray('  aiflow task next                — approve current gate + prep next session'));
-        console.log(chalk.gray('  aiflow task next --ticket <id>  — same, specify ticket explicitly'));
-    } catch (err) {
-        console.error(chalk.red(`Error: ${err.message}`));
-    }
-};
-
-// ──────────────────────────────────────────────────────────────
-// Commands
-// ──────────────────────────────────────────────────────────────
-
-async function showStatus() {
-    const tasks = await loadAllTaskStates();
-    const active  = tasks.filter(t => t.status === 'active');
-    const pending = tasks.filter(t => t.status === 'pending');
-
-    console.log(chalk.cyan('\nTask Status\n'));
-
-    if (active.length === 0) {
-        console.log(chalk.gray('  No active task. Run `aiflow use <ticket>` to start one.'));
-    } else {
-        const t = active[0];
-        console.log(`  ${chalk.green('● Active')}  ${chalk.white(t.taskId)}  ${chalk.gray('Gate ' + t.currentGate)}  ${chalk.gray(t.title.substring(0, 55))}`);
-    }
-
-    if (pending.length > 0) {
-        console.log(chalk.yellow(`\n  Pending (${pending.length}):`));
-        for (const t of pending) {
-            const paused = t.pausedAt ? chalk.gray(` paused ${formatRelative(t.pausedAt)}`) : '';
-            const note   = t.notes    ? chalk.gray(` — ${t.notes.substring(0, 40)}`) : '';
-            console.log(`  ${chalk.gray('○ Pending')} ${chalk.white(t.taskId.padEnd(14))} Gate ${t.currentGate}  ${chalk.gray(t.title.substring(0, 40))}${paused}${note}`);
-        }
-        console.log(chalk.gray('\n  Resume with: aiflow task resume <ticket-id>'));
-    }
-    console.log();
-}
-
-async function listTasks() {
-    const tasks = await loadAllTaskStates();
-    if (tasks.length === 0) {
-        console.log(chalk.gray('No saved tasks.'));
-        return;
-    }
-    console.log(chalk.cyan('\nAll Tasks:\n'));
-    for (const t of tasks) {
-        const icon  = t.status === 'active' ? chalk.green('●') : chalk.gray('○');
-        const label = t.status === 'active' ? chalk.green('active ') : chalk.yellow('pending');
-        console.log(`  ${icon} ${label}  ${chalk.white(t.taskId.padEnd(16))} Gate ${t.currentGate}  ${chalk.gray(t.title.substring(0, 50))}`);
-    }
-    console.log();
-}
-
-async function pauseTask(note = '') {
-    if (!(await fs.pathExists(CURRENT_FILE))) {
-        console.log(chalk.yellow('No active task to pause.'));
-        return;
-    }
-
-    const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
-    if (!ctx || !ctx.taskId) {
-        console.log(chalk.yellow('Current context has no taskId — cannot pause.'));
-        return;
-    }
-
-    const taskId  = ctx.taskId;
-    const taskDir = path.join(TASKS_DIR, taskId);
-    await fs.ensureDir(taskDir);
-
-    // Save context snapshot
-    await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
-
-    // Detect gate from plan/ artifacts
-    const currentGate = await detectCurrentGate(taskId);
-
-    // Write / update task-state.json
-    const statePath = path.join(taskDir, 'task-state.json');
-    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
-    const now = new Date().toISOString();
-
-    const taskState = {
-        ...existing,
-        taskId,
-        title:       ctx.title || '',
-        status:      'pending',
-        createdAt:   existing.createdAt || now,
-        updatedAt:   now,
-        pausedAt:    now,
-        currentGate,
-        gateApprovals: existing.gateApprovals || {},
-        notes:       note || existing.notes || '',
-    };
-    await fs.writeJson(statePath, taskState, { spaces: 2 });
-
-    // Remove current.json so no task is active
-    await fs.remove(CURRENT_FILE);
-
-    console.log(chalk.green(`✓ Task ${taskId} paused at Gate ${currentGate}.`));
-    if (note) console.log(chalk.gray(`  Note: ${note}`));
-    console.log(chalk.gray(`  Resume later: aiflow task resume ${taskId}`));
-}
-
-async function switchTask(targetId) {
-    // Pause current task if one is active
-    if (await fs.pathExists(CURRENT_FILE)) {
-        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
-        if (ctx && ctx.taskId && ctx.taskId !== targetId) {
-            console.log(chalk.gray(`  Pausing current task: ${ctx.taskId}`));
-            await pauseTask();
-        }
-    }
-
-    // Load target task if it exists
-    const statePath = path.join(TASKS_DIR, targetId, 'task-state.json');
-    if (await fs.pathExists(statePath)) {
-        await resumeTask(targetId);
-    } else {
-        console.log(chalk.yellow(`Task ${targetId} not found in saved tasks.`));
-        console.log(chalk.gray(`  Run 'aiflow use ${targetId}' to load it from your ticket system.`));
-    }
-}
-
-async function resumeTask(taskId) {
-    const taskDir   = path.join(TASKS_DIR, taskId);
-    const statePath = path.join(taskDir, 'task-state.json');
-    const ctxPath   = path.join(taskDir, 'context.json');
-
-    if (!(await fs.pathExists(statePath))) {
-        console.log(chalk.red(`Task not found: ${taskId}`));
-        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
-        return;
-    }
-
-    const taskState = await fs.readJson(statePath);
-    const ctx       = await fs.readJson(ctxPath).catch(() => null);
-
-    if (!ctx) {
-        console.log(chalk.red(`Context snapshot missing for task: ${taskId}`));
-        console.log(chalk.gray(`  Try loading it fresh: aiflow use ${taskId}`));
-        return;
-    }
-
-    // Restore context as current
-    await fs.ensureDir(CONTEXT_DIR);
-    await fs.writeJson(CURRENT_FILE, ctx, { spaces: 2 });
-
-    // Deactivate any other tasks before marking this one active
-    await deactivateOtherTasks(taskId);
-
-    // Mark as active
-    const now = new Date().toISOString();
-    taskState.status    = 'active';
-    taskState.pausedAt  = null;
-    taskState.updatedAt = now;
-    await fs.writeJson(statePath, taskState, { spaces: 2 });
-
-    console.log(chalk.green(`✓ Resumed task: ${taskId}`));
-    console.log(`  ${chalk.white('Gate:')}   ${taskState.currentGate} (${gateLabel(taskState.currentGate)})`);
-    console.log(`  ${chalk.white('Title:')}  ${ctx.title.substring(0, 70)}`);
-    if (taskState.notes) console.log(`  ${chalk.white('Note:')}   ${taskState.notes}`);
-    console.log(chalk.gray('\n  Open Claude to resume: claude'));
-}
-
-async function resetTask(taskId) {
-    const taskDir   = path.join(TASKS_DIR, taskId);
-    const statePath = path.join(taskDir, 'task-state.json');
-
-    if (!(await fs.pathExists(statePath))) {
-        console.log(chalk.red(`Task not found: ${taskId}`));
-        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
-        return;
-    }
-
-    const taskState = await fs.readJson(statePath);
-    const planDir   = path.join(PROJECT_DIR, 'plan', taskId);
-    const planFiles = [];
-    if (await fs.pathExists(planDir)) {
-        const entries = await fs.readdir(planDir);
-        for (const f of entries) planFiles.push(path.join('plan', taskId, f));
-    }
-
-    // Show what will happen
-    console.log(chalk.cyan(`\nReset task: ${taskId}\n`));
-    console.log(chalk.white('  The following will be reset:'));
-    console.log(chalk.gray(`  • .aiflow/tasks/${taskId}/task-state.json`));
-    console.log(chalk.gray(`    Gate: ${taskState.currentGate} → 1  |  Approvals cleared  |  Status: active`));
-
-    const deletePlan = planFiles.length > 0 && await confirm({
-        message: `Also delete ${planFiles.length} plan file(s) in plan/${taskId}/?`,
-        default: false,
-    });
-
-    if (deletePlan) {
-        console.log(chalk.gray('\n  Plan files to delete:'));
-        for (const f of planFiles) console.log(chalk.gray(`  • ${f}`));
-    }
-
-    console.log();
-    const ok = await confirm({ message: 'Proceed with reset?', default: false });
-    if (!ok) { console.log(chalk.gray('Cancelled.')); return; }
-
-    // Reset task-state.json to Gate 1
-    const now = new Date().toISOString();
-    await fs.writeJson(statePath, {
-        ...taskState,
-        status:        'active',
-        currentGate:   1,
-        gateApprovals: {},
-        pausedAt:      null,
-        updatedAt:     now,
-        notes:         '',
-    }, { spaces: 2 });
-
-    // Restore as current context
-    const ctxPath = path.join(taskDir, 'context.json');
-    if (await fs.pathExists(ctxPath)) {
-        await fs.ensureDir(CONTEXT_DIR);
-        await fs.copy(ctxPath, CURRENT_FILE, { overwrite: true });
-    }
-
-    if (deletePlan) await fs.remove(planDir);
-
-    console.log(chalk.green(`✓ Task ${taskId} reset to Gate 1.`));
-    if (deletePlan) console.log(chalk.gray(`  Deleted plan/${taskId}/`));
-    console.log(chalk.gray('  Open Claude to start Gate 1: claude'));
-}
-
-async function removeTask(taskId) {
-    const taskDir   = path.join(TASKS_DIR, taskId);
-    const statePath = path.join(taskDir, 'task-state.json');
-
-    if (!(await fs.pathExists(statePath))) {
-        console.log(chalk.red(`Task not found: ${taskId}`));
-        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
-        return;
-    }
-
-    const isActive  = await fs.pathExists(CURRENT_FILE) &&
-        ((await fs.readJson(CURRENT_FILE).catch(() => ({}))).taskId === taskId);
-    const planDir   = path.join(PROJECT_DIR, 'plan', taskId);
-    const hasPlan   = await fs.pathExists(planDir);
-    const planFiles = hasPlan ? await fs.readdir(planDir) : [];
-
-    // Show what will be deleted
-    console.log(chalk.cyan(`\nRemove task: ${taskId}\n`));
-    console.log(chalk.white('  The following will be permanently deleted:'));
-    console.log(chalk.gray(`  • .aiflow/tasks/${taskId}/  (context snapshot + gate state)`));
-    if (isActive) console.log(chalk.gray(`  • .aiflow/context/current.json  (this is the active task)`));
-    if (hasPlan) {
-        console.log(chalk.gray(`  • plan/${taskId}/  (${planFiles.length} file(s)):`));
-        for (const f of planFiles) console.log(chalk.gray(`      - plan/${taskId}/${f}`));
-    }
-
-    console.log();
-    const ok = await confirm({
-        message: chalk.red('This cannot be undone. Proceed?'),
-        default: false,
-    });
-    if (!ok) { console.log(chalk.gray('Cancelled.')); return; }
-
-    await fs.remove(taskDir);
-    if (isActive) await fs.remove(CURRENT_FILE);
-    if (hasPlan)  await fs.remove(planDir);
-
-    console.log(chalk.green(`✓ Task ${taskId} removed.`));
-}
-
-async function nextGate(taskId) {
-    // Resolve ticket ID
-    let resolvedId = taskId;
-    if (!resolvedId) {
-        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
-        resolvedId = ctx && ctx.taskId ? ctx.taskId : null;
-    }
-    if (!resolvedId) {
-        console.log(chalk.yellow('No ticket ID provided and no active task found.'));
-        console.log(chalk.gray('  Usage: aiflow task next --ticket PROJ-123'));
-        return;
-    }
-
-    const taskDir   = path.join(TASKS_DIR, resolvedId);
-    const statePath = path.join(taskDir, 'task-state.json');
-    await fs.ensureDir(taskDir);
-
-    // Load or create task state
-    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
-    const currentGate = existing.currentGate || await detectCurrentGate(resolvedId);
-    const now = new Date().toISOString();
-
-    // Record gate approval
-    const gateApprovals = existing.gateApprovals || {};
-    gateApprovals[String(currentGate)] = now;
-    const nextGateNum = currentGate + 1;
-
-    // Save context snapshot if current task is active
-    if (await fs.pathExists(CURRENT_FILE)) {
-        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
-        if (ctx && ctx.taskId === resolvedId) {
-            await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
-        }
-    }
-
-    const taskState = {
-        ...existing,
-        taskId:        resolvedId,
-        status:        'pending',
-        updatedAt:     now,
-        pausedAt:      now,
-        currentGate:   nextGateNum,
-        gateApprovals,
-    };
-    await fs.writeJson(statePath, taskState, { spaces: 2 });
-
-    // Clear current context so no task is "active"
-    await fs.remove(CURRENT_FILE);
-
-    // Generate cumulative task-summary.md
-    const planDir = path.join(PROJECT_DIR, 'plan', resolvedId);
-    await fs.ensureDir(planDir);
-    const summaryPath = path.join(planDir, 'task-summary.md');
-    const summaryContent = await generateMarkdownSummary(taskState);
-    await fs.writeFile(summaryPath, summaryContent, 'utf-8');
-
-    const gateLabels = { 1: 'Analyze', 2: 'Plan', 3: 'Code', 4: 'Review', 5: 'PR' };
-    console.log(chalk.green(`✓ Gate ${currentGate} approved for ${resolvedId}.`));
-    console.log(chalk.gray(`  Summary saved to: plan/${resolvedId}/task-summary.md`));
-    console.log(chalk.white(`\n  Next: Gate ${nextGateNum} — ${gateLabels[nextGateNum] || 'Done'}`));
-    console.log(chalk.cyan(`\n  To continue in a fresh session (Recommended to avoid context pollution):`));
-    console.log(chalk.gray(`    1. Open a NEW chatbox or terminal session.`));
-    console.log(chalk.gray(`    2. Run: aiflow task resume ${resolvedId} (to load context).`));
-    console.log(chalk.gray(`    3. Type "start" or "continue from the current plan".`));
-    console.log(chalk.yellow(`       (Note: Gate 3 progress is saved via [x] checkboxes in plan.md.`));
-    console.log(chalk.yellow(`        The AI will automatically resume the exact task you left off.)`));
-    console.log();
-}
-
-// ──────────────────────────────────────────────────────────────
-// Internal helpers
-// ──────────────────────────────────────────────────────────────
-
-async function deactivateOtherTasks(exceptTaskId) {
-    if (!(await fs.pathExists(TASKS_DIR))) return;
-    const entries = await fs.readdir(TASKS_DIR, { withFileTypes: true });
-    for (const entry of entries) {
-        if (!entry.isDirectory() || entry.name === exceptTaskId) continue;
-        const statePath = path.join(TASKS_DIR, entry.name, 'task-state.json');
-        if (!(await fs.pathExists(statePath))) continue;
-        const state = await fs.readJson(statePath).catch(() => null);
-        if (state && state.status === 'active') {
-            const now = new Date().toISOString();
-            await fs.writeJson(statePath, { ...state, status: 'pending', updatedAt: now, pausedAt: now }, { spaces: 2 });
-        }
-    }
-}
-
-async function loadAllTaskStates() {
-    if (!(await fs.pathExists(TASKS_DIR))) return [];
-    const entries = await fs.readdir(TASKS_DIR, { withFileTypes: true });
-    const states  = [];
-    for (const entry of entries) {
-        if (!entry.isDirectory()) continue;
-        const statePath = path.join(TASKS_DIR, entry.name, 'task-state.json');
-        if (await fs.pathExists(statePath)) {
-            const s = await fs.readJson(statePath).catch(() => null);
-            if (s) states.push(s);
-        }
-    }
-    return states.sort((a, b) => (b.updatedAt || '').localeCompare(a.updatedAt || ''));
-}
-
-async function detectCurrentGate(taskId) {
-    const statePath = path.join(TASKS_DIR, taskId, 'task-state.json');
-    if (await fs.pathExists(statePath)) {
-        const state = await fs.readJson(statePath).catch(() => ({}));
-        if (state.currentGate) return state.currentGate;
-    }
-
-    const planDir = path.join(PROJECT_DIR, 'plan', taskId);
-    if (!(await fs.pathExists(planDir))) return 1;
-    if (await fs.pathExists(path.join(planDir, 'summary.md')))     return 5;
-    if (await fs.pathExists(path.join(planDir, 'plan.md')))        return 3;
-    if (await fs.pathExists(path.join(planDir, 'requirement.md'))) return 2;
-    return 1;
-}
-
-async function generateMarkdownSummary(taskState) {
-    const lines = [];
-    lines.push(`# Task Summary: ${taskState.taskId}`);
-    lines.push(`**Title:** ${taskState.title}`);
-    lines.push(`**Status:** ${taskState.status}`);
-    lines.push(`**Current Gate:** ${taskState.currentGate} (${gateLabel(taskState.currentGate)})`);
-    lines.push(`**Updated At:** ${new Date().toLocaleString()}`);
-    lines.push(``);
-    lines.push(`## Gate History`);
-    lines.push(`| Gate | Status | Approved At |`);
-    lines.push(`|------|--------|-------------|`);
-    for (let i = 1; i <= 5; i++) {
-        const approvedAt = taskState.gateApprovals && taskState.gateApprovals[String(i)];
-        const status = approvedAt ? '✅ Approved' : (i === taskState.currentGate ? '⏳ In Progress' : '⚪ Pending');
-        const timeStr = approvedAt ? new Date(approvedAt).toLocaleString() : '-';
-        lines.push(`| Gate ${i} (${gateLabel(i)}) | ${status} | ${timeStr} |`);
-    }
-    lines.push(``);
-    lines.push(`---`);
-    lines.push(`*Auto-generated by aiflow task next*`);
-    return lines.join('\n');
-}
-
-function gateLabel(n) {
-    const labels = {
-        1: 'AI Analyze Requirement',
-        2: 'Implementation Plan',
-        3: 'Code Generation',
-        4: 'AI Self-Review',
-        5: 'Peer Review & PR',
-    };
-    return labels[n] || `Gate ${n}`;
-}
-
-function formatRelative(isoStr) {
-    const ms   = Date.now() - new Date(isoStr).getTime();
-    const mins = Math.floor(ms / 60000);
-    if (mins < 60)  return `${mins}m ago`;
-    const hrs = Math.floor(mins / 60);
-    if (hrs  < 24)  return `${hrs}h ago`;
-    return `${Math.floor(hrs / 24)}d ago`;
-}
-
-// ──────────────────────────────────────────────────────────────
-// Export helper used by session-start hook and use.js
-// ──────────────────────────────────────────────────────────────
-
-module.exports.getActiveTaskState = async function getActiveTaskState() {
-    if (!(await fs.pathExists(CURRENT_FILE))) return null;
-    const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
-    if (!ctx || !ctx.taskId) return null;
-    const statePath = path.join(TASKS_DIR, ctx.taskId, 'task-state.json');
-    return (await fs.pathExists(statePath)) ? fs.readJson(statePath).catch(() => null) : null;
-};
-
-module.exports.createOrActivateTaskState = async function createOrActivateTaskState(ctx) {
-    const taskId  = ctx.taskId;
-    if (!taskId) return;
-    const taskDir   = path.join(TASKS_DIR, taskId);
-    const statePath = path.join(taskDir, 'task-state.json');
-    await fs.ensureDir(taskDir);
-
-    // Save context snapshot
-    await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
-
-    // Detect current gate
-    const currentGate = await detectCurrentGate(taskId);
-
-    // Deactivate any other tasks before marking this one active
-    await deactivateOtherTasks(taskId);
-
-    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
-    const now = new Date().toISOString();
-    const taskState = {
-        ...existing,
-        taskId,
-        title:         ctx.title || '',
-        status:        'active',
-        createdAt:     existing.createdAt || now,
-        updatedAt:     now,
-        pausedAt:      null,
-        currentGate:   Math.max(existing.currentGate || 1, currentGate),
-        gateApprovals: existing.gateApprovals || {},
-        notes:         existing.notes || '',
-    };
-    await fs.writeJson(statePath, taskState, { spaces: 2 });
-};
-
-/**
- * Called by `aiflow gate <N> <start|approved>` to keep task-state.json in sync
- * with the actual gate progress — without requiring the user to run `task next`.
- *
- * - 'start'   : advances currentGate to N (if N is higher than stored value)
- * - 'approved': records approval timestamp + advances currentGate to N+1
- *
- * Silently no-ops when task-state.json does not exist yet (task not saved).
- */
-module.exports.updateTaskGateState = async function updateTaskGateState(ticketId, gateNum, action) {
-    if (!ticketId) return;
-    const taskDir   = path.join(TASKS_DIR, ticketId);
-    const statePath = path.join(taskDir, 'task-state.json');
-    if (!(await fs.pathExists(statePath))) return; // task not yet saved — nothing to update
-
-    const existing = await fs.readJson(statePath).catch(() => null);
-    if (!existing) return;
-
-    const now     = new Date().toISOString();
-    const updated = { ...existing, updatedAt: now };
-
-    if (action === 'start') {
-        // Advance currentGate only if this gate is higher (never go backwards)
-        if (gateNum > (existing.currentGate || 1)) {
-            updated.currentGate = gateNum;
-        }
-    } else if (action === 'approved') {
-        // Record the approval timestamp
-        const gateApprovals = { ...(existing.gateApprovals || {}) };
-        gateApprovals[String(gateNum)] = now;
-        updated.gateApprovals = gateApprovals;
-        // Advance currentGate to next gate (only if not already ahead)
-        if (gateNum >= (existing.currentGate || 1)) {
-            updated.currentGate = gateNum + 1;
-        }
-    }
-
-    await fs.writeJson(statePath, updated, { spaces: 2 });
-};
+const fs = require('fs-extra');
+const path = require('path');
+const chalk = require('chalk');
+const { confirm } = require('@inquirer/prompts');
+
+const PROJECT_DIR   = process.cwd();
+const AIFLOW_DIR    = path.join(PROJECT_DIR, '.aiflow');
+const CONTEXT_DIR   = path.join(AIFLOW_DIR, 'context');
+const CURRENT_FILE  = path.join(CONTEXT_DIR, 'current.json');
+const TASKS_DIR     = path.join(AIFLOW_DIR, 'tasks');
+
+// ──────────────────────────────────────────────────────────────
+// Entry point
+// ──────────────────────────────────────────────────────────────
+
+module.exports = async function task(action, options = {}) {
+    try {
+        if (action === 'list')                return await listTasks();
+        if (action === 'status')              return await showStatus();
+        if (action === 'pause')               return await pauseTask(options.note);
+        if (action === 'switch' && options.taskId) return await switchTask(options.taskId);
+        if (action === 'resume' && options.taskId) return await resumeTask(options.taskId);
+        if (action === 'reset'  && options.taskId) return await resetTask(options.taskId);
+        if (action === 'remove' && options.taskId) return await removeTask(options.taskId);
+        if (action === 'next')                return await nextGate(options.taskId);
+
+        console.log(chalk.yellow('Usage:'));
+        console.log(chalk.gray('  aiflow task status              — show active task and pending list'));
+        console.log(chalk.gray('  aiflow task list                — list all saved tasks'));
+        console.log(chalk.gray('  aiflow task pause               — pause current task'));
+        console.log(chalk.gray('  aiflow task pause --note "..."  — pause with a note'));
+        console.log(chalk.gray('  aiflow task switch <ticket-id>  — pause current + switch to another task'));
+        console.log(chalk.gray('  aiflow task resume <ticket-id>  — resume a paused task'));
+        console.log(chalk.gray('  aiflow task reset  <ticket-id>  — reset task to Gate 1 (keeps context)'));
+        console.log(chalk.gray('  aiflow task remove <ticket-id>  — permanently delete all task data'));
+        console.log(chalk.gray('  aiflow task next                — approve current gate + prep next session'));
+        console.log(chalk.gray('  aiflow task next --ticket <id>  — same, specify ticket explicitly'));
+    } catch (err) {
+        console.error(chalk.red(`Error: ${err.message}`));
+    }
+};
+
+// ──────────────────────────────────────────────────────────────
+// Commands
+// ──────────────────────────────────────────────────────────────
+
+async function showStatus() {
+    const tasks = await loadAllTaskStates();
+    const active  = tasks.filter(t => t.status === 'active');
+    const pending = tasks.filter(t => t.status === 'pending');
+
+    console.log(chalk.cyan('\nTask Status\n'));
+
+    if (active.length === 0) {
+        console.log(chalk.gray('  No active task. Run `aiflow use <ticket>` to start one.'));
+    } else {
+        const t = active[0];
+        console.log(`  ${chalk.green('● Active')}  ${chalk.white(t.taskId)}  ${chalk.gray('Gate ' + t.currentGate)}  ${chalk.gray(t.title.substring(0, 55))}`);
+    }
+
+    if (pending.length > 0) {
+        console.log(chalk.yellow(`\n  Pending (${pending.length}):`));
+        for (const t of pending) {
+            const paused = t.pausedAt ? chalk.gray(` paused ${formatRelative(t.pausedAt)}`) : '';
+            const note   = t.notes    ? chalk.gray(` — ${t.notes.substring(0, 40)}`) : '';
+            console.log(`  ${chalk.gray('○ Pending')} ${chalk.white(t.taskId.padEnd(14))} Gate ${t.currentGate}  ${chalk.gray(t.title.substring(0, 40))}${paused}${note}`);
+        }
+        console.log(chalk.gray('\n  Resume with: aiflow task resume <ticket-id>'));
+    }
+    console.log();
+}
+
+async function listTasks() {
+    const tasks = await loadAllTaskStates();
+    if (tasks.length === 0) {
+        console.log(chalk.gray('No saved tasks.'));
+        return;
+    }
+    console.log(chalk.cyan('\nAll Tasks:\n'));
+    for (const t of tasks) {
+        const icon  = t.status === 'active' ? chalk.green('●') : chalk.gray('○');
+        const label = t.status === 'active' ? chalk.green('active ') : chalk.yellow('pending');
+        console.log(`  ${icon} ${label}  ${chalk.white(t.taskId.padEnd(16))} Gate ${t.currentGate}  ${chalk.gray(t.title.substring(0, 50))}`);
+    }
+    console.log();
+}
+
+async function pauseTask(note = '') {
+    if (!(await fs.pathExists(CURRENT_FILE))) {
+        console.log(chalk.yellow('No active task to pause.'));
+        return;
+    }
+
+    const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
+    if (!ctx || !ctx.taskId) {
+        console.log(chalk.yellow('Current context has no taskId — cannot pause.'));
+        return;
+    }
+
+    const taskId  = ctx.taskId;
+    const taskDir = path.join(TASKS_DIR, taskId);
+    await fs.ensureDir(taskDir);
+
+    // Save context snapshot
+    await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
+
+    // Detect gate from plan/ artifacts
+    const currentGate = await detectCurrentGate(taskId);
+
+    // Write / update task-state.json
+    const statePath = path.join(taskDir, 'task-state.json');
+    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
+    const now = new Date().toISOString();
+
+    const taskState = {
+        ...existing,
+        taskId,
+        title:       ctx.title || '',
+        status:      'pending',
+        createdAt:   existing.createdAt || now,
+        updatedAt:   now,
+        pausedAt:    now,
+        currentGate,
+        gateApprovals: existing.gateApprovals || {},
+        notes:       note || existing.notes || '',
+    };
+    await fs.writeJson(statePath, taskState, { spaces: 2 });
+
+    // Remove current.json so no task is active
+    await fs.remove(CURRENT_FILE);
+
+    console.log(chalk.green(`✓ Task ${taskId} paused at Gate ${currentGate}.`));
+    if (note) console.log(chalk.gray(`  Note: ${note}`));
+    console.log(chalk.gray(`  Resume later: aiflow task resume ${taskId}`));
+}
+
+async function switchTask(targetId) {
+    // Pause current task if one is active
+    if (await fs.pathExists(CURRENT_FILE)) {
+        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
+        if (ctx && ctx.taskId && ctx.taskId !== targetId) {
+            console.log(chalk.gray(`  Pausing current task: ${ctx.taskId}`));
+            await pauseTask();
+        }
+    }
+
+    // Load target task if it exists
+    const statePath = path.join(TASKS_DIR, targetId, 'task-state.json');
+    if (await fs.pathExists(statePath)) {
+        await resumeTask(targetId);
+    } else {
+        console.log(chalk.yellow(`Task ${targetId} not found in saved tasks.`));
+        console.log(chalk.gray(`  Run 'aiflow use ${targetId}' to load it from your ticket system.`));
+    }
+}
+
+async function resumeTask(taskId) {
+    const taskDir   = path.join(TASKS_DIR, taskId);
+    const statePath = path.join(taskDir, 'task-state.json');
+    const ctxPath   = path.join(taskDir, 'context.json');
+
+    if (!(await fs.pathExists(statePath))) {
+        console.log(chalk.red(`Task not found: ${taskId}`));
+        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
+        return;
+    }
+
+    const taskState = await fs.readJson(statePath);
+    const ctx       = await fs.readJson(ctxPath).catch(() => null);
+
+    if (!ctx) {
+        console.log(chalk.red(`Context snapshot missing for task: ${taskId}`));
+        console.log(chalk.gray(`  Try loading it fresh: aiflow use ${taskId}`));
+        return;
+    }
+
+    // Restore context as current
+    await fs.ensureDir(CONTEXT_DIR);
+    await fs.writeJson(CURRENT_FILE, ctx, { spaces: 2 });
+
+    // Deactivate any other tasks before marking this one active
+    await deactivateOtherTasks(taskId);
+
+    // Mark as active
+    const now = new Date().toISOString();
+    taskState.status    = 'active';
+    taskState.pausedAt  = null;
+    taskState.updatedAt = now;
+    await fs.writeJson(statePath, taskState, { spaces: 2 });
+
+    console.log(chalk.green(`✓ Resumed task: ${taskId}`));
+    console.log(`  ${chalk.white('Gate:')}   ${taskState.currentGate} (${gateLabel(taskState.currentGate)})`);
+    console.log(`  ${chalk.white('Title:')}  ${ctx.title.substring(0, 70)}`);
+    if (taskState.notes) console.log(`  ${chalk.white('Note:')}   ${taskState.notes}`);
+    console.log(chalk.gray('\n  Open Claude to resume: claude'));
+}
+
+async function resetTask(taskId) {
+    const taskDir   = path.join(TASKS_DIR, taskId);
+    const statePath = path.join(taskDir, 'task-state.json');
+
+    if (!(await fs.pathExists(statePath))) {
+        console.log(chalk.red(`Task not found: ${taskId}`));
+        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
+        return;
+    }
+
+    const taskState = await fs.readJson(statePath);
+    const planDir   = path.join(PROJECT_DIR, 'plan', taskId);
+    const planFiles = [];
+    if (await fs.pathExists(planDir)) {
+        const entries = await fs.readdir(planDir);
+        for (const f of entries) planFiles.push(path.join('plan', taskId, f));
+    }
+
+    // Show what will happen
+    console.log(chalk.cyan(`\nReset task: ${taskId}\n`));
+    console.log(chalk.white('  The following will be reset:'));
+    console.log(chalk.gray(`  • .aiflow/tasks/${taskId}/task-state.json`));
+    console.log(chalk.gray(`    Gate: ${taskState.currentGate} → 1  |  Approvals cleared  |  Status: active`));
+
+    const deletePlan = planFiles.length > 0 && await confirm({
+        message: `Also delete ${planFiles.length} plan file(s) in plan/${taskId}/?`,
+        default: false,
+    });
+
+    if (deletePlan) {
+        console.log(chalk.gray('\n  Plan files to delete:'));
+        for (const f of planFiles) console.log(chalk.gray(`  • ${f}`));
+    }
+
+    console.log();
+    const ok = await confirm({ message: 'Proceed with reset?', default: false });
+    if (!ok) { console.log(chalk.gray('Cancelled.')); return; }
+
+    // Reset task-state.json to Gate 1
+    const now = new Date().toISOString();
+    await fs.writeJson(statePath, {
+        ...taskState,
+        status:        'active',
+        currentGate:   1,
+        gateApprovals: {},
+        pausedAt:      null,
+        updatedAt:     now,
+        notes:         '',
+    }, { spaces: 2 });
+
+    // Restore as current context
+    const ctxPath = path.join(taskDir, 'context.json');
+    if (await fs.pathExists(ctxPath)) {
+        await fs.ensureDir(CONTEXT_DIR);
+        await fs.copy(ctxPath, CURRENT_FILE, { overwrite: true });
+    }
+
+    if (deletePlan) await fs.remove(planDir);
+
+    console.log(chalk.green(`✓ Task ${taskId} reset to Gate 1.`));
+    if (deletePlan) console.log(chalk.gray(`  Deleted plan/${taskId}/`));
+    console.log(chalk.gray('  Open Claude to start Gate 1: claude'));
+}
+
+async function removeTask(taskId) {
+    const taskDir   = path.join(TASKS_DIR, taskId);
+    const statePath = path.join(taskDir, 'task-state.json');
+
+    if (!(await fs.pathExists(statePath))) {
+        console.log(chalk.red(`Task not found: ${taskId}`));
+        console.log(chalk.gray('  Run `aiflow task list` to see available tasks.'));
+        return;
+    }
+
+    const isActive  = await fs.pathExists(CURRENT_FILE) &&
+        ((await fs.readJson(CURRENT_FILE).catch(() => ({}))).taskId === taskId);
+    const planDir   = path.join(PROJECT_DIR, 'plan', taskId);
+    const hasPlan   = await fs.pathExists(planDir);
+    const planFiles = hasPlan ? await fs.readdir(planDir) : [];
+
+    // Show what will be deleted
+    console.log(chalk.cyan(`\nRemove task: ${taskId}\n`));
+    console.log(chalk.white('  The following will be permanently deleted:'));
+    console.log(chalk.gray(`  • .aiflow/tasks/${taskId}/  (context snapshot + gate state)`));
+    if (isActive) console.log(chalk.gray(`  • .aiflow/context/current.json  (this is the active task)`));
+    if (hasPlan) {
+        console.log(chalk.gray(`  • plan/${taskId}/  (${planFiles.length} file(s)):`));
+        for (const f of planFiles) console.log(chalk.gray(`      - plan/${taskId}/${f}`));
+    }
+
+    console.log();
+    const ok = await confirm({
+        message: chalk.red('This cannot be undone. Proceed?'),
+        default: false,
+    });
+    if (!ok) { console.log(chalk.gray('Cancelled.')); return; }
+
+    await fs.remove(taskDir);
+    if (isActive) await fs.remove(CURRENT_FILE);
+    if (hasPlan)  await fs.remove(planDir);
+
+    console.log(chalk.green(`✓ Task ${taskId} removed.`));
+}
+
+async function nextGate(taskId) {
+    // Resolve ticket ID
+    let resolvedId = taskId;
+    if (!resolvedId) {
+        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
+        resolvedId = ctx && ctx.taskId ? ctx.taskId : null;
+    }
+    if (!resolvedId) {
+        console.log(chalk.yellow('No ticket ID provided and no active task found.'));
+        console.log(chalk.gray('  Usage: aiflow task next --ticket PROJ-123'));
+        return;
+    }
+
+    const taskDir   = path.join(TASKS_DIR, resolvedId);
+    const statePath = path.join(taskDir, 'task-state.json');
+    await fs.ensureDir(taskDir);
+
+    // Load or create task state
+    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
+    const currentGate = existing.currentGate || await detectCurrentGate(resolvedId);
+    const now = new Date().toISOString();
+
+    // Record gate approval
+    const gateApprovals = existing.gateApprovals || {};
+    gateApprovals[String(currentGate)] = now;
+    const nextGateNum = currentGate + 1;
+
+    // Save context snapshot if current task is active
+    if (await fs.pathExists(CURRENT_FILE)) {
+        const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
+        if (ctx && ctx.taskId === resolvedId) {
+            await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
+        }
+    }
+
+    const taskState = {
+        ...existing,
+        taskId:        resolvedId,
+        status:        'pending',
+        updatedAt:     now,
+        pausedAt:      now,
+        currentGate:   nextGateNum,
+        gateApprovals,
+    };
+    await fs.writeJson(statePath, taskState, { spaces: 2 });
+
+    // Clear current context so no task is "active"
+    await fs.remove(CURRENT_FILE);
+
+    // Generate cumulative task-summary.md
+    const planDir = path.join(PROJECT_DIR, 'plan', resolvedId);
+    await fs.ensureDir(planDir);
+    const summaryPath = path.join(planDir, 'task-summary.md');
+    const summaryContent = await generateMarkdownSummary(taskState);
+    await fs.writeFile(summaryPath, summaryContent, 'utf-8');
+
+    const gateLabels = { 1: 'Analyze', 2: 'Plan', 3: 'Code', 4: 'Review', 5: 'PR' };
+    console.log(chalk.green(`✓ Gate ${currentGate} approved for ${resolvedId}.`));
+    console.log(chalk.gray(`  Summary saved to: plan/${resolvedId}/task-summary.md`));
+    console.log(chalk.white(`\n  Next: Gate ${nextGateNum} — ${gateLabels[nextGateNum] || 'Done'}`));
+    console.log(chalk.cyan(`\n  To continue in a fresh session (Recommended to avoid context pollution):`));
+    console.log(chalk.gray(`    1. Open a NEW chatbox or terminal session.`));
+    console.log(chalk.gray(`    2. Run: aiflow task resume ${resolvedId} (to load context).`));
+    console.log(chalk.gray(`    3. Type "start" or "continue from the current plan".`));
+    console.log(chalk.yellow(`       (Note: Gate 3 progress is saved via [x] checkboxes in plan.md.`));
+    console.log(chalk.yellow(`        The AI will automatically resume the exact task you left off.)`));
+    console.log();
+}
+
+// ──────────────────────────────────────────────────────────────
+// Internal helpers
+// ──────────────────────────────────────────────────────────────
+
+async function deactivateOtherTasks(exceptTaskId) {
+    if (!(await fs.pathExists(TASKS_DIR))) return;
+    const entries = await fs.readdir(TASKS_DIR, { withFileTypes: true });
+    for (const entry of entries) {
+        if (!entry.isDirectory() || entry.name === exceptTaskId) continue;
+        const statePath = path.join(TASKS_DIR, entry.name, 'task-state.json');
+        if (!(await fs.pathExists(statePath))) continue;
+        const state = await fs.readJson(statePath).catch(() => null);
+        if (state && state.status === 'active') {
+            const now = new Date().toISOString();
+            await fs.writeJson(statePath, { ...state, status: 'pending', updatedAt: now, pausedAt: now }, { spaces: 2 });
+        }
+    }
+}
+
+async function loadAllTaskStates() {
+    if (!(await fs.pathExists(TASKS_DIR))) return [];
+    const entries = await fs.readdir(TASKS_DIR, { withFileTypes: true });
+    const states  = [];
+    for (const entry of entries) {
+        if (!entry.isDirectory()) continue;
+        const statePath = path.join(TASKS_DIR, entry.name, 'task-state.json');
+        if (await fs.pathExists(statePath)) {
+            const s = await fs.readJson(statePath).catch(() => null);
+            if (s) states.push(s);
+        }
+    }
+    return states.sort((a, b) => (b.updatedAt || '').localeCompare(a.updatedAt || ''));
+}
+
+async function detectCurrentGate(taskId) {
+    const statePath = path.join(TASKS_DIR, taskId, 'task-state.json');
+    if (await fs.pathExists(statePath)) {
+        const state = await fs.readJson(statePath).catch(() => ({}));
+        if (state.currentGate) return state.currentGate;
+    }
+
+    const planDir = path.join(PROJECT_DIR, 'plan', taskId);
+    if (!(await fs.pathExists(planDir))) return 1;
+    if (await fs.pathExists(path.join(planDir, 'summary.md')))     return 5;
+    if (await fs.pathExists(path.join(planDir, 'plan.md')))        return 3;
+    if (await fs.pathExists(path.join(planDir, 'requirement.md'))) return 2;
+    return 1;
+}
+
+async function generateMarkdownSummary(taskState) {
+    const lines = [];
+    lines.push(`# Task Summary: ${taskState.taskId}`);
+    lines.push(`**Title:** ${taskState.title}`);
+    lines.push(`**Status:** ${taskState.status}`);
+    lines.push(`**Current Gate:** ${taskState.currentGate} (${gateLabel(taskState.currentGate)})`);
+    lines.push(`**Updated At:** ${new Date().toLocaleString()}`);
+    lines.push(``);
+    lines.push(`## Gate History`);
+    lines.push(`| Gate | Status | Approved At |`);
+    lines.push(`|------|--------|-------------|`);
+    for (let i = 1; i <= 5; i++) {
+        const approvedAt = taskState.gateApprovals && taskState.gateApprovals[String(i)];
+        const status = approvedAt ? '✅ Approved' : (i === taskState.currentGate ? '⏳ In Progress' : '⚪ Pending');
+        const timeStr = approvedAt ? new Date(approvedAt).toLocaleString() : '-';
+        lines.push(`| Gate ${i} (${gateLabel(i)}) | ${status} | ${timeStr} |`);
+    }
+    lines.push(``);
+    lines.push(`---`);
+    lines.push(`*Auto-generated by aiflow task next*`);
+    return lines.join('\n');
+}
+
+function gateLabel(n) {
+    const labels = {
+        1: 'AI Analyze Requirement',
+        2: 'Implementation Plan',
+        3: 'Code Generation',
+        4: 'AI Self-Review',
+        5: 'Peer Review & PR',
+    };
+    return labels[n] || `Gate ${n}`;
+}
+
+function formatRelative(isoStr) {
+    const ms   = Date.now() - new Date(isoStr).getTime();
+    const mins = Math.floor(ms / 60000);
+    if (mins < 60)  return `${mins}m ago`;
+    const hrs = Math.floor(mins / 60);
+    if (hrs  < 24)  return `${hrs}h ago`;
+    return `${Math.floor(hrs / 24)}d ago`;
+}
+
+// ──────────────────────────────────────────────────────────────
+// Export helper used by session-start hook and use.js
+// ──────────────────────────────────────────────────────────────
+
+module.exports.getActiveTaskState = async function getActiveTaskState() {
+    if (!(await fs.pathExists(CURRENT_FILE))) return null;
+    const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
+    if (!ctx || !ctx.taskId) return null;
+    const statePath = path.join(TASKS_DIR, ctx.taskId, 'task-state.json');
+    return (await fs.pathExists(statePath)) ? fs.readJson(statePath).catch(() => null) : null;
+};
+
+module.exports.createOrActivateTaskState = async function createOrActivateTaskState(ctx) {
+    const taskId  = ctx.taskId;
+    if (!taskId) return;
+    const taskDir   = path.join(TASKS_DIR, taskId);
+    const statePath = path.join(taskDir, 'task-state.json');
+    await fs.ensureDir(taskDir);
+
+    // Save context snapshot
+    await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
+
+    // Detect current gate
+    const currentGate = await detectCurrentGate(taskId);
+
+    // Deactivate any other tasks before marking this one active
+    await deactivateOtherTasks(taskId);
+
+    const existing  = (await fs.pathExists(statePath)) ? await fs.readJson(statePath).catch(() => ({})) : {};
+    const now = new Date().toISOString();
+    const taskState = {
+        ...existing,
+        taskId,
+        title:         ctx.title || '',
+        status:        'active',
+        createdAt:     existing.createdAt || now,
+        updatedAt:     now,
+        pausedAt:      null,
+        currentGate:   Math.max(existing.currentGate || 1, currentGate),
+        gateApprovals: existing.gateApprovals || {},
+        notes:         existing.notes || '',
+    };
+    await fs.writeJson(statePath, taskState, { spaces: 2 });
+};
+
+/**
+ * Called by `aiflow gate <N> <start|approved>` to keep task-state.json in sync
+ * with the actual gate progress — without requiring the user to run `task next`.
+ *
+ * - 'start'   : advances currentGate to N (if N is higher than stored value)
+ * - 'approved': records approval timestamp + advances currentGate to N+1
+ *
+ * Silently no-ops when task-state.json does not exist yet (task not saved).
+ */
+module.exports.updateTaskGateState = async function updateTaskGateState(ticketId, gateNum, action) {
+    if (!ticketId) return;
+    const taskDir   = path.join(TASKS_DIR, ticketId);
+    const statePath = path.join(taskDir, 'task-state.json');
+    if (!(await fs.pathExists(statePath))) return; // task not yet saved — nothing to update
+
+    const existing = await fs.readJson(statePath).catch(() => null);
+    if (!existing) return;
+
+    const now     = new Date().toISOString();
+    const updated = { ...existing, updatedAt: now };
+
+    if (action === 'start') {
+        // Advance currentGate only if this gate is higher (never go backwards)
+        if (gateNum > (existing.currentGate || 1)) {
+            updated.currentGate = gateNum;
+        }
+    } else if (action === 'approved') {
+        // Record the approval timestamp
+        const gateApprovals = { ...(existing.gateApprovals || {}) };
+        gateApprovals[String(gateNum)] = now;
+        updated.gateApprovals = gateApprovals;
+        // Advance currentGate to next gate (only if not already ahead)
+        if (gateNum >= (existing.currentGate || 1)) {
+            updated.currentGate = gateNum + 1;
+        }
+    }
+
+    await fs.writeJson(statePath, updated, { spaces: 2 });
+};