@jyork0828/pi-pilot 0.0.1 → 0.0.2

package/dist/index.js

@@ -486,6 +486,62 @@ import {
   SessionManager
 } from "@earendil-works/pi-coding-agent";
 
+// src/extensions/plan/schema.ts
+import { Type } from "typebox";
+var planItemStatusSchema = Type.Union([
+  Type.Literal("pending"),
+  Type.Literal("in_progress"),
+  Type.Literal("completed")
+]);
+var planItemSchema = Type.Object({
+  id: Type.String({ description: 'Short stable identifier for the item (kebab-case, e.g. "wire-factory").' }),
+  title: Type.String({ description: "One-line description of the step. Specific and verifiable." }),
+  status: planItemStatusSchema,
+  note: Type.Optional(Type.String({ description: "Optional short context \u2014 blocker, decision, or follow-up." }))
+});
+var updatePlanParamsSchema = Type.Object({
+  items: Type.Array(planItemSchema, {
+    description: "The full ordered plan. Always send the complete list; previous tool calls are not merged."
+  })
+});
+
+// src/extensions/plan/factory.ts
+var planExtensionFactory = (pi) => {
+  pi.registerTool({
+    name: "update_plan",
+    label: "Plan",
+    description: "Publish or refresh the working plan for the current task. Use for any task that takes 3+ discrete steps. Always send the full ordered list; previous calls are replaced, not merged.",
+    parameters: updatePlanParamsSchema,
+    promptSnippet: "update_plan: maintain a live checklist for multi-step tasks; update statuses as you progress.",
+    promptGuidelines: [
+      "For tasks with 3+ discrete steps, call update_plan once with the full list before starting work.",
+      "After completing each step, call update_plan again with refreshed statuses, then continue immediately \u2014 do not pause the turn just because the plan was updated.",
+      'Plan items should be specific and verifiable (e.g. "Add typebox dependency to packages/server"), not vague ("Set up infrastructure").',
+      "Exactly one item should be in_progress at a time. Mark completed only when the work is actually done."
+    ],
+    execute: async (_toolCallId, params) => ({
+      content: [
+        {
+          type: "text",
+          text: `Plan updated (${params.items.length} item${params.items.length === 1 ? "" : "s"}).`
+        }
+      ],
+      details: params
+    })
+  });
+  pi.registerCommand("plan", {
+    description: "Ask the agent to draft a plan for the current task.",
+    handler: async (args) => {
+      const task = args.trim();
+      const message = task ? `Draft a plan for: ${task}. Use the update_plan tool to publish it before starting any work.` : "Draft a plan for the current task. Use the update_plan tool to publish it before starting any work.";
+      pi.sendUserMessage(message);
+    }
+  });
+};
+
+// src/extensions/index.ts
+var builtinExtensionFactories = [planExtensionFactory];
+
 // src/ws/extension-ui.ts
 var ExtensionUIBridge = class {
   /** Symmetric with the old bridge so workspace-manager's dispose path
@@ -576,7 +632,10 @@ var createRuntime = async ({
 }) => {
   const services = await createAgentSessionServices({
     cwd,
-    resourceLoaderOptions: EXTENSIONS_ENABLED ? void 0 : { noExtensions: true }
+    resourceLoaderOptions: {
+      noExtensions: !EXTENSIONS_ENABLED,
+      extensionFactories: builtinExtensionFactories
+    }
   });
   const sessionResult = await createAgentSessionFromServices({
     services,
@@ -1265,6 +1324,17 @@ async function snapshot(workspaceId, roots, workspaceCwd) {
   const { skills } = loader.getSkills();
   const { prompts } = loader.getPrompts();
   const extResult = loader.getExtensions();
+  const resolvedExtCommandsByPath = /* @__PURE__ */ new Map();
+  const resolvedCommands = runtime.session.extensionRunner?.getRegisteredCommands() ?? [];
+  for (const cmd of resolvedCommands) {
+    const key = cmd.sourceInfo.path;
+    let list = resolvedExtCommandsByPath.get(key);
+    if (!list) {
+      list = [];
+      resolvedExtCommandsByPath.set(key, list);
+    }
+    list.push(cmd.invocationName);
+  }
   const skillsOut = skills.map((s) => ({
     name: s.name,
     description: s.description,
@@ -1282,15 +1352,18 @@ async function snapshot(workspaceId, roots, workspaceCwd) {
     source: toResourceSource(p.sourceInfo),
     managed: scopeFor(p.filePath, roots) !== void 0
   }));
-  const extensionsOut = extResult.extensions.map((e) => ({
-    path: e.path,
-    resolvedPath: e.resolvedPath,
-    source: toResourceSource(e.sourceInfo),
-    tools: [...e.tools.keys()],
-    commands: [...e.commands.keys()],
-    flags: [...e.flags.keys()],
-    shortcuts: [...e.shortcuts.keys()]
-  }));
+  const extensionsOut = extResult.extensions.map((e) => {
+    const resolved = resolvedExtCommandsByPath.get(e.sourceInfo.path);
+    return {
+      path: e.path,
+      resolvedPath: e.resolvedPath,
+      source: toResourceSource(e.sourceInfo),
+      tools: [...e.tools.keys()],
+      commands: resolved ?? [...e.commands.keys()],
+      flags: [...e.flags.keys()],
+      shortcuts: [...e.shortcuts.keys()]
+    };
+  });
   const extensionErrors = extResult.errors.map((err) => ({
     path: err.path,
     error: err.error
@@ -1935,8 +2008,11 @@ function translatePiEvent(ev) {
       return { kind: "turn_start" };
     case "turn_end":
       return { kind: "turn_end" };
-    case "message_start":
-      return { kind: "message_start", role: roleOf(ev.message) };
+    case "message_start": {
+      const role = roleOf(ev.message);
+      const text = role === "user" ? extractUserText2(ev.message) : void 0;
+      return { kind: "message_start", role, text };
+    }
     case "message_end":
       return { kind: "message_end", role: roleOf(ev.message) };
     case "message_update": {
@@ -2023,6 +2099,20 @@ function roleOf(message) {
   }
   return "assistant";
 }
+function extractUserText2(message) {
+  if (!message || typeof message !== "object") return void 0;
+  const content = message.content;
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return void 0;
+  const parts = [];
+  for (const block of content) {
+    if (block && typeof block === "object" && block.type === "text") {
+      const text = block.text;
+      if (typeof text === "string") parts.push(text);
+    }
+  }
+  return parts.length === 0 ? void 0 : parts.join("");
+}
 function extractText(result) {
   if (!result || typeof result !== "object") return void 0;
   const content = result.content;