@aliou/pi-dev-kit 0.6.4 → 0.7.0

package/src/tools/docs-tool.ts

@@ -4,17 +4,19 @@ import { ToolBody, ToolCallHeader, ToolFooter } from "@aliou/pi-utils-ui";
 import type {
   AgentToolResult,
   ExtensionAPI,
-  ExtensionContext,
   Theme,
-  ToolRenderResultOptions,
-} from "@mariozechner/pi-coding-agent";
-import { keyHint } from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+} from "@earendil-works/pi-coding-agent";
+import { defineTool, keyHint } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import { type Static, Type } from "typebox";
 import { findPiInstallation } from "./utils";
 
 const DocsParamsSchema = Type.Object({});
-type DocsParams = Record<string, never>;
+type DocsParams = Static<typeof DocsParamsSchema>;
+
+function renderDocsCall(_args: DocsParams, theme: Theme) {
+  return new ToolCallHeader({ toolName: "Pi Docs" }, theme);
+}
 
 interface DocsDetails {
   /** Relative paths from the pi install root, markdown only. */
@@ -38,144 +40,139 @@ function listFilesRecursive(dir: string, prefix = ""): string[] {
   return results;
 }
 
-export function setupDocsTool(pi: ExtensionAPI) {
-  pi.registerTool<typeof DocsParamsSchema, DocsDetails>({
-    name: "pi_docs",
-    label: "Pi Documentation",
-    description:
-      "List Pi markdown documentation files (README, docs/, examples/)",
-
-    promptSnippet: "List Pi documentation files",
-    promptGuidelines: [
-      "Use pi_docs to discover available Pi documentation",
-      "pi_docs returns markdown files from README.md, docs/, and examples/",
-    ],
-
-    parameters: DocsParamsSchema,
-
-    async execute(
-      _toolCallId: string,
-      _params: DocsParams,
-      _signal: AbortSignal | undefined,
-      _onUpdate: unknown,
-      _ctx: ExtensionContext,
-    ): Promise<AgentToolResult<DocsDetails>> {
-      const piPath = findPiInstallation();
-      if (!piPath) {
-        throw new Error("Could not locate running Pi installation directory");
-      }
+const docsTool = defineTool({
+  name: "pi_docs",
+  label: "Pi Documentation",
+  description:
+    "List Pi markdown documentation files (README, docs/, examples/)",
+
+  promptSnippet: "List Pi documentation files",
+  promptGuidelines: [
+    "Use pi_docs to discover available Pi documentation",
+    "pi_docs returns markdown files from README.md, docs/, and examples/",
+  ],
+
+  parameters: DocsParamsSchema,
+
+  async execute(
+    _toolCallId,
+    _params,
+    _signal,
+    _onUpdate,
+    _ctx,
+  ): Promise<AgentToolResult<DocsDetails>> {
+    const piPath = findPiInstallation();
+    if (!piPath) {
+      throw new Error("Could not locate running Pi installation directory");
+    }
 
-      const readmePath = path.join(piPath, "README.md");
-      const docsDir = path.join(piPath, "docs");
-      const examplesDir = path.join(piPath, "examples");
+    const readmePath = path.join(piPath, "README.md");
+    const docsDir = path.join(piPath, "docs");
+    const examplesDir = path.join(piPath, "examples");
 
-      const docFiles: string[] = [];
+    const docFiles: string[] = [];
 
-      if (fs.existsSync(readmePath)) {
-        docFiles.push("README.md");
-      }
+    if (fs.existsSync(readmePath)) {
+      docFiles.push("README.md");
+    }
 
-      if (fs.existsSync(docsDir)) {
-        for (const file of listFilesRecursive(docsDir)) {
-          if (file.endsWith(".md")) {
-            docFiles.push(`docs/${file}`);
-          }
+    if (fs.existsSync(docsDir)) {
+      for (const file of listFilesRecursive(docsDir)) {
+        if (file.endsWith(".md")) {
+          docFiles.push(`docs/${file}`);
         }
       }
+    }
 
-      if (fs.existsSync(examplesDir)) {
-        for (const file of listFilesRecursive(examplesDir)) {
-          if (file.endsWith(".md")) {
-            docFiles.push(`examples/${file}`);
-          }
+    if (fs.existsSync(examplesDir)) {
+      for (const file of listFilesRecursive(examplesDir)) {
+        if (file.endsWith(".md")) {
+          docFiles.push(`examples/${file}`);
         }
       }
+    }
 
-      if (docFiles.length === 0) {
-        throw new Error(
-          `No markdown documentation found in Pi installation at ${piPath}`,
-        );
-      }
+    if (docFiles.length === 0) {
+      throw new Error(
+        `No markdown documentation found in Pi installation at ${piPath}`,
+      );
+    }
 
-      // Content sent to LLM: full relative paths so it can read them.
-      const lines = docFiles.map((rel) => `${path.join(piPath, rel)} (${rel})`);
-      const message = `${docFiles.length} markdown files:\n${lines.join("\n")}`;
+    // Content sent to LLM: full relative paths so it can read them.
+    const lines = docFiles.map((rel) => `${path.join(piPath, rel)} (${rel})`);
+    const message = `${docFiles.length} markdown files:\n${lines.join("\n")}`;
+
+    return {
+      content: [{ type: "text", text: message }],
+      details: {
+        docFiles,
+        installPath: piPath,
+      },
+    };
+  },
+
+  renderCall(args, theme) {
+    return renderDocsCall(args, theme);
+  },
+
+  renderResult(result, options, theme) {
+    const { details } = result;
+    const { isPartial } = options;
+
+    // Handle isPartial first (this tool doesn't stream, but keep the pattern)
+    if (isPartial) {
+      return new Text(theme.fg("dim", "Loading..."), 0, 0);
+    }
 
-      return {
-        content: [{ type: "text", text: message }],
-        details: {
-          docFiles,
-          installPath: piPath,
-        },
-      };
-    },
-
-    renderCall(_args: DocsParams, theme: Theme) {
-      return new ToolCallHeader({ toolName: "Pi Docs" }, theme);
-    },
-
-    renderResult(
-      result: AgentToolResult<DocsDetails>,
-      options: ToolRenderResultOptions,
-      theme: Theme,
-    ) {
-      const { details } = result;
-      const { isPartial } = options;
-
-      // Handle isPartial first (this tool doesn't stream, but keep the pattern)
-      if (isPartial) {
-        return new Text(theme.fg("dim", "Loading..."), 0, 0);
-      }
+    // Check for missing expected fields in details to detect errors
+    if (!details || !details.docFiles) {
+      const text = result.content[0];
+      return new Text(
+        text?.type === "text" && text.text ? text.text : "No result",
+        0,
+        0,
+      );
+    }
 
-      // Check for missing expected fields in details to detect errors
-      if (!details || !details.docFiles) {
-        const text = result.content[0];
-        return new Text(
-          text?.type === "text" && text.text ? text.text : "No result",
-          0,
-          0,
-        );
+    const { docFiles } = details;
+    const fields: Array<
+      { label: string; value: string; showCollapsed?: boolean } | Text
+    > = [];
+
+    if (options.expanded) {
+      // Expanded view: show full file list
+      const lines: string[] = [];
+      lines.push(theme.fg("accent", `${docFiles.length} markdown files:`), "");
+      for (const rel of docFiles) {
+        lines.push(theme.fg("dim", `  ${rel}`));
       }
+      fields.push(new Text(lines.join("\n"), 0, 0));
+    } else {
+      // Collapsed view: show file count + expand hint
+      fields.push({
+        label: "Files",
+        value:
+          theme.fg("accent", `${docFiles.length} markdown files`) +
+          ` (${keyHint("app.tools.expand", "to expand")})`,
+        showCollapsed: true,
+      });
+    }
 
-      const { docFiles } = details;
-      const fields: Array<
-        { label: string; value: string; showCollapsed?: boolean } | Text
-      > = [];
-
-      if (options.expanded) {
-        // Expanded view: show full file list
-        const lines: string[] = [];
-        lines.push(
-          theme.fg("accent", `${docFiles.length} markdown files:`),
-          "",
-        );
-        for (const rel of docFiles) {
-          lines.push(theme.fg("dim", `  ${rel}`));
-        }
-        fields.push(new Text(lines.join("\n"), 0, 0));
-      } else {
-        // Collapsed view: show file count + expand hint
-        fields.push({
-          label: "Files",
-          value:
-            theme.fg("accent", `${docFiles.length} markdown files`) +
-            ` (${keyHint("app.tools.expand", "to expand")})`,
-          showCollapsed: true,
-        });
-      }
+    // Only show footer if there are items worth showing
+    const footer = new ToolFooter(theme, {
+      items: [
+        {
+          label: "docs",
+          value: String(docFiles.length),
+          tone: "accent",
+        },
+      ],
+    });
 
-      // Only show footer if there are items worth showing
-      const footer = new ToolFooter(theme, {
-        items: [
-          {
-            label: "docs",
-            value: String(docFiles.length),
-            tone: "accent",
-          },
-        ],
-      });
+    return new ToolBody({ fields, footer }, options, theme);
+  },
+});
 
-      return new ToolBody({ fields, footer }, options, theme);
-    },
-  });
+export function setupDocsTool(pi: ExtensionAPI) {
+  pi.registerTool(docsTool);
 }

package/src/tools/index.ts

@@ -1,4 +1,4 @@
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { setupChangelogTool } from "./changelog-tool";
 import { setupDocsTool } from "./docs-tool";
 import { setupPackageManagerTool } from "./package-manager-tool";
@@ -10,3 +10,7 @@ export function setupTools(pi: ExtensionAPI) {
   setupDocsTool(pi);
   setupChangelogTool(pi);
 }
+
+export default function toolsExtension(pi: ExtensionAPI) {
+  setupTools(pi);
+}

package/src/tools/package-manager-tool.ts

@@ -4,15 +4,18 @@ import { ToolCallHeader } from "@aliou/pi-utils-ui";
 import type {
   AgentToolResult,
   ExtensionAPI,
-  ExtensionContext,
   Theme,
-  ToolRenderResultOptions,
-} from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+} from "@earendil-works/pi-coding-agent";
+import { defineTool } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import { type Static, Type } from "typebox";
 
 const Params = Type.Object({});
-type PackageManagerParams = Record<string, never>;
+type PackageManagerParams = Static<typeof Params>;
+
+function renderPackageManagerCall(_args: PackageManagerParams, theme: Theme) {
+  return new ToolCallHeader({ toolName: "Detect Package Manager" }, theme);
+}
 
 interface PackageManagerDetails {
   packageManager?: string;
@@ -38,157 +41,159 @@ const COMMANDS: Record<string, { install: string; run: string }> = {
   bun: { install: "bun install", run: "bun run" },
 };
 
-export function setupPackageManagerTool(pi: ExtensionAPI) {
-  pi.registerTool<typeof Params, PackageManagerDetails>({
-    name: "detect_package_manager",
-    label: "Package Manager",
-    description:
-      "Detect the package manager used in the current project by checking lockfiles and package.json",
-    promptSnippet: "Detect the package manager for this project",
-    promptGuidelines: [
-      "Use detect_package_manager when you need to know which package manager (npm, yarn, pnpm, bun) the project uses",
-      "detect_package_manager is helpful before running install commands or scripts",
-    ],
-
-    parameters: Params,
-
-    async execute(
-      _toolCallId: string,
-      _params: PackageManagerParams,
-      _signal: AbortSignal | undefined,
-      _onUpdate: unknown,
-      ctx: ExtensionContext,
-    ): Promise<AgentToolResult<PackageManagerDetails>> {
-      const cwd = ctx.cwd;
-
-      const packageJsonPath = path.join(cwd, "package.json");
-      if (!fs.existsSync(packageJsonPath)) {
-        throw new Error(`No package.json found in ${cwd}`);
-      }
-
-      // Walk up from cwd to repo root, collecting packageManager and lockfile.
-      // Stop at .git boundary to avoid escaping the repository.
-      let declaredPm: string | undefined;
-      let declaredVersion: string | undefined;
-      let lockfile: string | undefined;
-      let lockfilePm: string | undefined;
-
-      let searchDir = cwd;
-      while (true) {
-        // Check packageManager field in package.json
-        if (!declaredPm) {
-          const pkgPath = path.join(searchDir, "package.json");
-          try {
-            if (fs.existsSync(pkgPath)) {
-              const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
-              if (typeof pkg.packageManager === "string") {
-                const match = pkg.packageManager.match(/^([^@]+)@?(.*)?$/);
-                if (match) {
-                  declaredPm = match[1];
-                  declaredVersion = match[2] || undefined;
-                }
+const packageManagerTool = defineTool({
+  name: "detect_package_manager",
+  label: "Package Manager",
+  description:
+    "Detect the package manager used in the current project by checking lockfiles and package.json",
+  promptSnippet: "Detect the package manager for this project",
+  promptGuidelines: [
+    "Use detect_package_manager when you need to know which package manager (npm, yarn, pnpm, bun) the project uses",
+    "detect_package_manager is helpful before running install commands or scripts",
+  ],
+
+  parameters: Params,
+
+  async execute(
+    _toolCallId,
+    _params,
+    _signal,
+    _onUpdate,
+    ctx,
+  ): Promise<AgentToolResult<PackageManagerDetails>> {
+    const cwd = ctx.cwd;
+
+    const packageJsonPath = path.join(cwd, "package.json");
+    if (!fs.existsSync(packageJsonPath)) {
+      throw new Error(`No package.json found in ${cwd}`);
+    }
+
+    // Walk up from cwd to repo root, collecting packageManager and lockfile.
+    // Stop at .git boundary to avoid escaping the repository.
+    let declaredPm: string | undefined;
+    let declaredVersion: string | undefined;
+    let lockfile: string | undefined;
+    let lockfilePm: string | undefined;
+
+    let searchDir = cwd;
+    while (true) {
+      // Check packageManager field in package.json
+      if (!declaredPm) {
+        const pkgPath = path.join(searchDir, "package.json");
+        try {
+          if (fs.existsSync(pkgPath)) {
+            const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+            if (typeof pkg.packageManager === "string") {
+              const match = pkg.packageManager.match(/^([^@]+)@?(.*)?$/);
+              if (match) {
+                declaredPm = match[1];
+                declaredVersion = match[2] || undefined;
               }
             }
-          } catch {
-            // Ignore parse errors
           }
+        } catch {
+          // Ignore parse errors
         }
+      }
 
-        // Check lockfiles
-        if (!lockfilePm) {
-          for (const [filename, pm] of Object.entries(LOCKFILES)) {
-            if (fs.existsSync(path.join(searchDir, filename))) {
-              lockfilePm = pm;
-              lockfile = filename;
-              break;
-            }
+      // Check lockfiles
+      if (!lockfilePm) {
+        for (const [filename, pm] of Object.entries(LOCKFILES)) {
+          if (fs.existsSync(path.join(searchDir, filename))) {
+            lockfilePm = pm;
+            lockfile = filename;
+            break;
           }
         }
-
-        // Stop if we found both, hit .git, or hit filesystem root
-        if (
-          (declaredPm && lockfilePm) ||
-          fs.existsSync(path.join(searchDir, ".git"))
-        ) {
-          break;
-        }
-        const parent = path.dirname(searchDir);
-        if (parent === searchDir) break;
-        searchDir = parent;
       }
 
-      const pm = declaredPm || lockfilePm || "npm";
-      const fallback = { install: `${pm} install`, run: pm };
-      const commands = COMMANDS[pm] ?? fallback;
-
-      const parts: string[] = [];
-      parts.push(`Package manager: ${pm}`);
-      if (declaredVersion) {
-        parts.push(`Declared version: ${declaredVersion}`);
-      }
-      if (lockfile) {
-        parts.push(`Lockfile: ${lockfile}`);
-      }
-      if (!lockfile && !declaredPm) {
-        parts.push(
-          "No lockfile or packageManager field found, defaulting to npm",
-        );
-      }
-      parts.push(`Install: ${commands.install}`);
-      parts.push(`Run: ${commands.run}`);
-
-      const message = parts.join("\n");
-
-      return {
-        content: [{ type: "text", text: message }],
-        details: {
-          packageManager: pm,
-          version: declaredVersion,
-          lockfile,
-          installCommand: commands.install,
-          runCommand: commands.run,
-          cwd,
-        },
-      };
-    },
-
-    renderCall(_args: PackageManagerParams, theme: Theme) {
-      return new ToolCallHeader({ toolName: "Detect Package Manager" }, theme);
-    },
-
-    renderResult(
-      result: AgentToolResult<PackageManagerDetails>,
-      _options: ToolRenderResultOptions,
-      theme: Theme,
-    ): Text {
-      const { details } = result;
-
-      // Check for missing expected fields (framework passes {} on error)
-      if (!details?.packageManager) {
-        // Extract error message from result.content
-        const text = result.content[0];
-        const errorMessage =
-          text?.type === "text" && text.text
-            ? text.text
-            : "Failed to detect package manager";
-        return new Text(theme.fg("error", errorMessage), 0, 0);
+      // Stop if we found both, hit .git, or hit filesystem root
+      if (
+        (declaredPm && lockfilePm) ||
+        fs.existsSync(path.join(searchDir, ".git"))
+      ) {
+        break;
       }
-
-      const lines: string[] = [];
-      lines.push(
-        theme.fg(
-          "success",
-          `Package manager: ${theme.bold(details.packageManager)}`,
-        ),
+      const parent = path.dirname(searchDir);
+      if (parent === searchDir) break;
+      searchDir = parent;
+    }
+
+    const pm = declaredPm || lockfilePm || "npm";
+    const fallback = { install: `${pm} install`, run: pm };
+    const commands = COMMANDS[pm] ?? fallback;
+
+    const parts: string[] = [];
+    parts.push(`Package manager: ${pm}`);
+    if (declaredVersion) {
+      parts.push(`Declared version: ${declaredVersion}`);
+    }
+    if (lockfile) {
+      parts.push(`Lockfile: ${lockfile}`);
+    }
+    if (!lockfile && !declaredPm) {
+      parts.push(
+        "No lockfile or packageManager field found, defaulting to npm",
       );
-      if (details.version) {
-        lines.push(theme.fg("dim", `Version: ${details.version}`));
-      }
-      if (details.lockfile) {
-        lines.push(theme.fg("dim", `Lockfile: ${details.lockfile}`));
-      }
+    }
+    parts.push(`Install: ${commands.install}`);
+    parts.push(`Run: ${commands.run}`);
+
+    const message = parts.join("\n");
+
+    return {
+      content: [{ type: "text", text: message }],
+      details: {
+        packageManager: pm,
+        version: declaredVersion,
+        lockfile,
+        installCommand: commands.install,
+        runCommand: commands.run,
+        cwd,
+      },
+    };
+  },
+
+  renderCall(args, theme) {
+    return renderPackageManagerCall(args, theme);
+  },
+
+  renderResult(result, options, theme) {
+    if (options.isPartial) {
+      return new Text(theme.fg("dim", "Package Manager: detecting..."), 0, 0);
+    }
+
+    const { details } = result;
+
+    // Check for missing expected fields (framework passes {} on error)
+    if (!details?.packageManager) {
+      // Extract error message from result.content
+      const text = result.content[0];
+      const errorMessage =
+        text?.type === "text" && text.text
+          ? text.text
+          : "Failed to detect package manager";
+      return new Text(theme.fg("error", errorMessage), 0, 0);
+    }
+
+    const lines: string[] = [];
+    lines.push(
+      theme.fg(
+        "success",
+        `Package manager: ${theme.bold(details.packageManager)}`,
+      ),
+    );
+    if (details.version) {
+      lines.push(theme.fg("dim", `Version: ${details.version}`));
+    }
+    if (details.lockfile) {
+      lines.push(theme.fg("dim", `Lockfile: ${details.lockfile}`));
+    }
+
+    return new Text(lines.join("\n"), 0, 0);
+  },
+});
 
-      return new Text(lines.join("\n"), 0, 0);
-    },
-  });
+export function setupPackageManagerTool(pi: ExtensionAPI) {
+  pi.registerTool(packageManagerTool);
 }