pi-updater 0.2.0

package/README.md

@@ -0,0 +1,38 @@
+# pi-updater
+
+A Codex-style auto-updater for Pi.
+
+> **Note:** Currently supports npm installations only.
+
+<img width="800" height="482" alt="Screenshot 2026-02-28 at 09 01 37" src="https://github.com/user-attachments/assets/89df2dad-8d91-464b-b3cb-dfd15bce1c06" />
+
+## What it does
+
+**On startup:** if a newer version is available, shows a prompt:
+- **Update now** — install with npm, then restart pi
+- **Skip** — dismiss until next session
+- **Skip this version** — don't ask again until a newer version appears
+
+**`/update`:** manually check for updates (always fetches fresh from npm)
+
+Version checks are cached. Latest version is fetched in the background on startup.
+
+## Install
+
+```bash
+pi install git:github.com/tonze/pi-updater
+```
+
+## Usage
+
+Use `/update` inside pi to manually check for updates and install them.
+
+## Updating this package
+
+```bash
+pi update
+```
+
+## License
+
+MIT

package/index.ts

@@ -0,0 +1,210 @@
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+} from "@mariozechner/pi-coding-agent";
+import { VERSION, BorderedLoader } from "@mariozechner/pi-coding-agent";
+import { readFileSync, writeFileSync, mkdirSync } from "node:fs";
+import { join, dirname } from "node:path";
+import { homedir } from "node:os";
+
+const PACKAGE_NAME = "@mariozechner/pi-coding-agent";
+const REGISTRY_URL = `https://registry.npmjs.org/${PACKAGE_NAME}/latest`;
+const CACHE_FILE = join(homedir(), ".pi", "agent", "update-cache.json");
+
+interface VersionCache {
+  latestVersion: string;
+  dismissedVersion?: string;
+}
+
+function readCache(): VersionCache | undefined {
+  try {
+    return JSON.parse(readFileSync(CACHE_FILE, "utf-8"));
+  } catch {
+    return undefined;
+  }
+}
+
+function writeCache(cache: VersionCache) {
+  try {
+    mkdirSync(dirname(CACHE_FILE), { recursive: true });
+    writeFileSync(CACHE_FILE, JSON.stringify(cache) + "\n");
+  } catch {}
+}
+
+function parseVersion(v: string): [number, number, number] | undefined {
+  const parts = v.trim().split(".");
+  if (parts.length !== 3) return undefined;
+  const nums = parts.map(Number);
+  if (nums.some(isNaN)) return undefined;
+  return nums as [number, number, number];
+}
+
+function isNewer(latest: string, current: string): boolean {
+  const l = parseVersion(latest);
+  const c = parseVersion(current);
+  if (!l || !c) return false;
+  if (l[0] !== c[0]) return l[0] > c[0];
+  if (l[1] !== c[1]) return l[1] > c[1];
+  return l[2] > c[2];
+}
+
+async function fetchLatestVersion(): Promise<string | undefined> {
+  try {
+    const res = await fetch(REGISTRY_URL, {
+      signal: AbortSignal.timeout(10_000),
+    });
+    if (!res.ok) return undefined;
+    return ((await res.json()) as { version?: string }).version;
+  } catch {
+    return undefined;
+  }
+}
+
+/**
+ * Returns the cached latest version if an upgrade is available and not dismissed.
+ * Always kicks off a background fetch to refresh the cache for the next run.
+ */
+function getUpgradeVersion(): string | undefined {
+  const cache = readCache();
+
+  void fetchLatestVersion().then((latest) => {
+    if (!latest) return;
+    // Re-read cache to avoid overwriting a dismissal that happened during the fetch
+    writeCache({
+      latestVersion: latest,
+      dismissedVersion: readCache()?.dismissedVersion,
+    });
+  });
+
+  if (!cache) return undefined;
+  if (!isNewer(cache.latestVersion, VERSION)) return undefined;
+  if (cache.dismissedVersion === cache.latestVersion) return undefined;
+  return cache.latestVersion;
+}
+
+function dismissVersion(version: string) {
+  const cache = readCache();
+  if (!cache) return;
+  cache.dismissedVersion = version;
+  writeCache(cache);
+}
+
+function getInstallCommand(version: string): { program: string; args: string[] } {
+  return {
+    program: "npm",
+    args: ["install", "-g", `${PACKAGE_NAME}@${version}`],
+  };
+}
+
+function fmtCmd(cmd: { program: string; args: string[] }): string {
+  return `${cmd.program} ${cmd.args.join(" ")}`;
+}
+
+export default function (pi: ExtensionAPI) {
+  async function doInstall(
+    ctx: ExtensionContext,
+    latest: string,
+    cmd: { program: string; args: string[] },
+  ) {
+    const success = await ctx.ui.custom<boolean>((tui, theme, _kb, done) => {
+      const loader = new BorderedLoader(tui, theme, `Installing ${latest}...`);
+      loader.onAbort = () => done(false);
+
+      const run = async () => {
+        if (cmd.program === "echo") {
+          ctx.ui.notify(cmd.args[0], "info");
+          return false;
+        }
+        const result = await pi.exec(cmd.program, cmd.args, {
+          timeout: 120_000,
+        });
+        if (result.code !== 0) {
+          ctx.ui.notify(
+            `Update failed (exit ${result.code}): ${result.stderr || result.stdout}`,
+            "error",
+          );
+          return false;
+        }
+        return true;
+      };
+
+      run()
+        .then(done)
+        .catch(() => done(false));
+      return loader;
+    });
+
+    if (!success) return;
+
+    const ok = await ctx.ui.confirm(
+      `Updated to ${latest}!`,
+      "Shut down pi? (Use pi -c to continue this session)",
+    );
+    if (ok) ctx.shutdown();
+  }
+
+  async function showUpdatePrompt(ctx: ExtensionContext, latest: string) {
+    const cmd = getInstallCommand(latest);
+    const choice = await ctx.ui.select(`Update ${VERSION} → ${latest}`, [
+      `Update now (${fmtCmd(cmd)})`,
+      "Skip",
+      "Skip this version",
+    ]);
+
+    if (!choice || choice === "Skip") return;
+    if (choice === "Skip this version") {
+      dismissVersion(latest);
+      return;
+    }
+    await doInstall(ctx, latest, cmd);
+  }
+
+  pi.on("session_start", async (_event, ctx) => {
+    if (!ctx.hasUI) return;
+    const latest = getUpgradeVersion();
+    if (latest) void showUpdatePrompt(ctx, latest);
+  });
+
+  pi.on("session_switch", async (_event, ctx) => {
+    if (!ctx.hasUI) return;
+    const latest = getUpgradeVersion();
+    if (latest) void showUpdatePrompt(ctx, latest);
+  });
+
+  pi.registerCommand("update", {
+    description: "Check for pi updates and install",
+    handler: async (_args, ctx) => {
+      const latest = await ctx.ui.custom<string | null>(
+        (tui, theme, _kb, done) => {
+          const loader = new BorderedLoader(
+            tui,
+            theme,
+            "Checking for updates...",
+          );
+          loader.onAbort = () => done(null);
+          fetchLatestVersion()
+            .then((v) => done(v ?? null))
+            .catch(() => done(null));
+          return loader;
+        },
+      );
+
+      if (!latest) {
+        ctx.ui.notify("Could not reach npm registry.", "error");
+        return;
+      }
+
+      writeCache({
+        latestVersion: latest,
+        dismissedVersion: readCache()?.dismissedVersion,
+      });
+
+      if (!isNewer(latest, VERSION)) {
+        ctx.ui.notify(`Already on latest version (${VERSION}).`, "info");
+        return;
+      }
+
+      await showUpdatePrompt(ctx, latest);
+    },
+  });
+}

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "pi-updater",
+  "version": "0.2.0",
+  "description": "A /update command for pi — check for new versions, install updates with npm, and resume with pi -c after restart",
+  "type": "module",
+  "license": "MIT",
+  "author": "Toms clanker",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/tonze/pi-updater.git"
+  },
+  "keywords": [
+    "pi-package",
+    "update"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ],
+    "image": "https://github.com/user-attachments/assets/89df2dad-8d91-464b-b3cb-dfd15bce1c06"
+  },
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@mariozechner/pi-coding-agent": "^0.55.1",
+    "@types/node": "^25.3.2",
+    "typescript": "^5.9.3"
+  }
+}