Commit f4617e7b19830111d88b1d914b3ca4e1396912a3 | the-knife

author	Noah Gregory <noah@wts.dev> 2026-03-28 18:44:13-04:00
committer	GitHub <noreply@github.com> 2026-03-28 15:44:13-07:00
log	f4617e7b19830111d88b1d914b3ca4e1396912a3
tree	27f554f53d0e22d84ac46a56e36b698a1731145a
parent	c426452239e06417ec8e231b7883f9c2292df63c
signature	Signed by PGP key B5690EEEBB952194

feat: add mock update server (#1180)

Co-authored-by: Julius Marminge <julius0216@outlook.com>

7 files changed, 121 insertions(+), 6 deletions(-)

.gitignore
apps
- desktop
  - src
    main.ts
- web
  - src
    components
    desktopUpdate.logic.test.ts
    desktopUpdate.logic.ts
package.json
scripts
- build-desktop-artifact.ts
- mock-update-server.ts

.gitignore+1

...	@@ -12,6 +12,7 @@ packages/*/dist	...	@@ -12,6 +12,7 @@ packages/*/dist
12	build/	12	build/
13	.logs/	13	.logs/
14	release/	14	release/
		15	release-mock/
15	.t3	16	.t3
16	.idea/	17	.idea/
17	apps/web/.playwright	18	apps/web/.playwright

apps/desktop/src/main.ts+23-1

 @@ -287,10 +287,12 @@ let updatePollTimer: ReturnType<typeof setInterval> | null = null;
 let updateStartupTimer: ReturnType<typeof setTimeout> | null = null;
 let updateCheckInFlight = false;
 let updateDownloadInFlight = false;
+let updateInstallInFlight = false;
 let updaterConfigured = false;
 let updateState: DesktopUpdateState = initialUpdateState();
 function resolveUpdaterErrorContext(): DesktopUpdateErrorContext {
+  if (updateInstallInFlight) return "install";
   if (updateDownloadInFlight) return "download";
   if (updateCheckInFlight) return "check";
   return updateState.errorContext;
 @@ -807,6 +809,7 @@ async function installDownloadedUpdate(): Promise<{ accepted: boolean; completed
   }
   isQuitting = true;
+  updateInstallInFlight = true;
   clearUpdatePollTimer();
   try {
     await stopBackendAndWaitForExit();
 @@ -814,10 +817,14 @@ async function installDownloadedUpdate(): Promise<{ accepted: boolean; completed
     for (const win of BrowserWindow.getAllWindows()) {
       win.destroy();
     }
+    // `quitAndInstall()` only starts the handoff to the updater. The actual
+    // install may still fail asynchronously, so keep the action incomplete
+    // until we either quit or receive an updater error.
     autoUpdater.quitAndInstall(true, true);
-    return { accepted: true, completed: true };
+    return { accepted: true, completed: false };
   } catch (error: unknown) {
     const message = formatErrorMessage(error);
+    updateInstallInFlight = false;
     isQuitting = false;
     setUpdateState(reduceDesktopUpdateStateOnInstallFailure(updateState, message));
     console.error(`[desktop-updater] Failed to install update: ${message}`);
 @@ -854,6 +861,13 @@ function configureAutoUpdater(): void {
     }
   }
+  if (process.env.T3CODE_DESKTOP_MOCK_UPDATES) {
+    autoUpdater.setFeedURL({
+      provider: "generic",
+      url: `http://localhost:${process.env.T3CODE_DESKTOP_MOCK_UPDATE_SERVER_PORT ?? 3000}`,
+    });
+  }
   autoUpdater.autoDownload = false;
   autoUpdater.autoInstallOnAppQuit = false;
   // Keep alpha branding, but force all installs onto the stable update track.
 @@ -890,6 +904,13 @@ function configureAutoUpdater(): void {
   });
   autoUpdater.on("error", (error) => {
     const message = formatErrorMessage(error);
+    if (updateInstallInFlight) {
+      updateInstallInFlight = false;
+      isQuitting = false;
+      setUpdateState(reduceDesktopUpdateStateOnInstallFailure(updateState, message));
+      console.error(`[desktop-updater] Updater error: ${message}`);
+      return;
+    }
     if (!updateCheckInFlight && !updateDownloadInFlight) {
       setUpdateState({
         status: "error",
 @@ -1365,6 +1386,7 @@ async function bootstrap(): Promise<void> {
 app.on("before-quit", () => {
   isQuitting = true;
+  updateInstallInFlight = false;
   writeDesktopLogHeader("before-quit received");
   clearUpdatePollTimer();
   stopBackend();

apps/web/src/components/desktopUpdate.logic.test.ts+10-3

 @@ -145,19 +145,26 @@ describe("getDesktopUpdateActionError", () => {
 });
 describe("desktop update UI helpers", () => {
-  it("toasts only for accepted incomplete actions", () => {
+  it("toasts only for actionable updater errors", () => {
     expect(
       shouldToastDesktopUpdateActionResult({
         accepted: true,
         completed: false,
-        state: baseState,
+        state: { ...baseState, message: "checksum mismatch" },
       }),
     ).toBe(true);
+    expect(
+      shouldToastDesktopUpdateActionResult({
+        accepted: true,
+        completed: false,
+        state: { ...baseState, message: null },
+      }),
+    ).toBe(false);
     expect(
       shouldToastDesktopUpdateActionResult({
         accepted: true,
         completed: true,
-        state: baseState,
+        state: { ...baseState, message: "checksum mismatch" },
       }),
     ).toBe(false);
   });

apps/web/src/components/desktopUpdate.logic.ts+1-1

...	@@ -87,7 +87,7 @@ export function getDesktopUpdateActionError(result: DesktopUpdateActionResult):	...	@@ -87,7 +87,7 @@ export function getDesktopUpdateActionError(result: DesktopUpdateActionResult):
87	}	87	}
88		88
89	export function shouldToastDesktopUpdateActionResult(result: DesktopUpdateActionResult): boolean {	89	export function shouldToastDesktopUpdateActionResult(result: DesktopUpdateActionResult): boolean {
90	return result.accepted && !result.completed;	90	return getDesktopUpdateActionError(result) !== null;
91	}	91	}
92		92
93	export function shouldHighlightDesktopUpdateError(state: DesktopUpdateState \| null): boolean {	93	export function shouldHighlightDesktopUpdateError(state: DesktopUpdateState \| null): boolean {

package.json+1

...	@@ -30,6 +30,7 @@	...	@@ -30,6 +30,7 @@
30	"start": "turbo run start --filter=t3",	30	"start": "turbo run start --filter=t3",
31	"start:desktop": "turbo run start --filter=@t3tools/desktop",	31	"start:desktop": "turbo run start --filter=@t3tools/desktop",
32	"start:marketing": "turbo run preview --filter=@t3tools/marketing",	32	"start:marketing": "turbo run preview --filter=@t3tools/marketing",
		33	"start:mock-update-server": "bun run scripts/mock-update-server.ts",
33	"build": "turbo run build",	34	"build": "turbo run build",
34	"build:marketing": "turbo run build --filter=@t3tools/marketing",	35	"build:marketing": "turbo run build --filter=@t3tools/marketing",
35	"build:desktop": "turbo run build --filter=@t3tools/desktop --filter=t3",	36	"build:desktop": "turbo run build --filter=@t3tools/desktop --filter=t3",

scripts/build-desktop-artifact.ts+41-1

 @@ -74,6 +74,8 @@ interface BuildCliInput {
   readonly keepStage: Option.Option<boolean>;
   readonly signed: Option.Option<boolean>;
   readonly verbose: Option.Option<boolean>;
+  readonly mockUpdates: Option.Option<boolean>;
+  readonly mockUpdateServerPort: Option.Option<string>;
 }
 function detectHostBuildPlatform(hostPlatform: string): typeof BuildPlatform.Type | undefined {
 @@ -162,6 +164,8 @@ interface ResolvedBuildOptions {
   readonly keepStage: boolean;
   readonly signed: boolean;
   readonly verbose: boolean;
+  readonly mockUpdates: boolean;
+  readonly mockUpdateServerPort: string | undefined;
 }
 interface StagePackageJson {
 @@ -204,6 +208,8 @@ const BuildEnvConfig = Config.all({
   keepStage: Config.boolean("T3CODE_DESKTOP_KEEP_STAGE").pipe(Config.withDefault(false)),
   signed: Config.boolean("T3CODE_DESKTOP_SIGNED").pipe(Config.withDefault(false)),
   verbose: Config.boolean("T3CODE_DESKTOP_VERBOSE").pipe(Config.withDefault(false)),
+  mockUpdates: Config.boolean("T3CODE_DESKTOP_MOCK_UPDATES").pipe(Config.withDefault(false)),
+  mockUpdateServerPort: Config.string("T3CODE_DESKTOP_MOCK_UPDATE_SERVER_PORT").pipe(Config.option),
 });
 const resolveBooleanFlag = (flag: Option.Option<boolean>, envValue: boolean) =>
 @@ -231,13 +237,26 @@ const resolveBuildOptions = Effect.fn("resolveBuildOptions")(function* (input: B
   const target = mergeOptions(input.target, env.target, PLATFORM_CONFIG[platform].defaultTarget);
   const arch = mergeOptions(input.arch, env.arch, getDefaultArch(platform));
   const version = mergeOptions(input.buildVersion, env.version, undefined);
-  const outputDir = path.resolve(repoRoot, mergeOptions(input.outputDir, env.outputDir, "release"));
+  const releaseDir = resolveBooleanFlag(input.mockUpdates, env.mockUpdates)
+    ? "release-mock"
+    : "release";
+  const outputDir = path.resolve(
+    repoRoot,
+    mergeOptions(input.outputDir, env.outputDir, releaseDir),
+  );
   const skipBuild = resolveBooleanFlag(input.skipBuild, env.skipBuild);
   const keepStage = resolveBooleanFlag(input.keepStage, env.keepStage);
   const signed = resolveBooleanFlag(input.signed, env.signed);
   const verbose = resolveBooleanFlag(input.verbose, env.verbose);
+  const mockUpdates = resolveBooleanFlag(input.mockUpdates, env.mockUpdates);
+  const mockUpdateServerPort = mergeOptions(
+    input.mockUpdateServerPort,
+    env.mockUpdateServerPort,
+    undefined,
+  );
   return {
     platform,
     target,
 @@ -248,6 +267,8 @@ const resolveBuildOptions = Effect.fn("resolveBuildOptions")(function* (input: B
     keepStage,
     signed,
     verbose,
+    mockUpdates,
+    mockUpdateServerPort,
   } satisfies ResolvedBuildOptions;
 });
 @@ -447,6 +468,8 @@ const createBuildConfig = Effect.fn("createBuildConfig")(function* (
   target: string,
   productName: string,
   signed: boolean,
+  mockUpdates: boolean,
+  mockUpdateServerPort: string | undefined,
 ) {
   const buildConfig: Record<string, unknown> = {
     appId: "com.t3tools.t3code",
 @@ -459,6 +482,13 @@ const createBuildConfig = Effect.fn("createBuildConfig")(function* (
   const publishConfig = resolveGitHubPublishConfig();
   if (publishConfig) {
     buildConfig.publish = [publishConfig];
+  } else if (mockUpdates) {
+    buildConfig.publish = [
+      {
+        provider: "generic",
+        url: `http://localhost:${mockUpdateServerPort ?? 3000}`,
+      },
+    ];
   }
   if (platform === "mac") {
 @@ -631,6 +661,8 @@ const buildDesktopArtifact = Effect.fn("buildDesktopArtifact")(function* (
       options.target,
       desktopPackageJson.productName ?? "T3 Code",
       options.signed,
+      options.mockUpdates,
+      options.mockUpdateServerPort,
     ),
     dependencies: {
       ...resolvedServerDependencies,
 @@ -769,6 +801,14 @@ const buildDesktopArtifactCli = Command.make("build-desktop-artifact", {
     Flag.withDescription("Stream subprocess stdout (env: T3CODE_DESKTOP_VERBOSE)."),
     Flag.optional,
   ),
+  mockUpdates: Flag.boolean("mock-updates").pipe(
+    Flag.withDescription("Enable mock updates (env: T3CODE_DESKTOP_MOCK_UPDATES)."),
+    Flag.optional,
+  ),
+  mockUpdateServerPort: Flag.string("mock-update-server-port").pipe(
+    Flag.withDescription("Mock update server port (env: T3CODE_DESKTOP_MOCK_UPDATE_SERVER_PORT)."),
+    Flag.optional,
+  ),
 }).pipe(
   Command.withDescription("Build a desktop artifact for T3 Code."),
   Command.withHandler((input) => Effect.flatMap(resolveBuildOptions(input), buildDesktopArtifact)),

scripts/mock-update-server.ts created+44

...	@@ -0,0 +1,44 @@
1	import { resolve, relative } from "node:path";
2	import { realpathSync } from "node:fs";
3
4	const port = Number(process.env.T3CODE_DESKTOP_MOCK_UPDATE_SERVER_PORT ?? 3000);
5	const root =
6	process.env.T3CODE_DESKTOP_MOCK_UPDATE_SERVER_ROOT ??
7	resolve(import.meta.dirname, "..", "release-mock");
8
9	const mockServerLog = (level: "info" \| "warn" \| "error" = "info", message: string) => {
10	console[level](`[mock-update-server] ${message}`);
11	};
12
13	function isWithinRoot(filePath: string): boolean {
14	try {
15	return !relative(realpathSync(root), realpathSync(filePath)).startsWith(".");
16	} catch (error) {
17	mockServerLog("error", `Error checking if file is within root: ${error}`);
18	return false;
19	}
20	}
21
22	Bun.serve({
23	port,
24	hostname: "localhost",
25	fetch: async (request) => {
26	const url = new URL(request.url);
27	const path = url.pathname;
28	mockServerLog("info", `Request received for path: ${path}`);
29	const filePath = resolve(root, `.${path}`);
30	if (!isWithinRoot(filePath)) {
31	mockServerLog("warn", `Attempted to access file outside of root: ${filePath}`);
32	return new Response("Not Found", { status: 404 });
33	}
34	const file = Bun.file(filePath);
35	if (!(await file.exists())) {
36	mockServerLog("warn", `Attempted to access non-existent file: ${filePath}`);
37	return new Response("Not Found", { status: 404 });
38	}
39	mockServerLog("info", `Serving file: ${filePath}`);
40	return new Response(file.stream());
41	},
42	});
43
44	mockServerLog("info", `running on http://localhost:${port}`);