feat: enhance update handling and success message

Author: qorzj

src/cli.tsx

@@ -78,6 +78,9 @@ void main();
 
 async function main(): Promise<void> {
   const updatePromptResult = await promptForPendingUpdate(packageInfo);
+  if (updatePromptResult.installed) {
+    process.exit(0);
+  }
 
   const restartRef: { current: (() => void) | null } = { current: null };
 
@@ -110,9 +113,7 @@ async function main(): Promise<void> {
     });
   }
 
-  if (!updatePromptResult.installed) {
-    void checkForNpmUpdate(packageInfo);
-  }
+  void checkForNpmUpdate(packageInfo);
 
   startApp();
 }

src/common/update-check.ts

@@ -4,7 +4,6 @@ import * as fs from "fs";
 import * as os from "os";
 import * as path from "path";
 import { render, type Instance } from "ink";
-import chalk from "chalk";
 import { UpdatePrompt, type UpdatePromptChoice } from "../ui";
 import { killProcessTree } from "./process-tree";
 
@@ -27,6 +26,7 @@ const UPDATE_STATE_FILE = "update-check.json";
 const NPM_VIEW_TIMEOUT_MS = 5000;
 const MAX_NPM_VIEW_OUTPUT_CHARS = 64 * 1024;
 const TENCENT_MIRROR_REGISTRY = "https://mirrors.cloud.tencent.com/npm/";
+export const UPDATE_SUCCESS_MESSAGE = "🎉 Update ran successfully! Please restart Deep Code.";
 
 export async function promptForPendingUpdate(packageInfo: PackageInfo): Promise<{ installed: boolean }> {
   const state = readUpdateState();
@@ -57,9 +57,7 @@ export async function promptForPendingUpdate(packageInfo: PackageInfo): Promise<
     const ok = await runNpmInstallGlobal(installSpec);
     if (ok) {
       writeUpdateState({ ...state, pending: null });
-      process.stdout.write(
-        `\n${chalk.red("Deep Code has been updated. Please restart the CLI to use the new version.")}\n\n`
-      );
+      process.stdout.write(`${UPDATE_SUCCESS_MESSAGE}\n\n`);
     }
     return { installed: ok };
   }

src/tests/update-check.test.ts

@@ -1,6 +1,6 @@
 import { test } from "node:test";
 import assert from "node:assert/strict";
-import { compareVersions, parseNpmViewVersion } from "../common/update-check";
+import { UPDATE_SUCCESS_MESSAGE, compareVersions, parseNpmViewVersion } from "../common/update-check";
 
 test("compareVersions orders semantic versions", () => {
   assert.equal(compareVersions("0.1.4", "0.1.3"), 1);
@@ -14,3 +14,7 @@ test("parseNpmViewVersion parses npm view JSON and plain output", () => {
   assert.equal(parseNpmViewVersion("0.1.5\n"), "0.1.5");
   assert.equal(parseNpmViewVersion("\n"), null);
 });
+
+test("UPDATE_SUCCESS_MESSAGE tells the user to restart Deep Code", () => {
+  assert.equal(UPDATE_SUCCESS_MESSAGE, "🎉 Update ran successfully! Please restart Deep Code.");
+});