fix(agent-tars): app updater not work (#328)

Author: ycjcl868

apps/agent-tars/dev-app-update.yml

@@ -50,7 +50,7 @@
     "@electron-toolkit/preload": "^3.0.1",
     "@electron-toolkit/utils": "^4.0.0",
     "@electron/asar": "^3.2.18",
-    "update-electron-app": "^3.1.0",
+    "electron-updater": "^6.3.9",
     "openai": "^4.86.2",
     "dotenv": "16.4.7",
     "@agent-infra/shared": "workspace:*",
@@ -94,5 +94,17 @@
     "vite-tsconfig-paths": "^5.1.4",
     "vitest": "^3.0.8",
     "vite-plugin-singlefile": "2.2.0"
+  },
+  "build": {
+    "publish": [
+      {
+        "provider": "github",
+        "owner": "bytedance",
+        "repo": "UI-TARS-desktop"
+      }
+    ],
+    "electronDownload": {
+      "mirror": "https://npmmirror.com/mirrors/electron/"
+    }
   }
 }

apps/agent-tars/package.json

@@ -1,32 +1,15 @@
 import { app, shell, BrowserWindow, ipcMain } from 'electron';
 import { join } from 'path';
 import { registerIpcMain } from '@ui-tars/electron-ipc/main';
-import { updateElectronApp, UpdateSourceType } from 'update-electron-app';
 import { electronApp, optimizer, is } from '@electron-toolkit/utils';
 import { ipcRoutes } from './ipcRoutes';
 import icon from '../../resources/icon.png?asset';
 import MenuBuilder from './menu';
 import { logger } from './utils/logger';
 import { ErrorReporter } from './utils/errorReporter';
+import { AppUpdater } from './utils/updateApp';
 
-class AppUpdater {
-  constructor() {
-    // autoUpdater.logger = logger;
-    // autoUpdater.checkForUpdatesAndNotify();
-    if (process.env.CI !== 'e2e') {
-      updateElectronApp({
-        updateSource: {
-          type: UpdateSourceType.ElectronPublicUpdateService,
-          repo: 'bytedance/UI-TARS-desktop',
-          host: 'https://update.electronjs.org',
-        },
-        updateInterval: '20 minutes',
-      });
-    }
-  }
-}
-
-function createWindow(): void {
+function createWindow(): BrowserWindow {
   // Create the browser window.
   const mainWindow = new BrowserWindow({
     width: 1200,
@@ -86,6 +69,8 @@ function createWindow(): void {
   // Set up the application menu
   const menuBuilder = new MenuBuilder(mainWindow);
   menuBuilder.buildMenu();
+
+  return mainWindow;
 }
 
 const initializeApp = async () => {
@@ -122,8 +107,6 @@ app.whenReady().then(async () => {
     optimizer.watchWindowShortcuts(window);
   });
 
-  new AppUpdater();
-
   await initializeApp();
 
   // IPC test
@@ -133,7 +116,9 @@ app.whenReady().then(async () => {
   });
   registerIpcMain(ipcRoutes);
 
-  createWindow();
+  const mainWindow = createWindow();
+
+  new AppUpdater(mainWindow);
 
   app.on('activate', function () {
     // On macOS it's common to re-create a window in the app when the

apps/agent-tars/src/main/index.ts

@@ -0,0 +1,30 @@
+import { UpdateInfo } from 'builder-util-runtime';
+import { app, BrowserWindow } from 'electron';
+import { logger } from '@main/utils/logger';
+import {
+  AppUpdater as ElectronAppUpdater,
+  autoUpdater,
+} from 'electron-updater';
+
+export class AppUpdater {
+  autoUpdater: ElectronAppUpdater = autoUpdater;
+  constructor(mainWindow: BrowserWindow) {
+    autoUpdater.logger = logger;
+    autoUpdater.autoDownload = true;
+    autoUpdater.forceDevUpdateConfig = !app.isPackaged;
+
+    autoUpdater.on('error', (error) => {
+      logger.error('Update_Error', error);
+      mainWindow.webContents.send('main:error', error);
+    });
+
+    autoUpdater.on('update-available', (releaseInfo: UpdateInfo) => {
+      logger.info('new version', releaseInfo);
+      mainWindow.webContents.send('app-update-available', releaseInfo);
+    });
+
+    this.autoUpdater = autoUpdater;
+
+    this.autoUpdater.checkForUpdatesAndNotify();
+  }
+}

apps/agent-tars/src/main/utils/updateApp.ts

@@ -136,6 +136,8 @@ declare global {
       ipcRenderer: {
         invoke: (channel: string, ...args: any[]) => Promise<any>;
         on: (channel: string, listener: (...args: any[]) => void) => void;
+        once: (channel: string, listener: (...args: any[]) => void) => void;
+        off: (channel: string, listener: (...args: any[]) => void) => void;
       };
     };
   }
@@ -148,5 +150,11 @@ window.electron = {
     on: async () => {
       // noop
     },
+    once: async () => {
+      // noop
+    },
+    off: async () => {
+      // noop
+    },
   },
 };

apps/agent-tars/src/renderer/mock/ipc.ts

@@ -37,6 +37,7 @@
     "registry": "https://registry.npmjs.org"
   },
   "dependencies": {
+    "electron-updater": "6.3.9",
     "flora-colossus": "2.0.0",
     "@common/configs": "workspace:*"
   },