@wizdear/atlas-code 0.2.4 → 0.2.6

package/dist/system-architect.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"system-architect.d.ts","sourceRoot":"","sources":["../src/system-architect.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAItD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAI7C,0CAA0C;AAC1C,MAAM,WAAW,sBAAsB;IACtC,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,EAAE,UAAU,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,WAAW,CAAC;IACxB,qEAAqE;IACrE,KAAK,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,iDAAiD;IACjD,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,CAAC;IAC3C,mDAAmD;IACnD,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,sEAAsE;IACtE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,oEAAoE;IACpE,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,qEAAqE;IACrE,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACxC,0DAA0D;IAC1D,eAAe,CAAC,EAAE,MAAM,IAAI,CAAC;CAC7B;AAED,yCAAyC;AACzC,MAAM,WAAW,qBAAqB;IACrC,8CAA8C;IAC9C,SAAS,EAAE,OAAO,CAAC;CACnB;AAID;;;;;;;;;;GAUG;AACH,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAuExG","sourcesContent":["import type { Agent, AgentEvent } from \"@mariozechner/pi-agent-core\";\nimport type { Model } from \"@mariozechner/pi-ai\";\nimport type { GetApiKeyFn } from \"./agent-factory.js\";\nimport { createRoleAgent, runAgent } from \"./agent-factory.js\";\nimport { getSystemPromptForRole } from \"./roles/index.js\";\nimport { extractArtifactContent } from \"./step-executor.js\";\nimport type { VibeStore } from \"./store.js\";\nimport type { VibeConfig } from \"./types.js\";\n\n// ─── Types ───────────────────────────────────────────────────────────────────\n\n/** System Architect execution options. */\nexport interface SystemArchitectOptions {\n\tstore: VibeStore;\n\tconfig: VibeConfig;\n\tprojectRoot: string;\n\tgetApiKey?: GetApiKeyFn;\n\t/** Model to use. Uses the Agent's default model if not specified. */\n\tmodel?: Model<any>;\n\t/** Agent event callback for progress display. */\n\tonAgentEvent?: (event: AgentEvent) => void;\n\t/** Callback to relay agent messages externally. */\n\tonMessage?: (message: string) => void;\n\t/** User feedback on rejection. Re-generates with feedback applied. */\n\tfeedback?: string;\n\t/** Lightweight skill index. Included in the agent system prompt. */\n\tavailableSkills?: string;\n\t/** Callback invoked when the agent is created (for abort wiring). */\n\tonAgentCreated?: (agent: Agent) => void;\n\t/** Callback invoked when the agent finishes execution. */\n\tonAgentFinished?: () => void;\n}\n\n/** System Architect execution result. */\nexport interface SystemArchitectResult {\n\t/** Whether system-design.md was generated. */\n\tcompleted: boolean;\n}\n\n// ─── Runner ──────────────────────────────────────────────────────────────────\n\n/**\n * Runs the System Architect agent to generate system-design.md\n * based on requirements.md and project-context.md.\n *\n * 1. Load requirements.md and project-context.md from the store\n * 2. Optionally load existing system-design.md for iterative refinement\n * 3. Create systemArchitect agent\n * 4. Run the agent with a prompt (includes feedback if re-running)\n * 5. Extract system-design.md from the response\n * 6. Write to store\n */\nexport async function runSystemArchitect(options: SystemArchitectOptions): Promise<SystemArchitectResult> {\n\tconst { store, config, projectRoot, getApiKey, model, onAgentEvent, onMessage, feedback } = options;\n\n\t// 1. Load inputs\n\tlet requirements: string | undefined;\n\tif (await store.hasRequirements()) {\n\t\trequirements = await store.readRequirements();\n\t}\n\n\tlet projectContext: string | undefined;\n\tif (await store.hasProjectContext()) {\n\t\tprojectContext = await store.readProjectContext();\n\t}\n\n\tlet existingDesign: string | undefined;\n\tif (await store.hasSystemDesign()) {\n\t\texistingDesign = await store.readSystemDesign();\n\t}\n\n\t// 2. Build feature context (injected into system prompt)\n\tconst featureContext = buildSystemArchitectContext(requirements, projectContext, existingDesign);\n\n\t// 3. Create agent\n\tconst systemPrompt = getSystemPromptForRole(\"systemArchitect\");\n\tconst agent = await createRoleAgent({\n\t\trole: \"systemArchitect\",\n\t\tsystemPrompt,\n\t\tconfig,\n\t\tprojectRoot,\n\t\tmodel,\n\t\tfeatureContext: featureContext || undefined,\n\t\tgetApiKey,\n\t\tavailableSkills: options.availableSkills,\n\t});\n\n\toptions.onAgentCreated?.(agent);\n\n\t// 4. Build prompt and execute\n\ttry {\n\t\tlet prompt: string;\n\t\tif (feedback) {\n\t\t\tprompt =\n\t\t\t\t`The previous system design was rejected with the following feedback:\\n\\n${feedback}\\n\\n` +\n\t\t\t\t\"Please revise the system design based on this feedback.\\n\\n\" +\n\t\t\t\tbuildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t} else {\n\t\t\tprompt = buildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t}\n\n\t\tconst response = await runAgent(agent, prompt, onAgentEvent ? { onEvent: onAgentEvent } : undefined);\n\t\tonMessage?.(response);\n\n\t\t// 5. Extract system-design.md\n\t\tconst content = extractArtifactContent(response, \"system-design.md\", false);\n\t\tif (!content) {\n\t\t\treturn { completed: false };\n\t\t}\n\n\t\t// 6. Write to store\n\t\tawait store.writeSystemDesign(content);\n\n\t\treturn { completed: true };\n\t} catch (error) {\n\t\tconst msg = error instanceof Error ? error.message : String(error);\n\t\tif (msg.includes(\"aborted\")) {\n\t\t\treturn { completed: false };\n\t\t}\n\t\tthrow error;\n\t} finally {\n\t\toptions.onAgentFinished?.();\n\t}\n}\n\n// ─── Context & Prompt Builders ───────────────────────────────────────────────\n\nfunction buildSystemArchitectContext(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements\\n\");\n\t\tparts.push(requirements);\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\\n\");\n\t\tparts.push(projectContext);\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design (for refinement)\\n\");\n\t\tparts.push(existingDesign);\n\t}\n\n\treturn parts.join(\"\\n\\n\");\n}\n\nfunction buildSystemArchitectPrompt(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tparts.push(\n\t\t\"Analyze the following inputs and produce a system-design.md document that establishes component boundaries, API contracts, shared data models, and common patterns.\",\n\t);\n\tparts.push(\"\");\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements Document\");\n\t\tparts.push(\"\");\n\t\tparts.push(requirements);\n\t\tparts.push(\"\");\n\t} else {\n\t\tparts.push(\"Note: No requirements.md found. Use the project context to infer system architecture.\");\n\t\tparts.push(\"\");\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\");\n\t\tparts.push(\"\");\n\t\tparts.push(projectContext);\n\t\tparts.push(\"\");\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design\");\n\t\tparts.push(\"\");\n\t\tparts.push(\n\t\t\t\"A previous system-design.md exists. Refine and update it based on the current requirements rather than starting from scratch.\",\n\t\t);\n\t\tparts.push(\"\");\n\t\tparts.push(existingDesign);\n\t\tparts.push(\"\");\n\t}\n\n\tparts.push(\"## Instructions\");\n\tparts.push(\"\");\n\tparts.push(\"1. Identify all system components and their boundaries\");\n\tparts.push(\"2. Define API contracts between components\");\n\tparts.push(\"3. Establish shared data models\");\n\tparts.push(\"4. Define common patterns (auth, error handling, logging)\");\n\tparts.push(\"5. Make infrastructure decisions based on the System Scope requirements\");\n\tparts.push(\"6. Note any decisions deferred to feature-level architects\");\n\tparts.push(\"7. Output the result in a ```system-design.md code block\");\n\n\treturn parts.join(\"\\n\");\n}\n"]}
+{"version":3,"file":"system-architect.d.ts","sourceRoot":"","sources":["../src/system-architect.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAItD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAI7C,0CAA0C;AAC1C,MAAM,WAAW,sBAAsB;IACtC,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,EAAE,UAAU,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,WAAW,CAAC;IACxB,qEAAqE;IACrE,KAAK,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,iDAAiD;IACjD,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,CAAC;IAC3C,mDAAmD;IACnD,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,sEAAsE;IACtE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,qEAAqE;IACrE,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACxC,0DAA0D;IAC1D,eAAe,CAAC,EAAE,MAAM,IAAI,CAAC;CAC7B;AAED,yCAAyC;AACzC,MAAM,WAAW,qBAAqB;IACrC,8CAA8C;IAC9C,SAAS,EAAE,OAAO,CAAC;IACnB,8BAA8B;IAC9B,KAAK,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;CACpH;AAID;;;;;;;;;;GAUG;AACH,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAuExG","sourcesContent":["import type { Agent, AgentEvent } from \"@mariozechner/pi-agent-core\";\nimport type { Model } from \"@mariozechner/pi-ai\";\nimport type { GetApiKeyFn } from \"./agent-factory.js\";\nimport { createRoleAgent, runAgent } from \"./agent-factory.js\";\nimport { getSystemPromptForRole } from \"./roles/index.js\";\nimport { aggregateUsage, extractArtifactContent } from \"./step-executor.js\";\nimport type { VibeStore } from \"./store.js\";\nimport type { VibeConfig } from \"./types.js\";\n\n// ─── Types ───────────────────────────────────────────────────────────────────\n\n/** System Architect execution options. */\nexport interface SystemArchitectOptions {\n\tstore: VibeStore;\n\tconfig: VibeConfig;\n\tprojectRoot: string;\n\tgetApiKey?: GetApiKeyFn;\n\t/** Model to use. Uses the Agent's default model if not specified. */\n\tmodel?: Model<any>;\n\t/** Agent event callback for progress display. */\n\tonAgentEvent?: (event: AgentEvent) => void;\n\t/** Callback to relay agent messages externally. */\n\tonMessage?: (message: string) => void;\n\t/** User feedback on rejection. Re-generates with feedback applied. */\n\tfeedback?: string;\n\t/** Callback invoked when the agent is created (for abort wiring). */\n\tonAgentCreated?: (agent: Agent) => void;\n\t/** Callback invoked when the agent finishes execution. */\n\tonAgentFinished?: () => void;\n}\n\n/** System Architect execution result. */\nexport interface SystemArchitectResult {\n\t/** Whether system-design.md was generated. */\n\tcompleted: boolean;\n\t/** Aggregated token usage. */\n\tusage?: { input: number; output: number; cacheRead: number; cacheWrite: number; totalTokens: number; cost: number };\n}\n\n// ─── Runner ──────────────────────────────────────────────────────────────────\n\n/**\n * Runs the System Architect agent to generate system-design.md\n * based on requirements.md and project-context.md.\n *\n * 1. Load requirements.md and project-context.md from the store\n * 2. Optionally load existing system-design.md for iterative refinement\n * 3. Create systemArchitect agent\n * 4. Run the agent with a prompt (includes feedback if re-running)\n * 5. Extract system-design.md from the response\n * 6. Write to store\n */\nexport async function runSystemArchitect(options: SystemArchitectOptions): Promise<SystemArchitectResult> {\n\tconst { store, config, projectRoot, getApiKey, model, onAgentEvent, onMessage, feedback } = options;\n\n\t// 1. Load inputs\n\tlet requirements: string | undefined;\n\tif (await store.hasRequirements()) {\n\t\trequirements = await store.readRequirements();\n\t}\n\n\tlet projectContext: string | undefined;\n\tif (await store.hasProjectContext()) {\n\t\tprojectContext = await store.readProjectContext();\n\t}\n\n\tlet existingDesign: string | undefined;\n\tif (await store.hasSystemDesign()) {\n\t\texistingDesign = await store.readSystemDesign();\n\t}\n\n\t// 2. Build feature context (injected into system prompt)\n\tconst featureContext = buildSystemArchitectContext(requirements, existingDesign);\n\n\t// 3. Create agent\n\tconst systemPrompt = getSystemPromptForRole(\"systemArchitect\");\n\tconst agent = await createRoleAgent({\n\t\trole: \"systemArchitect\",\n\t\tsystemPrompt,\n\t\tconfig,\n\t\tprojectRoot,\n\t\tmodel,\n\t\tfeatureContext: featureContext || undefined,\n\t\tgetApiKey,\n\t});\n\n\toptions.onAgentCreated?.(agent);\n\n\t// 4. Build prompt and execute\n\ttry {\n\t\tlet prompt: string;\n\t\tif (feedback) {\n\t\t\tprompt =\n\t\t\t\t`The previous system design was rejected with the following feedback:\\n\\n${feedback}\\n\\n` +\n\t\t\t\t\"Please revise the system design based on this feedback.\\n\\n\" +\n\t\t\t\tbuildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t} else {\n\t\t\tprompt = buildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t}\n\n\t\tconst response = await runAgent(agent, prompt, onAgentEvent ? { onEvent: onAgentEvent } : undefined);\n\t\tconst usage = aggregateUsage(agent.state.messages);\n\t\tonMessage?.(response);\n\n\t\t// 5. Extract system-design.md\n\t\tconst content = extractArtifactContent(response, \"system-design.md\", false);\n\t\tif (!content) {\n\t\t\treturn { completed: false, usage };\n\t\t}\n\n\t\t// 6. Write to store\n\t\tawait store.writeSystemDesign(content);\n\n\t\treturn { completed: true, usage };\n\t} catch (error) {\n\t\tconst msg = error instanceof Error ? error.message : String(error);\n\t\tif (msg.includes(\"aborted\")) {\n\t\t\treturn { completed: false };\n\t\t}\n\t\tthrow error;\n\t} finally {\n\t\toptions.onAgentFinished?.();\n\t}\n}\n\n// ─── Context & Prompt Builders ───────────────────────────────────────────────\n\nfunction buildSystemArchitectContext(requirements?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements\\n\");\n\t\tparts.push(requirements);\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design (for refinement)\\n\");\n\t\tparts.push(existingDesign);\n\t}\n\n\treturn parts.join(\"\\n\\n\");\n}\n\nfunction buildSystemArchitectPrompt(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tparts.push(\n\t\t\"Analyze the following inputs and produce a system-design.md document that establishes component boundaries, API contracts, shared data models, and common patterns.\",\n\t);\n\tparts.push(\"\");\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements Document\");\n\t\tparts.push(\"\");\n\t\tparts.push(requirements);\n\t\tparts.push(\"\");\n\t} else {\n\t\tparts.push(\"Note: No requirements.md found. Use the project context to infer system architecture.\");\n\t\tparts.push(\"\");\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\");\n\t\tparts.push(\"\");\n\t\tparts.push(projectContext);\n\t\tparts.push(\"\");\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design\");\n\t\tparts.push(\"\");\n\t\tparts.push(\n\t\t\t\"A previous system-design.md exists. Refine and update it based on the current requirements rather than starting from scratch.\",\n\t\t);\n\t\tparts.push(\"\");\n\t\tparts.push(existingDesign);\n\t\tparts.push(\"\");\n\t}\n\n\tparts.push(\"## Instructions\");\n\tparts.push(\"\");\n\tparts.push(\"1. Identify all system components and their boundaries\");\n\tparts.push(\"2. Define API contracts between components\");\n\tparts.push(\"3. Establish shared data models\");\n\tparts.push(\"4. Define common patterns (auth, error handling, logging)\");\n\tparts.push(\"5. Make infrastructure decisions based on the System Scope requirements\");\n\tparts.push(\"6. Note any decisions deferred to feature-level architects\");\n\tparts.push(\"7. Output the result in a ```system-design.md code block\");\n\n\treturn parts.join(\"\\n\");\n}\n"]}

package/dist/system-architect.js

@@ -1,6 +1,6 @@
 import { createRoleAgent, runAgent } from "./agent-factory.js";
 import { getSystemPromptForRole } from "./roles/index.js";
-import { extractArtifactContent } from "./step-executor.js";
+import { aggregateUsage, extractArtifactContent } from "./step-executor.js";
 // ─── Runner ──────────────────────────────────────────────────────────────────
 /**
  * Runs the System Architect agent to generate system-design.md
@@ -29,7 +29,7 @@ export async function runSystemArchitect(options) {
         existingDesign = await store.readSystemDesign();
     }
     // 2. Build feature context (injected into system prompt)
-    const featureContext = buildSystemArchitectContext(requirements, projectContext, existingDesign);
+    const featureContext = buildSystemArchitectContext(requirements, existingDesign);
     // 3. Create agent
     const systemPrompt = getSystemPromptForRole("systemArchitect");
     const agent = await createRoleAgent({
@@ -40,7 +40,6 @@ export async function runSystemArchitect(options) {
         model,
         featureContext: featureContext || undefined,
         getApiKey,
-        availableSkills: options.availableSkills,
     });
     options.onAgentCreated?.(agent);
     // 4. Build prompt and execute
@@ -56,15 +55,16 @@ export async function runSystemArchitect(options) {
             prompt = buildSystemArchitectPrompt(requirements, projectContext, existingDesign);
         }
         const response = await runAgent(agent, prompt, onAgentEvent ? { onEvent: onAgentEvent } : undefined);
+        const usage = aggregateUsage(agent.state.messages);
         onMessage?.(response);
         // 5. Extract system-design.md
         const content = extractArtifactContent(response, "system-design.md", false);
         if (!content) {
-            return { completed: false };
+            return { completed: false, usage };
         }
         // 6. Write to store
         await store.writeSystemDesign(content);
-        return { completed: true };
+        return { completed: true, usage };
     }
     catch (error) {
         const msg = error instanceof Error ? error.message : String(error);
@@ -78,16 +78,12 @@ export async function runSystemArchitect(options) {
     }
 }
 // ─── Context & Prompt Builders ───────────────────────────────────────────────
-function buildSystemArchitectContext(requirements, projectContext, existingDesign) {
+function buildSystemArchitectContext(requirements, existingDesign) {
     const parts = [];
     if (requirements) {
         parts.push("## Requirements\n");
         parts.push(requirements);
     }
-    if (projectContext) {
-        parts.push("## Project Context\n");
-        parts.push(projectContext);
-    }
     if (existingDesign) {
         parts.push("## Existing System Design (for refinement)\n");
         parts.push(existingDesign);

package/dist/system-architect.js.map

@@ -1 +1 @@
-{"version":3,"file":"system-architect.js","sourceRoot":"","sources":["../src/system-architect.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAkC5D,0NAAgF;AAEhF;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,OAA+B,EAAkC;IACzG,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAEpG,iBAAiB;IACjB,IAAI,YAAgC,CAAC;IACrC,IAAI,MAAM,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC;QACnC,YAAY,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,cAAkC,CAAC;IACvC,IAAI,MAAM,KAAK,CAAC,iBAAiB,EAAE,EAAE,CAAC;QACrC,cAAc,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACnD,CAAC;IAED,IAAI,cAAkC,CAAC;IACvC,IAAI,MAAM,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC;QACnC,cAAc,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE,CAAC;IACjD,CAAC;IAED,yDAAyD;IACzD,MAAM,cAAc,GAAG,2BAA2B,CAAC,YAAY,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;IAEjG,kBAAkB;IAClB,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;QACnC,IAAI,EAAE,iBAAiB;QACvB,YAAY;QACZ,MAAM;QACN,WAAW;QACX,KAAK;QACL,cAAc,EAAE,cAAc,IAAI,SAAS;QAC3C,SAAS;QACT,eAAe,EAAE,OAAO,CAAC,eAAe;KACxC,CAAC,CAAC;IAEH,OAAO,CAAC,cAAc,EAAE,CAAC,KAAK,CAAC,CAAC;IAEhC,8BAA8B;IAC9B,IAAI,CAAC;QACJ,IAAI,MAAc,CAAC;QACnB,IAAI,QAAQ,EAAE,CAAC;YACd,MAAM;gBACL,2EAA2E,QAAQ,MAAM;oBACzF,6DAA6D;oBAC7D,0BAA0B,CAAC,YAAY,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;QAC3E,CAAC;aAAM,CAAC;YACP,MAAM,GAAG,0BAA0B,CAAC,YAAY,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;QACnF,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACrG,SAAS,EAAE,CAAC,QAAQ,CAAC,CAAC;QAEtB,8BAA8B;QAC9B,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC5E,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC7B,CAAC;QAED,oBAAoB;QACpB,MAAM,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAEvC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC7B,CAAC;QACD,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC;IAC7B,CAAC;AAAA,CACD;AAED,oLAAgF;AAEhF,SAAS,2BAA2B,CAAC,YAAqB,EAAE,cAAuB,EAAE,cAAuB,EAAU;IACrH,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QAC3D,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAAA,CAC1B;AAED,SAAS,0BAA0B,CAAC,YAAqB,EAAE,cAAuB,EAAE,cAAuB,EAAU;IACpH,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,CAAC,IAAI,CACT,qKAAqK,CACrK,CAAC;IACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,IAAI,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QACvC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;SAAM,CAAC;QACP,KAAK,CAAC,IAAI,CAAC,uFAAuF,CAAC,CAAC;QACpG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACxC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACT,+HAA+H,CAC/H,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;IACrE,KAAK,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;IACzD,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IAC9C,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;IACxE,KAAK,CAAC,IAAI,CAAC,yEAAyE,CAAC,CAAC;IACtF,KAAK,CAAC,IAAI,CAAC,4DAA4D,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;IAEvE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB","sourcesContent":["import type { Agent, AgentEvent } from \"@mariozechner/pi-agent-core\";\nimport type { Model } from \"@mariozechner/pi-ai\";\nimport type { GetApiKeyFn } from \"./agent-factory.js\";\nimport { createRoleAgent, runAgent } from \"./agent-factory.js\";\nimport { getSystemPromptForRole } from \"./roles/index.js\";\nimport { extractArtifactContent } from \"./step-executor.js\";\nimport type { VibeStore } from \"./store.js\";\nimport type { VibeConfig } from \"./types.js\";\n\n// ─── Types ───────────────────────────────────────────────────────────────────\n\n/** System Architect execution options. */\nexport interface SystemArchitectOptions {\n\tstore: VibeStore;\n\tconfig: VibeConfig;\n\tprojectRoot: string;\n\tgetApiKey?: GetApiKeyFn;\n\t/** Model to use. Uses the Agent's default model if not specified. */\n\tmodel?: Model<any>;\n\t/** Agent event callback for progress display. */\n\tonAgentEvent?: (event: AgentEvent) => void;\n\t/** Callback to relay agent messages externally. */\n\tonMessage?: (message: string) => void;\n\t/** User feedback on rejection. Re-generates with feedback applied. */\n\tfeedback?: string;\n\t/** Lightweight skill index. Included in the agent system prompt. */\n\tavailableSkills?: string;\n\t/** Callback invoked when the agent is created (for abort wiring). */\n\tonAgentCreated?: (agent: Agent) => void;\n\t/** Callback invoked when the agent finishes execution. */\n\tonAgentFinished?: () => void;\n}\n\n/** System Architect execution result. */\nexport interface SystemArchitectResult {\n\t/** Whether system-design.md was generated. */\n\tcompleted: boolean;\n}\n\n// ─── Runner ──────────────────────────────────────────────────────────────────\n\n/**\n * Runs the System Architect agent to generate system-design.md\n * based on requirements.md and project-context.md.\n *\n * 1. Load requirements.md and project-context.md from the store\n * 2. Optionally load existing system-design.md for iterative refinement\n * 3. Create systemArchitect agent\n * 4. Run the agent with a prompt (includes feedback if re-running)\n * 5. Extract system-design.md from the response\n * 6. Write to store\n */\nexport async function runSystemArchitect(options: SystemArchitectOptions): Promise<SystemArchitectResult> {\n\tconst { store, config, projectRoot, getApiKey, model, onAgentEvent, onMessage, feedback } = options;\n\n\t// 1. Load inputs\n\tlet requirements: string | undefined;\n\tif (await store.hasRequirements()) {\n\t\trequirements = await store.readRequirements();\n\t}\n\n\tlet projectContext: string | undefined;\n\tif (await store.hasProjectContext()) {\n\t\tprojectContext = await store.readProjectContext();\n\t}\n\n\tlet existingDesign: string | undefined;\n\tif (await store.hasSystemDesign()) {\n\t\texistingDesign = await store.readSystemDesign();\n\t}\n\n\t// 2. Build feature context (injected into system prompt)\n\tconst featureContext = buildSystemArchitectContext(requirements, projectContext, existingDesign);\n\n\t// 3. Create agent\n\tconst systemPrompt = getSystemPromptForRole(\"systemArchitect\");\n\tconst agent = await createRoleAgent({\n\t\trole: \"systemArchitect\",\n\t\tsystemPrompt,\n\t\tconfig,\n\t\tprojectRoot,\n\t\tmodel,\n\t\tfeatureContext: featureContext || undefined,\n\t\tgetApiKey,\n\t\tavailableSkills: options.availableSkills,\n\t});\n\n\toptions.onAgentCreated?.(agent);\n\n\t// 4. Build prompt and execute\n\ttry {\n\t\tlet prompt: string;\n\t\tif (feedback) {\n\t\t\tprompt =\n\t\t\t\t`The previous system design was rejected with the following feedback:\\n\\n${feedback}\\n\\n` +\n\t\t\t\t\"Please revise the system design based on this feedback.\\n\\n\" +\n\t\t\t\tbuildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t} else {\n\t\t\tprompt = buildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t}\n\n\t\tconst response = await runAgent(agent, prompt, onAgentEvent ? { onEvent: onAgentEvent } : undefined);\n\t\tonMessage?.(response);\n\n\t\t// 5. Extract system-design.md\n\t\tconst content = extractArtifactContent(response, \"system-design.md\", false);\n\t\tif (!content) {\n\t\t\treturn { completed: false };\n\t\t}\n\n\t\t// 6. Write to store\n\t\tawait store.writeSystemDesign(content);\n\n\t\treturn { completed: true };\n\t} catch (error) {\n\t\tconst msg = error instanceof Error ? error.message : String(error);\n\t\tif (msg.includes(\"aborted\")) {\n\t\t\treturn { completed: false };\n\t\t}\n\t\tthrow error;\n\t} finally {\n\t\toptions.onAgentFinished?.();\n\t}\n}\n\n// ─── Context & Prompt Builders ───────────────────────────────────────────────\n\nfunction buildSystemArchitectContext(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements\\n\");\n\t\tparts.push(requirements);\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\\n\");\n\t\tparts.push(projectContext);\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design (for refinement)\\n\");\n\t\tparts.push(existingDesign);\n\t}\n\n\treturn parts.join(\"\\n\\n\");\n}\n\nfunction buildSystemArchitectPrompt(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tparts.push(\n\t\t\"Analyze the following inputs and produce a system-design.md document that establishes component boundaries, API contracts, shared data models, and common patterns.\",\n\t);\n\tparts.push(\"\");\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements Document\");\n\t\tparts.push(\"\");\n\t\tparts.push(requirements);\n\t\tparts.push(\"\");\n\t} else {\n\t\tparts.push(\"Note: No requirements.md found. Use the project context to infer system architecture.\");\n\t\tparts.push(\"\");\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\");\n\t\tparts.push(\"\");\n\t\tparts.push(projectContext);\n\t\tparts.push(\"\");\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design\");\n\t\tparts.push(\"\");\n\t\tparts.push(\n\t\t\t\"A previous system-design.md exists. Refine and update it based on the current requirements rather than starting from scratch.\",\n\t\t);\n\t\tparts.push(\"\");\n\t\tparts.push(existingDesign);\n\t\tparts.push(\"\");\n\t}\n\n\tparts.push(\"## Instructions\");\n\tparts.push(\"\");\n\tparts.push(\"1. Identify all system components and their boundaries\");\n\tparts.push(\"2. Define API contracts between components\");\n\tparts.push(\"3. Establish shared data models\");\n\tparts.push(\"4. Define common patterns (auth, error handling, logging)\");\n\tparts.push(\"5. Make infrastructure decisions based on the System Scope requirements\");\n\tparts.push(\"6. Note any decisions deferred to feature-level architects\");\n\tparts.push(\"7. Output the result in a ```system-design.md code block\");\n\n\treturn parts.join(\"\\n\");\n}\n"]}
+{"version":3,"file":"system-architect.js","sourceRoot":"","sources":["../src/system-architect.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAkC5E,0NAAgF;AAEhF;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,OAA+B,EAAkC;IACzG,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAEpG,iBAAiB;IACjB,IAAI,YAAgC,CAAC;IACrC,IAAI,MAAM,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC;QACnC,YAAY,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,cAAkC,CAAC;IACvC,IAAI,MAAM,KAAK,CAAC,iBAAiB,EAAE,EAAE,CAAC;QACrC,cAAc,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACnD,CAAC;IAED,IAAI,cAAkC,CAAC;IACvC,IAAI,MAAM,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC;QACnC,cAAc,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE,CAAC;IACjD,CAAC;IAED,yDAAyD;IACzD,MAAM,cAAc,GAAG,2BAA2B,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;IAEjF,kBAAkB;IAClB,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;QACnC,IAAI,EAAE,iBAAiB;QACvB,YAAY;QACZ,MAAM;QACN,WAAW;QACX,KAAK;QACL,cAAc,EAAE,cAAc,IAAI,SAAS;QAC3C,SAAS;KACT,CAAC,CAAC;IAEH,OAAO,CAAC,cAAc,EAAE,CAAC,KAAK,CAAC,CAAC;IAEhC,8BAA8B;IAC9B,IAAI,CAAC;QACJ,IAAI,MAAc,CAAC;QACnB,IAAI,QAAQ,EAAE,CAAC;YACd,MAAM;gBACL,2EAA2E,QAAQ,MAAM;oBACzF,6DAA6D;oBAC7D,0BAA0B,CAAC,YAAY,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;QAC3E,CAAC;aAAM,CAAC;YACP,MAAM,GAAG,0BAA0B,CAAC,YAAY,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC;QACnF,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACrG,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnD,SAAS,EAAE,CAAC,QAAQ,CAAC,CAAC;QAEtB,8BAA8B;QAC9B,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC5E,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACpC,CAAC;QAED,oBAAoB;QACpB,MAAM,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAEvC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC7B,CAAC;QACD,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC;IAC7B,CAAC;AAAA,CACD;AAED,oLAAgF;AAEhF,SAAS,2BAA2B,CAAC,YAAqB,EAAE,cAAuB,EAAU;IAC5F,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QAC3D,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAAA,CAC1B;AAED,SAAS,0BAA0B,CAAC,YAAqB,EAAE,cAAuB,EAAE,cAAuB,EAAU;IACpH,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,CAAC,IAAI,CACT,qKAAqK,CACrK,CAAC;IACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,IAAI,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QACvC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;SAAM,CAAC;QACP,KAAK,CAAC,IAAI,CAAC,uFAAuF,CAAC,CAAC;QACpG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,cAAc,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACxC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACT,+HAA+H,CAC/H,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;IACrE,KAAK,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;IACzD,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IAC9C,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;IACxE,KAAK,CAAC,IAAI,CAAC,yEAAyE,CAAC,CAAC;IACtF,KAAK,CAAC,IAAI,CAAC,4DAA4D,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;IAEvE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB","sourcesContent":["import type { Agent, AgentEvent } from \"@mariozechner/pi-agent-core\";\nimport type { Model } from \"@mariozechner/pi-ai\";\nimport type { GetApiKeyFn } from \"./agent-factory.js\";\nimport { createRoleAgent, runAgent } from \"./agent-factory.js\";\nimport { getSystemPromptForRole } from \"./roles/index.js\";\nimport { aggregateUsage, extractArtifactContent } from \"./step-executor.js\";\nimport type { VibeStore } from \"./store.js\";\nimport type { VibeConfig } from \"./types.js\";\n\n// ─── Types ───────────────────────────────────────────────────────────────────\n\n/** System Architect execution options. */\nexport interface SystemArchitectOptions {\n\tstore: VibeStore;\n\tconfig: VibeConfig;\n\tprojectRoot: string;\n\tgetApiKey?: GetApiKeyFn;\n\t/** Model to use. Uses the Agent's default model if not specified. */\n\tmodel?: Model<any>;\n\t/** Agent event callback for progress display. */\n\tonAgentEvent?: (event: AgentEvent) => void;\n\t/** Callback to relay agent messages externally. */\n\tonMessage?: (message: string) => void;\n\t/** User feedback on rejection. Re-generates with feedback applied. */\n\tfeedback?: string;\n\t/** Callback invoked when the agent is created (for abort wiring). */\n\tonAgentCreated?: (agent: Agent) => void;\n\t/** Callback invoked when the agent finishes execution. */\n\tonAgentFinished?: () => void;\n}\n\n/** System Architect execution result. */\nexport interface SystemArchitectResult {\n\t/** Whether system-design.md was generated. */\n\tcompleted: boolean;\n\t/** Aggregated token usage. */\n\tusage?: { input: number; output: number; cacheRead: number; cacheWrite: number; totalTokens: number; cost: number };\n}\n\n// ─── Runner ──────────────────────────────────────────────────────────────────\n\n/**\n * Runs the System Architect agent to generate system-design.md\n * based on requirements.md and project-context.md.\n *\n * 1. Load requirements.md and project-context.md from the store\n * 2. Optionally load existing system-design.md for iterative refinement\n * 3. Create systemArchitect agent\n * 4. Run the agent with a prompt (includes feedback if re-running)\n * 5. Extract system-design.md from the response\n * 6. Write to store\n */\nexport async function runSystemArchitect(options: SystemArchitectOptions): Promise<SystemArchitectResult> {\n\tconst { store, config, projectRoot, getApiKey, model, onAgentEvent, onMessage, feedback } = options;\n\n\t// 1. Load inputs\n\tlet requirements: string | undefined;\n\tif (await store.hasRequirements()) {\n\t\trequirements = await store.readRequirements();\n\t}\n\n\tlet projectContext: string | undefined;\n\tif (await store.hasProjectContext()) {\n\t\tprojectContext = await store.readProjectContext();\n\t}\n\n\tlet existingDesign: string | undefined;\n\tif (await store.hasSystemDesign()) {\n\t\texistingDesign = await store.readSystemDesign();\n\t}\n\n\t// 2. Build feature context (injected into system prompt)\n\tconst featureContext = buildSystemArchitectContext(requirements, existingDesign);\n\n\t// 3. Create agent\n\tconst systemPrompt = getSystemPromptForRole(\"systemArchitect\");\n\tconst agent = await createRoleAgent({\n\t\trole: \"systemArchitect\",\n\t\tsystemPrompt,\n\t\tconfig,\n\t\tprojectRoot,\n\t\tmodel,\n\t\tfeatureContext: featureContext || undefined,\n\t\tgetApiKey,\n\t});\n\n\toptions.onAgentCreated?.(agent);\n\n\t// 4. Build prompt and execute\n\ttry {\n\t\tlet prompt: string;\n\t\tif (feedback) {\n\t\t\tprompt =\n\t\t\t\t`The previous system design was rejected with the following feedback:\\n\\n${feedback}\\n\\n` +\n\t\t\t\t\"Please revise the system design based on this feedback.\\n\\n\" +\n\t\t\t\tbuildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t} else {\n\t\t\tprompt = buildSystemArchitectPrompt(requirements, projectContext, existingDesign);\n\t\t}\n\n\t\tconst response = await runAgent(agent, prompt, onAgentEvent ? { onEvent: onAgentEvent } : undefined);\n\t\tconst usage = aggregateUsage(agent.state.messages);\n\t\tonMessage?.(response);\n\n\t\t// 5. Extract system-design.md\n\t\tconst content = extractArtifactContent(response, \"system-design.md\", false);\n\t\tif (!content) {\n\t\t\treturn { completed: false, usage };\n\t\t}\n\n\t\t// 6. Write to store\n\t\tawait store.writeSystemDesign(content);\n\n\t\treturn { completed: true, usage };\n\t} catch (error) {\n\t\tconst msg = error instanceof Error ? error.message : String(error);\n\t\tif (msg.includes(\"aborted\")) {\n\t\t\treturn { completed: false };\n\t\t}\n\t\tthrow error;\n\t} finally {\n\t\toptions.onAgentFinished?.();\n\t}\n}\n\n// ─── Context & Prompt Builders ───────────────────────────────────────────────\n\nfunction buildSystemArchitectContext(requirements?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements\\n\");\n\t\tparts.push(requirements);\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design (for refinement)\\n\");\n\t\tparts.push(existingDesign);\n\t}\n\n\treturn parts.join(\"\\n\\n\");\n}\n\nfunction buildSystemArchitectPrompt(requirements?: string, projectContext?: string, existingDesign?: string): string {\n\tconst parts: string[] = [];\n\n\tparts.push(\n\t\t\"Analyze the following inputs and produce a system-design.md document that establishes component boundaries, API contracts, shared data models, and common patterns.\",\n\t);\n\tparts.push(\"\");\n\n\tif (requirements) {\n\t\tparts.push(\"## Requirements Document\");\n\t\tparts.push(\"\");\n\t\tparts.push(requirements);\n\t\tparts.push(\"\");\n\t} else {\n\t\tparts.push(\"Note: No requirements.md found. Use the project context to infer system architecture.\");\n\t\tparts.push(\"\");\n\t}\n\n\tif (projectContext) {\n\t\tparts.push(\"## Project Context\");\n\t\tparts.push(\"\");\n\t\tparts.push(projectContext);\n\t\tparts.push(\"\");\n\t}\n\n\tif (existingDesign) {\n\t\tparts.push(\"## Existing System Design\");\n\t\tparts.push(\"\");\n\t\tparts.push(\n\t\t\t\"A previous system-design.md exists. Refine and update it based on the current requirements rather than starting from scratch.\",\n\t\t);\n\t\tparts.push(\"\");\n\t\tparts.push(existingDesign);\n\t\tparts.push(\"\");\n\t}\n\n\tparts.push(\"## Instructions\");\n\tparts.push(\"\");\n\tparts.push(\"1. Identify all system components and their boundaries\");\n\tparts.push(\"2. Define API contracts between components\");\n\tparts.push(\"3. Establish shared data models\");\n\tparts.push(\"4. Define common patterns (auth, error handling, logging)\");\n\tparts.push(\"5. Make infrastructure decisions based on the System Scope requirements\");\n\tparts.push(\"6. Note any decisions deferred to feature-level architects\");\n\tparts.push(\"7. Output the result in a ```system-design.md code block\");\n\n\treturn parts.join(\"\\n\");\n}\n"]}

package/dist/telegram/bridge.d.ts

@@ -0,0 +1,39 @@
+/**
+ * Telegram bridge — mirrors TUI state to Telegram and accepts simple commands.
+ * Activated automatically when token is available (env var or .vibe/config.json).
+ * Chat ID is auto-discovered from the first incoming message if not configured.
+ */
+import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from "@mariozechner/pi-coding-agent";
+interface TelegramBridgeConfig {
+    token: string;
+    chatId?: string;
+}
+/**
+ * Resolve Telegram config from env vars and .vibe/config.json.
+ * Priority: env var > config file.
+ */
+export declare function resolveConfig(vibeDir: string): TelegramBridgeConfig | undefined;
+/** Registered command entry for Telegram dispatch. */
+export interface TelegramCommand {
+    name: string;
+    description?: string;
+    handler: (args: string, ctx: ExtensionCommandContext) => Promise<void>;
+    /** Subcommands rendered as separate entries in Telegram bot menu (e.g., vibe_resume). */
+    subcommands?: Array<{
+        name: string;
+        description: string;
+    }>;
+}
+/**
+ * Create a bridge manager that supports dynamic mid-session activation.
+ * Call `tryStart()` on session_start and agent_end. Call `stop()` on session_shutdown.
+ *
+ * @param commands Shared array of commands. Items can be added after creation;
+ *   the polling loop reads from this array by reference on every message.
+ */
+export declare function createTelegramBridgeManager(pi: ExtensionAPI, ctx: ExtensionContext, vibeDir: string, commands?: TelegramCommand[]): {
+    tryStart: () => void;
+    stop: () => void;
+};
+export {};
+//# sourceMappingURL=bridge.d.ts.map

package/dist/telegram/bridge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bridge.d.ts","sourceRoot":"","sources":["../../src/telegram/bridge.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,KAAK,EAAE,YAAY,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AA6B7G,UAAU,oBAAoB;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,oBAAoB,GAAG,SAAS,CAqB/E;AAoBD,sDAAsD;AACtD,MAAM,WAAW,eAAe;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,uBAAuB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACvE,yFAAyF;IACzF,WAAW,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAC3D;AAED;;;;;;GAMG;AACH,wBAAgB,2BAA2B,CAC1C,EAAE,EAAE,YAAY,EAChB,GAAG,EAAE,gBAAgB,EACrB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,eAAe,EAAO,GAC9B;IAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAoB5C","sourcesContent":["/**\n * Telegram bridge — mirrors TUI state to Telegram and accepts simple commands.\n * Activated automatically when token is available (env var or .vibe/config.json).\n * Chat ID is auto-discovered from the first incoming message if not configured.\n */\n\nimport { readFileSync, writeFileSync } from \"node:fs\";\nimport { join } from \"node:path\";\n\nimport type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from \"@mariozechner/pi-coding-agent\";\n\nimport { formatToolLabel } from \"./formatter.js\";\nimport type { StepDisplay } from \"./renderer.js\";\nimport { ResponseRenderer } from \"./renderer.js\";\nimport { TelegramAPI } from \"./telegram-api.js\";\n\n// ─── Vibe event shapes (local, not exported) ────────────────────────────────\n\ninterface VibePipelineEvent {\n\ttype: string;\n\tfeatureId: string;\n\tstep?: { agent: string; action: string };\n\tdata?: Record<string, unknown>;\n}\n\ninterface VibeActivityEvent {\n\trole: string;\n\ttext: string;\n\ttype: string;\n}\n\ninterface VibeGateEvent {\n\tfeatureId: string;\n\tsummary: string;\n}\n\n// ─── Config resolution ───────────────────────────────────────────────────────\n\ninterface TelegramBridgeConfig {\n\ttoken: string;\n\tchatId?: string;\n}\n\n/**\n * Resolve Telegram config from env vars and .vibe/config.json.\n * Priority: env var > config file.\n */\nexport function resolveConfig(vibeDir: string): TelegramBridgeConfig | undefined {\n\tlet fileToken: string | undefined;\n\tlet fileChatId: string | undefined;\n\n\ttry {\n\t\tconst configPath = join(vibeDir, \"config.json\");\n\t\tconst raw = JSON.parse(readFileSync(configPath, \"utf-8\")) as Record<string, unknown>;\n\t\tconst tg = raw.telegram as Record<string, unknown> | undefined;\n\t\tif (tg) {\n\t\t\tfileToken = typeof tg.token === \"string\" ? tg.token : undefined;\n\t\t\tfileChatId = typeof tg.chatId === \"string\" ? tg.chatId : undefined;\n\t\t}\n\t} catch {\n\t\t// Config file may not exist\n\t}\n\n\tconst token = process.env.VIBE_TELEGRAM_TOKEN ?? fileToken;\n\tif (!token) return undefined;\n\n\tconst chatId = process.env.VIBE_TELEGRAM_CHAT_ID ?? fileChatId;\n\treturn { token, chatId: chatId || undefined };\n}\n\n/** Save discovered chat ID to .vibe/config.json. */\nfunction saveDiscoveredChatId(vibeDir: string, chatId: string): void {\n\tconst configPath = join(vibeDir, \"config.json\");\n\tlet config: Record<string, unknown> = {};\n\ttry {\n\t\tconfig = JSON.parse(readFileSync(configPath, \"utf-8\")) as Record<string, unknown>;\n\t} catch {\n\t\t// File may not exist\n\t}\n\tif (!config.telegram || typeof config.telegram !== \"object\") {\n\t\tconfig.telegram = {};\n\t}\n\t(config.telegram as Record<string, unknown>).chatId = chatId;\n\twriteFileSync(configPath, `${JSON.stringify(config, null, \"\\t\")}\\n`, \"utf-8\");\n}\n\n// ─── Bridge Manager ──────────────────────────────────────────────────────────\n\n/** Registered command entry for Telegram dispatch. */\nexport interface TelegramCommand {\n\tname: string;\n\tdescription?: string;\n\thandler: (args: string, ctx: ExtensionCommandContext) => Promise<void>;\n\t/** Subcommands rendered as separate entries in Telegram bot menu (e.g., vibe_resume). */\n\tsubcommands?: Array<{ name: string; description: string }>;\n}\n\n/**\n * Create a bridge manager that supports dynamic mid-session activation.\n * Call `tryStart()` on session_start and agent_end. Call `stop()` on session_shutdown.\n *\n * @param commands Shared array of commands. Items can be added after creation;\n *   the polling loop reads from this array by reference on every message.\n */\nexport function createTelegramBridgeManager(\n\tpi: ExtensionAPI,\n\tctx: ExtensionContext,\n\tvibeDir: string,\n\tcommands: TelegramCommand[] = [],\n): { tryStart: () => void; stop: () => void } {\n\tlet cleanup: (() => void) | undefined;\n\tlet started = false;\n\n\tfunction tryStart(): void {\n\t\tif (started) return;\n\t\tconst config = resolveConfig(vibeDir);\n\t\tif (!config) return;\n\n\t\tcleanup = startBridge(pi, ctx, config, vibeDir, commands);\n\t\tstarted = true;\n\t}\n\n\tfunction stop(): void {\n\t\tcleanup?.();\n\t\tcleanup = undefined;\n\t\tstarted = false;\n\t}\n\n\treturn { tryStart, stop };\n}\n\n// ─── Bridge core ─────────────────────────────────────────────────────────────\n\nfunction startBridge(\n\tpi: ExtensionAPI,\n\tctx: ExtensionContext,\n\tconfig: TelegramBridgeConfig,\n\tvibeDir: string,\n\tcommands: TelegramCommand[],\n): () => void {\n\tconst tg = new TelegramAPI(config.token, config.chatId);\n\tconst renderer = new ResponseRenderer(tg);\n\tconst abortController = new AbortController();\n\tconst unsubs: Array<() => void> = [];\n\n\t// Pipeline/gate state tracking — prevents Telegram input during vibe workflows\n\t// and preserves renderer state between pipeline steps.\n\tlet pipelineActive = false;\n\tlet gateWaiting = false;\n\n\t// 1. Agent event mirroring\n\tpi.on(\"message_update\", (event) => {\n\t\tif (event.assistantMessageEvent.type === \"text_delta\") {\n\t\t\trenderer.appendText(event.assistantMessageEvent.delta);\n\t\t}\n\t});\n\n\tpi.on(\"tool_execution_start\", (event) => {\n\t\tconst label = formatToolLabel(event.toolName, event.args as Record<string, unknown>);\n\t\trenderer.appendToolInfo(`\\u2192 ${label}`);\n\t});\n\n\tpi.on(\"agent_end\", () => {\n\t\t// During pipeline execution, don't finalize — the pipeline events manage renderer lifecycle.\n\t\t// Finalizing here would reset the step table between steps.\n\t\tif (!pipelineActive) {\n\t\t\trenderer.finalize().catch(() => {});\n\t\t}\n\t});\n\n\t// 1b. Vibe custom messages (agent responses, milestones, errors, etc.)\n\t// pi.sendMessage() → sendCustomMessage() only fires _emit() (TUI listeners),\n\t// not _emitExtensionEvent(), so pi.on(\"message_end\") never fires.\n\t// Instead, sendVibeMessage() emits vibe:message on the EventBus.\n\tunsubs.push(\n\t\tpi.events.on(\"vibe:message\", (raw: unknown) => {\n\t\t\tconst event = raw as {\n\t\t\t\tcontent: string;\n\t\t\t\tcategory: string;\n\t\t\t\tfullText?: string;\n\t\t\t};\n\n\t\t\t// Skip categories already handled by pipeline/gate event handlers\n\t\t\tif (event.category === \"gate\") return;\n\t\t\tif (!event.content) return;\n\n\t\t\tconst icons: Record<string, string> = {\n\t\t\t\tmilestone: \"\\u2728\",\n\t\t\t\terror: \"\\u274C\",\n\t\t\t\twarning: \"\\u26A0\\uFE0F\",\n\t\t\t\tcommand: \"\\u25B6\",\n\t\t\t\tevent: \"\\u{1F4AC}\",\n\t\t\t\t\"agent-response\": \"\\u{1F916}\",\n\t\t\t\t\"agent-text\": \"\\u270F\\uFE0F\",\n\t\t\t};\n\t\t\tconst icon = icons[event.category] ?? \"\\u{1F4AC}\";\n\n\t\t\t// For agent-text, show the first line of fullText instead of the generic\n\t\t\t// \"[role] ✎ (N lines)\" summary. This gives meaningful context (e.g., file\n\t\t\t// being worked on) instead of identical-looking messages.\n\t\t\tlet displayText = event.content;\n\t\t\tif (event.category === \"agent-text\" && event.fullText) {\n\t\t\t\tconst firstLine = event.fullText.split(\"\\n\")[0].trim();\n\t\t\t\tif (firstLine) {\n\t\t\t\t\tdisplayText = firstLine;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Send as plain text to avoid Markdown parse failures from LLM output\n\t\t\ttg.sendMessage(`${icon} ${displayText}`, { parse_mode: null }).catch(() => {});\n\t\t}),\n\t);\n\n\t// 2. Vibe event mirroring\n\tunsubs.push(\n\t\tpi.events.on(\"vibe:pipeline\", (raw: unknown) => {\n\t\t\tconst event = raw as VibePipelineEvent;\n\t\t\tif (event.type === \"feature_start\") {\n\t\t\t\tpipelineActive = true;\n\t\t\t} else if (\n\t\t\t\tevent.type === \"pipeline_complete\" ||\n\t\t\t\tevent.type === \"feature_complete\" ||\n\t\t\t\tevent.type === \"orchestration_complete\" ||\n\t\t\t\tevent.type === \"feature_aborted\"\n\t\t\t) {\n\t\t\t\tpipelineActive = false;\n\t\t\t\tgateWaiting = false;\n\t\t\t}\n\t\t\thandlePipelineEvent(event, renderer);\n\t\t}),\n\t\tpi.events.on(\"vibe:activity\", (raw: unknown) => {\n\t\t\tconst event = raw as VibeActivityEvent;\n\t\t\trenderer.addActivity(event.text);\n\t\t}),\n\t\tpi.events.on(\"vibe:gate\", (raw: unknown) => {\n\t\t\tconst event = raw as VibeGateEvent;\n\t\t\tgateWaiting = true;\n\t\t\tconst keyboard = {\n\t\t\t\tinline_keyboard: [\n\t\t\t\t\t[\n\t\t\t\t\t\t{ text: \"\\u2705 Approve\", callback_data: \"Approve\" },\n\t\t\t\t\t\t{ text: \"\\u274C Reject\", callback_data: \"Reject\" },\n\t\t\t\t\t],\n\t\t\t\t\t[\n\t\t\t\t\t\t{ text: \"\\u23ED Skip\", callback_data: \"Skip\" },\n\t\t\t\t\t\t{ text: \"\\u26D4 Abort\", callback_data: \"Abort\" },\n\t\t\t\t\t],\n\t\t\t\t],\n\t\t\t};\n\t\t\ttg.sendMessage(`\\u23F3 Gate: ${event.summary}`, { reply_markup: keyboard }).catch(() => {});\n\t\t}),\n\t);\n\n\t// Reset gate flag when pipeline resumes after gate approval\n\tunsubs.push(\n\t\tpi.events.on(\"vibe:pipeline\", (raw: unknown) => {\n\t\t\tconst event = raw as VibePipelineEvent;\n\t\t\tif (event.type === \"step_start\") {\n\t\t\t\tgateWaiting = false;\n\t\t\t}\n\t\t}),\n\t);\n\n\t// 3. Polling loop (fire-and-forget)\n\tstartPolling(tg, pi, ctx, abortController.signal, vibeDir, {\n\t\tisVibeBusy: () => pipelineActive || gateWaiting,\n\t\tisGateWaiting: () => gateWaiting,\n\t\tclearGateWaiting: () => {\n\t\t\tgateWaiting = false;\n\t\t},\n\t\tcommands,\n\t});\n\n\t// 4. Register bot menu commands (including subcommands as separate entries)\n\tconst menuCommands: Array<{ command: string; description: string }> = [\n\t\t{ command: \"stop\", description: \"Stop current execution\" },\n\t];\n\tfor (const cmd of commands) {\n\t\tmenuCommands.push({ command: cmd.name, description: cmd.description ?? `/${cmd.name}` });\n\t\tif (cmd.subcommands) {\n\t\t\tfor (const sub of cmd.subcommands) {\n\t\t\t\t// Telegram command names: lowercase, digits, underscores only\n\t\t\t\tmenuCommands.push({ command: `${cmd.name}_${sub.name}`, description: sub.description });\n\t\t\t}\n\t\t}\n\t}\n\ttg.setMyCommands(menuCommands).catch(() => {});\n\n\t// 5. Startup\n\tif (tg.hasChatId()) {\n\t\tconst model = ctx.model;\n\t\tconst startupMsg = model\n\t\t\t? `Telegram bridge active. Model: ${model.provider}/${model.id}`\n\t\t\t: \"Telegram bridge active. No model.\";\n\t\ttg.sendMessage(startupMsg).catch(() => {});\n\t} else {\n\t\tctx.ui.notify(\"[Telegram] Waiting for first message from your bot\", \"info\");\n\t}\n\n\t// Return cleanup function\n\treturn () => {\n\t\tabortController.abort();\n\t\tfor (const unsub of unsubs) unsub();\n\t};\n}\n\n// ─── Polling ─────────────────────────────────────────────────────────────────\n\nfunction sleep(ms: number): Promise<void> {\n\treturn new Promise((resolve) => setTimeout(resolve, ms));\n}\n\ninterface PollingState {\n\tisVibeBusy: () => boolean;\n\tisGateWaiting: () => boolean;\n\tclearGateWaiting: () => void;\n\tcommands: TelegramCommand[];\n}\n\nasync function startPolling(\n\ttg: TelegramAPI,\n\tpi: ExtensionAPI,\n\tctx: ExtensionContext,\n\tsignal: AbortSignal,\n\tvibeDir: string,\n\tstate: PollingState,\n): Promise<void> {\n\twhile (!signal.aborted) {\n\t\ttry {\n\t\t\tconst updates = await tg.getUpdates();\n\t\t\tfor (const update of updates) {\n\t\t\t\tif (signal.aborted) break;\n\n\t\t\t\t// Handle callback queries (gate button clicks)\n\t\t\t\tif (update.callback_query) {\n\t\t\t\t\tconst data = update.callback_query.data ?? \"\";\n\t\t\t\t\ttg.answerCallbackQuery(update.callback_query.id).catch(() => {});\n\t\t\t\t\tif (state.isGateWaiting()) {\n\t\t\t\t\t\tpi.events.emit(\"vibe:gate_response\", { choice: data });\n\t\t\t\t\t\tstate.clearGateWaiting();\n\t\t\t\t\t}\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\n\t\t\t\t// Auto-discover chat ID from first incoming message\n\t\t\t\tif (!tg.hasChatId() && update.message?.chat.id) {\n\t\t\t\t\tconst discoveredId = String(update.message.chat.id);\n\t\t\t\t\ttg.setChatId(discoveredId);\n\t\t\t\t\tsaveDiscoveredChatId(vibeDir, discoveredId);\n\t\t\t\t\tconst model = ctx.model;\n\t\t\t\t\tconst msg = model\n\t\t\t\t\t\t? `Telegram bridge connected. Model: ${model.provider}/${model.id}`\n\t\t\t\t\t\t: \"Telegram bridge connected.\";\n\t\t\t\t\tawait tg.sendMessage(msg).catch(() => {});\n\t\t\t\t\tctx.ui.notify(\"[Telegram] Connected\", \"info\");\n\t\t\t\t\tcontinue; // Skip processing discovery message as a command\n\t\t\t\t}\n\n\t\t\t\tconst text = update.message?.text;\n\t\t\t\tif (!text) continue;\n\n\t\t\t\tif (text === \"/stop\") {\n\t\t\t\t\tctx.abort();\n\t\t\t\t\tawait tg.sendMessage(\"\\u23F8 Stopped.\").catch(() => {});\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\n\t\t\t\t// Block input during pipeline execution or gate waiting\n\t\t\t\tif (state.isVibeBusy()) {\n\t\t\t\t\tawait tg.sendMessage(\"\\u23F3 Pipeline running. Wait or /stop.\").catch(() => {});\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\n\t\t\t\tif (ctx.isIdle()) {\n\t\t\t\t\t// Try to dispatch as extension command first.\n\t\t\t\t\t// Fire-and-forget: command handlers may block on gates/UI,\n\t\t\t\t\t// and the polling loop must keep running to receive callback queries.\n\t\t\t\t\tif (text.startsWith(\"/\") && tryDispatchCommandSync(text, state.commands, ctx)) {\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\t\t\t\t\tpi.sendUserMessage(text);\n\t\t\t\t} else {\n\t\t\t\t\tawait tg.sendMessage(\"\\u23F3 Busy. Wait or /stop.\").catch(() => {});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch {\n\t\t\tif (signal.aborted) break;\n\t\t\tawait sleep(5000);\n\t\t}\n\t}\n}\n\n// ─── Command dispatch ────────────────────────────────────────────────────────\n\n/**\n * Try to dispatch a slash command to a registered handler.\n * Handles both direct commands (/vibe resume) and Telegram-style\n * subcommands (/vibe_resume) by mapping underscores back to spaces.\n *\n * Returns true if the command was matched. The handler runs as\n * fire-and-forget so the polling loop is never blocked (handlers\n * may await gates/UI dialogs indefinitely).\n */\nfunction tryDispatchCommandSync(text: string, commands: TelegramCommand[], ctx: ExtensionContext): boolean {\n\t// Strip leading / and optional @BotName suffix (Telegram adds this in groups)\n\tconst raw = text.slice(1).split(\"@\")[0];\n\tconst spaceIndex = raw.indexOf(\" \");\n\tconst fullName = spaceIndex === -1 ? raw : raw.slice(0, spaceIndex);\n\tconst trailingArgs = spaceIndex === -1 ? \"\" : raw.slice(spaceIndex + 1);\n\n\t// 1. Direct match: /vibe resume → command=\"vibe\", args=\"resume\"\n\tconst directMatch = commands.find((c) => c.name === fullName);\n\tif (directMatch) {\n\t\tdirectMatch.handler(trailingArgs, ctx as unknown as ExtensionCommandContext).catch(() => {});\n\t\treturn true;\n\t}\n\n\t// 2. Subcommand match: /vibe_resume → command=\"vibe\", args=\"resume\"\n\tfor (const cmd of commands) {\n\t\tif (!cmd.subcommands) continue;\n\t\tconst prefix = `${cmd.name}_`;\n\t\tif (fullName.startsWith(prefix)) {\n\t\t\tconst subName = fullName.slice(prefix.length);\n\t\t\tif (cmd.subcommands.some((s) => s.name === subName)) {\n\t\t\t\tconst combinedArgs = trailingArgs ? `${subName} ${trailingArgs}` : subName;\n\t\t\t\tcmd.handler(combinedArgs, ctx as unknown as ExtensionCommandContext).catch(() => {});\n\t\t\t\treturn true;\n\t\t\t}\n\t\t}\n\t}\n\n\treturn false;\n}\n\n// ─── Pipeline event handler ─────────────────────────────────────────────────\n\nfunction handlePipelineEvent(event: VibePipelineEvent, renderer: ResponseRenderer): void {\n\tswitch (event.type) {\n\t\tcase \"feature_start\": {\n\t\t\trenderer.setPipelineMode(true);\n\t\t\tconst title = event.data?.title as string | undefined;\n\t\t\trenderer.setPhase(title ? `${event.featureId}: ${title}` : event.featureId);\n\n\t\t\tconst totalSteps = event.data?.totalSteps as number | undefined;\n\t\t\tif (totalSteps) {\n\t\t\t\tconst steps: StepDisplay[] = [];\n\t\t\t\tfor (let i = 0; i < totalSteps; i++) {\n\t\t\t\t\tsteps.push({ label: `Step ${i + 1}`, status: \"pending\" });\n\t\t\t\t}\n\t\t\t\trenderer.setSteps(steps);\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\n\t\tcase \"step_start\": {\n\t\t\tconst stepIndex = event.data?.stepIndex as number | undefined;\n\t\t\tif (stepIndex !== undefined) {\n\t\t\t\trenderer.updateStep(stepIndex, \"running\");\n\t\t\t}\n\t\t\tif (event.step) {\n\t\t\t\trenderer.updateStepLabel(event.data?.stepIndex as number | undefined, event.step.agent);\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\n\t\tcase \"step_complete\": {\n\t\t\tconst stepIndex = event.data?.stepIndex as number | undefined;\n\t\t\tif (stepIndex !== undefined) {\n\t\t\t\tconst elapsed = event.data?.elapsed as number | undefined;\n\t\t\t\tconst usage = event.data?.usage as { cost?: number } | undefined;\n\t\t\t\trenderer.updateStep(stepIndex, \"done\", elapsed !== undefined ? elapsed / 1000 : undefined, usage?.cost);\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\n\t\tcase \"step_failed\": {\n\t\t\tconst stepIndex = event.data?.stepIndex as number | undefined;\n\t\t\tif (stepIndex !== undefined) {\n\t\t\t\trenderer.updateStep(stepIndex, \"failed\");\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\n\t\tcase \"pipeline_complete\":\n\t\tcase \"feature_complete\":\n\t\tcase \"orchestration_complete\":\n\t\tcase \"feature_aborted\": {\n\t\t\trenderer.setPipelineMode(false);\n\t\t\trenderer.finalize().catch(() => {});\n\t\t\tbreak;\n\t\t}\n\t}\n}\n"]}

package/dist/telegram/bridge.js

@@ -0,0 +1,380 @@
+/**
+ * Telegram bridge — mirrors TUI state to Telegram and accepts simple commands.
+ * Activated automatically when token is available (env var or .vibe/config.json).
+ * Chat ID is auto-discovered from the first incoming message if not configured.
+ */
+import { readFileSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { formatToolLabel } from "./formatter.js";
+import { ResponseRenderer } from "./renderer.js";
+import { TelegramAPI } from "./telegram-api.js";
+/**
+ * Resolve Telegram config from env vars and .vibe/config.json.
+ * Priority: env var > config file.
+ */
+export function resolveConfig(vibeDir) {
+    let fileToken;
+    let fileChatId;
+    try {
+        const configPath = join(vibeDir, "config.json");
+        const raw = JSON.parse(readFileSync(configPath, "utf-8"));
+        const tg = raw.telegram;
+        if (tg) {
+            fileToken = typeof tg.token === "string" ? tg.token : undefined;
+            fileChatId = typeof tg.chatId === "string" ? tg.chatId : undefined;
+        }
+    }
+    catch {
+        // Config file may not exist
+    }
+    const token = process.env.VIBE_TELEGRAM_TOKEN ?? fileToken;
+    if (!token)
+        return undefined;
+    const chatId = process.env.VIBE_TELEGRAM_CHAT_ID ?? fileChatId;
+    return { token, chatId: chatId || undefined };
+}
+/** Save discovered chat ID to .vibe/config.json. */
+function saveDiscoveredChatId(vibeDir, chatId) {
+    const configPath = join(vibeDir, "config.json");
+    let config = {};
+    try {
+        config = JSON.parse(readFileSync(configPath, "utf-8"));
+    }
+    catch {
+        // File may not exist
+    }
+    if (!config.telegram || typeof config.telegram !== "object") {
+        config.telegram = {};
+    }
+    config.telegram.chatId = chatId;
+    writeFileSync(configPath, `${JSON.stringify(config, null, "\t")}\n`, "utf-8");
+}
+/**
+ * Create a bridge manager that supports dynamic mid-session activation.
+ * Call `tryStart()` on session_start and agent_end. Call `stop()` on session_shutdown.
+ *
+ * @param commands Shared array of commands. Items can be added after creation;
+ *   the polling loop reads from this array by reference on every message.
+ */
+export function createTelegramBridgeManager(pi, ctx, vibeDir, commands = []) {
+    let cleanup;
+    let started = false;
+    function tryStart() {
+        if (started)
+            return;
+        const config = resolveConfig(vibeDir);
+        if (!config)
+            return;
+        cleanup = startBridge(pi, ctx, config, vibeDir, commands);
+        started = true;
+    }
+    function stop() {
+        cleanup?.();
+        cleanup = undefined;
+        started = false;
+    }
+    return { tryStart, stop };
+}
+// ─── Bridge core ─────────────────────────────────────────────────────────────
+function startBridge(pi, ctx, config, vibeDir, commands) {
+    const tg = new TelegramAPI(config.token, config.chatId);
+    const renderer = new ResponseRenderer(tg);
+    const abortController = new AbortController();
+    const unsubs = [];
+    // Pipeline/gate state tracking — prevents Telegram input during vibe workflows
+    // and preserves renderer state between pipeline steps.
+    let pipelineActive = false;
+    let gateWaiting = false;
+    // 1. Agent event mirroring
+    pi.on("message_update", (event) => {
+        if (event.assistantMessageEvent.type === "text_delta") {
+            renderer.appendText(event.assistantMessageEvent.delta);
+        }
+    });
+    pi.on("tool_execution_start", (event) => {
+        const label = formatToolLabel(event.toolName, event.args);
+        renderer.appendToolInfo(`\u2192 ${label}`);
+    });
+    pi.on("agent_end", () => {
+        // During pipeline execution, don't finalize — the pipeline events manage renderer lifecycle.
+        // Finalizing here would reset the step table between steps.
+        if (!pipelineActive) {
+            renderer.finalize().catch(() => { });
+        }
+    });
+    // 1b. Vibe custom messages (agent responses, milestones, errors, etc.)
+    // pi.sendMessage() → sendCustomMessage() only fires _emit() (TUI listeners),
+    // not _emitExtensionEvent(), so pi.on("message_end") never fires.
+    // Instead, sendVibeMessage() emits vibe:message on the EventBus.
+    unsubs.push(pi.events.on("vibe:message", (raw) => {
+        const event = raw;
+        // Skip categories already handled by pipeline/gate event handlers
+        if (event.category === "gate")
+            return;
+        if (!event.content)
+            return;
+        const icons = {
+            milestone: "\u2728",
+            error: "\u274C",
+            warning: "\u26A0\uFE0F",
+            command: "\u25B6",
+            event: "\u{1F4AC}",
+            "agent-response": "\u{1F916}",
+            "agent-text": "\u270F\uFE0F",
+        };
+        const icon = icons[event.category] ?? "\u{1F4AC}";
+        // For agent-text, show the first line of fullText instead of the generic
+        // "[role] ✎ (N lines)" summary. This gives meaningful context (e.g., file
+        // being worked on) instead of identical-looking messages.
+        let displayText = event.content;
+        if (event.category === "agent-text" && event.fullText) {
+            const firstLine = event.fullText.split("\n")[0].trim();
+            if (firstLine) {
+                displayText = firstLine;
+            }
+        }
+        // Send as plain text to avoid Markdown parse failures from LLM output
+        tg.sendMessage(`${icon} ${displayText}`, { parse_mode: null }).catch(() => { });
+    }));
+    // 2. Vibe event mirroring
+    unsubs.push(pi.events.on("vibe:pipeline", (raw) => {
+        const event = raw;
+        if (event.type === "feature_start") {
+            pipelineActive = true;
+        }
+        else if (event.type === "pipeline_complete" ||
+            event.type === "feature_complete" ||
+            event.type === "orchestration_complete" ||
+            event.type === "feature_aborted") {
+            pipelineActive = false;
+            gateWaiting = false;
+        }
+        handlePipelineEvent(event, renderer);
+    }), pi.events.on("vibe:activity", (raw) => {
+        const event = raw;
+        renderer.addActivity(event.text);
+    }), pi.events.on("vibe:gate", (raw) => {
+        const event = raw;
+        gateWaiting = true;
+        const keyboard = {
+            inline_keyboard: [
+                [
+                    { text: "\u2705 Approve", callback_data: "Approve" },
+                    { text: "\u274C Reject", callback_data: "Reject" },
+                ],
+                [
+                    { text: "\u23ED Skip", callback_data: "Skip" },
+                    { text: "\u26D4 Abort", callback_data: "Abort" },
+                ],
+            ],
+        };
+        tg.sendMessage(`\u23F3 Gate: ${event.summary}`, { reply_markup: keyboard }).catch(() => { });
+    }));
+    // Reset gate flag when pipeline resumes after gate approval
+    unsubs.push(pi.events.on("vibe:pipeline", (raw) => {
+        const event = raw;
+        if (event.type === "step_start") {
+            gateWaiting = false;
+        }
+    }));
+    // 3. Polling loop (fire-and-forget)
+    startPolling(tg, pi, ctx, abortController.signal, vibeDir, {
+        isVibeBusy: () => pipelineActive || gateWaiting,
+        isGateWaiting: () => gateWaiting,
+        clearGateWaiting: () => {
+            gateWaiting = false;
+        },
+        commands,
+    });
+    // 4. Register bot menu commands (including subcommands as separate entries)
+    const menuCommands = [
+        { command: "stop", description: "Stop current execution" },
+    ];
+    for (const cmd of commands) {
+        menuCommands.push({ command: cmd.name, description: cmd.description ?? `/${cmd.name}` });
+        if (cmd.subcommands) {
+            for (const sub of cmd.subcommands) {
+                // Telegram command names: lowercase, digits, underscores only
+                menuCommands.push({ command: `${cmd.name}_${sub.name}`, description: sub.description });
+            }
+        }
+    }
+    tg.setMyCommands(menuCommands).catch(() => { });
+    // 5. Startup
+    if (tg.hasChatId()) {
+        const model = ctx.model;
+        const startupMsg = model
+            ? `Telegram bridge active. Model: ${model.provider}/${model.id}`
+            : "Telegram bridge active. No model.";
+        tg.sendMessage(startupMsg).catch(() => { });
+    }
+    else {
+        ctx.ui.notify("[Telegram] Waiting for first message from your bot", "info");
+    }
+    // Return cleanup function
+    return () => {
+        abortController.abort();
+        for (const unsub of unsubs)
+            unsub();
+    };
+}
+// ─── Polling ─────────────────────────────────────────────────────────────────
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function startPolling(tg, pi, ctx, signal, vibeDir, state) {
+    while (!signal.aborted) {
+        try {
+            const updates = await tg.getUpdates();
+            for (const update of updates) {
+                if (signal.aborted)
+                    break;
+                // Handle callback queries (gate button clicks)
+                if (update.callback_query) {
+                    const data = update.callback_query.data ?? "";
+                    tg.answerCallbackQuery(update.callback_query.id).catch(() => { });
+                    if (state.isGateWaiting()) {
+                        pi.events.emit("vibe:gate_response", { choice: data });
+                        state.clearGateWaiting();
+                    }
+                    continue;
+                }
+                // Auto-discover chat ID from first incoming message
+                if (!tg.hasChatId() && update.message?.chat.id) {
+                    const discoveredId = String(update.message.chat.id);
+                    tg.setChatId(discoveredId);
+                    saveDiscoveredChatId(vibeDir, discoveredId);
+                    const model = ctx.model;
+                    const msg = model
+                        ? `Telegram bridge connected. Model: ${model.provider}/${model.id}`
+                        : "Telegram bridge connected.";
+                    await tg.sendMessage(msg).catch(() => { });
+                    ctx.ui.notify("[Telegram] Connected", "info");
+                    continue; // Skip processing discovery message as a command
+                }
+                const text = update.message?.text;
+                if (!text)
+                    continue;
+                if (text === "/stop") {
+                    ctx.abort();
+                    await tg.sendMessage("\u23F8 Stopped.").catch(() => { });
+                    continue;
+                }
+                // Block input during pipeline execution or gate waiting
+                if (state.isVibeBusy()) {
+                    await tg.sendMessage("\u23F3 Pipeline running. Wait or /stop.").catch(() => { });
+                    continue;
+                }
+                if (ctx.isIdle()) {
+                    // Try to dispatch as extension command first.
+                    // Fire-and-forget: command handlers may block on gates/UI,
+                    // and the polling loop must keep running to receive callback queries.
+                    if (text.startsWith("/") && tryDispatchCommandSync(text, state.commands, ctx)) {
+                        continue;
+                    }
+                    pi.sendUserMessage(text);
+                }
+                else {
+                    await tg.sendMessage("\u23F3 Busy. Wait or /stop.").catch(() => { });
+                }
+            }
+        }
+        catch {
+            if (signal.aborted)
+                break;
+            await sleep(5000);
+        }
+    }
+}
+// ─── Command dispatch ────────────────────────────────────────────────────────
+/**
+ * Try to dispatch a slash command to a registered handler.
+ * Handles both direct commands (/vibe resume) and Telegram-style
+ * subcommands (/vibe_resume) by mapping underscores back to spaces.
+ *
+ * Returns true if the command was matched. The handler runs as
+ * fire-and-forget so the polling loop is never blocked (handlers
+ * may await gates/UI dialogs indefinitely).
+ */
+function tryDispatchCommandSync(text, commands, ctx) {
+    // Strip leading / and optional @BotName suffix (Telegram adds this in groups)
+    const raw = text.slice(1).split("@")[0];
+    const spaceIndex = raw.indexOf(" ");
+    const fullName = spaceIndex === -1 ? raw : raw.slice(0, spaceIndex);
+    const trailingArgs = spaceIndex === -1 ? "" : raw.slice(spaceIndex + 1);
+    // 1. Direct match: /vibe resume → command="vibe", args="resume"
+    const directMatch = commands.find((c) => c.name === fullName);
+    if (directMatch) {
+        directMatch.handler(trailingArgs, ctx).catch(() => { });
+        return true;
+    }
+    // 2. Subcommand match: /vibe_resume → command="vibe", args="resume"
+    for (const cmd of commands) {
+        if (!cmd.subcommands)
+            continue;
+        const prefix = `${cmd.name}_`;
+        if (fullName.startsWith(prefix)) {
+            const subName = fullName.slice(prefix.length);
+            if (cmd.subcommands.some((s) => s.name === subName)) {
+                const combinedArgs = trailingArgs ? `${subName} ${trailingArgs}` : subName;
+                cmd.handler(combinedArgs, ctx).catch(() => { });
+                return true;
+            }
+        }
+    }
+    return false;
+}
+// ─── Pipeline event handler ─────────────────────────────────────────────────
+function handlePipelineEvent(event, renderer) {
+    switch (event.type) {
+        case "feature_start": {
+            renderer.setPipelineMode(true);
+            const title = event.data?.title;
+            renderer.setPhase(title ? `${event.featureId}: ${title}` : event.featureId);
+            const totalSteps = event.data?.totalSteps;
+            if (totalSteps) {
+                const steps = [];
+                for (let i = 0; i < totalSteps; i++) {
+                    steps.push({ label: `Step ${i + 1}`, status: "pending" });
+                }
+                renderer.setSteps(steps);
+            }
+            break;
+        }
+        case "step_start": {
+            const stepIndex = event.data?.stepIndex;
+            if (stepIndex !== undefined) {
+                renderer.updateStep(stepIndex, "running");
+            }
+            if (event.step) {
+                renderer.updateStepLabel(event.data?.stepIndex, event.step.agent);
+            }
+            break;
+        }
+        case "step_complete": {
+            const stepIndex = event.data?.stepIndex;
+            if (stepIndex !== undefined) {
+                const elapsed = event.data?.elapsed;
+                const usage = event.data?.usage;
+                renderer.updateStep(stepIndex, "done", elapsed !== undefined ? elapsed / 1000 : undefined, usage?.cost);
+            }
+            break;
+        }
+        case "step_failed": {
+            const stepIndex = event.data?.stepIndex;
+            if (stepIndex !== undefined) {
+                renderer.updateStep(stepIndex, "failed");
+            }
+            break;
+        }
+        case "pipeline_complete":
+        case "feature_complete":
+        case "orchestration_complete":
+        case "feature_aborted": {
+            renderer.setPipelineMode(false);
+            renderer.finalize().catch(() => { });
+            break;
+        }
+    }
+}
+//# sourceMappingURL=bridge.js.map