pi-archimedes 0.2.0

package/package.json

@@ -0,0 +1,32 @@
+{
+  "name": "pi-archimedes",
+  "version": "0.2.0",
+  "type": "module",
+  "keywords": [
+    "pi-package"
+  ],
+  "description": "Visual polish and useful context for the Pi coding agent TUI",
+  "files": [
+    "src"
+  ],
+  "main": "./src/index.ts",
+  "dependencies": {
+    "@pi-archimedes/core": "0.2.0",
+    "@pi-archimedes/diff": "0.2.0",
+    "@pi-archimedes/footer": "0.2.0",
+    "@pi-archimedes/image-paste": "0.2.0"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": ">=0.1.0",
+    "@earendil-works/pi-tui": ">=0.1.0"
+  },
+  "devDependencies": {
+    "typescript": "^6.0.0"
+  },
+  "pi": {
+    "extensions": [
+      "./src/index.ts"
+    ],
+    "image": "https://raw.githubusercontent.com/danielcherubini/pi-ui-hephaestus/main/docs/splash-screen.png"
+  }
+}

package/src/config.ts

@@ -0,0 +1,87 @@
+import { readFileSync, writeFileSync, existsSync } from "node:fs";
+import { join } from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+
+// ── Re-export core config ──────────────────────────────────────────────
+
+import {
+  loadCoreConfig,
+  saveCoreConfig,
+  DEFAULT_CORE_CONFIG,
+  ANIMATION_STYLES,
+  type CoreConfig,
+} from "@pi-archimedes/core/config";
+export {
+  loadCoreConfig,
+  saveCoreConfig,
+  DEFAULT_CORE_CONFIG,
+  ANIMATION_STYLES,
+  type CoreConfig,
+} from "@pi-archimedes/core/config";
+
+// ── Re-export footer config ────────────────────────────────────────────
+
+import {
+  loadFooterConfig,
+  saveFooterConfig,
+  DEFAULT_FOOTER_CONFIG,
+  type FooterConfig,
+} from "@pi-archimedes/footer/config";
+export {
+  loadFooterConfig,
+  saveFooterConfig,
+  DEFAULT_FOOTER_CONFIG,
+  type FooterConfig,
+} from "@pi-archimedes/footer/config";
+
+// ── Diff config ────────────────────────────────────────────────────────
+
+import type { DiffConfig } from "@pi-archimedes/diff";
+export type { DiffConfig } from "@pi-archimedes/diff";
+
+export const DEFAULT_DIFF_CONFIG: DiffConfig = {
+  diffTheme: "github-dark",
+  diffSplitMinWidth: 150,
+  diffSplitMinCodeWidth: 60,
+};
+
+const SETTINGS_PATH = join(getAgentDir(), "settings.json");
+
+export function loadDiffConfig(): DiffConfig {
+  if (existsSync(SETTINGS_PATH)) {
+    try {
+      const full = JSON.parse(readFileSync(SETTINGS_PATH, "utf-8"));
+      return { ...DEFAULT_DIFF_CONFIG, ...(full["archimedes.diff"] ?? {}) };
+    } catch {
+      /* ignore corrupt file */
+    }
+  }
+  return DEFAULT_DIFF_CONFIG;
+}
+
+export function saveDiffConfig(config: DiffConfig): void {
+  let full: Record<string, unknown> = {};
+  if (existsSync(SETTINGS_PATH)) {
+    try {
+      full = JSON.parse(readFileSync(SETTINGS_PATH, "utf-8"));
+    } catch {
+      /* ignore corrupt file */
+    }
+  }
+  full["archimedes.diff"] = config;
+  writeFileSync(SETTINGS_PATH, JSON.stringify(full, null, 2), "utf-8");
+}
+
+// ── Composed config loader ─────────────────────────────────────────────
+
+export function loadAllConfig(): {
+  core: CoreConfig;
+  footer: FooterConfig;
+  diff: DiffConfig;
+} {
+  return {
+    core: loadCoreConfig(),
+    footer: loadFooterConfig(),
+    diff: loadDiffConfig(),
+  };
+}

package/src/index.ts

@@ -0,0 +1,38 @@
+import type { ExtensionAPI, ExtensionContext, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+
+import { registerCore } from "@pi-archimedes/core";
+import { registerFooter } from "@pi-archimedes/footer";
+import { registerDiffTools } from "@pi-archimedes/diff";
+import { registerImagePaste, initImagePasteSession, shutdownImagePaste } from "@pi-archimedes/image-paste";
+import { loadDiffConfig } from "./config.js";
+import { openSettings } from "./settings.js";
+
+export default function (pi: ExtensionAPI): void {
+  // Register all component extensions
+  registerCore(pi);
+  registerFooter(pi);
+
+  // Register image paste (shortcuts, input handler, preview renderer)
+  registerImagePaste(pi);
+
+  // session_shutdown handler (top-level to prevent accumulation on /reload)
+  pi.on("session_shutdown", (_event, _ctx) => {
+    shutdownImagePaste();
+  });
+
+  pi.on("session_start", (_event, ctx: ExtensionContext) => {
+    // Register diff tools (needs getTheme + readConfig callbacks)
+    registerDiffTools(pi, () => ctx.ui.theme, () => loadDiffConfig());
+
+    // Initialize image paste for this session
+    initImagePasteSession(ctx);
+  });
+
+  // Register /archimedes command
+  pi.registerCommand("archimedes", {
+    description: "Open Archimedes settings",
+    handler: async (_args: string, ctx: ExtensionCommandContext) => {
+      openSettings(pi, ctx);
+    },
+  });
+}

package/src/settings.ts

@@ -0,0 +1,203 @@
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { getSettingsListTheme } from "@earendil-works/pi-coding-agent";
+import type { Theme } from "@earendil-works/pi-coding-agent";
+import { SettingsList, type SettingItem, TUI } from "@earendil-works/pi-tui";
+
+import { getCoreSettingsItems } from "@pi-archimedes/core";
+import { getFooterSettingsItems } from "@pi-archimedes/footer/config";
+import { getDiffSettingsItems } from "@pi-archimedes/diff";
+import {
+  loadAllConfig,
+  saveCoreConfig,
+  saveFooterConfig,
+  saveDiffConfig,
+  ANIMATION_STYLES,
+  type CoreConfig,
+  type FooterConfig,
+  type DiffConfig,
+} from "./config.js";
+
+// ── Factory: text submenu ───────────────────────────────────────────────
+
+function createTextSubmenu(opts: {
+  label: string;
+  cancelHint?: string;
+  confirmHint?: string;
+}): SettingItem["submenu"] {
+  return (currentValue: string, done: (selectedValue?: string) => void) => {
+    const state = { value: currentValue };
+    return {
+      invalidate(): void { /* no-op */ },
+      render(): string[] {
+        const hints: string[] = [];
+        if (opts.cancelHint) hints.push(opts.cancelHint);
+        if (opts.confirmHint) hints.push(opts.confirmHint);
+        return [
+          opts.label,
+          "",
+          `  ${state.value}`,
+          "",
+          hints.join(" | "),
+        ];
+      },
+      handleInput(data: string): void {
+        if (data === "\x1b") { done(); return; }
+        if (data === "\r" || data === "\n") { done(state.value); return; }
+        if (data === "\x7f" || data === "\x08") { state.value = state.value.slice(0, -1); }
+        else if (data.length === 1) { state.value += data; }
+      },
+    };
+  };
+}
+
+// ── Factory: number submenu ─────────────────────────────────────────────
+
+function createNumberSubmenu(opts: {
+  label: string;
+  cancelHint?: string;
+  confirmHint?: string;
+  min?: number;
+}): SettingItem["submenu"] {
+  return (currentValue: string, done: (selectedValue?: string) => void) => {
+    const state = { value: currentValue };
+    return {
+      invalidate(): void { /* no-op */ },
+      render(): string[] {
+        const hints: string[] = [];
+        if (opts.cancelHint) hints.push(opts.cancelHint);
+        if (opts.confirmHint) hints.push(opts.confirmHint);
+        return [
+          opts.label,
+          "",
+          `  ${state.value}`,
+          "",
+          hints.join(" | "),
+        ];
+      },
+      handleInput(data: string): void {
+        if (data === "\x1b") { done(); return; }
+        if (data === "\r" || data === "\n") {
+          const n = parseInt(state.value, 10);
+          if (Number.isFinite(n) && (!opts.min || n >= opts.min)) done(String(n));
+          else done();
+          return;
+        }
+        if (data === "\x7f" || data === "\x08") { state.value = state.value.slice(0, -1); }
+        else if (/^\d$/.test(data)) { state.value += data; }
+      },
+    };
+  };
+}
+
+// ── Settings UI ─────────────────────────────────────────────────────────
+
+export function openSettings(pi: ExtensionAPI, ctx: ExtensionContext): void {
+  const allConfig = loadAllConfig();
+
+  const coreConfig: CoreConfig = { ...allConfig.core };
+  const footerConfig: FooterConfig = { ...allConfig.footer };
+  const diffConfig: DiffConfig = { ...allConfig.diff };
+
+  // Build composed items from sub-packages
+  const coreItems = getCoreSettingsItems(coreConfig);
+  const footerItems = getFooterSettingsItems();
+  const diffItems = getDiffSettingsItems();
+
+  // Add submenus for text/number fields
+  const addSubmenus = (items: SettingItem[]) => {
+    for (const item of items) {
+      if (item.id === "labelText") {
+        item.submenu = createTextSubmenu({
+          label: "Enter label text (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "ENTER: confirm",
+        });
+      } else if (item.id === "labelColor") {
+        item.submenu = createTextSubmenu({
+          label: "Enter RGB color (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "ENTER: confirm",
+        });
+      } else if (item.id === "diffTheme") {
+        item.submenu = createTextSubmenu({
+          label: "Enter Shiki theme (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "ENTER: confirm",
+        });
+      } else if (item.id === "diffSplitMinWidth") {
+        item.submenu = createNumberSubmenu({
+          label: "Enter min width (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "min 100",
+          min: 100,
+        });
+      } else if (item.id === "diffSplitMinCodeWidth") {
+        item.submenu = createNumberSubmenu({
+          label: "Enter min code width (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "min 30",
+          min: 30,
+        });
+      } else if (item.id === "splitThreshold") {
+        item.submenu = createNumberSubmenu({
+          label: "Enter split threshold (ESC to cancel):",
+          cancelHint: "ESC: cancel",
+          confirmHint: "min 80",
+          min: 80,
+        });
+      }
+    }
+  };
+
+  addSubmenus(coreItems);
+  addSubmenus(diffItems);
+  addSubmenus(footerItems);
+
+  const items: SettingItem[] = [
+    ...coreItems,
+    ...footerItems,
+    ...diffItems,
+    {
+      id: "save",
+      label: "Save",
+      description: "Save changes and exit",
+      currentValue: "",
+      values: ["Save"],
+    },
+  ];
+
+  ctx.ui.custom((tui: TUI, theme: Theme, _keybindings, done) => {
+    const settingsList = new SettingsList(items, 10, getSettingsListTheme(), (id: string, newValue: string) => {
+      switch (id) {
+        // ── Core settings ──
+        case "mutedTheme": coreConfig.mutedTheme = newValue === "On"; break;
+        case "codeUnindent": coreConfig.codeUnindent = newValue === "On"; break;
+        case "labelText": coreConfig.labelText = newValue; break;
+        case "labelColor": coreConfig.labelColor = newValue; break;
+        case "animationStyle": coreConfig.animationStyle = newValue as CoreConfig["animationStyle"]; break;
+
+        // ── Footer settings ──
+        case "splitThreshold": footerConfig.splitThreshold = parseInt(newValue, 10); break;
+
+        // ── Diff settings ──
+        case "diffTheme": diffConfig.diffTheme = newValue; break;
+        case "diffSplitMinWidth": diffConfig.diffSplitMinWidth = parseInt(newValue, 10); break;
+        case "diffSplitMinCodeWidth": diffConfig.diffSplitMinCodeWidth = parseInt(newValue, 10); break;
+
+        // ── Save ──
+        case "save": {
+          saveCoreConfig(coreConfig);
+          saveFooterConfig(footerConfig);
+          saveDiffConfig(diffConfig);
+          done(undefined);
+          return;
+        }
+      }
+    }, () => {
+      // ESC cancels without saving
+      done(undefined);
+    });
+
+    return settingsList;
+  });
+}