386 lines
13 KiB
JavaScript
386 lines
13 KiB
JavaScript
|
// 桌面端检查更新：拉取 /api/version/latest，应用内下载安装包并启动安装程序
|
|||
|
|
const { app, BrowserWindow, dialog, net, ipcMain, shell } = require('electron')
|
|||
|
|
const { spawn } = require('child_process')
|
|||
|
|
const path = require('path')
|
|||
|
|
const fs = require('fs')
|
|||
|
|
|
|||
|
|
const DOWNLOAD_TIMEOUT_MS = 15 * 60 * 1000
|
|||
|
|
|
|||
|
|
let deps = {
|
|||
|
|
getServerBase: () => '',
|
|||
|
|
getMainWindow: () => null,
|
|||
|
|
setQuitting: () => {},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let busy = false
|
|||
|
|
let abortDownload = null
|
|||
|
|
|
|||
|
|
function init(next) {
|
|||
|
|
deps = { ...deps, ...next }
|
|||
|
|
ipcMain.handle('nlg:check-update', (_e, silent) => checkUpdate(!!silent))
|
|||
|
|
ipcMain.handle('nlg:cancel-update-download', () => {
|
|||
|
|
if (abortDownload) abortDownload()
|
|||
|
|
return true
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function compareVer(a, b) {
|
|||
|
|
const pa = String(a || '0').split('.').map((n) => parseInt(n, 10) || 0)
|
|||
|
|
const pb = String(b || '0').split('.').map((n) => parseInt(n, 10) || 0)
|
|||
|
|
const len = Math.max(pa.length, pb.length)
|
|||
|
|
for (let i = 0; i < len; i++) {
|
|||
|
|
const va = pa[i] || 0
|
|||
|
|
const vb = pb[i] || 0
|
|||
|
|
if (va > vb) return 1
|
|||
|
|
if (va < vb) return -1
|
|||
|
|
}
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function resolveDownloadUrl(url) {
|
|||
|
|
const base = String(deps.getServerBase() || '').replace(/\/+$/, '')
|
|||
|
|
if (!url) return ''
|
|||
|
|
try {
|
|||
|
|
const u = new URL(url, base + '/')
|
|||
|
|
const m = String(u.pathname || '').match(/\/(?:api\/)?download\/([^/]+)$/i)
|
|||
|
|
if (m) {
|
|||
|
|
return base + '/api/download/' + encodeURIComponent(decodeURIComponent(m[1]))
|
|||
|
|
}
|
|||
|
|
return u.href
|
|||
|
|
} catch {
|
|||
|
|
return url
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function formatBytes(n) {
|
|||
|
|
n = Number(n) || 0
|
|||
|
|
if (n < 1024) return n + ' B'
|
|||
|
|
if (n < 1024 * 1024) return (n / 1024).toFixed(1) + ' KB'
|
|||
|
|
return (n / 1024 / 1024).toFixed(1) + ' MB'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function headerVal(headers, key) {
|
|||
|
|
const raw = headers[key] || headers[key.toLowerCase()] || headers[key.toUpperCase()]
|
|||
|
|
return Array.isArray(raw) ? raw[0] : (raw || '')
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function fetchLatestVersion() {
|
|||
|
|
return new Promise((resolve) => {
|
|||
|
|
const base = String(deps.getServerBase() || '').replace(/\/+$/, '')
|
|||
|
|
if (!base) return resolve({ ok: false, msg: '未配置服务器地址' })
|
|||
|
|
let settled = false
|
|||
|
|
const done = (v) => {
|
|||
|
|
if (settled) return
|
|||
|
|
settled = true
|
|||
|
|
resolve(v)
|
|||
|
|
}
|
|||
|
|
const req = net.request({
|
|||
|
|
method: 'GET',
|
|||
|
|
url: base + '/api/version/latest',
|
|||
|
|
redirect: 'follow',
|
|||
|
|
})
|
|||
|
|
let body = ''
|
|||
|
|
req.on('response', (resp) => {
|
|||
|
|
const code = resp.statusCode || 0
|
|||
|
|
if (code < 200 || code >= 300) {
|
|||
|
|
done({ ok: false, msg: '服务器返回 ' + code })
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
resp.on('data', (chunk) => { body += chunk.toString('utf8') })
|
|||
|
|
resp.on('end', () => {
|
|||
|
|
try {
|
|||
|
|
const parsed = JSON.parse(body)
|
|||
|
|
if (parsed && typeof parsed.code === 'number' && parsed.code !== 0) {
|
|||
|
|
done({ ok: false, msg: parsed.msg || '检查更新失败' })
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const data = parsed && parsed.data ? parsed.data : parsed
|
|||
|
|
done({ ok: true, data })
|
|||
|
|
} catch {
|
|||
|
|
done({ ok: false, msg: '解析版本信息失败' })
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
req.on('error', (err) => done({ ok: false, msg: '网络错误：' + err.message }))
|
|||
|
|
const timer = setTimeout(() => {
|
|||
|
|
try { req.abort() } catch { /* ignore */ }
|
|||
|
|
done({ ok: false, msg: '请求超时' })
|
|||
|
|
}, 10000)
|
|||
|
|
req.on('close', () => clearTimeout(timer))
|
|||
|
|
req.end()
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function downloadInstaller(url, dest, onProgress) {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
let settled = false
|
|||
|
|
const finish = (err, val) => {
|
|||
|
|
if (settled) return
|
|||
|
|
settled = true
|
|||
|
|
abortDownload = null
|
|||
|
|
if (err) reject(err)
|
|||
|
|
else resolve(val)
|
|||
|
|
}
|
|||
|
|
const req = net.request({ method: 'GET', url, redirect: 'follow' })
|
|||
|
|
abortDownload = () => {
|
|||
|
|
try { req.abort() } catch { /* ignore */ }
|
|||
|
|
const e = new Error('已取消')
|
|||
|
|
e.cancelled = true
|
|||
|
|
finish(e)
|
|||
|
|
}
|
|||
|
|
const timer = setTimeout(() => {
|
|||
|
|
try { req.abort() } catch { /* ignore */ }
|
|||
|
|
finish(new Error('下载超时，请检查网络后重试'))
|
|||
|
|
}, DOWNLOAD_TIMEOUT_MS)
|
|||
|
|
|
|||
|
|
req.on('response', (resp) => {
|
|||
|
|
const code = resp.statusCode || 0
|
|||
|
|
if (code < 200 || code >= 300) {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
finish(new Error('下载失败（HTTP ' + code + '）'))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const ct = String(headerVal(resp.headers, 'content-type'))
|
|||
|
|
if (/text\/html/i.test(ct)) {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
finish(new Error('服务器返回了网页而不是安装包。请确认后台已上传安装包。'))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const total = parseInt(headerVal(resp.headers, 'content-length'), 10) || 0
|
|||
|
|
fs.mkdirSync(path.dirname(dest), { recursive: true })
|
|||
|
|
const ws = fs.createWriteStream(dest)
|
|||
|
|
let loaded = 0
|
|||
|
|
resp.on('data', (chunk) => {
|
|||
|
|
loaded += chunk.length
|
|||
|
|
if (!ws.write(chunk) && typeof resp.pause === 'function') {
|
|||
|
|
resp.pause()
|
|||
|
|
ws.once('drain', () => { if (typeof resp.resume === 'function') resp.resume() })
|
|||
|
|
}
|
|||
|
|
if (onProgress) onProgress(loaded, total)
|
|||
|
|
})
|
|||
|
|
resp.on('end', () => { ws.end() })
|
|||
|
|
resp.on('error', (err) => {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
try { ws.destroy() } catch { /* ignore */ }
|
|||
|
|
finish(err)
|
|||
|
|
})
|
|||
|
|
ws.on('finish', () => {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
finish(null, { size: loaded, total })
|
|||
|
|
})
|
|||
|
|
ws.on('error', (err) => {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
finish(err)
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
req.on('error', (err) => {
|
|||
|
|
clearTimeout(timer)
|
|||
|
|
finish(err)
|
|||
|
|
})
|
|||
|
|
req.end()
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function verifyInstaller(filePath, expectSize) {
|
|||
|
|
let st
|
|||
|
|
try {
|
|||
|
|
st = fs.statSync(filePath)
|
|||
|
|
} catch {
|
|||
|
|
return '安装包不存在'
|
|||
|
|
}
|
|||
|
|
// 网页被当成 exe 保存时通常只有几 KB；正式安装包约 100MB
|
|||
|
|
if (st.size < 64 * 1024) {
|
|||
|
|
return '文件过小，不是完整安装包（常见原因：下载地址被网站首页顶替）'
|
|||
|
|
}
|
|||
|
|
if (expectSize && Math.abs(st.size - expectSize) > 2048) {
|
|||
|
|
return `文件大小不符（已下 ${formatBytes(st.size)}，期望 ${formatBytes(expectSize)}）`
|
|||
|
|
}
|
|||
|
|
const ext = path.extname(filePath).toLowerCase()
|
|||
|
|
if (ext === '.exe') {
|
|||
|
|
const buf = Buffer.alloc(2)
|
|||
|
|
const fd = fs.openSync(filePath, 'r')
|
|||
|
|
fs.readSync(fd, buf, 0, 2, 0)
|
|||
|
|
fs.closeSync(fd)
|
|||
|
|
if (buf[0] !== 0x4d || buf[1] !== 0x5a) {
|
|||
|
|
return '安装包损坏（不是有效的 Windows 程序）'
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return ''
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function launchInstaller(filePath) {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
if (process.platform === 'win32') {
|
|||
|
|
// start 脱离本进程，避免 app.quit() 把安装程序一起带走
|
|||
|
|
const cmd = process.env.ComSpec || 'cmd.exe'
|
|||
|
|
const child = spawn(cmd, ['/c', 'start', '', filePath], {
|
|||
|
|
detached: true,
|
|||
|
|
stdio: 'ignore',
|
|||
|
|
windowsHide: true,
|
|||
|
|
})
|
|||
|
|
child.once('error', reject)
|
|||
|
|
child.unref()
|
|||
|
|
setTimeout(resolve, 400)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
shell.openPath(filePath).then((err) => {
|
|||
|
|
if (err) reject(new Error(err))
|
|||
|
|
else resolve()
|
|||
|
|
}).catch(reject)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function openProgressWin(version) {
|
|||
|
|
const parent = deps.getMainWindow()
|
|||
|
|
const win = new BrowserWindow({
|
|||
|
|
width: 440,
|
|||
|
|
height: 230,
|
|||
|
|
parent: parent && !parent.isDestroyed() ? parent : undefined,
|
|||
|
|
modal: !!(parent && !parent.isDestroyed()),
|
|||
|
|
resizable: false,
|
|||
|
|
minimizable: false,
|
|||
|
|
maximizable: false,
|
|||
|
|
autoHideMenuBar: true,
|
|||
|
|
backgroundColor: '#12142a',
|
|||
|
|
title: '正在下载更新',
|
|||
|
|
webPreferences: {
|
|||
|
|
preload: path.join(__dirname, 'preload.cjs'),
|
|||
|
|
contextIsolation: true,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="UTF-8"><title>下载更新</title>
|
|||
|
|
<style>
|
|||
|
|
body{margin:0;background:#12142a;color:#e8e9f5;font:14px/1.6 "Microsoft YaHei",sans-serif;padding:22px 26px}
|
|||
|
|
h3{margin:0 0 10px;font-size:15px}
|
|||
|
|
.bar{height:8px;border-radius:99px;background:#262a52;overflow:hidden;margin:12px 0 8px}
|
|||
|
|
.bar i{display:block;height:100%;width:0;background:linear-gradient(135deg,#6d6ff0,#9b6df0);transition:width .15s}
|
|||
|
|
.msg{font-size:12px;color:#8d90b3;min-height:18px}
|
|||
|
|
.row{display:flex;justify-content:flex-end;margin-top:16px}
|
|||
|
|
button{border:0;border-radius:8px;padding:7px 18px;font-size:13px;cursor:pointer;background:#262a52;color:#c6c8e8}
|
|||
|
|
</style></head><body>
|
|||
|
|
<h3>正在下载 ${String(version || '').replace(/[<>]/g, '')}</h3>
|
|||
|
|
<div class="bar"><i id="bar"></i></div>
|
|||
|
|
<div class="msg" id="msg">正在连接服务器…</div>
|
|||
|
|
<div class="row"><button id="cancel">取消</button></div>
|
|||
|
|
<script>
|
|||
|
|
window.__setProgress = (pct, text) => {
|
|||
|
|
document.getElementById('bar').style.width = Math.max(0, Math.min(100, pct)) + '%'
|
|||
|
|
if (text) document.getElementById('msg').textContent = text
|
|||
|
|
}
|
|||
|
|
document.getElementById('cancel').onclick = async () => {
|
|||
|
|
try { await window.desktop.cancelUpdateDownload() } catch (e) {}
|
|||
|
|
window.close()
|
|||
|
|
}
|
|||
|
|
</script>
|
|||
|
|
</body></html>`
|
|||
|
|
win.loadURL('data:text/html;charset=utf-8,' + encodeURIComponent(html))
|
|||
|
|
return win
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function setProgressUI(win, pct, text) {
|
|||
|
|
if (!win || win.isDestroyed()) return
|
|||
|
|
const js = `window.__setProgress && window.__setProgress(${Number(pct) || 0}, ${JSON.stringify(text)})`
|
|||
|
|
win.webContents.executeJavaScript(js).catch(() => {})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function downloadAndInstall({ version, url, size, filename }) {
|
|||
|
|
const destName = (filename && path.basename(String(filename))) || ('PixelArcade-Setup-' + version + '.exe')
|
|||
|
|
const dest = path.join(app.getPath('temp'), destName)
|
|||
|
|
try { fs.unlinkSync(dest) } catch { /* ignore */ }
|
|||
|
|
|
|||
|
|
const win = openProgressWin(version)
|
|||
|
|
win.on('closed', () => {
|
|||
|
|
if (abortDownload) abortDownload()
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
setProgressUI(win, 0, '正在连接服务器…')
|
|||
|
|
const onProg = (loaded, total) => {
|
|||
|
|
const t = total || size || 0
|
|||
|
|
const pct = t ? Math.min(99, Math.round((loaded / t) * 100)) : 0
|
|||
|
|
setProgressUI(win, pct, `已下载 ${formatBytes(loaded)}${t ? ' / ' + formatBytes(t) : ''}`)
|
|||
|
|
}
|
|||
|
|
let r
|
|||
|
|
try {
|
|||
|
|
r = await downloadInstaller(url, dest, onProg)
|
|||
|
|
} catch (e) {
|
|||
|
|
if (e && e.cancelled) throw e
|
|||
|
|
const alt = String(url).replace('/api/download/', '/download/')
|
|||
|
|
if (alt === url) throw e
|
|||
|
|
setProgressUI(win, 0, '正在尝试备用下载地址…')
|
|||
|
|
r = await downloadInstaller(alt, dest, onProg)
|
|||
|
|
}
|
|||
|
|
setProgressUI(win, 99, '正在校验安装包…')
|
|||
|
|
const bad = verifyInstaller(dest, size || r.total || 0)
|
|||
|
|
if (bad) throw new Error(bad)
|
|||
|
|
setProgressUI(win, 100, '下载完成，即将启动安装程序…')
|
|||
|
|
await launchInstaller(dest)
|
|||
|
|
deps.setQuitting(true)
|
|||
|
|
app.quit()
|
|||
|
|
} catch (e) {
|
|||
|
|
try { if (!win.isDestroyed()) win.close() } catch { /* ignore */ }
|
|||
|
|
if (e && e.cancelled) return
|
|||
|
|
try { fs.unlinkSync(dest) } catch { /* ignore */ }
|
|||
|
|
const parent = deps.getMainWindow()
|
|||
|
|
const { response } = await dialog.showMessageBox(parent && !parent.isDestroyed() ? parent : undefined, {
|
|||
|
|
type: 'error',
|
|||
|
|
title: '下载更新失败',
|
|||
|
|
message: '无法下载或安装更新',
|
|||
|
|
detail: (e && e.message) || String(e),
|
|||
|
|
buttons: ['在浏览器中打开', '确定'],
|
|||
|
|
defaultId: 0,
|
|||
|
|
cancelId: 1,
|
|||
|
|
noLink: true,
|
|||
|
|
})
|
|||
|
|
if (response === 0 && url) shell.openExternal(url)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function checkUpdate(silent = false) {
|
|||
|
|
if (busy) return
|
|||
|
|
const local = app.getVersion()
|
|||
|
|
const parent = () => {
|
|||
|
|
const w = deps.getMainWindow()
|
|||
|
|
return w && !w.isDestroyed() ? w : undefined
|
|||
|
|
}
|
|||
|
|
const r = await fetchLatestVersion()
|
|||
|
|
if (!r.ok) {
|
|||
|
|
if (!silent) dialog.showErrorBox('检查更新失败', r.msg || '无法连接服务器')
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const { version, download_url, release_notes: notes, force, size, filename } = r.data || {}
|
|||
|
|
if (compareVer(version, local) <= 0) {
|
|||
|
|
if (!silent) {
|
|||
|
|
dialog.showMessageBox(parent(), {
|
|||
|
|
type: 'info',
|
|||
|
|
title: '已是最新版本',
|
|||
|
|
message: '当前已是最新版本',
|
|||
|
|
detail: `版本 ${local}`,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const url = resolveDownloadUrl(download_url)
|
|||
|
|
const buttons = url ? ['立即更新', force ? '确定' : '稍后提醒'] : [force ? '确定' : '稍后提醒']
|
|||
|
|
const { response } = await dialog.showMessageBox(parent(), {
|
|||
|
|
type: force ? 'warning' : 'info',
|
|||
|
|
title: '发现新版本',
|
|||
|
|
message: `发现新版本 ${version}`,
|
|||
|
|
detail: notes || '将下载安装包并启动安装程序，安装时请先关闭本应用。',
|
|||
|
|
buttons,
|
|||
|
|
defaultId: 0,
|
|||
|
|
cancelId: buttons.length - 1,
|
|||
|
|
noLink: true,
|
|||
|
|
})
|
|||
|
|
if (response === 0 && url) {
|
|||
|
|
busy = true
|
|||
|
|
try {
|
|||
|
|
await downloadAndInstall({ version, url, size, filename })
|
|||
|
|
} finally {
|
|||
|
|
busy = false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { init, checkUpdate, compareVer, resolveDownloadUrl }
|