pi-updater 0.3.0 → 0.3.2

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # Changelog
 
+## 0.3.2 - 2026-05-02
+
+- Use pi's native `pi update --self` installer on pi 0.70.3+ and keep npm install as the fallback for older pi versions.
+- Use pi's `https://pi.dev/api/latest-version` update endpoint with a `pi/<version>` user agent.
+- Keep pi-updater's interactive startup prompt on pi 0.70.3+ while avoiding pi's duplicate built-in version notice.
+
+## 0.3.1 - 2026-04-04
+
+- Compatibility with pi 0.65+: use `session_start` instead of legacy `session_switch` for automatic checks. See [pi-mono v0.65.0](https://github.com/badlogic/pi-mono/releases/tag/v0.65.0).
+- Store cache and dismissed-version state in pi's configured agent directory.
+- Preserve `--no-session` mode when restarting after an update and show the correct manual restart hint.
+
 ## 0.3.0 - 2026-03-23
 
 - Auto-restart pi after a successful update. Asks to restart, then seamlessly relaunches on the current session.

package/README.md

@@ -5,29 +5,29 @@ A lightweight, Codex-style auto-updater for pi with fast, cache-first startup ch
 - npm: https://www.npmjs.com/package/pi-updater
 - repo: https://github.com/tonze/pi-updater
 
-> **Note:** Automatic installation currently supports npm-based pi installs only.
+> **Note:** On pi 0.70.3+, pi-updater delegates installation to pi's native `pi update --self` command. Older pi versions fall back to npm-based installation.
 
 <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:
+**On pi 0.70.3+:** pi-updater is native-aware. It keeps the interactive startup update prompt, checks pi's update service, installs with `pi update --self`, then offers to restart the current session. Pi's built-in updater only shows a notice with a command to run; pi-updater provides the clickable update/restart flow.
+
+**On older pi versions:** if a newer version is available, pi-updater shows a startup prompt:
 - **Update now** — install with npm, then auto-restart pi on the current session
 - **Skip** — dismiss until next session
 - **Skip this version** — don't ask again until a newer version appears
 
-After a successful update, pi-updater asks whether to restart immediately. If confirmed, pi relaunches seamlessly on the current session. In non-interactive modes or if auto-restart fails, it falls back to a manual restart message.
-
-**In the background (once per run):** performs one live npm check and can show the prompt in the same session when a new release is detected.
+After a successful update, pi-updater asks whether to restart immediately. If confirmed, pi relaunches seamlessly on the current session. In non-interactive modes or if auto-restart fails, it falls back to a manual restart message. Ephemeral `--no-session` runs stay ephemeral on restart.
 
-**`/update`:** manually check for updates (always fetches fresh from npm, unless `PI_OFFLINE` is set).
+**`/update`:** manually check for updates (always fetches fresh from pi's update service, unless `PI_OFFLINE` is set). On pi 0.70.3+ it installs with `pi update --self`; on older pi it falls back to npm.
 
 ## How version checks work
 
 pi-updater uses a cache-first approach to keep startup fast:
 
 1. On startup, cached version data is checked instantly.
-2. One background live fetch refreshes the cache.
+2. One background live fetch refreshes the cache from pi's update service.
 3. If the background fetch finds a newer version, pi-updater can prompt in the same session.
 4. Automatic checks are skipped when `PI_SKIP_VERSION_CHECK` or `PI_OFFLINE` is set.
 
@@ -47,6 +47,8 @@ pi install git:github.com/tonze/pi-updater
 
 Use `/update` inside pi to manually check for updates and install them.
 
+Cache and dismissed-version state are stored in pi's configured agent directory and respect `PI_CODING_AGENT_DIR`.
+
 ## Environment flags
 
 Disable automatic version checks:

package/index.ts

@@ -2,18 +2,19 @@ import type {
   ExtensionAPI,
   ExtensionContext,
 } from "@mariozechner/pi-coding-agent";
-import { VERSION, BorderedLoader } from "@mariozechner/pi-coding-agent";
+import { VERSION, BorderedLoader, getAgentDir } from "@mariozechner/pi-coding-agent";
 import { spawnSync } from "node:child_process";
 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");
+const LATEST_VERSION_URL = "https://pi.dev/api/latest-version";
+const NATIVE_SELF_UPDATE_MIN_VERSION = "0.70.3";
+const CACHE_FILE = join(getAgentDir(), "update-cache.json");
 
 const ENV_SKIP_VERSION_CHECK = "PI_SKIP_VERSION_CHECK";
 const ENV_OFFLINE = "PI_OFFLINE";
+const ENV_INTERNAL_SKIP = "PI_UPDATER_SUPPRESSED_NATIVE_VERSION_CHECK";
 
 interface VersionCache {
   latestVersion: string;
@@ -36,35 +37,75 @@ function writeCache(cache: VersionCache) {
   } 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];
+interface ParsedVersion {
+  major: number;
+  minor: number;
+  patch: number;
+  prerelease?: string;
+}
+
+function parseVersion(version: string): ParsedVersion | undefined {
+  const match = version
+    .trim()
+    .match(/^v?(\d+)\.(\d+)\.(\d+)(?:-([0-9A-Za-z.-]+))?(?:\+.*)?$/);
+  if (!match) return undefined;
+  return {
+    major: Number.parseInt(match[1], 10),
+    minor: Number.parseInt(match[2], 10),
+    patch: Number.parseInt(match[3], 10),
+    prerelease: match[4],
+  };
+}
+
+function compareVersions(leftVersion: string, rightVersion: string): number | undefined {
+  const left = parseVersion(leftVersion);
+  const right = parseVersion(rightVersion);
+  if (!left || !right) return undefined;
+  if (left.major !== right.major) return left.major - right.major;
+  if (left.minor !== right.minor) return left.minor - right.minor;
+  if (left.patch !== right.patch) return left.patch - right.patch;
+  if (left.prerelease === right.prerelease) return 0;
+  if (!left.prerelease) return 1;
+  if (!right.prerelease) return -1;
+  return left.prerelease.localeCompare(right.prerelease);
 }
 
 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];
+  const comparison = compareVersions(latest, current);
+  return comparison !== undefined && comparison > 0;
+}
+
+function isAtLeast(version: string, minimum: string): boolean {
+  const comparison = compareVersions(version, minimum);
+  return comparison !== undefined && comparison >= 0;
 }
 
 function isEnvSet(name: string): boolean {
   return Boolean(process.env[name]);
 }
 
+const userSkippedVersionCheck =
+  isEnvSet(ENV_SKIP_VERSION_CHECK) && !isEnvSet(ENV_INTERNAL_SKIP);
+
 function shouldSkipAutoChecks(): boolean {
-  return isEnvSet(ENV_SKIP_VERSION_CHECK) || isEnvSet(ENV_OFFLINE);
+  return userSkippedVersionCheck || isEnvSet(ENV_OFFLINE);
 }
 
 function isOffline(): boolean {
   return isEnvSet(ENV_OFFLINE);
 }
 
+function piUserAgent(): string {
+  const runtime = process.versions.bun
+    ? `bun/${process.versions.bun}`
+    : `node/${process.version}`;
+  return `pi/${VERSION} (${process.platform}; ${runtime}; ${process.arch})`;
+}
+
+function hasNativeSelfUpdate(): boolean {
+  return isAtLeast(VERSION, NATIVE_SELF_UPDATE_MIN_VERSION);
+}
+
 function saveLatestToCache(latest: string) {
   const prev = readCache();
   writeCache({
@@ -76,11 +117,18 @@ function saveLatestToCache(latest: string) {
 
 async function fetchLatestVersion(): Promise<string | undefined> {
   try {
-    const res = await fetch(REGISTRY_URL, {
+    const res = await fetch(LATEST_VERSION_URL, {
+      headers: {
+        "User-Agent": piUserAgent(),
+        accept: "application/json",
+      },
       signal: AbortSignal.timeout(10_000),
     });
     if (!res.ok) return undefined;
-    return ((await res.json()) as { version?: string }).version;
+    const version = ((await res.json()) as { version?: string }).version;
+    return typeof version === "string" && version.trim()
+      ? version.trim()
+      : undefined;
   } catch {
     return undefined;
   }
@@ -95,7 +143,7 @@ function getCachedUpgradeVersion(): string | undefined {
   return cache.latestVersion;
 }
 
-/** Fetch latest from npm and refresh cache. */
+/** Fetch latest from Pi's update endpoint and refresh cache. */
 async function refreshLatestVersionInCache(): Promise<string | undefined> {
   const latest = await fetchLatestVersion();
   if (!latest) return undefined;
@@ -112,18 +160,39 @@ function dismissVersion(version: string) {
   });
 }
 
-function getInstallCommand(version: string): { program: string; args: string[] } {
+interface InstallCommand {
+  program: string;
+  args: string[];
+  display: string;
+}
+
+function getInstallCommand(version: string): InstallCommand {
+  if (hasNativeSelfUpdate()) {
+    return {
+      program: "pi",
+      args: ["update", "--self"],
+      display: "pi update --self",
+    };
+  }
+
   return {
     program: "npm",
     args: ["install", "-g", `${PACKAGE_NAME}@${version}`],
+    display: `npm install -g ${PACKAGE_NAME}@${version}`,
   };
 }
 
-function fmtCmd(cmd: { program: string; args: string[] }): string {
-  return `${cmd.program} ${cmd.args.join(" ")}`;
+function fmtCmd(cmd: InstallCommand): string {
+  return cmd.display;
 }
 
 export default function (pi: ExtensionAPI) {
+  const suppressNativeCheck = hasNativeSelfUpdate() && !userSkippedVersionCheck;
+  if (suppressNativeCheck) {
+    process.env[ENV_SKIP_VERSION_CHECK] = "1";
+    process.env[ENV_INTERNAL_SKIP] = "1";
+  }
+
   let promptOpen = false;
   const promptedVersions = new Set<string>();
   let liveCheckStarted = false;
@@ -144,13 +213,18 @@ export default function (pi: ExtensionAPI) {
   async function restartPi(ctx: ExtensionContext): Promise<boolean> {
     const piBinary = await findPiBinary();
     const sessionFile = ctx.sessionManager.getSessionFile();
-    const restartArgs = sessionFile ? ["--session", sessionFile] : ["-c"];
+    const restartArgs = sessionFile ? ["--session", sessionFile] : ["--no-session"];
+    const env = { ...process.env };
+    if (suppressNativeCheck) {
+      delete env[ENV_SKIP_VERSION_CHECK];
+      delete env[ENV_INTERNAL_SKIP];
+    }
 
     return ctx.ui.custom<boolean>((tui, _theme, _kb, done) => {
       tui.stop();
       const result = spawnSync(piBinary, restartArgs, {
         cwd: ctx.cwd,
-        env: process.env,
+        env,
         stdio: "inherit",
         shell: process.platform === "win32",
         windowsHide: false,
@@ -165,17 +239,35 @@ export default function (pi: ExtensionAPI) {
   async function doInstall(
     ctx: ExtensionContext,
     latest: string,
-    cmd: { program: string; args: string[] },
+    cmd: InstallCommand,
   ) {
     const success = await ctx.ui.custom<boolean>((tui, theme, _kb, done) => {
-      const loader = new BorderedLoader(tui, theme, `Installing ${latest}...`);
+      const loader = new BorderedLoader(tui, theme, `Running ${cmd.display}...`);
       loader.onAbort = () => done(false);
 
-      pi.exec(cmd.program, cmd.args, { timeout: 120_000 })
+      const runUpdateCommand = async () => {
+        if (suppressNativeCheck && cmd.program === "pi") {
+          delete process.env[ENV_SKIP_VERSION_CHECK];
+          delete process.env[ENV_INTERNAL_SKIP];
+          try {
+            return await pi.exec(cmd.program, cmd.args, { timeout: 120_000 });
+          } finally {
+            process.env[ENV_SKIP_VERSION_CHECK] = "1";
+            process.env[ENV_INTERNAL_SKIP] = "1";
+          }
+        }
+        return pi.exec(cmd.program, cmd.args, { timeout: 120_000 });
+      };
+
+      runUpdateCommand()
         .then((result) => {
           if (result.code !== 0) {
+            const output = [result.stderr, result.stdout]
+              .filter(Boolean)
+              .join("\n")
+              .trim();
             ctx.ui.notify(
-              `Update failed (exit ${result.code}): ${result.stderr || result.stdout}`,
+              `Update failed (exit ${result.code})${output ? `: ${output}` : ""}`,
               "error",
             );
             done(false);
@@ -183,16 +275,26 @@ export default function (pi: ExtensionAPI) {
             done(true);
           }
         })
-        .catch(() => done(false));
+        .catch((error) => {
+          ctx.ui.notify(
+            `Update failed: ${error instanceof Error ? error.message : String(error)}`,
+            "error",
+          );
+          done(false);
+        });
 
       return loader;
     });
 
     if (!success) return;
 
+    const restartTip = ctx.sessionManager.getSessionFile()
+      ? "Tip: run `pi -c` to continue this session."
+      : "Tip: run `pi --no-session` to continue without a saved session.";
+
     if (!canAutoRestart(ctx)) {
       ctx.ui.notify(
-        `Updated to ${latest}! Please restart pi.\nTip: run \`pi -c\` to continue this session.`,
+        `Updated to ${latest}! Please restart pi.\n${restartTip}`,
         "info",
       );
       return;
@@ -212,7 +314,7 @@ export default function (pi: ExtensionAPI) {
     }
 
     ctx.ui.notify(
-      `Updated to ${latest}! Auto-restart failed. Please restart pi manually.\nTip: run \`pi -c\` to continue this session.`,
+      `Updated to ${latest}! Auto-restart failed. Please restart pi manually.\n${restartTip}`,
       "error",
     );
   }
@@ -272,16 +374,13 @@ export default function (pi: ExtensionAPI) {
       .catch(() => {});
   }
 
-  pi.on("session_start", async (_event, ctx) => {
-    runAutoChecks(ctx);
-  });
-
-  pi.on("session_switch", async (_event, ctx) => {
+  pi.on("session_start", async (event, ctx) => {
+    if (event.reason === "reload" || event.reason === "fork") return;
     runAutoChecks(ctx);
   });
 
   pi.registerCommand("update", {
-    description: "Check for pi updates and install",
+    description: "Check for pi updates and install via native updater when available",
     handler: async (rawArgs, ctx) => {
       // /update --test — simulate the full UI flow without a real install
       if (rawArgs?.trim() === "--test") {
@@ -295,7 +394,7 @@ export default function (pi: ExtensionAPI) {
         if (!choice || choice === "Skip" || choice === "Skip this version") return;
 
         await ctx.ui.custom<void>((tui, theme, _kb, done) => {
-          const loader = new BorderedLoader(tui, theme, `Installing ${fakeLatest}...`);
+          const loader = new BorderedLoader(tui, theme, `Running ${cmd.display}...`);
           loader.onAbort = () => done();
           setTimeout(() => done(), 1500);
           return loader;
@@ -339,7 +438,7 @@ export default function (pi: ExtensionAPI) {
       );
 
       if (!latest) {
-        ctx.ui.notify("Could not reach npm registry.", "error");
+        ctx.ui.notify("Could not reach Pi update service.", "error");
         return;
       }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-updater",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "Codex-style auto-updater for pi. Checks for new versions on startup and prompts to install.",
   "type": "module",
   "license": "MIT",
@@ -29,7 +29,7 @@
     "@mariozechner/pi-coding-agent": "*"
   },
   "devDependencies": {
-    "@mariozechner/pi-coding-agent": "^0.55.1",
+    "@mariozechner/pi-coding-agent": "^0.65.0",
     "@types/node": "^25.3.2",
     "typescript": "^5.9.3"
   }