fix(agent-tars): autoUpdater lack of latest-mac.yml (#341)

Author: ycjcl868

apps/agent-tars/forge.config.ts

@@ -12,6 +12,7 @@ import { rimraf, rimrafSync } from 'rimraf';
 import {
   getModuleRoot,
   getExternalPkgsDependencies,
+  hooks,
 } from '@common/electron-build';
 
 import pkg from './package.json';
@@ -229,6 +230,11 @@ const config: ForgeConfig = {
           }),
         ]),
   ],
+  hooks: {
+    postMake: async (forgeConfig, makeResults) => {
+      return await hooks.postMake?.(forgeConfig, makeResults);
+    },
+  },
 };
 
 export default config;

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

@@ -9,22 +9,23 @@ import {
 export class AppUpdater {
   autoUpdater: ElectronAppUpdater = autoUpdater;
   constructor(mainWindow: BrowserWindow) {
-    autoUpdater.logger = logger;
-    autoUpdater.autoDownload = true;
-    autoUpdater.forceDevUpdateConfig = !app.isPackaged;
+    if (app.isPackaged) {
+      autoUpdater.logger = logger;
+      autoUpdater.autoDownload = true;
 
-    autoUpdater.on('error', (error) => {
-      logger.error('Update_Error', error);
-      mainWindow.webContents.send('main:error', error);
-    });
+      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);
-    });
+      autoUpdater.on('update-available', (releaseInfo: UpdateInfo) => {
+        logger.info('new version', releaseInfo);
+        mainWindow.webContents.send('app-update-available', releaseInfo);
+      });
 
-    this.autoUpdater = autoUpdater;
+      this.autoUpdater = autoUpdater;
 
-    this.autoUpdater.checkForUpdatesAndNotify();
+      this.autoUpdater.checkForUpdatesAndNotify();
+    }
   }
 }

packages/common/electron-build/package.json

@@ -42,6 +42,7 @@
     "@common/configs": "workspace:*"
   },
   "devDependencies": {
+    "@electron-forge/shared-types": "^7.8.0",
     "tsx": "^4.19.2",
     "@rslib/core": "^0.5.4",
     "typescript": "^5.7.2",

packages/common/electron-build/src/hooks/index.ts

@@ -0,0 +1,6 @@
+import { type ForgeHookMap } from '@electron-forge/shared-types';
+import { postMake } from './postMake';
+
+export const hooks: ForgeHookMap = {
+  postMake,
+};

packages/common/electron-build/src/hooks/postMake.ts

@@ -0,0 +1,90 @@
+/**
+ * Copyright (c) 2025 Bytedance, Inc. and its affiliates.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import crypto from 'node:crypto';
+import fs from 'node:fs';
+import path from 'node:path';
+import yaml from 'js-yaml';
+import type { ForgeHookMap } from '@electron-forge/shared-types';
+
+const artifactRegex = /.*\.(?:exe|dmg|AppImage|zip)$/;
+const platformNamesMap = {
+  darwin: 'macos',
+  linux: 'linux',
+  win32: 'windows',
+};
+const updateYmlMap = {
+  darwin: 'latest-mac.yml',
+  linux: 'latest-linux.yml',
+  win32: 'latest.yml',
+};
+
+export const postMake: ForgeHookMap['postMake'] = async (
+  _forgeConfig,
+  makeResults,
+) => {
+  const yml: {
+    version?: string;
+    files: {
+      url: string;
+      sha512: string;
+      size: number;
+    }[];
+    releaseDate?: string;
+  } = {
+    version: makeResults[0]?.packageJSON?.version,
+    files: [],
+  };
+
+  makeResults = makeResults.map((result) => {
+    result.artifacts = result.artifacts.map((artifact) => {
+      if (artifactRegex.test(artifact)) {
+        try {
+          const fileData = fs.readFileSync(artifact);
+          const hash = crypto
+            .createHash('sha512')
+            .update(fileData)
+            .digest('base64');
+          const { size } = fs.statSync(artifact);
+
+          yml.files.push({
+            url: path.basename(artifact),
+            sha512: hash,
+            size,
+          });
+        } catch {
+          console.error(`Failed to hash ${artifact}`);
+        }
+        return artifact;
+      } else {
+        return artifact;
+      }
+    });
+    return result;
+  });
+  yml.releaseDate = new Date().toISOString();
+
+  const firstResult = makeResults[0];
+  if (!firstResult?.artifacts?.[0] || !firstResult.platform) {
+    throw new Error('Missing required artifact or platform information');
+  }
+
+  const ymlPath = `${path.dirname(firstResult.artifacts[0])}/${
+    updateYmlMap[firstResult.platform as keyof typeof updateYmlMap]
+  }`;
+
+  const ymlStr = yaml.dump(yml, {
+    lineWidth: -1,
+  });
+  fs.writeFileSync(ymlPath, ymlStr);
+
+  makeResults.push({
+    artifacts: [ymlPath],
+    platform: makeResults[0].platform,
+    arch: makeResults[0].arch,
+    packageJSON: makeResults[0].packageJSON,
+  });
+
+  return makeResults;
+};

packages/common/electron-build/src/index.ts

@@ -6,3 +6,4 @@ export {
   getModuleRoot,
   getExternalPkgsDependencies,
 } from './getPackageDependencies';
+export { hooks } from './hooks/index';