newpr 0.5.5 → 0.5.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "newpr",
-  "version": "0.5.5",
+  "version": "0.5.6",
   "description": "AI-powered large PR review tool - understand PRs with 1000+ lines of changes",
   "module": "src/cli/index.ts",
   "type": "module",

package/src/web/client/components/AppShell.tsx

@@ -256,43 +256,43 @@ export function AppShell({
 					</button>
 				</div>
 
-				{update.needsUpdate && (
+				{(update.needsUpdate || update.restarting) && (
 					<div className="shrink-0 mx-2 mt-2 rounded-lg border bg-blue-500/5 px-3 py-2.5">
-						<div className="flex items-center gap-2 mb-2">
-							<Download className="h-3 w-3 text-blue-500 shrink-0" />
-							<span className="text-[11px] font-medium text-blue-600 dark:text-blue-400">
-								v{update.latest} available
-							</span>
-						</div>
-						<button
-							type="button"
-							disabled={update.updating}
-							onClick={update.doUpdate}
-							className="w-full flex items-center justify-center gap-1.5 rounded-md bg-blue-500 hover:bg-blue-600 text-white text-[11px] font-medium py-1.5 transition-colors disabled:opacity-50"
-						>
-							{update.updating ? (
-								<><Loader2 className="h-3 w-3 animate-spin" /> Updating...</>
-							) : (
-								<><Download className="h-3 w-3" /> Update now</>
-							)}
-						</button>
-						{update.error && (
-							<p className="text-[10px] text-red-500 mt-1.5">{update.error}</p>
+						{update.restarting ? (
+							<div className="flex items-center gap-2">
+								<Loader2 className="h-3 w-3 animate-spin text-blue-500 shrink-0" />
+								<span className="text-[11px] text-blue-600 dark:text-blue-400">
+									Restarting...
+								</span>
+							</div>
+						) : (
+							<>
+								<div className="flex items-center gap-2 mb-2">
+									<Download className="h-3 w-3 text-blue-500 shrink-0" />
+									<span className="text-[11px] font-medium text-blue-600 dark:text-blue-400">
+										v{update.latest} available
+									</span>
+								</div>
+								<button
+									type="button"
+									disabled={update.updating}
+									onClick={update.doUpdate}
+									className="w-full flex items-center justify-center gap-1.5 rounded-md bg-blue-500 hover:bg-blue-600 text-white text-[11px] font-medium py-1.5 transition-colors disabled:opacity-50"
+								>
+									{update.updating ? (
+										<><Loader2 className="h-3 w-3 animate-spin" /> Updating...</>
+									) : (
+										<><Download className="h-3 w-3" /> Update &amp; restart</>
+									)}
+								</button>
+								{update.error && (
+									<p className="text-[10px] text-red-500 mt-1.5">{update.error}</p>
+								)}
+							</>
 						)}
 					</div>
 				)}
 
-				{update.updated && (
-					<div className="shrink-0 mx-2 mt-2 rounded-lg border border-green-500/20 bg-green-500/5 px-3 py-2.5">
-						<div className="flex items-center gap-2">
-							<Check className="h-3 w-3 text-green-500 shrink-0" />
-							<span className="text-[11px] text-green-600 dark:text-green-400">
-								Updated! Restart newpr to apply.
-							</span>
-						</div>
-					</div>
-				)}
-
 				<SessionList
 					sessions={sessions}
 					activeSessionId={activeSessionId}

package/src/web/client/hooks/useUpdateCheck.ts

@@ -6,10 +6,22 @@ interface UpdateState {
 	current: string;
 	latest: string;
 	updating: boolean;
-	updated: boolean;
+	restarting: boolean;
 	error: string | null;
 }
 
+async function waitForServer(maxWait = 30000): Promise<boolean> {
+	const start = Date.now();
+	while (Date.now() - start < maxWait) {
+		await new Promise((r) => setTimeout(r, 1000));
+		try {
+			const res = await fetch("/api/features", { signal: AbortSignal.timeout(2000) });
+			if (res.ok) return true;
+		} catch {}
+	}
+	return false;
+}
+
 export function useUpdateCheck() {
 	const [state, setState] = useState<UpdateState>({
 		checking: true,
@@ -17,35 +29,49 @@ export function useUpdateCheck() {
 		current: "",
 		latest: "",
 		updating: false,
-		updated: false,
+		restarting: false,
 		error: null,
 	});
 
 	useEffect(() => {
-		fetch("/api/update-check")
-			.then((r) => r.json())
-			.then((data) => {
-				const d = data as { current: string; latest: string; needsUpdate: boolean };
-				setState((s) => ({
-					...s,
-					checking: false,
-					current: d.current,
-					latest: d.latest,
-					needsUpdate: d.needsUpdate,
-				}));
-			})
-			.catch(() => setState((s) => ({ ...s, checking: false })));
+		const check = () => {
+			fetch("/api/update-check")
+				.then((r) => r.json())
+				.then((data) => {
+					const d = data as { current: string; latest: string; needsUpdate: boolean };
+					setState((s) => ({
+						...s,
+						checking: false,
+						current: d.current,
+						latest: d.latest,
+						needsUpdate: s.restarting ? s.needsUpdate : d.needsUpdate,
+					}));
+				})
+				.catch(() => setState((s) => ({ ...s, checking: false })));
+		};
+		check();
+		const interval = setInterval(check, 60 * 60 * 1000);
+		return () => clearInterval(interval);
 	}, []);
 
 	const doUpdate = useCallback(async () => {
 		setState((s) => ({ ...s, updating: true, error: null }));
 		try {
 			const res = await fetch("/api/update", { method: "POST" });
-			const data = await res.json() as { ok: boolean; error?: string };
-			if (data.ok) {
-				setState((s) => ({ ...s, updating: false, updated: true, needsUpdate: false }));
-			} else {
+			const data = await res.json() as { ok: boolean; restarting?: boolean; error?: string };
+			if (!data.ok) {
 				setState((s) => ({ ...s, updating: false, error: data.error ?? "Update failed" }));
+				return;
+			}
+
+			if (data.restarting) {
+				setState((s) => ({ ...s, updating: false, restarting: true }));
+				const up = await waitForServer();
+				if (up) {
+					window.location.reload();
+				} else {
+					setState((s) => ({ ...s, restarting: false, error: "Server did not come back up. Try restarting manually." }));
+				}
 			}
 		} catch (err) {
 			setState((s) => ({ ...s, updating: false, error: err instanceof Error ? err.message : String(err) }));

package/src/web/server/routes.ts

@@ -675,7 +675,18 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 				if (exitCode !== 0) {
 					return json({ ok: false, error: stderr.trim() || stdout.trim() }, 500);
 				}
-				return json({ ok: true, message: "Updated. Restart newpr to use the new version." });
+
+				setTimeout(() => {
+					Bun.spawn(["newpr", ...process.argv.slice(2)], {
+						cwd: process.cwd(),
+						stdin: "inherit",
+						stdout: "inherit",
+						stderr: "inherit",
+					});
+					setTimeout(() => process.exit(0), 500);
+				}, 1000);
+
+				return json({ ok: true, restarting: true });
 			} catch (err) {
 				return json({ ok: false, error: err instanceof Error ? err.message : String(err) }, 500);
 			}