main.mjs 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683
  1. import { app, BrowserWindow, ipcMain, Menu, Tray, nativeImage, shell, utilityProcess, powerMonitor, systemPreferences, dialog, safeStorage, net, session as electronSession } from 'electron'
  2. import { appendFileSync, mkdirSync, readFileSync, realpathSync } from 'node:fs'
  3. import { join, dirname } from 'node:path'
  4. import { homedir, hostname } from 'node:os'
  5. import { fileURLToPath, pathToFileURL } from 'node:url'
  6. import { DesktopServiceManager } from './service-manager.mjs'
  7. import { DesktopRunnerManager } from './runner-manager.mjs'
  8. import { DesktopPreferences } from './preferences.mjs'
  9. import { DesktopCredentials } from './credentials.mjs'
  10. import { synchronizeLocalService, stopLocalService, migrateLocalData } from './service-sync.mjs'
  11. import { resolveDataHome } from './data-home.mjs'
  12. import { DesktopEnvironmentHost, DesktopHostCore } from './environment-host.mjs'
  13. import { DesktopHostManager } from './host-manager.mjs'
  14. import { remoteRegistration, remoteSession, enrollDesktopHost, inspectServer, persistRemoteSession, sessionCookieDetails } from './remote-login.mjs'
  15. import { DesktopLoginSessions } from './login-sessions.mjs'
  16. import { DesktopOnboarding } from './onboarding.mjs'
  17. import { DesktopUpdateManager } from './update-manager.mjs'
  18. import { DesktopAutoUpdateManager } from './auto-update-manager.mjs'
  19. import { createRequire } from 'node:module'
  20. import { installComputerViewer } from './computer-viewer.mjs'
  21. import { desktopPlatform, desktopMenu, loginItemOptions, clientOnlyRuntime } from './platform.mjs'
  22. const desktopRoot = dirname(fileURLToPath(import.meta.url))
  23. const fixtureHome = process.env.HERMES_YAOYAO_DESKTOP_TEST_HOME
  24. // Packaged: the runtime sits in Resources/runtime; dev: staged build roots.
  25. const clientOnly = app.isPackaged
  26. ? clientOnlyRuntime(join(process.resourcesPath, 'runtime'))
  27. : clientOnlyRuntime(join(app.getAppPath(), '.desktop-build-client'))
  28. const platform = desktopPlatform(process.platform, { clientOnly })
  29. const loginOptions = loginItemOptions()
  30. // Fixture identity also isolates Chromium cookies, window state and the app lock.
  31. // The shell package name distinguishes the variants: the client keeps the
  32. // historical 夭夭 identity (userData, logs, single-instance lock) so upgrades
  33. // stay in place, while the full dual-mode build uses its own 夭夭完整版 home.
  34. const productName = JSON.parse(readFileSync(join(desktopRoot, 'package.json'), 'utf8')).productName || '夭夭'
  35. if (fixtureHome) app.setPath('userData', join(fixtureHome, 'desktop'))
  36. app.setName(productName)
  37. if (platform.windows) app.setAppUserModelId('cn.samien.yaoyao.desktop')
  38. if (!app.requestSingleInstanceLock()) { app.quit() }
  39. else {
  40. let window, tray, manager, runnerManager, quitting = false, closing = false, timer
  41. let updateWindow, updater, environmentHost, hostManager, onboarding
  42. let remoteMode = false, remoteURL = '', serverModeActive = false, switchingMode = false
  43. let recoveringService = false
  44. let resumeLocalAfterUpdate = false
  45. let checkRemoteAuthorization = async () => {}
  46. const serviceURL = () => remoteMode ? remoteURL : manager?.state.url
  47. // Use the same Chromium networking as server detection/navigation. Auth owns
  48. // its cookie jar: do not send or overwrite another account's browser cookies.
  49. const authFetch = (url, options) => net.fetch(url, { ...options, credentials: 'omit' })
  50. const serviceOrigins = () => {
  51. try { return remoteMode && remoteURL ? [new URL(remoteURL).origin] : manager?.state.url ? [new URL(manager.state.url).origin] : [] }
  52. catch { return [] }
  53. }
  54. const closeComputerViewer = installComputerViewer({ owner: () => window, origin: () => serviceURL(),
  55. preload: join(desktopRoot, 'preload.cjs'), quitting: () => quitting })
  56. app.on('will-quit', closeComputerViewer)
  57. const updateURL = pathToFileURL(join(desktopRoot, 'update.html')).href
  58. const bootURL = pathToFileURL(join(desktopRoot, 'boot.html')).href
  59. const root = app.isPackaged ? join(process.resourcesPath, 'runtime')
  60. : join(app.getAppPath(), clientOnly ? '.desktop-build-client' : '.desktop-build')
  61. const logRoot = join(app.getPath('logs'), fixtureHome ? 'verification' : 'service')
  62. mkdirSync(logRoot, { recursive: true })
  63. const logFile = join(logRoot, 'server.log')
  64. const log = text => appendFileSync(logFile, `${new Date().toISOString()} ${text.trim()}\n`, { mode: 0o600 })
  65. const home = fixtureHome || resolveDataHome(process.env.YAOYAO_HOME || process.env.HERMES_YAOYAO_HOME)
  66. const preferences = new DesktopPreferences(home)
  67. let startHidden = false
  68. const port = Number(process.env.HERMES_YAOYAO_DESKTOP_PORT || 15300)
  69. if (!Number.isInteger(port) || port < 1 || port > 65535) throw new Error('桌面服务端口无效')
  70. const trustedBoot = event => !closing && !quitting && event.sender === window?.webContents
  71. && event.senderFrame === window.webContents.mainFrame && event.senderFrame.url === bootURL
  72. ipcMain.handle('desktop:status', event => {
  73. if (!trustedBoot(event)) throw new Error('不允许此页面访问桌面服务')
  74. return onboarding.snapshot()
  75. })
  76. ipcMain.handle('desktop:retry', async event => {
  77. if (!trustedBoot(event)) throw new Error('不允许此页面启动桌面服务')
  78. if (!platform.server) throw new Error('此客户端不支持在本机运行服务器')
  79. manager.restarts = []
  80. return onboarding.prepare()
  81. })
  82. ipcMain.handle('desktop:logs', event => {
  83. if (!trustedBoot(event)) throw new Error('不允许此页面读取日志')
  84. shell.showItemInFolder(logFile)
  85. })
  86. ipcMain.handle('desktop:force-sync', async event => {
  87. if (!trustedBoot(event) || event.senderFrame !== window.webContents.mainFrame || closing || quitting || !manager?.canForceSynchronization)
  88. throw new Error('不允许此页面覆盖本机 Web')
  89. return onboarding.prepare({ force: true })
  90. })
  91. for (const action of ['select', 'prepare', 'submit']) {
  92. ipcMain.handle(`desktop:onboarding-${action}`, (event, input) => {
  93. if (!trustedBoot(event)) throw new Error('不允许此页面操作安装与登录')
  94. return onboarding[action](input)
  95. })
  96. }
  97. const trustedUpdate = event => event.sender === updateWindow?.webContents
  98. && event.senderFrame === updateWindow.webContents.mainFrame && event.senderFrame.url === updateURL
  99. ipcMain.handle('desktop:updates', async event => {
  100. let trusted = false
  101. try {
  102. trusted = event.sender === window?.webContents && event.senderFrame === window.webContents.mainFrame
  103. && serviceOrigins().includes(new URL(event.senderFrame.url).origin)
  104. } catch { /* Only the active server's main page may open the local updater. */ }
  105. if (!trusted || closing || quitting || !updater) throw new Error('不允许此页面打开 App 更新')
  106. await showUpdates()
  107. })
  108. for (const action of ['state', 'check', 'download', 'cancel', 'install', 'open', 'folder', 'release']) {
  109. ipcMain.handle(`desktop-update:${action}`, async event => {
  110. if (!trustedUpdate(event)) throw new Error('不允许此页面操作 App 更新')
  111. if (action === 'state') return updater.snapshot()
  112. if (closing || quitting) throw new Error('App 正在重启或退出')
  113. if (action === 'check') return updater.check()
  114. if (action === 'download') return updater.download()
  115. if (action === 'install') return updater.install()
  116. if (action === 'cancel') { updater.cancel(); return updater.snapshot() }
  117. if (action === 'release') { safeExternal(updater.snapshot().releasePageUrl || 'https://github.com/Lsamien/yaoyao/releases'); return }
  118. const path = await updater.verifiedFile()
  119. if (closing || quitting || !trustedUpdate(event)) return
  120. if (action === 'folder') shell.showItemInFolder(path)
  121. else { const error = await shell.openPath(path); if (error) throw new Error(error) }
  122. })
  123. }
  124. async function showUpdates() {
  125. if (closing || quitting) return
  126. if (updateWindow && !updateWindow.isDestroyed()) { updateWindow.show(); updateWindow.focus(); return }
  127. updateWindow = new BrowserWindow({ title: 'App 更新 · 夭夭', width: 640, height: 620, minWidth: 360, minHeight: 440, show: false,
  128. webPreferences: { contextIsolation: true, nodeIntegration: false, sandbox: true, webSecurity: true,
  129. preload: join(desktopRoot, 'update-preload.cjs') } })
  130. updateWindow.webContents.setWindowOpenHandler(() => ({ action: 'deny' }))
  131. updateWindow.webContents.on('will-navigate', event => event.preventDefault())
  132. updateWindow.webContents.on('will-redirect', event => event.preventDefault())
  133. updateWindow.webContents.on('will-attach-webview', event => event.preventDefault())
  134. updateWindow.on('closed', () => { updateWindow = undefined })
  135. updateWindow.once('ready-to-show', () => updateWindow?.show())
  136. await updateWindow.loadURL(updateURL)
  137. }
  138. function show() {
  139. if (closing || quitting) return
  140. if (!window || window.isDestroyed()) createWindow()
  141. window.show(); window.focus()
  142. if (serverModeActive && manager) void manager.reconnect().then(() => stateChanged(manager.state)).catch(error => log(error.message))
  143. }
  144. function safeExternal(url) {
  145. try { const parsed = new URL(url); if (['https:', 'http:', 'mailto:'].includes(parsed.protocol)) void shell.openExternal(url) }
  146. catch { /* No arbitrary file, shell or custom URL dispatch. */ }
  147. }
  148. function createWindow() {
  149. window = new BrowserWindow({ width: 1280, height: 860, minWidth: 720, minHeight: 560, show: false,
  150. title: '夭夭', backgroundColor: '#f7f7f7',
  151. webPreferences: { contextIsolation: true, nodeIntegration: false, sandbox: true,
  152. webSecurity: true, preload: join(desktopRoot, 'preload.cjs') } })
  153. window.webContents.setWindowOpenHandler(({ url }) => {
  154. try {
  155. if (serviceOrigins().includes(new URL(url).origin)) void window.loadURL(url)
  156. else safeExternal(url)
  157. } catch { /* Invalid navigation is denied. */ }
  158. return { action: 'deny' }
  159. })
  160. const guardNavigation = (event, url) => {
  161. try { if (url === bootURL || serviceOrigins().includes(new URL(url).origin)) return }
  162. catch { /* Invalid navigation is denied. */ }
  163. event.preventDefault(); safeExternal(url)
  164. }
  165. window.webContents.on('will-navigate', guardNavigation)
  166. window.webContents.on('will-redirect', guardNavigation)
  167. window.webContents.on('will-attach-webview', event => event.preventDefault())
  168. const trustedService = (contents, url) => {
  169. try { return contents === window?.webContents && serviceOrigins().includes(new URL(url).origin) }
  170. catch { return false }
  171. }
  172. window.webContents.session.setPermissionCheckHandler((contents, permission, origin) =>
  173. closeComputerViewer.owns(contents) ? ['fullscreen', 'clipboard-sanitized-write'].includes(permission)
  174. : trustedService(contents, origin) && ['media', 'notifications', 'clipboard-sanitized-write'].includes(permission))
  175. window.webContents.session.setPermissionRequestHandler((contents, permission, callback, details) => {
  176. if (closeComputerViewer.owns(contents)) { callback(['fullscreen', 'clipboard-sanitized-write'].includes(permission)); return }
  177. if (!trustedService(contents, details.requestingUrl)) { callback(false); return }
  178. if (permission === 'media' && details.mediaTypes?.length && details.mediaTypes.every(type => type === 'audio')) {
  179. if (platform.windows) callback(systemPreferences.getMediaAccessStatus('microphone') !== 'denied')
  180. else void systemPreferences.askForMediaAccess('microphone').then(callback, () => callback(false))
  181. } else callback(['notifications', 'clipboard-sanitized-write', 'fullscreen'].includes(permission))
  182. })
  183. window.on('close', event => { if (!quitting) { event.preventDefault(); window.hide() } })
  184. window.once('ready-to-show', () => { if (!startHidden) window.show() })
  185. if (onboarding?.state.active) return window.loadURL(bootURL)
  186. if (remoteMode && remoteURL) return window.loadURL(remoteURL)
  187. if (manager?.state.phase === 'ready') return window.loadURL(manager.state.url)
  188. return window.loadFile(join(desktopRoot, 'boot.html'))
  189. }
  190. function stateChanged(state) {
  191. log(`${state.phase}: ${state.message}`)
  192. onboarding?.serviceChanged(state)
  193. // Quit owns the window from this point on. A detached connection is not a
  194. // stopped server and must never navigate the closing window to boot.html.
  195. if (closing || quitting || remoteMode) return
  196. const serviceStatus=Menu.getApplicationMenu()?.getMenuItemById('desktop-service-status')
  197. if(serviceStatus)serviceStatus.label=state.phase==='ready'?'后台服务运行中':state.message
  198. const localVmMenu=Menu.getApplicationMenu()?.getMenuItemById('desktop-local-vm')
  199. if(localVmMenu)localVmMenu.enabled=state.phase==='ready'
  200. const notice = Menu.getApplicationMenu()?.getMenuItemById('desktop-sync-notice')
  201. if (notice) notice.visible = Boolean(state.updateNotice)
  202. const retrySync = Menu.getApplicationMenu()?.getMenuItemById('desktop-sync-retry')
  203. if (retrySync) retrySync.enabled = state.phase === 'ready' && Boolean(manager?.options.synchronize)
  204. const forceSync = Menu.getApplicationMenu()?.getMenuItemById('desktop-sync-force')
  205. if (forceSync) forceSync.enabled = !closing && !quitting && manager?.canForceSynchronization === true
  206. tray?.setToolTip(`夭夭 · ${state.message}`)
  207. if (!window || window.isDestroyed()) return
  208. if (onboarding?.state.active) {
  209. if (recoveringService && state.phase === 'ready') {
  210. recoveringService = false
  211. if (!onboarding.busy) void onboarding.prepare({ autoEnter: true }).catch(error => log(error.message))
  212. }
  213. return
  214. }
  215. if (state.phase === 'ready') {
  216. const current = window.webContents.getURL()
  217. if (!current.startsWith(`${state.url}/`) && current !== state.url) void window.loadURL(state.url)
  218. } else if (window.webContents.getURL() !== bootURL) {
  219. recoveringService = true
  220. onboarding.open({ mode: 'local', remember: preferences.value.startupChoice === 'local', restoring: true })
  221. onboarding.serviceChanged(state)
  222. void window.loadURL(bootURL)
  223. }
  224. }
  225. app.on('second-instance', show)
  226. app.on('activate', show)
  227. app.on('window-all-closed', () => {})
  228. function requestQuit(stopBackground = false) {
  229. if (closing || quitting) return
  230. const resumeLocal = serverModeActive && manager?.activationRequested === true
  231. closing = true; clearInterval(timer)
  232. updater?.stopChecking?.(); updater?.cancel(); updateWindow?.close()
  233. window?.hide()
  234. // Native Cmd+Q can enter before-quit from Cocoa's termination callback.
  235. // Both cleanup and the final quit must run after that callback unwinds.
  236. setImmediate(() => { void (async()=>{
  237. try {
  238. await environmentHost?.close()
  239. await hostManager?.stop()
  240. await runnerManager?.stop()
  241. if (stopBackground) await manager?.stopBackground()
  242. else await manager?.stop()
  243. quitting = true
  244. setImmediate(() => app.quit())
  245. } catch(error) {
  246. log(error.message)
  247. window?.show()
  248. const choice=await dialog.showMessageBox(window,{type:'warning',message:stopBackground?'后台服务停止未完成':'App 退出清理未完成',detail:error.message,buttons:['返回应用','仅退出 App'],defaultId:0,cancelId:0})
  249. if(choice.response===1){quitting=true;setImmediate(()=>app.quit())}
  250. else {
  251. closing=false
  252. if(remoteMode)await hostManager?.start().catch(error=>log(error.message))
  253. else if(serverModeActive){
  254. await manager?.reconnect().catch(error=>log(error.message))
  255. if(resumeLocal){
  256. await manager?.activate().catch(error=>log(error.message))
  257. environmentHost?.start()
  258. await runnerManager?.start().catch(error=>log(error.message))
  259. }
  260. }
  261. if (manager) stateChanged(manager.state)
  262. timer=setInterval(()=>{if(serverModeActive)void manager.check();else void checkRemoteAuthorization()},10000);timer.unref()
  263. updater?.startChecking?.()
  264. show()
  265. }
  266. }
  267. })() })
  268. }
  269. async function prepareUpdateRestart() {
  270. if (closing || quitting) throw new Error('App 正在退出,请稍后重试')
  271. resumeLocalAfterUpdate = serverModeActive && manager?.activationRequested === true
  272. closing = true; clearInterval(timer)
  273. await environmentHost?.close()
  274. await hostManager?.stop()
  275. await runnerManager?.stop()
  276. await manager?.stop()
  277. // Cleanup has completed before Squirrel begins quitting. Releasing the
  278. // single-instance lock lets the newly installed app start immediately.
  279. quitting = true
  280. app.releaseSingleInstanceLock()
  281. }
  282. async function recoverUpdateRestart() {
  283. quitting = false; closing = false
  284. if (!app.requestSingleInstanceLock()) { quitting = true; app.quit(); return }
  285. if (remoteMode) await hostManager?.start().catch(error => log(error.message))
  286. else if (serverModeActive) {
  287. await manager?.reconnect().catch(error => log(error.message))
  288. if (resumeLocalAfterUpdate) {
  289. await manager?.activate().catch(error => log(error.message))
  290. environmentHost?.start()
  291. await runnerManager?.start().catch(error => log(error.message))
  292. }
  293. }
  294. clearInterval(timer)
  295. timer = setInterval(() => { if (serverModeActive) void manager?.check(); else void checkRemoteAuthorization() }, 10000); timer.unref()
  296. window?.show(); updateWindow?.show()
  297. }
  298. app.on('before-quit', event => {
  299. if (quitting) return
  300. event.preventDefault()
  301. requestQuit()
  302. })
  303. app.whenReady().then(async () => {
  304. await preferences.load()
  305. startHidden = preferences.value.backgroundAtLogin && (platform.windows
  306. ? process.argv.includes('--yaoyao-login') : app.getLoginItemSettings().wasOpenedAtLogin)
  307. const packageVersion = JSON.parse(readFileSync(join(root, 'release.json'), 'utf8')).webVersion
  308. const buildInfo=JSON.parse(readFileSync(join(root,'build-info.json'),'utf8'))
  309. const releases = await import(pathToFileURL(join(root, 'github-release.mjs')).href)
  310. if ((app.isPackaged && !fixtureHome) || (fixtureHome && process.env.HERMES_YAOYAO_DESKTOP_TEST_AUTO_UPDATE === '1')) {
  311. const updateModule = createRequire(import.meta.url)(join(root, 'electron-updater.cjs'))
  312. const autoUpdater = platform.windows ? new updateModule.NsisUpdater() : updateModule.autoUpdater
  313. autoUpdater.logger = Object.fromEntries(['info', 'warn', 'error', 'debug'].map(level => [level, (...values) => log(`[更新:${level}] ${values.join(' ')}`)]))
  314. updater = new DesktopAutoUpdateManager({ driver: autoUpdater, version: packageVersion, platform: platform.platform,
  315. prepareInstall: prepareUpdateRestart, recoverInstall: recoverUpdateRestart })
  316. updater.startChecking()
  317. } else {
  318. updater = new DesktopUpdateManager({ version: packageVersion, cacheRoot: join(home, 'updates', 'desktop-downloads'),
  319. source: releases.DEFAULT_RELEASE_SOURCE, inspect: releases.inspectGitHubRelease, compare: releases.compareReleaseVersions,
  320. fetchImpl: (...args) => net.fetch(...args) })
  321. }
  322. app.setAboutPanelOptions({applicationName:'夭夭',applicationVersion:packageVersion,version:`${String(buildInfo.commit).slice(0,12)}${buildInfo.dirty?' · 工作区快照':''}`})
  323. const credentials=new DesktopCredentials({home,platform:platform.platform,safeStorage,helper:join(root,app.isPackaged?'keychain-helper':'keychain-helper-dev'),legacyDecrypt:async bytes=>(await safeStorage.decryptStringAsync(bytes)).result})
  324. const loginSessions = new DesktopLoginSessions({ home, encrypt: value => credentials.encrypt(value), decrypt: value => credentials.decrypt(value) })
  325. async function installSessionCookies(server, cookies) {
  326. for (const cookie of cookies) await electronSession.defaultSession.cookies.set({ url: server, ...cookie,
  327. httpOnly: true, secure: cookie.secure || server.startsWith('https://') })
  328. await electronSession.defaultSession.cookies.flushStore()
  329. }
  330. // Serialize vault + Chromium changes so an in-flight restore cannot undo logout.
  331. let loginOperation = Promise.resolve()
  332. function synchronizeLogin(run) {
  333. const next = loginOperation.then(run)
  334. loginOperation = next.catch(() => {})
  335. return next
  336. }
  337. async function clearLogin(server) {
  338. await loginSessions.forget(server)
  339. await electronSession.defaultSession.cookies.remove(server, 'hermes_yaoyao_session')
  340. await electronSession.defaultSession.cookies.flushStore()
  341. }
  342. function forgetLogin(server) { return synchronizeLogin(() => clearLogin(server)) }
  343. async function inspectRemote(server) {
  344. return synchronizeLogin(async () => {
  345. const saved = await loginSessions.read(server)
  346. const info = await inspectServer(server, saved
  347. ? (url, options) => authFetch(url, { ...options, headers: { ...options.headers, cookie: `${saved.cookie.name}=${saved.cookie.value}` } })
  348. : (...args) => net.fetch(...args))
  349. if (!info.authenticated) {
  350. if (saved) await clearLogin(server)
  351. return info
  352. }
  353. if (saved) {
  354. await installSessionCookies(server, [{ ...saved.cookie, expirationDate: Date.now() / 1000 + 400 * 86400 }])
  355. } else {
  356. const cookies = await electronSession.defaultSession.cookies.get({ url: server })
  357. const persistent = await persistRemoteSession(server, info, cookies, authFetch)
  358. await loginSessions.save(server, persistent)
  359. await installSessionCookies(server, persistent)
  360. }
  361. return info
  362. })
  363. }
  364. if (platform.server) {
  365. manager = new DesktopServiceManager({ home, port, version: packageVersion, log, onState: stateChanged,
  366. prepareHome: !fixtureHome ? onProgress => migrateLocalData({ home, port, root, onProgress }) : undefined,
  367. stopBackground: (app.isPackaged && !fixtureHome) || (fixtureHome && process.env.HERMES_YAOYAO_DESKTOP_TEST_SYNC === '1')
  368. ? onProgress => stopLocalService({home,port,root,fixture:Boolean(fixtureHome),onProgress}) : undefined,
  369. synchronize: (app.isPackaged && !fixtureHome) || (fixtureHome && process.env.HERMES_YAOYAO_DESKTOP_TEST_SYNC === '1')
  370. ? (onProgress, { force = false } = {}) => synchronizeLocalService({ home, port, root, fixture: Boolean(fixtureHome), onProgress, force,
  371. environment: { HERMES_YAOYAO_UPSTREAM: process.env.HERMES_YAOYAO_UPSTREAM || 'http://127.0.0.1:9119', HERMES_YAOYAO_SUPERVISE_DASHBOARD: fixtureHome ? '0' : '1', HERMES_YAOYAO_DEFER_HERMES: '1' } })
  372. : undefined,
  373. fork: ({ home, port }) => {
  374. const env = { ...process.env, NODE_ENV: 'production', NODE_USE_ENV_PROXY: '0',
  375. PATH: [join(homedir(), '.local/bin'), '/opt/homebrew/bin', '/usr/local/bin', process.env.PATH || '/usr/bin:/bin'].join(':'),
  376. HERMES_YAOYAO_DESKTOP: '1', HERMES_YAOYAO_HOME: home, HERMES_YAOYAO_HOST: '127.0.0.1',
  377. HERMES_YAOYAO_PORT: String(port), HERMES_YAOYAO_STATIC_DIR: join(root, 'ui'),
  378. HERMES_YAOYAO_SUPERVISE_DASHBOARD: fixtureHome ? '0' : '1',
  379. HERMES_YAOYAO_DEFER_HERMES: '1',
  380. HERMES_YAOYAO_UPSTREAM: process.env.HERMES_YAOYAO_UPSTREAM || 'http://127.0.0.1:9119' }
  381. delete env.ELECTRON_RUN_AS_NODE
  382. delete env.HERMES_YAOYAO_TLS_CERT; delete env.HERMES_YAOYAO_TLS_KEY
  383. return utilityProcess.fork(join(root, 'server.mjs'), [], { cwd: root, env, stdio: 'pipe' })
  384. },
  385. })
  386. environmentHost=new DesktopEnvironmentHost({manager,root,dataRoot:app.getPath('userData')})
  387. const { parseRunnerConfiguration } = await import(pathToFileURL(join(root,'runner-config.mjs')).href)
  388. runnerManager = new DesktopRunnerManager({home,validate:parseRunnerConfiguration,
  389. encrypt:value=>credentials.encrypt(value),
  390. decrypt:bytes=>credentials.decrypt(bytes),
  391. onState:message=>{log(`Runner: ${message}`);const item=Menu.getApplicationMenu()?.getMenuItemById('runner-status');if(item)item.label=message},
  392. fork:()=>utilityProcess.fork(join(root,'runner.mjs'),['--desktop-ipc'],{cwd:root,stdio:'pipe',env:{HOME:homedir(),PATH:[join(homedir(),'.local/bin'),join(homedir(),'.orbstack/bin'),'/opt/homebrew/bin','/usr/local/bin',process.env.PATH||'/usr/bin:/bin'].join(':'),NODE_ENV:'production',NODE_USE_ENV_PROXY:'0'}}),
  393. })
  394. }
  395. async function runnerAction(action){try{await action()}catch(error){await dialog.showMessageBox(window,{type:'error',message:'执行节点操作未完成',detail:error.message})}}
  396. hostManager=new DesktopHostManager({home,root,dataRoot:app.getPath('userData'),
  397. // Device polling must share the Chromium transport used to log in to LAN servers.
  398. fetchImpl:authFetch,
  399. createCore:({root,dataRoot})=>new DesktopHostCore({root,dataRoot}),
  400. encrypt:value=>credentials.encrypt(value),
  401. decrypt:bytes=>credentials.decrypt(bytes),
  402. onState:message=>{log(`电脑: ${message}`);const item=Menu.getApplicationMenu()?.getMenuItemById('desktop-host-status');if(item)item.label=message}})
  403. ipcMain.handle('desktop:device-host', async event => {
  404. try {
  405. const trusted = event.sender === window?.webContents && event.senderFrame === event.sender.mainFrame
  406. && serviceOrigins().includes(new URL(event.senderFrame.url).origin)
  407. if (!trusted || closing || quitting) throw new Error('不允许此页面读取设备身份')
  408. } catch (error) {
  409. if (error.message === '不允许此页面读取设备身份') throw error
  410. throw new Error('不允许此页面读取设备身份')
  411. }
  412. if (remoteMode) {
  413. const enrolled = await hostManager.read().catch(() => undefined)
  414. return { deviceHost: enrolled && new URL(enrolled.serverURL).origin === new URL(remoteURL).origin ? enrolled.hostId : null }
  415. }
  416. return { deviceHost: 'local' }
  417. })
  418. async function hostAction(action){try{await action()}catch(error){await dialog.showMessageBox(window,{type:'error',message:'电脑操作未完成',detail:error.message})}}
  419. const appDialog = options => window && !window.isDestroyed() ? dialog.showMessageBox(window, options) : dialog.showMessageBox(options)
  420. let authorizedOnboardingServer = ''
  421. async function authorizeComputer(server,session){
  422. const previous=await hostManager.read().catch(()=>undefined)
  423. const previousHostId=previous&&new URL(previous.serverURL).origin===new URL(server).origin?previous.hostId:undefined
  424. const registration=await enrollDesktopHost(session,{name:hostname(),installId:await hostManager.installIdentity(),previousHostId},authFetch)
  425. await hostManager.adopt({protocol:1,serverURL:server,hostId:registration.host.id,token:registration.token,...(server.startsWith('http://')?{allowInsecureLan:true}:{})})
  426. return registration
  427. }
  428. function requireModePage(event){
  429. let trusted=false
  430. try{trusted=event.sender===window?.webContents&&event.senderFrame===window.webContents.mainFrame
  431. &&serviceOrigins().includes(new URL(event.senderFrame.url).origin)}catch{}
  432. if(!trusted||closing||quitting)throw new Error('不允许此页面切换桌面运行模式')
  433. }
  434. ipcMain.handle('desktop:mode-state',event=>{
  435. requireModePage(event)
  436. return {mode:remoteMode||!platform.server?'client':'server',serverURL:serviceURL()||'',switching:switchingMode,
  437. platform:platform.platform,supportedModes:platform.supportedModes}
  438. })
  439. ipcMain.handle('desktop:mode-switch',async(event,mode)=>{
  440. requireModePage(event)
  441. if(mode!=='client'&&mode!=='server')throw new Error('运行模式无效')
  442. return switchMode(mode==='client'?'remote':'local')
  443. })
  444. ipcMain.handle('desktop:remote-login',event=>{
  445. requireModePage(event)
  446. if(switchingMode)throw new Error('正在切换运行模式,请稍候')
  447. return openRemoteLogin()
  448. })
  449. ipcMain.handle('desktop:computer-authorize',async(event,csrfToken)=>{
  450. requireModePage(event)
  451. if(!remoteMode||!remoteURL)return {registered:false}
  452. try{
  453. const token=String(csrfToken??'').trim()
  454. if(!token)throw new Error('登录状态缺少安全令牌,请重新登录')
  455. const cookies=await electronSession.defaultSession.cookies.get({url:remoteURL})
  456. const registration=await authorizeComputer(remoteURL,{origin:remoteURL,csrfToken:token,cookies:cookies.map(cookie=>[cookie.name,cookie.value])})
  457. return {registered:true,hostId:registration.host.id}
  458. }catch(error){
  459. const message=error instanceof Error?error.message:'电脑授权恢复失败'
  460. hostManager.publish(`电脑授权未恢复:${message}`)
  461. return {registered:false}
  462. }
  463. })
  464. async function importDesktopHost(){
  465. const selection=window&&!window.isDestroyed()
  466. ?await dialog.showOpenDialog(window,{title:'导入电脑配置',properties:['openFile'],filters:[{name:'电脑配置',extensions:['json']}]})
  467. :await dialog.showOpenDialog({title:'导入电脑配置',properties:['openFile'],filters:[{name:'电脑配置',extensions:['json']}]})
  468. if(!selection.canceled&&selection.filePaths[0]&&!closing)await hostAction(()=>hostManager.importFile(selection.filePaths[0]))
  469. }
  470. if (!preferences.value.remoteServer) {
  471. const enrolled = await hostManager.read().catch(() => undefined)
  472. if (enrolled?.serverURL) await preferences.setRemoteServer(enrolled.serverURL)
  473. }
  474. onboarding = new DesktopOnboarding({
  475. platform: platform.platform, supportedModes: platform.supportedModes,
  476. remoteServer: () => preferences.value.remoteServer,
  477. inspect: inspectRemote,
  478. localSession: async server => {
  479. const session = await manager.adminSession()
  480. await installSessionCookies(server, sessionCookieDetails(session.setCookies))
  481. return { authenticated: true, user: session.user }
  482. },
  483. prepareLocal: async ({ force }) => {
  484. if (!platform.server) throw new Error('此客户端不支持在本机运行服务器')
  485. serverModeActive = true; remoteMode = false; remoteURL = ''
  486. await hostManager.stop()
  487. manager.restarts = []
  488. if (force) await manager.retrySynchronization({ force: true })
  489. else await manager.start()
  490. if (manager.state.phase !== 'ready') throw new Error(manager.state.message)
  491. return manager.state.url
  492. },
  493. register: ({ serverURL, username, password }) => remoteRegistration(serverURL, { username, password }, authFetch),
  494. authenticate: async ({ mode, serverURL, setup, username, password }) => {
  495. authorizedOnboardingServer = ''
  496. const auth = await remoteSession(serverURL, { setup, username, password }, authFetch)
  497. if (mode === 'remote') await loginSessions.save(serverURL, auth.cookieDetails)
  498. await installSessionCookies(serverURL, auth.cookieDetails)
  499. let warning
  500. if (mode === 'remote' && auth.user.role === 'admin') {
  501. try { await authorizeComputer(serverURL, auth); authorizedOnboardingServer = serverURL }
  502. catch (error) { warning = `账号已登录,但电脑授权未恢复:${error.message}。可先进入夭夭,稍后在“电脑”菜单重新授权。` }
  503. }
  504. return { username: auth.user.username, warning }
  505. },
  506. activate: async ({ mode, serverURL, remember }) => {
  507. if (closing || quitting) throw new Error('App 正在退出')
  508. const alreadyAuthorized = authorizedOnboardingServer === serverURL
  509. authorizedOnboardingServer = ''
  510. if (mode === 'remote') {
  511. // Detach only; an independently installed local service keeps running.
  512. serverModeActive = false
  513. await environmentHost?.close()
  514. await runnerManager?.stop()
  515. await manager?.stop()
  516. remoteMode = true; remoteURL = serverURL
  517. await preferences.setRemoteServer(serverURL)
  518. if (!alreadyAuthorized) {
  519. // A remembered browser session skips password authentication. Restore
  520. // its device authorization only on entry, never during detection.
  521. try {
  522. const current = await inspectServer(serverURL, (...args) => net.fetch(...args))
  523. if (current.authenticated && current.user?.role === 'admin') {
  524. const cookies = await electronSession.defaultSession.cookies.get({ url: serverURL })
  525. await authorizeComputer(serverURL, { origin: serverURL, csrfToken: current.csrfToken,
  526. cookies: cookies.map(cookie => [cookie.name, cookie.value]) })
  527. }
  528. } catch (error) { hostManager.publish(`电脑授权未恢复:${error.message}`) }
  529. }
  530. const enrolled = await hostManager.read().catch(error => { log(error.message); return undefined })
  531. if (enrolled && new URL(enrolled.serverURL).origin === new URL(serverURL).origin)
  532. await hostManager.start().catch(error => hostManager.publish(error.message))
  533. } else {
  534. await manager.activate()
  535. if (closing || quitting) throw new Error('App 正在退出')
  536. environmentHost.start()
  537. await runnerManager.start().catch(error => runnerManager.publish(error.message))
  538. }
  539. await preferences.setStartupChoice(remember ? mode : 'ask')
  540. syncModeMenu()
  541. const status = Menu.getApplicationMenu()?.getMenuItemById('desktop-service-status')
  542. if (status) status.label = mode === 'remote' ? `已登录远程 · ${new URL(serverURL).host}` : '后台服务运行中'
  543. tray?.setToolTip(mode === 'remote' ? `夭夭 · 远程 ${new URL(serverURL).host}` : '夭夭 · 本机运行')
  544. },
  545. navigate: url => {
  546. if (closing || quitting) throw new Error('App 正在退出')
  547. return window.loadURL(url)
  548. },
  549. })
  550. async function openOnboarding(mode, { autoPrepare = false, remember = preferences.value.startupChoice !== 'ask', forceLogin = false } = {}) {
  551. if (closing || quitting) throw new Error('App 正在退出')
  552. const prepare = autoPrepare && (mode === 'local' || Boolean(preferences.value.remoteServer))
  553. onboarding.open({ mode, serverURL: preferences.value.remoteServer, remember, forceLogin, restoring: prepare && !forceLogin })
  554. if (!window || window.isDestroyed()) await createWindow()
  555. else if (window.webContents.getURL() !== bootURL) await window.loadURL(bootURL)
  556. window.show(); window.focus()
  557. if (prepare)
  558. await onboarding.prepare({ autoEnter: true })
  559. return onboarding.snapshot()
  560. }
  561. function openRemoteLogin() {
  562. return openOnboarding('remote', { forceLogin: true })
  563. }
  564. ipcMain.handle('desktop:open-login', event => {
  565. requireModePage(event)
  566. // A session expiry returns to the same main-page flow for the active server.
  567. return openOnboarding(remoteMode ? 'remote' : 'local', { autoPrepare: true })
  568. })
  569. ipcMain.handle('desktop:forget-login', async event => {
  570. requireModePage(event)
  571. if (remoteMode && remoteURL) {
  572. await forgetLogin(remoteURL)
  573. await hostManager.stop()
  574. }
  575. })
  576. let checkingAuthorization = false
  577. checkRemoteAuthorization = async () => {
  578. if (checkingAuthorization || closing || quitting || !remoteMode || onboarding.state.active) return
  579. checkingAuthorization = true
  580. const server = remoteURL
  581. try {
  582. const info = await inspectServer(server, (...args) => net.fetch(...args))
  583. if (!info.authenticated && remoteURL === server && !onboarding.state.active && !closing && !quitting) {
  584. // Chromium may expire its copy; the native authorization remains authoritative.
  585. if ((await inspectRemote(server)).authenticated) return
  586. await forgetLogin(server)
  587. await hostManager.stop()
  588. await openOnboarding('remote', { autoPrepare: true })
  589. }
  590. } catch { /* Network loss does not revoke a saved login. */ }
  591. finally { checkingAuthorization = false }
  592. }
  593. function syncModeMenu(){
  594. const menu=Menu.getApplicationMenu()
  595. const useRemote=menu?.getMenuItemById('desktop-host-use-remote'),useLocal=menu?.getMenuItemById('desktop-host-use-local'),ask=menu?.getMenuItemById('desktop-host-ask')
  596. if(useRemote)useRemote.enabled=!switchingMode&&!remoteMode
  597. if(useLocal)useLocal.enabled=!switchingMode&&!serverModeActive
  598. const status=menu?.getMenuItemById('desktop-mode-status')
  599. if(status)status.label=switchingMode?'正在切换运行模式…':remoteMode?'当前:客户端模式':serverModeActive?'当前:服务器模式':'请选择运行模式'
  600. if(ask)ask.checked=preferences.value.startupChoice==='ask'
  601. }
  602. async function switchMode(mode) {
  603. if (mode === 'local' && !platform.server) return { ok: false, error: '此客户端不支持在本机运行服务器' }
  604. if (closing || quitting) return { ok: false, error: 'App 正在退出' }
  605. if (switchingMode || onboarding.busy) return { ok: false, error: '当前步骤尚未完成,请稍候' }
  606. switchingMode = true; syncModeMenu()
  607. try {
  608. const state = await openOnboarding(mode, { autoPrepare: true, remember: true })
  609. return { ok: true, pendingLogin: state.active }
  610. } catch (error) {
  611. return { ok: false, error: error.message }
  612. } finally { switchingMode = false; syncModeMenu() }
  613. }
  614. async function importRunner(){
  615. const selection=await dialog.showOpenDialog(window,{title:'导入执行节点配置',properties:['openFile'],filters:[{name:'Runner 配置',extensions:['json']}]})
  616. if(!selection.canceled&&selection.filePaths[0]&&!closing)await runnerAction(()=>runnerManager.importFile(selection.filePaths[0]))
  617. }
  618. Menu.setApplicationMenu(Menu.buildFromTemplate(desktopMenu([
  619. { label: '夭夭', submenu: [{ role: 'about', label: '关于夭夭' }, { type: 'separator' },
  620. { id:'desktop-service-status',label:'正在连接后台服务…',enabled:false },
  621. { label: '显示窗口', click: show }, { label: '在浏览器中打开', click: () => { const url = serviceURL(); if (url) safeExternal(url) } },
  622. { id:'desktop-local-vm',label:'本地虚拟机设置…',enabled:false,click:()=>{if(manager.state.phase==='ready'){show();void window.webContents.executeJavaScript("window.dispatchEvent(new CustomEvent('yaoyao:local-vm-settings'))")}} },
  623. { label: '查看服务日志', click: () => shell.showItemInFolder(logFile) },
  624. { id:'desktop-sync-notice', label:'Web 同步待完成:服务忙碌', enabled:false, visible:false },
  625. { id:'desktop-sync-retry', label:'重试同步本机 Web…', enabled:false,
  626. click:async()=>{try{await manager.retrySynchronization()}catch(error){log(error.message)}} },
  627. { id:'desktop-sync-force', label:'使用当前 App 覆盖同版本 Web', enabled:false,
  628. click:async()=>{if(closing||quitting||!manager.canForceSynchronization)return;try{await manager.retrySynchronization({force:true})}catch(error){log(error.message)}} },
  629. { id:'desktop-login', label:'登录 macOS 时启动', type:'checkbox', enabled:app.isPackaged, checked:app.getLoginItemSettings(loginOptions).openAtLogin,
  630. click:async item=>{try{app.setLoginItemSettings({...loginOptions,openAtLogin:item.checked});item.checked=app.getLoginItemSettings(loginOptions).openAtLogin}catch(error){item.checked=app.getLoginItemSettings(loginOptions).openAtLogin;await dialog.showMessageBox(window,{type:'error',message:'无法修改登录启动设置',detail:error.message})}} },
  631. { id:'desktop-background', label:'登录启动时仅驻留菜单栏', type:'checkbox', checked:preferences.value.backgroundAtLogin,
  632. click:async item=>{try{await preferences.setBackgroundAtLogin(item.checked)}catch(error){item.checked=preferences.value.backgroundAtLogin;await dialog.showMessageBox(window,{type:'error',message:error.message})}} },
  633. { id:'desktop-update-check', label:'检查 App 更新…', click:showUpdates },
  634. { id:'desktop-update-help', label:'App 更新与回退…', click:()=>dialog.showMessageBox(window,{type:'info',message:`夭夭 App ${packageVersion}`,detail:`App 构建:${String(buildInfo.commit).slice(0,12)}\n\n通过“检查 App 更新”下载新版,准备完成后点击“重启更新”。此操作只更新当前电脑的 App,不升级远程服务器。开发运行时提供手动安装包下载。${platform.server ? "\n\n独立后台服务继续运行。服务器模式下,首次启动同步较旧的本机 Web,已有较新的 Web 保留。Web 降级通过服务器的回滚入口处理;数据库结构不兼容时不能直接降级。" : ""}\n\n数据目录:${home}`,buttons:['知道了','打开数据目录']}).then(result=>{if(result.response===1)shell.showItemInFolder(home)}) },
  635. { type: 'separator' },
  636. { role: 'hide', label: '隐藏夭夭' }, { role: 'hideOthers', label: '隐藏其他' }, { role: 'unhide', label: '显示全部' },
  637. { type: 'separator' },
  638. {label:'撤销本机控制授权',click:()=>{void environmentHost?.revoke();void hostManager?.revoke()}},
  639. { id:'desktop-quit',label:'退出夭夭(后台继续运行)',accelerator:'CommandOrControl+Q',click:()=>requestQuit() },
  640. { id:'desktop-stop-and-quit',label:'停止后台服务并退出',click:()=>requestQuit(true) }] },
  641. {label:'执行节点',submenu:[{id:'runner-status',label:'未配置执行节点',enabled:false},{type:'separator'},
  642. {id:'runner-import',label:'导入节点配置…',click:importRunner},
  643. {id:'runner-reconnect',label:'重新连接',click:()=>runnerAction(()=>runnerManager.start())},
  644. {id:'runner-forget',label:'断开并忘记配置',click:()=>runnerAction(()=>runnerManager.forget())}]},
  645. {label:'运行模式',submenu:[{id:'desktop-mode-status',label:'请选择运行模式',enabled:false},{type:'separator'},
  646. {id:'desktop-host-use-remote',label:'客户端模式(连接服务器)',enabled:!remoteMode,click:()=>void switchMode('remote')},
  647. {id:'desktop-host-use-local',label:'服务器模式(本机运行)',enabled:!serverModeActive,click:()=>void switchMode('local')},
  648. {label:'更换服务器…',click:()=>{void openRemoteLogin().catch(error=>log(error.message))}},{type:'separator'},
  649. {id:'desktop-host-ask',label:'启动时询问运行模式',type:'checkbox',checked:preferences.value.startupChoice==='ask',
  650. click:item=>{void preferences.setStartupChoice(item.checked?'ask':remoteMode?'remote':'local').catch(async error=>{item.checked=preferences.value.startupChoice==='ask';await appDialog({type:'error',message:'无法保存启动偏好',detail:error.message})})}}]},
  651. {label:'电脑',submenu:[{id:'desktop-host-status',label:'未配置电脑',enabled:false},{type:'separator'},
  652. {id:'desktop-host-authorize',label:'重新授权…',click:()=>{void openRemoteLogin().catch(error=>log(error.message))}},
  653. {id:'desktop-host-import',label:'导入电脑配置…',click:importDesktopHost},
  654. {id:'desktop-host-reconnect',label:'重新连接',click:()=>hostAction(()=>hostManager.start())},
  655. {id:'desktop-host-forget',label:'断开并忘记配置',click:()=>hostAction(()=>hostManager.forget())}]},
  656. { role: 'editMenu', label: '编辑' }, { role: 'viewMenu', label: '显示' }, { role: 'windowMenu', label: '窗口' },
  657. ], platform.platform, { clientOnly: !platform.server })))
  658. const icon = nativeImage.createFromPath(join(desktopRoot, 'icon.png')).resize({ width: 20, height: 20 })
  659. if (!platform.windows) icon.setTemplateImage(true)
  660. tray = new Tray(icon)
  661. tray.setToolTip('夭夭')
  662. tray.setContextMenu(Menu.buildFromTemplate(desktopMenu([{ label: '打开夭夭', click: show }, { label: '查看日志', click: () => shell.showItemInFolder(logFile) }, { type: 'separator' }, { label: '退出夭夭(后台继续运行)', click: () => requestQuit() }, { label: '停止后台服务并退出', click: () => requestQuit(true) }], platform.platform, { clientOnly: !platform.server })))
  663. tray.on('click', show)
  664. // Restore a saved session behind neutral loading feedback. The guide only
  665. // becomes visible if a connection or account actually needs attention.
  666. const startupMode = !platform.server ? 'remote' : preferences.value.startupChoice === 'ask'
  667. ? (fixtureHome && process.env.HERMES_YAOYAO_DESKTOP_TEST_MODE !== 'ask' ? 'local' : null)
  668. : preferences.value.startupChoice
  669. const restoreSession = Boolean(startupMode && (startupMode === 'local' || preferences.value.remoteServer)) && (platform.server || preferences.value.startupChoice !== 'ask')
  670. onboarding.open({ mode: startupMode, serverURL: preferences.value.remoteServer,
  671. remember: preferences.value.startupChoice !== 'ask', restoring: restoreSession })
  672. // Finish the initial navigation before a remembered session enters its server.
  673. await createWindow()
  674. powerMonitor.on('resume', () => { if (!closing && !quitting) { if (serverModeActive) void manager.check(); else void checkRemoteAuthorization() } })
  675. timer = setInterval(() => { if (serverModeActive) void manager.check(); else void checkRemoteAuthorization() }, 10_000); timer.unref()
  676. syncModeMenu()
  677. if (restoreSession)
  678. await onboarding.prepare({ autoEnter: true })
  679. }).catch(error => { log(error.stack || error.message); quitting = true; app.quit() })
  680. }