pim-agent 0.0.1 → 0.10.0

package/packages/core/src/attachments/Attachments.ts

@@ -0,0 +1,69 @@
+import type { PromptOptions } from "@earendil-works/pi-coding-agent";
+import { basename, extname } from "node:path";
+
+import type { StoredAttachment } from "./AttachmentStore";
+
+export type AttachmentPrompt = {
+  /** One line per file, appended to the prompt text. */
+  readonly lines: readonly string[];
+  readonly images: NonNullable<PromptOptions["images"]>;
+};
+
+/** One file a prompt carried, read back out of the text that carried it. */
+export type PromptAttachment = {
+  readonly path: string;
+  readonly isImage: boolean;
+};
+
+export type ParsedPrompt = {
+  /** The prompt with the marker lines removed: what a person actually said. */
+  readonly text: string;
+  readonly files: readonly PromptAttachment[];
+};
+
+// Must keep matching every spelling `render` has ever produced: old session files still carry them.
+const MARKER = /^\[(image )?attachment:\s*(.+)]$/i;
+
+const STAMP = /-\d{10,}$/;
+
+function render(files: readonly StoredAttachment[]): AttachmentPrompt {
+  const lines: string[] = [];
+  const images: NonNullable<PromptOptions["images"]> = [];
+  for (const file of files) {
+    if (file.imageBase64) {
+      images.push({
+        type: "image",
+        data: file.imageBase64,
+        mimeType: file.mimeType,
+      });
+      lines.push(`[Image attachment: ${file.path}]`);
+      continue;
+    }
+    lines.push(`[Attachment: ${file.path}]`);
+  }
+  return { lines, images };
+}
+
+function parse(text: string): ParsedPrompt {
+  const kept: string[] = [];
+  const files: PromptAttachment[] = [];
+  for (const line of text.split("\n")) {
+    const marker = MARKER.exec(line.trim());
+    if (marker) {
+      files.push({ path: marker[2]!, isImage: marker[1] !== undefined });
+      continue;
+    }
+    kept.push(line);
+  }
+  return files.length === 0
+    ? { text, files }
+    : { text: kept.join("\n").trim(), files };
+}
+
+function nameOf(path: string): string {
+  const file = basename(path);
+  const ext = extname(file);
+  return `${file.slice(0, file.length - ext.length).replace(STAMP, "")}${ext}`;
+}
+
+export const Attachments = { render, parse, nameOf };

package/packages/core/src/extensions/CoreExtensions.ts

@@ -0,0 +1,54 @@
+import type {
+  ExtensionFactory,
+  InlineExtension,
+} from "@earendil-works/pi-coding-agent";
+
+import {
+  ExtensionToggles,
+  type PimExtensionName,
+} from "../shared/ExtensionToggles";
+import type { Surface } from "../shared/Surface";
+import applyPatch from "./apply-patch/index";
+import bash from "./bash/index";
+import edit from "./edit/index";
+import glob from "./glob/index";
+import grep from "./grep/index";
+import read from "./read/index";
+import subagent from "./subagent/index";
+import systemPrompt from "./system-prompt/index";
+import todo from "./todo/index";
+import webFetch from "./web-fetch/index";
+import webSearch from "./web-search/index";
+import write from "./write/index";
+
+export type PimInlineExtension = {
+  readonly name: PimExtensionName;
+  readonly factory: ExtensionFactory;
+};
+
+// Enumerated, never globbed: the published tarball must not depend on a directory scan.
+function list(surface?: Surface): readonly PimInlineExtension[] {
+  return [
+    { name: "apply-patch", factory: applyPatch },
+    { name: "bash", factory: bash },
+    { name: "edit", factory: edit },
+    { name: "glob", factory: glob },
+    { name: "grep", factory: grep },
+    { name: "read", factory: read },
+    { name: "subagent", factory: subagent },
+    { name: "system-prompt", factory: systemPrompt(surface) },
+    { name: "todo", factory: todo },
+    { name: "web-fetch", factory: webFetch },
+    { name: "web-search", factory: webSearch },
+    { name: "write", factory: write },
+  ];
+}
+
+function gated(surface?: Surface): InlineExtension[] {
+  return list(surface).map(({ name, factory }) => ({
+    name,
+    factory: ExtensionToggles.gate(name, factory),
+  }));
+}
+
+export const CoreExtensions = { list, gated };

package/packages/core/src/extensions/apply-patch/coordinator.ts

@@ -0,0 +1,53 @@
+const EDIT_TOOL = "edit";
+const APPLY_PATCH_TOOL = "apply_patch";
+
+/** Reconcile the single `edit`/`apply_patch` slot; returns `active` by reference when nothing changes. */
+export function computeActiveTools(
+  available: readonly string[],
+  active: readonly string[],
+  preferApplyPatch: boolean
+): readonly string[] {
+  const hasEdit = active.includes(EDIT_TOOL);
+  const hasApplyPatch = active.includes(APPLY_PATCH_TOOL);
+  const canEdit = available.includes(EDIT_TOOL);
+  const canApplyPatch = available.includes(APPLY_PATCH_TOOL);
+
+  if (!hasEdit && !hasApplyPatch) {
+    if (canEdit !== canApplyPatch) {
+      return [...active, canEdit ? EDIT_TOOL : APPLY_PATCH_TOOL];
+    }
+    return active;
+  }
+
+  if (!canEdit && !canApplyPatch) {
+    return active;
+  }
+
+  const desired =
+    canApplyPatch && (preferApplyPatch || !canEdit)
+      ? APPLY_PATCH_TOOL
+      : EDIT_TOOL;
+  const drop = desired === EDIT_TOOL ? APPLY_PATCH_TOOL : EDIT_TOOL;
+
+  if (active.includes(desired) && !active.includes(drop)) {
+    return active;
+  }
+
+  const result: string[] = [];
+  let placed = false;
+  for (const tool of active) {
+    if (tool === EDIT_TOOL || tool === APPLY_PATCH_TOOL) {
+      if (!placed) {
+        result.push(desired);
+        placed = true;
+      }
+      continue;
+    }
+    result.push(tool);
+  }
+  if (!placed) {
+    result.push(desired);
+  }
+
+  return result;
+}