瀏覽代碼

add auto update check

Jeffrey Morgan 1 年之前
父節點
當前提交
594f07d729
共有 1 個文件被更改,包括 15 次插入1 次删除
  1. 15 1
      desktop/src/index.ts

+ 15 - 1
desktop/src/index.ts

@@ -1,4 +1,4 @@
-import { app, BrowserWindow, autoUpdater } from 'electron'
+import { app, BrowserWindow, autoUpdater, dialog } from 'electron'
 import { spawn } from 'child_process'
 import * as path from 'path'
 
@@ -87,3 +87,17 @@ autoUpdater.setFeedURL({ url: `https://updates.ollama.ai/update/${process.platfo
 setInterval(() => {
   autoUpdater.checkForUpdates()
 }, 60000)
+
+autoUpdater.on('update-downloaded', (event, releaseNotes, releaseName) => {
+  dialog
+    .showMessageBox({
+      type: 'info',
+      buttons: ['Restart Now', 'Later'],
+      title: 'New update available',
+      message: process.platform === 'win32' ? releaseNotes : releaseName,
+      detail: 'A new version of Ollama is available. Restart to apply the update.',
+    })
+    .then(returnValue => {
+      if (returnValue.response === 0) autoUpdater.quitAndInstall()
+    })
+})