70 lines
2.4 KiB
JavaScript
70 lines
2.4 KiB
JavaScript
|
// main/updater.js
|
|||
|
|
const { app, BrowserWindow, ipcMain } = require('electron')
|
|||
|
|
const { autoUpdater } = require('electron-updater')
|
|||
|
|
const log = require('electron-log')
|
|||
|
|
Object.assign(console, log.functions)
|
|||
|
|
|
|||
|
|
const updaterState = {
|
|||
|
|
updateInProgress: false,
|
|||
|
|
updateDownloaded: false,
|
|||
|
|
pendingNavs: [],
|
|||
|
|
}
|
|||
|
|
function flushPendingNavs() {
|
|||
|
|
const fns = updaterState.pendingNavs.slice()
|
|||
|
|
updaterState.pendingNavs = []
|
|||
|
|
for (const fn of fns) try { fn() } catch (e) { console.warn('[Nav defer err]', e) }
|
|||
|
|
}
|
|||
|
|
function broadcast(channel, payload) {
|
|||
|
|
BrowserWindow.getAllWindows().forEach(w => !w.isDestroyed() && w.webContents.send(channel, payload))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function registerUpdater() {
|
|||
|
|
autoUpdater.logger = log
|
|||
|
|
autoUpdater.logger.transports.file.level = 'info'
|
|||
|
|
autoUpdater.autoDownload = true
|
|||
|
|
autoUpdater.autoInstallOnAppQuit = false
|
|||
|
|
|
|||
|
|
autoUpdater.on('checking-for-update', () => console.log('[updater] checking...'))
|
|||
|
|
autoUpdater.on('update-available', (info) => {
|
|||
|
|
updaterState.updateInProgress = true
|
|||
|
|
console.log('[updater] available', info.version)
|
|||
|
|
broadcast('update:available', info)
|
|||
|
|
})
|
|||
|
|
autoUpdater.on('update-not-available', () => {
|
|||
|
|
console.log('[updater] not-available')
|
|||
|
|
updaterState.updateInProgress = false
|
|||
|
|
flushPendingNavs()
|
|||
|
|
})
|
|||
|
|
let lastSend = 0
|
|||
|
|
autoUpdater.on('download-progress', (p) => {
|
|||
|
|
updaterState.updateInProgress = true
|
|||
|
|
const now = Date.now()
|
|||
|
|
if (now - lastSend > 150) {
|
|||
|
|
lastSend = now
|
|||
|
|
broadcast('update:progress', p)
|
|||
|
|
}
|
|||
|
|
const win = BrowserWindow.getAllWindows()[0]
|
|||
|
|
if (win) win.setProgressBar(p.percent / 100)
|
|||
|
|
})
|
|||
|
|
autoUpdater.on('update-downloaded', (info) => {
|
|||
|
|
console.log('[updater] downloaded')
|
|||
|
|
updaterState.updateInProgress = true
|
|||
|
|
updaterState.updateDownloaded = true
|
|||
|
|
broadcast('update:downloaded', info)
|
|||
|
|
// 想要自动安装：autoUpdater.quitAndInstall(false, true)
|
|||
|
|
})
|
|||
|
|
autoUpdater.on('error', (err) => {
|
|||
|
|
console.error('[updater] error', err)
|
|||
|
|
broadcast('update:error', { message: String(err) })
|
|||
|
|
updaterState.updateInProgress = false
|
|||
|
|
flushPendingNavs()
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
ipcMain.handle('update:quitAndInstall', () => autoUpdater.quitAndInstall(false, true))
|
|||
|
|
ipcMain.handle('update:checkNow', () => autoUpdater.checkForUpdates())
|
|||
|
|
|
|||
|
|
autoUpdater.checkForUpdates()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { registerUpdater, updaterState }
|