add desktop tray and updater support (#1193)

packages/desktop/src/main/updater.ts

@@ -1,46 +1,165 @@
 import { app, dialog } from 'electron'
-import { autoUpdater } from 'electron-updater'
+import { autoUpdater, type ProgressInfo, type UpdateDownloadedEvent, type UpdateInfo } from 'electron-updater'
+import { t } from './desktop-i18n'
 
 let initialized = false
+let checking = false
+let updateDownloaded = false
 
-export function initAutoUpdater() {
+const LATEST_RELEASE_URL = 'https://api.github.com/repos/EKKOLearnAI/hermes-web-ui/releases/latest'
+const CLOUDFLARE_DOWNLOAD_BASE_URL = 'https://download.ekkolearnai.com'
+
+interface GitHubRelease {
+  tag_name?: string
+}
+
+interface AutoUpdaterOptions {
+  beforeQuitAndInstall?: () => void
+}
+
+let options: AutoUpdaterOptions = {}
+
+async function getLatestReleaseTag(): Promise<string> {
+  const res = await fetch(LATEST_RELEASE_URL, {
+    headers: {
+      Accept: 'application/vnd.github+json',
+      'User-Agent': `Hermes-Studio/${app.getVersion()}`,
+    },
+  })
+  if (!res.ok) throw new Error(`GitHub returned ${res.status}`)
+
+  const release = await res.json() as GitHubRelease
+  const tag = release.tag_name?.trim()
+  if (!tag) throw new Error('Latest release response did not include a tag')
+  return tag.startsWith('v') ? tag : `v${tag}`
+}
+
+async function configureFeedFromLatestRelease(): Promise<void> {
+  const tag = await getLatestReleaseTag()
+  autoUpdater.setFeedURL({
+    provider: 'generic',
+    url: `${CLOUDFLARE_DOWNLOAD_BASE_URL}/${tag}`,
+  })
+}
+
+function showUpToDate(info?: UpdateInfo) {
+  const version = info?.version || app.getVersion()
+  dialog.showMessageBox({
+    type: 'info',
+    title: t('update.upToDateTitle'),
+    message: t('update.upToDateMessage'),
+    detail: t('update.currentVersion', { version }),
+    buttons: [t('common.ok')],
+  }).catch(() => undefined)
+}
+
+function showUpdateCheckFailed(err: unknown) {
+  const detail = err instanceof Error ? err.message : String(err)
+  dialog.showMessageBox({
+    type: 'error',
+    title: t('update.failedTitle'),
+    message: t('update.failedMessage'),
+    detail,
+    buttons: [t('common.ok')],
+  }).catch(() => undefined)
+}
+
+export function initAutoUpdater(nextOptions: AutoUpdaterOptions = {}) {
+  options = { ...options, ...nextOptions }
   if (initialized) return
   initialized = true
 
   if (!app.isPackaged) return // dev mode: skip
-  if (process.env.HERMES_DESKTOP_ENABLE_AUTO_UPDATE !== 'true') return
 
   autoUpdater.autoDownload = true
   autoUpdater.autoInstallOnAppQuit = true
 
   autoUpdater.on('update-available', info => {
     console.log(`[updater] update available: ${info.version}`)
+    dialog.showMessageBox({
+      type: 'info',
+      title: t('update.availableTitle'),
+      message: t('update.availableMessage', { version: info.version }),
+      detail: t('update.downloading'),
+      buttons: [t('common.ok')],
+    }).catch(() => undefined)
   })
-  autoUpdater.on('update-not-available', () => {
+  autoUpdater.on('update-not-available', info => {
     console.log('[updater] up to date')
+    if (checking) showUpToDate(info)
   })
   autoUpdater.on('error', err => {
     console.error('[updater] error:', err)
+    if (checking) showUpdateCheckFailed(err)
   })
-  autoUpdater.on('update-downloaded', async info => {
+  autoUpdater.on('download-progress', (info: ProgressInfo) => {
+    console.log(`[updater] download ${Math.round(info.percent)}%`)
+  })
+  autoUpdater.on('update-downloaded', async (info: UpdateDownloadedEvent) => {
+    updateDownloaded = true
     const { response } = await dialog.showMessageBox({
       type: 'info',
-      title: 'Update ready',
-      message: `Hermes Studio ${info.version} is ready to install.`,
-      detail: 'Restart now to apply the update, or it will be installed on next quit.',
-      buttons: ['Restart now', 'Later'],
+      title: t('update.readyTitle'),
+      message: t('update.readyMessage', { version: info.version }),
+      detail: t('update.readyDetail'),
+      buttons: [t('update.restartNow'), t('update.later')],
       defaultId: 0,
       cancelId: 1,
     })
-    if (response === 0) autoUpdater.quitAndInstall()
+    if (response === 0) {
+      options.beforeQuitAndInstall?.()
+      autoUpdater.quitAndInstall()
+    }
   })
 
-  autoUpdater.checkForUpdates().catch(err => {
-    console.error('[updater] initial check failed:', err)
-  })
+  if (process.env.HERMES_DESKTOP_ENABLE_AUTO_UPDATE !== 'false') {
+    checkForDesktopUpdates(false).catch(err => {
+      console.error('[updater] initial check failed:', err)
+    })
+  }
 
   // Recheck every 6h while app is running
   setInterval(() => {
-    autoUpdater.checkForUpdates().catch(() => undefined)
+    checkForDesktopUpdates(false).catch(() => undefined)
   }, 6 * 60 * 60 * 1000)
 }
+
+export async function checkForDesktopUpdates(manual: boolean): Promise<void> {
+  if (!app.isPackaged) {
+    if (manual) {
+      await dialog.showMessageBox({
+        type: 'info',
+        title: t('update.checkingTitle'),
+        message: t('update.packagedOnlyMessage'),
+        buttons: [t('common.ok')],
+      })
+    }
+    return
+  }
+
+  if (updateDownloaded) {
+    options.beforeQuitAndInstall?.()
+    autoUpdater.quitAndInstall()
+    return
+  }
+
+  if (manual) {
+    await dialog.showMessageBox({
+      type: 'info',
+      title: t('update.checkingTitle'),
+      message: t('update.checkingMessage'),
+      buttons: [t('common.ok')],
+    })
+  }
+
+  checking = manual
+  try {
+    await configureFeedFromLatestRelease()
+    await autoUpdater.checkForUpdates()
+  } catch (err) {
+    if (manual) showUpdateCheckFailed(err)
+    throw err
+  } finally {
+    checking = false
+  }
+}