@relipa/ai-flow-kit 0.1.9 → 0.2.0-beta.1

package/scripts/link-resolver.js

@@ -3,8 +3,14 @@ const https = require('https');
 const chalk = require('chalk');
 
 const BACKLOG_RE = /https?:\/\/([\w.-]+\.backlog(?:tool)?\.com)\/view\/([A-Z][A-Z0-9_]+-\d+)(?:#comment-(\d+))?/i;
+// Backlog Document permalink — "Copy link" on a Document page gives /alias/document/{id}.
+// NOTE: Nulab's Document REST API is new (2026) — path unverified against a live space, adjust if it 404s.
+const BACKLOG_DOCUMENT_RE = /https?:\/\/([\w.-]+\.backlog(?:tool)?\.com)\/(?:alias\/)?document\/(\d+)(?:#comment-(\d+))?/i;
+// Backlog Wiki permalink — "Copy link" on a Wiki page gives /alias/wiki/{id}.
+const BACKLOG_WIKI_RE = /https?:\/\/([\w.-]+\.backlog(?:tool)?\.com)\/alias\/wiki\/(\d+)/i;
 const JIRA_RE = /https?:\/\/([\w.-]+\.atlassian\.net)\/browse\/([A-Z][A-Z0-9_]+-\d+)/i;
 const JIRA_COMMENT_RE = /[?&]focusedCommentId=(\d+)/;
+const SHAREPOINT_RE = /https?:\/\/([\w.-]+\.sharepoint\.com)\/\S*/i;
 const URL_RE = /https?:\/\/[^\s\)\]"<>]+/g;
 
 const MAX_AUTO_LINKS = 5;
@@ -13,25 +19,43 @@ function classifyLink(url) {
   const bm = url.match(BACKLOG_RE);
   if (bm) {
     const [, domain, ticketId, commentId] = bm;
-    return { adapter: 'backlog', domain, ticketId, commentId: commentId || null, isComment: !!commentId };
+    return { adapter: 'backlog', kind: 'ticket', domain, ticketId, commentId: commentId || null, isComment: !!commentId };
+  }
+  const dm = url.match(BACKLOG_DOCUMENT_RE);
+  if (dm) {
+    const [, domain, documentId, commentId] = dm;
+    return { adapter: 'backlog', kind: 'document', domain, documentId, commentId: commentId || null, isComment: !!commentId };
+  }
+  const wm = url.match(BACKLOG_WIKI_RE);
+  if (wm) {
+    const [, domain, wikiId] = wm;
+    return { adapter: 'backlog', kind: 'wiki', domain, wikiId };
   }
   const jm = url.match(JIRA_RE);
   if (jm) {
     const [, domain, ticketId] = jm;
     const cm = url.match(JIRA_COMMENT_RE);
     const commentId = cm ? cm[1] : null;
-    return { adapter: 'jira', domain, ticketId, commentId, isComment: !!commentId };
+    return { adapter: 'jira', kind: 'ticket', domain, ticketId, commentId, isComment: !!commentId };
+  }
+  const sm = url.match(SHAREPOINT_RE);
+  if (sm) {
+    const [, domain] = sm;
+    return { adapter: 'sharepoint', kind: 'unsupported', domain };
   }
   return null;
 }
 
+// Auto-resolve (from a ticket description, capped at MAX_AUTO_LINKS) only ever
+// followed ticket/comment links. Document/Wiki/SharePoint are explicit, single-URL
+// lookups only (`ak fetch-links <url>`) — never auto-fetched en masse.
 function scanLinks(text) {
   const found = [];
   let m;
   URL_RE.lastIndex = 0;
   while ((m = URL_RE.exec(text)) !== null) {
     const classified = classifyLink(m[0]);
-    if (classified) found.push({ url: m[0], ...classified });
+    if (classified && classified.kind === 'ticket') found.push({ url: m[0], ...classified });
   }
   return found;
 }
@@ -54,6 +78,41 @@ function httpsGet(url, headers) {
   });
 }
 
+// Generic POST — `bodyType` picks the wire format:
+//   'form' → application/x-www-form-urlencoded (Backlog's write API)
+//   'json' → application/json (Jira's write API)
+function httpsPost(url, payload, { headers = {}, bodyType = 'json' } = {}) {
+  return new Promise((resolve, reject) => {
+    const body = bodyType === 'form'
+      ? new URLSearchParams(payload).toString()
+      : JSON.stringify(payload);
+    const reqHeaders = {
+      ...headers,
+      'Content-Type': bodyType === 'form' ? 'application/x-www-form-urlencoded' : 'application/json',
+      'Content-Length': Buffer.byteLength(body),
+    };
+    const { hostname, pathname, search } = new URL(url);
+    const req = https.request(
+      { hostname, path: pathname + search, method: 'POST', headers: reqHeaders },
+      (res) => {
+        let data = '';
+        res.on('data', chunk => (data += chunk));
+        res.on('end', () => {
+          if (res.statusCode < 200 || res.statusCode >= 300) {
+            reject(new Error(`HTTP ${res.statusCode}: ${data.substring(0, 500)}`));
+            return;
+          }
+          try { resolve(data ? JSON.parse(data) : {}); }
+          catch (e) { reject(new Error('Invalid JSON from API')); }
+        });
+      }
+    );
+    req.on('error', reject);
+    req.write(body);
+    req.end();
+  });
+}
+
 async function fetchBacklogTicket(domain, apiKey, ticketId) {
   const issue = await httpsGet(`https://${domain}/api/v2/issues/${ticketId}?apiKey=${apiKey}`);
   return {
@@ -83,6 +142,77 @@ async function fetchBacklogComment(domain, apiKey, ticketId, commentId) {
   };
 }
 
+async function fetchBacklogDocument(domain, apiKey, documentId) {
+  const doc = await httpsGet(`https://${domain}/api/v2/documents/${documentId}?apiKey=${apiKey}`);
+  return {
+    sourceType: 'document',
+    sourceUrl: `https://${domain}/document/${documentId}`,
+    documentId: String(documentId),
+    title: doc.title || doc.name || '',
+    tag: (doc.tags || []).map(t => t.name).join(', '),
+    content: (doc.plainText || doc.content || '').substring(0, 4000),
+  };
+}
+
+async function fetchBacklogDocumentComment(domain, apiKey, documentId, commentId) {
+  const comments = await httpsGet(
+    `https://${domain}/api/v2/documents/${documentId}/comments?apiKey=${apiKey}&count=100&order=asc`
+  );
+  const comment = (comments || []).find(c => String(c.id) === String(commentId));
+  if (!comment) throw new Error(`Comment ${commentId} not found in document ${documentId}`);
+  const date = comment.created ? new Date(comment.created).toLocaleDateString('vi-VN') : '';
+  return {
+    sourceType: 'document-comment',
+    sourceUrl: `https://${domain}/document/${documentId}#comment-${commentId}`,
+    documentId: String(documentId),
+    commentId: String(commentId),
+    author: comment.createdUser?.name || 'Unknown',
+    date,
+    content: (comment.content || '').substring(0, 2000),
+  };
+}
+
+async function fetchBacklogWiki(domain, apiKey, wikiId) {
+  const wiki = await httpsGet(`https://${domain}/api/v2/wikis/${wikiId}?apiKey=${apiKey}`);
+  return {
+    sourceType: 'wiki',
+    sourceUrl: `https://${domain}/alias/wiki/${wikiId}`,
+    wikiId: String(wikiId),
+    title: wiki.name || '',
+    content: (wiki.content || '').substring(0, 4000),
+  };
+}
+
+// ── Backlog WRITE — issue creation (PM Workflow_v1.0.md "Vấn đề 5") ──────────
+
+async function fetchBacklogIssueTypes(domain, apiKey, projectIdOrKey) {
+  return httpsGet(`https://${domain}/api/v2/projects/${projectIdOrKey}/issueTypes?apiKey=${apiKey}`);
+}
+
+async function fetchBacklogPriorities(domain, apiKey) {
+  return httpsGet(`https://${domain}/api/v2/priorities?apiKey=${apiKey}`);
+}
+
+/**
+ * Creates a Backlog issue. Requires an apiKey with WRITE permission — distinct from the
+ * read-only key used elsewhere in this file for `ak fetch-links`/`ak use`.
+ * `projectId`, `summary`, `issueTypeId`, `priorityId` are mandatory per Backlog's API.
+ */
+async function createBacklogIssue(domain, apiKeyWrite, { projectId, summary, description, issueTypeId, priorityId }) {
+  const issue = await httpsPost(
+    `https://${domain}/api/v2/issues?apiKey=${apiKeyWrite}`,
+    {
+      projectId: String(projectId),
+      summary,
+      issueTypeId: String(issueTypeId),
+      priorityId: String(priorityId),
+      description: description || '',
+    },
+    { bodyType: 'form' }
+  );
+  return { ticketId: issue.issueKey, url: `https://${domain}/view/${issue.issueKey}`, raw: issue };
+}
+
 function jiraAuth(email, token) {
   return `Basic ${Buffer.from(`${email}:${token}`).toString('base64')}`;
 }
@@ -126,17 +256,67 @@ async function fetchJiraComment(domain, email, token, ticketId, commentId) {
   };
 }
 
+// ── Jira WRITE — issue creation (PM Workflow_v1.0.md "Vấn đề 5") ────────────
+
+/**
+ * Creates a Jira issue. `apiTokenWrite`/`emailWrite` are distinct from the read-only
+ * credentials used elsewhere in this file — same "separate WRITE key" policy as Backlog.
+ */
+async function createJiraIssue(domain, emailWrite, apiTokenWrite, { projectKey, summary, description, issueTypeName }) {
+  const issue = await httpsPost(
+    `https://${domain}/rest/api/3/issue`,
+    {
+      fields: {
+        project: { key: projectKey },
+        summary,
+        issuetype: { name: issueTypeName || 'Task' },
+        description: {
+          type: 'doc',
+          version: 1,
+          content: [{ type: 'paragraph', content: [{ type: 'text', text: description || '' }] }],
+        },
+      },
+    },
+    { headers: { Authorization: jiraAuth(emailWrite, apiTokenWrite), Accept: 'application/json' }, bodyType: 'json' }
+  );
+  return { ticketId: issue.key, url: `https://${domain}/browse/${issue.key}`, raw: issue };
+}
+
+function linkDedupeKey(classified) {
+  if (classified.kind === 'document') return 'doc:' + classified.documentId + (classified.commentId ? '#' + classified.commentId : '');
+  if (classified.kind === 'wiki') return 'wiki:' + classified.wikiId;
+  if (classified.ticketId) return classified.ticketId + (classified.commentId ? '#' + classified.commentId : '');
+  return null;
+}
+
 async function fetchLink(url, credentials = {}, alreadyLoadedIds = []) {
   const classified = classifyLink(url);
   if (!classified) return null;
 
-  const { adapter, domain, ticketId, commentId, isComment } = classified;
-  const dedupeKey = ticketId + (commentId ? '#' + commentId : '');
-  if (alreadyLoadedIds.includes(dedupeKey) || alreadyLoadedIds.includes(ticketId)) return null;
+  const dedupeKey = linkDedupeKey(classified);
+  if (dedupeKey && (alreadyLoadedIds.includes(dedupeKey) || alreadyLoadedIds.includes(classified.ticketId))) return null;
+
+  const { adapter, kind, domain, ticketId, commentId, documentId, wikiId, isComment } = classified;
+
+  if (adapter === 'sharepoint') {
+    return {
+      sourceType: 'unsupported',
+      reason: 'sharepoint-not-configured',
+      sourceUrl: url,
+      message:
+        'SharePoint chưa có connector (cần Microsoft Graph API + OAuth — xem "Vấn đề 4" trong PM Workflow_v1.0.md). ' +
+        'Copy/paste nội dung comment/tài liệu trực tiếp vào chat thay vì dán link.',
+    };
+  }
 
   if (adapter === 'backlog') {
     const apiKey = credentials.BACKLOG_API_KEY;
     if (!apiKey) throw new Error('BACKLOG_API_KEY not set');
+    if (kind === 'document') {
+      if (isComment) return await fetchBacklogDocumentComment(domain, apiKey, documentId, commentId);
+      return await fetchBacklogDocument(domain, apiKey, documentId);
+    }
+    if (kind === 'wiki') return await fetchBacklogWiki(domain, apiKey, wikiId);
     if (isComment) return await fetchBacklogComment(domain, apiKey, ticketId, commentId);
     return await fetchBacklogTicket(domain, apiKey, ticketId);
   }
@@ -176,4 +356,11 @@ async function resolveLinks(description, credentials = {}, alreadyLoadedIds = []
   return results;
 }
 
-module.exports = { classifyLink, scanLinks, fetchBacklogTicket, fetchBacklogComment, fetchJiraTicket, fetchJiraComment, fetchLink, resolveLinks };
+module.exports = {
+  classifyLink, scanLinks, fetchLink, resolveLinks,
+  fetchBacklogTicket, fetchBacklogComment,
+  fetchBacklogDocument, fetchBacklogDocumentComment, fetchBacklogWiki,
+  fetchJiraTicket, fetchJiraComment,
+  fetchBacklogIssueTypes, fetchBacklogPriorities, createBacklogIssue,
+  createJiraIssue,
+};

package/scripts/prompt.js

@@ -278,6 +278,150 @@ Write documentation based on approved outline.
 **INVOKE:** \`superpowers:verification-before-completion\`
 Verify docs match code. No outdated examples.
 Display: "GATE 3: Documentation complete. Type APPROVED."
+`,
+    },
+    'create-spec': {
+        header: 'Create UC Spec',
+        instruction: 'Create a complete UC Spec from the raw requirement described below, following the BA 4-Gate workflow.',
+        skillWorkflow: `
+## STRICT GATE WORKFLOW — [BA] 4-Gate Spec Creation
+
+Follow gates IN ORDER. Do NOT skip or merge gates.
+
+### GATE 1 — Analyze Requirement & Draft Q&A (auto-start)
+**INVOKE:** \`read-study-requirement\` skill, then the \`ba-skills\` sub-skills for analysis
+1. Confirm \`functionId\` and ensure AK-Docs is on branch \`feature/[functionId]/[ticketId]\`
+2. Read the raw requirement (ticket/backlog/file) + related source code
+3. Identify Gaps/Assumptions, draft a Q&A list — classify every item as Fact / Assumption / Gap
+4. Output \`AK-Docs/02.BA-Specs/01.QnA/[functionId]/[ticketId].md\`
+5. Display "GATE 1: Q&A draft ready" → wait for **APPROVED**
+
+### GATE 2 — Update Q&A (loop until all Confirmed)
+Integrate BA/stakeholder answers into the analysis doc. Loop: unresolved items remain → keep asking ONE question at a time. All Confirmed → display "GATE 2: All items Confirmed" → wait for **APPROVED**.
+
+### GATE 3 — UI Prototype (HTML/CSS)
+Only runs after Gate 2 APPROVED.
+1. List UI components from the confirmed requirement
+2. Design and write static HTML/CSS prototype (incl. error-state placeholders)
+3. Save under \`AK-Docs/02.BA-Specs/03.Prototypes/[functionId]/\`
+4. Display "GATE 3: Prototype ready" → wait for **APPROVED**
+
+### GATE 4 — Full UC Spec (wait for APPROVED)
+Only runs after Gate 3 APPROVED.
+1. Fill the UC Spec template: Flow of Events (Main/Alternative/Exception), UI Components, Mermaid activity diagram, Business Rules
+2. Save \`AK-Docs/02.BA-Specs/04.UC-Specs/[functionId]/UC-Spec_v[N].md\`
+3. Retrospect — propose a memory draft for any new business rule/decision learned
+4. Submit AK-Docs via Merge Request (title/description shown, explicit user confirmation) — PM reviews & merges into \`main\`
+5. Display "GATE 4: UC Spec ready" → wait for **APPROVED**
+`,
+    },
+    'create-testcase': {
+        header: 'Create Test Cases',
+        instruction: 'Create a complete, execution-ready Test Case set from the requirement described below, following the QA 4-Gate workflow.',
+        skillWorkflow: `
+## STRICT GATE WORKFLOW — [QA] 4-Gate TestCase Creation
+
+Follow gates IN ORDER. Do NOT skip or merge gates.
+
+### GATE 1 — Requirement & Risk Analysis (auto-start)
+**INVOKE:** \`read-study-requirement\` skill, then the \`test-skills\` sub-skills
+1. Confirm \`functionId\` and ensure AK-Docs is on branch \`feature/[functionId]/[ticketId]\`
+2. Read the requirement (SRS/Backlog/Jira/UC Spec) and assess risk areas
+3. Clarify with QA/stakeholder if anything is ambiguous — ask ONE question at a time
+4. Output requirement + risk analysis under \`AK-Docs/03.QA-Specs/01.Requirements/[functionId]/[ticketId].md\`
+5. Display "GATE 1: Requirement & risk analysis ready" → wait for **APPROVED**
+
+### GATE 2 — Scenario Building
+Only runs after Gate 1 APPROVED.
+Apply scenario-building skills to derive test scenarios (happy path, edge case, negative). Output scenario list. Display "GATE 2: Scenarios ready" → wait for **APPROVED**.
+
+### GATE 3 — Detailed Test Case Design
+Only runs after Gate 2 APPROVED.
+1. Apply TC design techniques (equivalence partitioning, boundary values, etc.)
+2. Design UI, business, and data test cases
+3. Output draft Test Cases under \`AK-Docs/03.QA-Specs/03.TestCases/[functionId]/\`
+4. Display "GATE 3: Draft Test Cases ready" → wait for **APPROVED**
+
+### GATE 4 — Review & Optimize (wait for APPROVED)
+Only runs after Gate 3 APPROVED.
+1. Review coverage against requirements, optimize/dedupe test cases
+2. Confirm sign-off status, generate the final TestCase artifact
+3. Retrospect — propose a memory draft for any lesson learned
+4. Submit AK-Docs via Merge Request (title/description shown, explicit user confirmation) — PM reviews & merges into \`main\`
+5. Display "GATE 4: Final Test Cases ready" → wait for **APPROVED**
+`,
+    },
+    'create-system-requirement': {
+        header: 'Create System Requirement',
+        instruction: 'Bridge the UC Spec into a Dev-facing System Requirement, following the 2-Gate workflow.',
+        skillWorkflow: `
+## STRICT GATE WORKFLOW — Create System Requirement (2-Gate Bridge)
+
+> Runs once per \`functionId\` per UC Spec version — NOT per ticket. Blocks coding Gate 1 until APPROVED.
+
+### GATE 1 — Investigate & Draft (auto-start)
+**INVOKE:** \`create-system-requirement\` skill
+1. Resolve \`functionId\`; locate the current UC Spec (\`AK-Docs/02.BA-Specs/04.UC-Specs/[functionId]/UC-Spec_v[N].md\`) — not found → STOP, tell DEV the UC Spec must exist and be BA-signed-off first
+2. Read the full UC Spec, then investigate source code (existing error-handling conventions, data model/API surface)
+3. Translate UC → Functional/Non-Functional Requirements, Business Rules → Validation Rules, Exception & Error Handling, Acceptance Test Scenarios — every item traced to a UC reference, no invented content
+4. Classify every item as Fact / Assumption / Gap; ask ONE question at a time until every Gap is Confirmed
+5. Display "GATE 1: All Gaps Confirmed" → wait for **APPROVED**
+
+### GATE 2 — Finalize & Approve (wait for APPROVED)
+Only runs after Gate 1 APPROVED.
+1. Decide single-file vs split (default: single file mirroring the UC Spec 1-1)
+2. Write \`AK-Docs/02.BA-Specs/00.Requirements/[functionId]/System-Requirement_v[N].md\` with the Traceability Matrix + all sections
+3. Display "GATE 2: System Requirement ready" with coverage counts (Open Gaps must be 0) → wait for **APPROVED**
+
+**On APPROVED:** coding Gate 1 (\`read-study-requirement\`) unlocks for this functionId.
+`,
+    },
+    'execute': {
+        header: 'Execute Test Cases',
+        instruction: 'Execute the Test Case file described below against the running app, following the EXECUTE 4-Gate workflow.',
+        skillWorkflow: `
+## STRICT GATE WORKFLOW — [EXECUTE] Executing Flow
+
+Follow gates IN ORDER. Do NOT skip or merge gates.
+
+### GATE 1 — Pre-flight & Work Plan (auto-start)
+**INVOKE:** \`execute-flow\` skill
+Check: MCP Playwright configured, TC file is parseable, \`ak-test/{repo}/\` scaffold exists, \`BASE_URL\` set, Playwright config exists. Display the work plan (TC file, repo, screen, BASE_URL) → wait for **APPROVED**.
+
+### GATE 2 — Script Sync
+Only runs after Gate 1 APPROVED.
+**INVOKE:** \`script-sync\` skill
+Hash-based TC↔script sync: new TC → generate Playwright test via MCP snapshot/locator (never fabricate selectors); changed TC → update; unchanged → skip; manual/blocked → mark Untest/Pending. Display new/updated/skipped/manual counts → wait for **APPROVED**.
+
+### GATE 3 — Execute & Evidence
+Only runs after Gate 2 APPROVED.
+Run \`BASE_URL=<url> npx playwright test scripts/{screenId}/{ScreenID}.spec.ts\`, organize evidence (screenshots/trace/result.md) into \`ak-test/{repo}/results/{screenId}/run-{N}/\`, auto-draft bug reports for failures, update R1/R2 columns in the TC file. Supports \`RETEST: [TC_ID]\` and \`PR: [url]\`. Display pass/fail/untest/pending counts → wait for **APPROVED** (all Critical/High bugs resolved).
+
+### GATE 4 — Report & Bug Logging (wait for APPROVED)
+Only runs after Gate 3 APPROVED.
+Generate \`testreport.md\` (Go/No-Go: 0 unresolved Critical = Go), log each failed TC as a bug one at a time (ask Y/N/edit before logging). Display decision + bug counts → wait for **APPROVED** to sign off.
+`,
+    },
+    'ingest-data': {
+        header: 'Ingest Data',
+        instruction: 'Ingest the customer communication (Backlog/Jira/SharePoint link or pasted text) described below into AK-Docs logs, following the 2-Gate workflow.',
+        skillWorkflow: `
+## STRICT GATE WORKFLOW — Ingest Data (2-Gate)
+
+**INVOKE:** \`ingest-data\` skill for both gates — it owns the classification rules, draft templates, and per-file format.
+
+### GATE 1 — Fetch, Classify, Draft (auto-start)
+1. Resolve the source: Backlog/Jira ticket/comment/Document/Wiki URL → \`ak fetch-links "<url>"\`; SharePoint → not yet connected, ask for pasted text instead; plain text → use as-is
+2. Classify into \`QnA-Log.md\` / \`Meetings-Log.md\` / \`Confirmations-Log.md\` — ask the user if ambiguous, never guess silently
+3. Draft the entry using the exact template for the target log file, including trace fields (Nguồn, functionId, Người tổng hợp, Người approve tại nguồn)
+4. Display the full draft → wait for **APPROVED** (or edit requests, looped until approved)
+
+### GATE 2 — Branch + Merge Request (runs immediately after APPROVED)
+1. Resolve \`functionId\` (tag from source → inferred from input → fallback \`general\`)
+2. Ensure AK-Docs is on branch \`feature/[functionId]/[ticketId]\` (create only after explicit confirmation)
+3. Append the approved entry to the target log file (create with header template if missing)
+4. Submit AK-Docs via Merge Request (title/description shown, explicit user confirmation) — PM reviews & merges into \`main\`
+5. Display "GATE 2 DONE: Entry written to AK-Docs/01.QnA/[file]"
 `,
     },
     'gen-doc': {

package/scripts/task.js

@@ -331,21 +331,27 @@ async function nextGate(taskId) {
     const nextGateNum = currentGate + 1;
 
     // Save context snapshot if current task is active
-    let taskType = null;
+    // taskType falls back to task-state.json's own stored value — current.json is removed
+    // at the end of this function, so on Gate 2+ of a multi-step task it's already gone.
+    let taskType = existing.taskType || null;
     if (await fs.pathExists(CURRENT_FILE)) {
         const ctx = await fs.readJson(CURRENT_FILE).catch(() => null);
         if (ctx && ctx.taskId === resolvedId) {
-            taskType = ctx.taskType || null;
+            taskType = ctx.taskType || taskType;
             await fs.writeJson(path.join(taskDir, 'context.json'), ctx, { spaces: 2 });
         }
     }
 
     const isGenDoc = taskType === 'gen-doc';
-    const maxGate = isGenDoc ? 2 : 5;
+    const isSystemRequirement = taskType === 'create-system-requirement';
+    const isIngestData = taskType === 'ingest-data';
+    const isTwoGateDoc = isGenDoc || isSystemRequirement || isIngestData;
+    const maxGate = isTwoGateDoc ? 2 : 5;
 
     const taskState = {
         ...existing,
         taskId:        resolvedId,
+        taskType:      taskType || existing.taskType || null,
         status:        nextGateNum > maxGate ? 'done' : 'pending',
         updatedAt:     now,
         pausedAt:      now,
@@ -365,7 +371,7 @@ async function nextGate(taskId) {
     console.log(chalk.green(`✓ Gate ${currentGate} approved for ${resolvedId}.`));
     console.log(chalk.gray(`  Summary saved to: .aiflow/tasks/${resolvedId}/task-summary.md`));
 
-    if (isGenDoc && currentGate >= maxGate) {
+    if (isTwoGateDoc && currentGate >= maxGate) {
         console.log(chalk.green('\n  ✅ Document workflow complete. Task is done.'));
     } else {
         const nextLabel = gateLabel(nextGateNum, taskType);
@@ -374,7 +380,7 @@ async function nextGate(taskId) {
         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".`));
-        if (!isGenDoc) {
+        if (!isTwoGateDoc) {
             console.log(chalk.yellow(`       (Note: Gate 3 progress is saved via [x] checkboxes in the Gate 2 plan doc (AK-Docs/04.Coding/02.Plans/).`));
             console.log(chalk.yellow(`        The AI will automatically resume the exact task you left off.)`));
         }
@@ -479,7 +485,7 @@ async function detectCurrentGate(taskId) {
 
 async function generateMarkdownSummary(taskState, taskType = null) {
     taskType = taskType || taskState.taskType || 'feature';
-    const maxGate = taskType === 'testing' ? 4 : (taskType === 'gen-doc' ? 2 : 5);
+    const maxGate = taskType === 'testing' ? 4 : ((taskType === 'gen-doc' || taskType === 'create-system-requirement' || taskType === 'ingest-data') ? 2 : 5);
     const lines = [];
     lines.push(`# Task Summary: ${taskState.taskId}`);
     lines.push(`**Title:** ${taskState.title}`);
@@ -517,6 +523,16 @@ function gateLabel(n, taskType = null) {
         if (n === 2) return 'Generate Document';
         if (n >= 3) return 'Done';
     }
+    if (taskType === 'create-system-requirement') {
+        if (n === 1) return 'Investigate UC & Draft';
+        if (n === 2) return 'Finalize & Approve';
+        if (n >= 3) return 'Done';
+    }
+    if (taskType === 'ingest-data') {
+        if (n === 1) return 'Fetch, Classify, Draft';
+        if (n === 2) return 'Branch + Merge Request';
+        if (n >= 3) return 'Done';
+    }
     const labels = {
         1: 'AI Analyze Requirement',
         2: 'Implementation Plan',