feat: 添加自动更新功能

ErlichLiu · ErlichLiu · commit 856b72b9ff52 · 2026-05-07T16:55:05.000+08:00
diff --git a/apps/electron/src/main/lib/github-release-service.ts b/apps/electron/src/main/lib/github-release-service.ts
@@ -26,8 +26,14 @@ interface ReleaseCache {
 
 let releaseCache: ReleaseCache | null = null
 
-/** 缓存有效期（5 分钟） */
-const CACHE_TTL = 5 * 60 * 1000
+/** 单个 Release 缓存（按 tag） */
+const tagCache = new Map<string, { data: GitHubRelease; timestamp: number }>()
+
+/** 缓存有效期（30 分钟） */
+const CACHE_TTL = 30 * 60 * 1000
+
+/** Rate limit 冷却标记 */
+let rateLimitUntil = 0
 
 /**
  * 从 GitHub API 获取 releases
@@ -36,6 +42,11 @@ const CACHE_TTL = 5 * 60 * 1000
  * @returns Release 数据
  */
 async function fetchFromGitHub<T>(endpoint: string): Promise<T> {
+  // Rate limit 冷却期内直接跳过
+  if (Date.now() < rateLimitUntil) {
+    throw new Error('GitHub API rate limit cooldown')
+  }
+
   const url = `${GITHUB_API_BASE}/repos/${GITHUB_REPO.owner}/${GITHUB_REPO.repo}${endpoint}`
 
   console.log(`[GitHub Release] 正在请求: ${url}`)
@@ -47,6 +58,12 @@ async function fetchFromGitHub<T>(endpoint: string): Promise<T> {
     },
   })
 
+  if (response.status === 403 || response.status === 429) {
+    // Rate limited — 冷却 15 分钟
+    rateLimitUntil = Date.now() + 15 * 60 * 1000
+    throw new Error('GitHub API rate limit exceeded, cooling down for 15 minutes')
+  }
+
   if (!response.ok) {
     const errorText = await response.text()
     throw new Error(
@@ -150,13 +167,24 @@ export async function listReleases(
  */
 export async function getReleaseByTag(tag: string): Promise<GitHubRelease | null> {
   try {
+    // 检查缓存
+    const cached = tagCache.get(tag)
+    if (cached && Date.now() - cached.timestamp < CACHE_TTL) {
+      return cached.data
+    }
+
     const release = await fetchFromGitHub<GitHubRelease>(
       `/releases/tags/${tag}`
     )
     console.log(`[GitHub Release] 获取 Release: ${tag}`)
+
+    tagCache.set(tag, { data: release, timestamp: Date.now() })
     return release
   } catch (error) {
     console.error(`[GitHub Release] 获取 Release ${tag} 失败:`, error)
+    // 返回过期缓存
+    const cached = tagCache.get(tag)
+    if (cached) return cached.data
     return null
   }
 }
diff --git a/apps/electron/src/main/lib/updater/auto-updater.ts b/apps/electron/src/main/lib/updater/auto-updater.ts
@@ -1,9 +1,7 @@
 /**
  * 自动更新核心模块
  *
- * 仅检测新版本并通知用户，不做自动下载/安装。
- * 用户通过 GitHub Releases 页面手动下载覆盖安装。
- *
+ * 检测新版本 → 自动后台下载 → 用户确认后重启安装。
  * 仅在打包后的生产环境中工作。
  */
 
@@ -45,6 +43,11 @@ export async function checkForUpdates(): Promise<void> {
   }
 }
 
+/** 退出并安装已下载的更新 */
+export function quitAndInstall(): void {
+  autoUpdater.quitAndInstall(false, true)
+}
+
 /** 清理更新器资源（定时器等） */
 export function cleanupUpdater(): void {
   if (checkInterval) {
@@ -61,17 +64,16 @@ export function cleanupUpdater(): void {
 export function initAutoUpdater(mainWindow: BrowserWindow): void {
   win = mainWindow
 
-  // 配置 electron-updater 日志，转发到 console
   autoUpdater.logger = {
     info: (...args: unknown[]) => console.log('[更新-updater]', ...args),
     warn: (...args: unknown[]) => console.warn('[更新-updater]', ...args),
     error: (...args: unknown[]) => console.error('[更新-updater]', ...args),
     debug: (...args: unknown[]) => console.log('[更新-updater:debug]', ...args),
   }
 
-  // 禁用自动下载和自动安装，仅做版本检测
-  autoUpdater.autoDownload = false
-  autoUpdater.autoInstallOnAppQuit = false
+  // 自动下载，退出时自动安装
+  autoUpdater.autoDownload = true
+  autoUpdater.autoInstallOnAppQuit = true
 
   // 监听更新事件
   autoUpdater.on('checking-for-update', () => {
@@ -90,6 +92,27 @@ export function initAutoUpdater(mainWindow: BrowserWindow): void {
     })
   })
 
+  autoUpdater.on('download-progress', (progress) => {
+    setStatus({
+      status: 'downloading',
+      version: (currentStatus as { version?: string }).version || '',
+      progress: {
+        percent: progress.percent,
+        transferred: progress.transferred,
+        total: progress.total,
+        bytesPerSecond: progress.bytesPerSecond,
+      },
+    })
+  })
+
+  autoUpdater.on('update-downloaded', (info) => {
+    console.log('[更新] 下载完成:', info.version)
+    setStatus({
+      status: 'downloaded',
+      version: info.version,
+    })
+  })
+
   autoUpdater.on('update-not-available', () => {
     console.log('[更新] 已是最新版本')
     setStatus({ status: 'not-available' })
@@ -124,5 +147,5 @@ export function initAutoUpdater(mainWindow: BrowserWindow): void {
     win = null
   })
 
-  console.log('[更新] 版本检测模块已初始化（仅检测，不自动下载/安装）')
+  console.log('[更新] 自动更新模块已初始化（自动下载，用户确认后安装）')
 }
diff --git a/apps/electron/src/main/lib/updater/updater-ipc.ts b/apps/electron/src/main/lib/updater/updater-ipc.ts
@@ -2,7 +2,6 @@
  * 自动更新 IPC 处理器
  *
  * 注册更新相关的 IPC 通道，供渲染进程调用。
- * 仅支持检查更新和获取状态，不提供下载/安装功能。
  */
 
 import { ipcMain } from 'electron'
@@ -11,27 +10,33 @@ import type { UpdateStatus } from './updater-types'
 import {
   checkForUpdates,
   getUpdateStatus,
+  quitAndInstall,
 } from './auto-updater'
 
 /** 注册更新 IPC 处理器 */
 export function registerUpdaterIpc(): void {
   console.log('[更新 IPC] 正在注册更新 IPC 处理器...')
 
-  // 检查更新
   ipcMain.handle(
     UPDATER_IPC_CHANNELS.CHECK_FOR_UPDATES,
     async (): Promise<void> => {
       await checkForUpdates()
     }
   )
 
-  // 获取当前更新状态
   ipcMain.handle(
     UPDATER_IPC_CHANNELS.GET_STATUS,
     async (): Promise<UpdateStatus> => {
       return getUpdateStatus()
     }
   )
 
+  ipcMain.handle(
+    UPDATER_IPC_CHANNELS.QUIT_AND_INSTALL,
+    (): void => {
+      quitAndInstall()
+    }
+  )
+
   console.log('[更新 IPC] 更新 IPC 处理器注册完成')
 }
diff --git a/apps/electron/src/main/lib/updater/updater-types.ts b/apps/electron/src/main/lib/updater/updater-types.ts
@@ -1,21 +1,35 @@
 /**
  * 自动更新相关类型定义
  *
- * 仅检测新版本并通知用户，不做自动下载/安装。
- * 用户通过 GitHub Releases 页面手动下载覆盖安装。
+ * 检测新版本 → 自动下载 → 用户确认后重启安装
  */
 
 /** 更新状态 */
-export interface UpdateStatus {
-  status: 'idle' | 'checking' | 'available' | 'not-available' | 'error'
-  version?: string
-  releaseNotes?: string
-  error?: string
+export type UpdateStatus =
+  | { status: 'idle' }
+  | { status: 'checking' }
+  | { status: 'available'; version: string; releaseNotes?: string }
+  | { status: 'downloading'; version: string; progress: DownloadProgress }
+  | { status: 'downloaded'; version: string }
+  | { status: 'not-available' }
+  | { status: 'error'; error: string }
+
+/** 下载进度 */
+export interface DownloadProgress {
+  /** 已下载百分比 0-100 */
+  percent: number
+  /** 已下载字节数 */
+  transferred: number
+  /** 总字节数 */
+  total: number
+  /** 下载速度（字节/秒） */
+  bytesPerSecond: number
 }
 
 /** 更新 IPC 通道常量 */
 export const UPDATER_IPC_CHANNELS = {
   CHECK_FOR_UPDATES: 'updater:check',
   GET_STATUS: 'updater:get-status',
   ON_STATUS_CHANGED: 'updater:status-changed',
+  QUIT_AND_INSTALL: 'updater:quit-and-install',
 } as const
diff --git a/apps/electron/src/preload/index.ts b/apps/electron/src/preload/index.ts
@@ -651,23 +651,26 @@ export interface ElectronAPI {
   /** 设置默认提示词 */
   setDefaultPrompt: (id: string | null) => Promise<void>
 
-  // ===== 版本检测相关（仅检测，不自动下载/安装） =====
+  // ===== 自动更新 =====
 
   /** 更新 API */
   updater?: {
     checkForUpdates: () => Promise<void>
     getStatus: () => Promise<{
-      status: 'idle' | 'checking' | 'available' | 'not-available' | 'error'
+      status: 'idle' | 'checking' | 'available' | 'downloading' | 'downloaded' | 'not-available' | 'error'
       version?: string
       releaseNotes?: string
+      progress?: { percent: number; transferred: number; total: number; bytesPerSecond: number }
       error?: string
     }>
     onStatusChanged: (callback: (status: {
-      status: 'idle' | 'checking' | 'available' | 'not-available' | 'error'
+      status: 'idle' | 'checking' | 'available' | 'downloading' | 'downloaded' | 'not-available' | 'error'
       version?: string
       releaseNotes?: string
+      progress?: { percent: number; transferred: number; total: number; bytesPerSecond: number }
       error?: string
     }) => void) => () => void
+    quitAndInstall: () => Promise<void>
   }
 
   // GitHub Release
@@ -1548,7 +1551,7 @@ const electronAPI: ElectronAPI = {
     return ipcRenderer.invoke(SYSTEM_PROMPT_IPC_CHANNELS.SET_DEFAULT, id)
   },
 
-  // 自动更新（仅版本检测，不自动下载/安装）
+  // 自动更新
   updater: {
     checkForUpdates: () => ipcRenderer.invoke('updater:check'),
     getStatus: () => ipcRenderer.invoke('updater:get-status'),
@@ -1557,6 +1560,7 @@ const electronAPI: ElectronAPI = {
       ipcRenderer.on('updater:status-changed', listener)
       return () => { ipcRenderer.removeListener('updater:status-changed', listener) }
     },
+    quitAndInstall: () => ipcRenderer.invoke('updater:quit-and-install'),
   },
 
   // GitHub Release
diff --git a/apps/electron/src/renderer/atoms/updater.ts b/apps/electron/src/renderer/atoms/updater.ts
@@ -2,27 +2,35 @@
  * 自动更新状态原子
  *
  * 管理应用更新状态，订阅主进程推送的更新事件。
- * 仅检测新版本并通知，不自动下载/安装。
  * 优雅降级：如果 window.electronAPI.updater 不存在（开源构建），状态保持 idle。
  */
 
 import { atom } from 'jotai'
 
+/** 下载进度 */
+export interface DownloadProgress {
+  percent: number
+  transferred: number
+  total: number
+  bytesPerSecond: number
+}
+
 /** 更新状态 */
 export interface UpdateStatus {
-  status: 'idle' | 'checking' | 'available' | 'not-available' | 'error'
+  status: 'idle' | 'checking' | 'available' | 'downloading' | 'downloaded' | 'not-available' | 'error'
   version?: string
   releaseNotes?: string
+  progress?: DownloadProgress
   error?: string
 }
 
 /** 更新状态 atom */
 export const updateStatusAtom = atom<UpdateStatus>({ status: 'idle' })
 
-/** 是否有可用更新 */
+/** 是否有可用更新（包含已下载完成） */
 export const hasUpdateAtom = atom((get) => {
   const { status } = get(updateStatusAtom)
-  return status === 'available'
+  return status === 'available' || status === 'downloading' || status === 'downloaded'
 })
 
 /** updater 是否可用 */
diff --git a/apps/electron/src/renderer/components/settings/UpdateDialog.tsx b/apps/electron/src/renderer/components/settings/UpdateDialog.tsx
