pi-web-ui 0.8.3 → 0.8.5

package/bin/pi-web-ui.mjs

@@ -2,8 +2,9 @@
 /**
  * pi-web-ui CLI.
  *
- *   pi-web-ui                              启动生产服务器（前台，Ctrl+C 停止）
+ *   pi-web-ui                              启动生产服务器（前台，Ctrl+C 停止，自动打开浏览器）
  *   pi-web-ui --port 9000 --cwd /path      同上，覆盖端口 / 工作目录 / 数据目录
+ *   pi-web-ui --no-browser                 启动但不自动打开浏览器
  *   pi-web-ui --version | --help
  *   pi-web-ui server install [选项]         安装系统服务（开机自启）并启动
  *   pi-web-ui server uninstall [选项]       卸载系统服务
@@ -21,6 +22,7 @@
  * PI_CODING_AGENT_DIR。
  */
 import { spawnSync } from "node:child_process";
+import { get as httpGet } from "node:http";
 import {
 	existsSync,
 	mkdirSync,
@@ -47,8 +49,9 @@ try {
 const HELP = `pi-web-ui v${pkg.version} — web chat for the pi coding agent
 
 用法:
-  pi-web-ui                               启动服务器（前台，Ctrl+C 停止）
+  pi-web-ui                               启动服务器（前台，Ctrl+C 停止，自动打开浏览器）
   pi-web-ui --port 9000 --cwd /path       启动并指定端口 / 工作目录 / 数据目录
+  pi-web-ui --no-browser                  启动但不自动打开浏览器
   pi-web-ui server install [选项]         安装系统服务（开机自启）并启动
   pi-web-ui server uninstall [选项]       卸载系统服务
   pi-web-ui server start|stop|restart|status [选项]
@@ -109,6 +112,7 @@ function parseFlags(argv) {
 		dataDir: undefined,
 		name: undefined,
 		print: false,
+		noBrowser: false,
 		help: false,
 	};
 	const positionals = [];
@@ -141,6 +145,9 @@ function parseFlags(argv) {
 			case "--print":
 				opts.print = true;
 				break;
+			case "--no-browser":
+				opts.noBrowser = true;
+				break;
 			case "--help":
 			case "-h":
 				opts.help = true;
@@ -157,11 +164,52 @@ function parseFlags(argv) {
 // 前台启动
 // ---------------------------------------------------------------------------
 
+/** Open a URL in the OS default browser; failures are ignored (best-effort). */
+function openBrowser(url) {
+	if (isMac) {
+		spawnSync("open", [url], { stdio: "ignore" });
+	} else if (isWin) {
+		spawnSync("cmd", ["/c", "start", "", url], { stdio: "ignore" });
+	} else {
+		spawnSync("xdg-open", [url], { stdio: "ignore" });
+	}
+}
+
+/**
+ * Poll `url` until the server answers (or ~15s pass), then open it in the
+ * default browser. The foreground server runs in this process, so the first
+ * HTTP response is the "listening" signal; if the server crashes meanwhile
+ * (e.g. port already taken), the pending timers die with the process and
+ * nothing is opened.
+ */
+function openBrowserWhenUp(url) {
+	const deadline = Date.now() + 15_000;
+	const attempt = () => {
+		const req = httpGet(url, (res) => {
+			res.resume();
+			console.log(`  🌐 已自动打开浏览器：${url}（--no-browser 可关闭）`);
+			openBrowser(url);
+		});
+		req.setTimeout(1000, () => {
+			req.destroy();
+			if (Date.now() < deadline) setTimeout(attempt, 150);
+		});
+		req.on("error", () => {
+			if (Date.now() < deadline) setTimeout(attempt, 150);
+		});
+	};
+	attempt();
+}
+
 async function startForeground(opts) {
 	if (opts.port) process.env.PORT = opts.port;
 	if (opts.cwd) process.env.PI_WEB_CWD = resolve(opts.cwd);
 	if (opts.dataDir) process.env.PI_WEB_DATA_DIR = resolve(opts.dataDir);
+	const url = `http://localhost:${String(
+		opts.port ?? process.env.PORT ?? "8787",
+	)}`;
 	await import(pathToFileURL(SERVER_ENTRY).href);
+	if (!opts.noBrowser) openBrowserWhenUp(url);
 }
 
 // ---------------------------------------------------------------------------
@@ -394,7 +442,7 @@ User=${process.env.SUDO_USER ?? userInfo().username}
 WorkingDirectory=${cwd}
 ${envLines}
 ExecStart=${JSON.stringify(NODE)} ${JSON.stringify(SERVER_ENTRY)}
-Restart=on-failure
+Restart=always
 RestartSec=5
 
 [Install]

package/deploy/com.xingshuyin.pi-web-ui.plist

@@ -1,48 +1,48 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
-<!--
-  pi-web-ui launchd agent — macOS auto-start at login.
-
-  Install:
-    npm i -g pi-web-ui
-    cp deploy/com.xingshuyin.pi-web-ui.plist ~/Library/LaunchAgents/
-    # edit ProgramArguments / WorkingDirectory / PI_WEB_CWD for your setup
-    launchctl load ~/Library/LaunchAgents/com.xingshuyin.pi-web-ui.plist
-    # (or: launchctl bootstrap gui/$(id -u) ~/Library/LaunchAgents/com.xingshuyin.pi-web-ui.plist)
-
-  Find the pi-web-ui binary path with: which pi-web-ui
--->
-<plist version="1.0">
-<dict>
-  <key>Label</key>
-  <string>com.xingshuyin.pi-web-ui</string>
-
-  <key>ProgramArguments</key>
-  <array>
-    <string>/usr/local/bin/pi-web-ui</string>
-  </array>
-
-  <key>RunAtLoad</key>
-  <true/>
-
-  <!-- Restart if it crashes -->
-  <key>KeepAlive</key>
-  <true/>
-
-  <key>WorkingDirectory</key>
-  <string>/Users/YOUR_USER</string>
-
-  <key>EnvironmentVariables</key>
-  <dict>
-    <key>PORT</key>
-    <string>8787</string>
-    <key>PI_WEB_CWD</key>
-    <string>/Users/YOUR_USER</string>
-  </dict>
-
-  <key>StandardOutPath</key>
-  <string>/tmp/pi-web-ui.log</string>
-  <key>StandardErrorPath</key>
-  <string>/tmp/pi-web-ui.err</string>
-</dict>
-</plist>
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<!--
+  pi-web-ui launchd agent — macOS auto-start at login.
+
+  Install:
+    npm i -g pi-web-ui
+    cp deploy/com.xingshuyin.pi-web-ui.plist ~/Library/LaunchAgents/
+    # edit ProgramArguments / WorkingDirectory / PI_WEB_CWD for your setup
+    launchctl load ~/Library/LaunchAgents/com.xingshuyin.pi-web-ui.plist
+    # (or: launchctl bootstrap gui/$(id -u) ~/Library/LaunchAgents/com.xingshuyin.pi-web-ui.plist)
+
+  Find the pi-web-ui binary path with: which pi-web-ui
+-->
+<plist version="1.0">
+<dict>
+  <key>Label</key>
+  <string>com.xingshuyin.pi-web-ui</string>
+
+  <key>ProgramArguments</key>
+  <array>
+    <string>/usr/local/bin/pi-web-ui</string>
+  </array>
+
+  <key>RunAtLoad</key>
+  <true/>
+
+  <!-- Restart if it crashes -->
+  <key>KeepAlive</key>
+  <true/>
+
+  <key>WorkingDirectory</key>
+  <string>/Users/YOUR_USER</string>
+
+  <key>EnvironmentVariables</key>
+  <dict>
+    <key>PORT</key>
+    <string>8787</string>
+    <key>PI_WEB_CWD</key>
+    <string>/Users/YOUR_USER</string>
+  </dict>
+
+  <key>StandardOutPath</key>
+  <string>/tmp/pi-web-ui.log</string>
+  <key>StandardErrorPath</key>
+  <string>/tmp/pi-web-ui.err</string>
+</dict>
+</plist>

package/deploy/pi-web-ui-task.xml

@@ -1,71 +1,71 @@
-<?xml version="1.0" encoding="UTF-16"?>
-<!--
-  pi-web-ui Task Scheduler task — Windows auto-start at logon.
-
-  The easy way (no admin needed, generates everything for you):
-    npm i -g pi-web-ui
-    pi-web-ui server install --port 8787 --cwd C:\path\to\project
-    pi-web-ui server status | restart | stop | uninstall
-
-  Manual install with this template (edit the paths below first):
-    schtasks /Create /TN "pi-web-ui" /XML pi-web-ui-task.xml /F
-    schtasks /Run /TN "pi-web-ui"
-
-  Notes:
-    - The task runs the PowerShell launcher the CLI generates at
-      %APPDATA%\pi-web-ui\pi-web-ui.ps1 with -WindowStyle Hidden, so the
-      server runs with no black console window (nothing to accidentally
-      close/kill). The ps1 sets PORT/PI_WEB_CWD, cd's to the workspace,
-      launches node, and appends output to %USERPROFILE%\pi-web-ui.log.
-      Preview both generated files with: pi-web-ui server install --print
-    - Save this file as UTF-16 LE (schtasks requires it; the CLI does this
-      automatically when it writes the task XML).
-    - LogonTrigger = starts when you log in, same as a launchd user agent.
-      For boot-start without login, use Docker instead (see README).
--->
-<Task version="1.2" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
-  <RegistrationInfo>
-    <Description>pi-web-ui — web chat for the pi coding agent (auto-start at logon)</Description>
-  </RegistrationInfo>
-  <Triggers>
-    <LogonTrigger>
-      <Enabled>true</Enabled>
-    </LogonTrigger>
-  </Triggers>
-  <Principals>
-    <Principal id="Author">
-      <LogonType>InteractiveToken</LogonType>
-      <RunLevel>LeastPrivilege</RunLevel>
-    </Principal>
-  </Principals>
-  <Settings>
-    <MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
-    <DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
-    <StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
-    <AllowHardTerminate>true</AllowHardTerminate>
-    <StartWhenAvailable>false</StartWhenAvailable>
-    <RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>
-    <IdleSettings>
-      <StopOnIdleEnd>false</StopOnIdleEnd>
-      <RestartOnIdle>false</RestartOnIdle>
-    </IdleSettings>
-    <AllowStartOnDemand>true</AllowStartOnDemand>
-    <Enabled>true</Enabled>
-    <Hidden>false</Hidden>
-    <RunOnlyIfIdle>false</RunOnlyIfIdle>
-    <WakeToRun>false</WakeToRun>
-    <ExecutionTimeLimit>PT0S</ExecutionTimeLimit>
-    <Priority>7</Priority>
-    <RestartOnFailure>
-      <Interval>PT1M</Interval>
-      <Count>3</Count>
-    </RestartOnFailure>
-  </Settings>
-  <Actions Context="Author">
-    <Exec>
-      <Command>C:\Windows\System32\WindowsPowerShell\v1.0\powershell.exe</Command>
-      <Arguments>-NoProfile -NonInteractive -ExecutionPolicy Bypass -WindowStyle Hidden -File "%APPDATA%\pi-web-ui\pi-web-ui.ps1"</Arguments>
-      <WorkingDirectory>%USERPROFILE%</WorkingDirectory>
-    </Exec>
-  </Actions>
-</Task>
+<?xml version="1.0" encoding="UTF-16"?>
+<!--
+  pi-web-ui Task Scheduler task — Windows auto-start at logon.
+
+  The easy way (no admin needed, generates everything for you):
+    npm i -g pi-web-ui
+    pi-web-ui server install --port 8787 --cwd C:\path\to\project
+    pi-web-ui server status | restart | stop | uninstall
+
+  Manual install with this template (edit the paths below first):
+    schtasks /Create /TN "pi-web-ui" /XML pi-web-ui-task.xml /F
+    schtasks /Run /TN "pi-web-ui"
+
+  Notes:
+    - The task runs the PowerShell launcher the CLI generates at
+      %APPDATA%\pi-web-ui\pi-web-ui.ps1 with -WindowStyle Hidden, so the
+      server runs with no black console window (nothing to accidentally
+      close/kill). The ps1 sets PORT/PI_WEB_CWD, cd's to the workspace,
+      launches node, and appends output to %USERPROFILE%\pi-web-ui.log.
+      Preview both generated files with: pi-web-ui server install --print
+    - Save this file as UTF-16 LE (schtasks requires it; the CLI does this
+      automatically when it writes the task XML).
+    - LogonTrigger = starts when you log in, same as a launchd user agent.
+      For boot-start without login, use Docker instead (see README).
+-->
+<Task version="1.2" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
+  <RegistrationInfo>
+    <Description>pi-web-ui — web chat for the pi coding agent (auto-start at logon)</Description>
+  </RegistrationInfo>
+  <Triggers>
+    <LogonTrigger>
+      <Enabled>true</Enabled>
+    </LogonTrigger>
+  </Triggers>
+  <Principals>
+    <Principal id="Author">
+      <LogonType>InteractiveToken</LogonType>
+      <RunLevel>LeastPrivilege</RunLevel>
+    </Principal>
+  </Principals>
+  <Settings>
+    <MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
+    <DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
+    <StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
+    <AllowHardTerminate>true</AllowHardTerminate>
+    <StartWhenAvailable>false</StartWhenAvailable>
+    <RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>
+    <IdleSettings>
+      <StopOnIdleEnd>false</StopOnIdleEnd>
+      <RestartOnIdle>false</RestartOnIdle>
+    </IdleSettings>
+    <AllowStartOnDemand>true</AllowStartOnDemand>
+    <Enabled>true</Enabled>
+    <Hidden>false</Hidden>
+    <RunOnlyIfIdle>false</RunOnlyIfIdle>
+    <WakeToRun>false</WakeToRun>
+    <ExecutionTimeLimit>PT0S</ExecutionTimeLimit>
+    <Priority>7</Priority>
+    <RestartOnFailure>
+      <Interval>PT1M</Interval>
+      <Count>3</Count>
+    </RestartOnFailure>
+  </Settings>
+  <Actions Context="Author">
+    <Exec>
+      <Command>C:\Windows\System32\WindowsPowerShell\v1.0\powershell.exe</Command>
+      <Arguments>-NoProfile -NonInteractive -ExecutionPolicy Bypass -WindowStyle Hidden -File "%APPDATA%\pi-web-ui\pi-web-ui.ps1"</Arguments>
+      <WorkingDirectory>%USERPROFILE%</WorkingDirectory>
+    </Exec>
+  </Actions>
+</Task>

package/deploy/pi-web-ui.service

@@ -1,31 +1,31 @@
-# pi-web-ui systemd unit — Linux auto-start on boot.
-#
-# Install (adjust User/WorkingDirectory/Environment to taste):
-#   sudo npm i -g pi-web-ui
-#   sudo cp deploy/pi-web-ui.service /etc/systemd/system/
-#   sudo systemctl daemon-reload
-#   sudo systemctl enable --now pi-web-ui
-#   sudo systemctl status pi-web-ui
-#
-# Logs: journalctl -u pi-web-ui -f
-
-[Unit]
-Description=pi-web-ui — web chat for the pi coding agent
-After=network.target
-
-[Service]
-Type=simple
-# Run as an unprivileged user (change to your user, e.g. yourname)
-User=YOUR_USER
-# The workspace the agent operates in (read/edit/bash/write)
-WorkingDirectory=/home/YOUR_USER
-Environment=PORT=8787
-# Point at your pi config dir if it's not the default ~/.pi/agent
-#Environment=PI_CODING_AGENT_DIR=/home/YOUR_USER/.pi/agent
-ExecStart=/usr/bin/pi-web-ui
-Restart=on-failure
-RestartSec=5
-# npm global bin may live elsewhere (nvm, etc.) — find with: which pi-web-ui
-
-[Install]
-WantedBy=multi-user.target
+# pi-web-ui systemd unit — Linux auto-start on boot.
+#
+# Install (adjust User/WorkingDirectory/Environment to taste):
+#   sudo npm i -g pi-web-ui
+#   sudo cp deploy/pi-web-ui.service /etc/systemd/system/
+#   sudo systemctl daemon-reload
+#   sudo systemctl enable --now pi-web-ui
+#   sudo systemctl status pi-web-ui
+#
+# Logs: journalctl -u pi-web-ui -f
+
+[Unit]
+Description=pi-web-ui — web chat for the pi coding agent
+After=network.target
+
+[Service]
+Type=simple
+# Run as an unprivileged user (change to your user, e.g. yourname)
+User=YOUR_USER
+# The workspace the agent operates in (read/edit/bash/write)
+WorkingDirectory=/home/YOUR_USER
+Environment=PORT=8787
+# Point at your pi config dir if it's not the default ~/.pi/agent
+#Environment=PI_CODING_AGENT_DIR=/home/YOUR_USER/.pi/agent
+ExecStart=/usr/bin/pi-web-ui
+Restart=on-failure
+RestartSec=5
+# npm global bin may live elsewhere (nvm, etc.) — find with: which pi-web-ui
+
+[Install]
+WantedBy=multi-user.target

package/dist/server/agent-service.js

@@ -946,6 +946,11 @@ export class ClientSession {
     }
     /** True once updateApp succeeded — the process must restart to run new code. */
     pendingRestart = false;
+    /**
+     * Set by index.ts: called after a successful self-update; returns whether
+     * the process is going to restart itself (so the notice can say so).
+     */
+    onUpdateReady = undefined;
     /** Ask the npm registry for the latest pi-web-ui version and report it. */
     async checkUpdate() {
         const current = ClientSession.currentAppVersion();
@@ -993,10 +998,13 @@ export class ClientSession {
                     ok: true,
                     detail: out.slice(0, 400),
                 });
+                const autoRestart = this.onUpdateReady?.() ?? false;
                 this.emit({
                     type: "notice",
                     level: "info",
-                    text: "✅ 已更新 pi-web-ui，重启服务后生效（pi-web-ui server restart）",
+                    text: autoRestart
+                        ? "✅ 已更新 pi-web-ui，正在自动重启…"
+                        : "✅ 已更新 pi-web-ui，重启服务后生效（pi-web-ui server restart）",
                 });
             }
             else {
@@ -2350,6 +2358,11 @@ export class AgentService {
     clients = new Map();
     pending = new Map();
     stateStore;
+    /**
+     * Set by index.ts: called by a client session after a successful
+     * self-update; returns whether the process will restart itself.
+     */
+    onUpdateReady = undefined;
     constructor(cwd, sessionDirRoot, stateFile) {
         this.cwd = cwd;
         this.sessionDirRoot = sessionDirRoot;
@@ -2395,6 +2408,8 @@ export class AgentService {
             }
         }
         cs.attachSink(send);
+        // Forward the update hook (set once by index.ts) to every session.
+        cs.onUpdateReady = this.onUpdateReady;
         return cs;
     }
     /** Remove a socket from a client's broadcast set (called on socket close). */

package/dist/server/index.js

@@ -15,6 +15,8 @@
 import { existsSync } from "node:fs";
 import { stat } from "node:fs/promises";
 import { createServer } from "node:http";
+import { createConnection } from "node:net";
+import { spawn } from "node:child_process";
 import { basename, dirname, join, resolve } from "node:path";
 import { fileURLToPath } from "node:url";
 import { randomUUID } from "node:crypto";
@@ -109,6 +111,51 @@ const heartbeatTimer = setInterval(() => {
 const service = new AgentService(CWD, SESSION_DIR_ROOT, 
 // Per-client persisted UI state: last-used workspace + recent projects.
 join(DATA_DIR, "client-state.json"));
+// ---------------------------------------------------------------------------
+// Self-update auto-restart
+// ---------------------------------------------------------------------------
+// npm i -g writes new code to disk but the running process keeps the old
+// code in memory — so a successful in-app update hands the process over:
+//   macOS launchd (KeepAlive) and systemd (Restart) relaunch us on exit;
+//   foreground runs get a replacement child that waits for our port to free.
+// Docker containers can't self-restart (the orchestrator owns that), so they
+// keep the manual-restart notice.
+const RESTART_CHILD_ENV = "PI_WEB_RESTART_CHILD";
+function scheduleUpdateRestart() {
+    const isLaunchd = process.platform === "darwin" && process.ppid === 1;
+    const isSystemd = process.platform === "linux" && !!process.env.INVOCATION_ID;
+    const inDocker = existsSync("/.dockerenv");
+    if (isLaunchd || isSystemd || inDocker) {
+        // Supervisors relaunch on exit; Docker restarts externally. Nothing to
+        // spawn — just exit after the notice has flushed.
+        if (isLaunchd || isSystemd) {
+            setTimeout(() => {
+                console.log("update applied — auto-restarting…");
+                if (isSystemd) {
+                    // Non-zero exit: legacy units use Restart=on-failure.
+                    process.exit(3);
+                }
+                void shutdown();
+            }, 1500);
+            return true;
+        }
+        return false;
+    }
+    // Foreground / Windows: spawn a replacement from the updated install and
+    // exit. Same stdio (logs keep flowing), same args/env (port, cwd, data
+    // dir…); the child waits for this port to free before binding.
+    setTimeout(() => {
+        console.log("update applied — spawning replacement…");
+        spawn(process.execPath, process.argv.slice(1), {
+            stdio: "inherit",
+            env: { ...process.env, [RESTART_CHILD_ENV]: "1" },
+            ...(process.platform === "win32" ? { windowsHide: true } : {}),
+        });
+        void shutdown();
+    }, 1500);
+    return true;
+}
+service.onUpdateReady = scheduleUpdateRestart;
 wss.on("connection", (ws) => {
     let clientId = null;
     let closed = false;
@@ -279,6 +326,29 @@ wss.on("connection", (ws) => {
             service.detach(clientId, send);
     });
 });
+// When spawned by the old process as an auto-restart replacement, wait for
+// the old instance to release the port before binding (it exits right after
+// spawning us). Probe by attempting a connection: refused = free.
+if (process.env[RESTART_CHILD_ENV] === "1") {
+    const deadline = Date.now() + 20_000;
+    const portFree = () => new Promise((resolve) => {
+        const sock = createConnection({ port: PORT, host: "127.0.0.1" });
+        sock.once("connect", () => {
+            sock.destroy();
+            resolve(false); // busy — old instance still up
+        });
+        sock.once("error", () => resolve(true)); // refused → free
+        sock.setTimeout(500, () => {
+            sock.destroy();
+            resolve(false);
+        });
+    });
+    while (Date.now() < deadline) {
+        if (await portFree())
+            break;
+        await new Promise((r) => setTimeout(r, 300));
+    }
+}
 httpServer.listen(PORT, () => {
     console.log("");
     console.log("  ⚡ pi-web-ui — web chat for the pi coding agent");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-web-ui",
-  "version": "0.8.3",
+  "version": "0.8.5",
   "description": "Web chat interface for the pi coding agent, powered by the pi SDK (@earendil-works/pi-coding-agent) — one-command run, Docker/systemd/launchd deployable",
   "license": "MIT",
   "type": "module",