refactor: Move getDiskInfo to worker to reduce dist size

Author: ci010

xmcl-electron-app/main/controllers/setupWindow.ts

@@ -1,19 +1,19 @@
 import { ElectronController } from '@/ElectronController'
-import { ControllerPlugin } from './plugin'
-import { getDiskInfo } from 'node-disk-info'
+import { kSetupWorker } from '@/setupWorker'
 import { ipcMain } from 'electron'
 import { join, parse } from 'path'
-import type Drive from 'node-disk-info/dist/classes/drive'
+import { setTimeout } from 'timers/promises'
+import { ControllerPlugin } from './plugin'
 
 /**
  * Handle setup window preset request
  */
 export const setupWindow: ControllerPlugin = function (this: ElectronController) {
   ipcMain.handle('preset', async () => {
-    const drives = await new Promise<Drive[]>((resolve) => {
-      getDiskInfo().then(resolve, () => resolve([]))
-      setTimeout(() => { resolve([]) }, 4000)
-    })
+    const drives = await Promise.race([
+      this.app.registry.get(kSetupWorker).then(w => w.getDiskInfo()),
+      setTimeout(4000).then(() => []),
+    ])
     const defaultPath = join(this.app.host.getPath('home'), '.xmcl')
     const getPath = (driveSymbol: string) => {
       const parsedHome = parse(defaultPath)

xmcl-electron-app/main/definedPlugins.ts

@@ -31,6 +31,7 @@ import { pluginYggdrasilHandler } from '@xmcl/runtime/yggdrasilServer/pluginYggd
 
 import { LauncherAppPlugin } from '~/app'
 import { definedServices } from './definedServices'
+import { pluginSetupWorker } from './pluginSetupWorker'
 
 export const definedPlugins: LauncherAppPlugin[] = [
   pluginAutoUpdate,
@@ -39,6 +40,7 @@ export const definedPlugins: LauncherAppPlugin[] = [
   pluginIconProtocol,
   pluginResourceWorker,
   pluginEncodingWorker,
+  pluginSetupWorker,
 
   pluginMediaProtocol,
   pluginResourcePackLink,

xmcl-electron-app/main/pluginSetupWorker.ts

@@ -0,0 +1,24 @@
+import { checkUpdate, createLazyWorker } from '@xmcl/runtime/worker'
+import { LauncherAppPlugin } from '~/app'
+import { SetupWorker, kSetupWorker } from './setupWorker'
+import createSetupWorker, { path as setupWorkerPath } from './setupWorkerEntry?worker'
+import Drive from 'node-disk-info/dist/classes/drive'
+
+export const pluginSetupWorker: LauncherAppPlugin = async (app) => {
+  const logger = app.getLogger('SetupWorker')
+
+  const worker: SetupWorker = createLazyWorker(createSetupWorker, ['getDiskInfo'], logger)
+  app.registry.register(kSetupWorker, {
+    getDiskInfo: async () => {
+      const infos = await worker.getDiskInfo()
+      for (const i of infos) {
+        Object.setPrototypeOf(i, Drive.prototype)
+      }
+      return infos
+    },
+  })
+
+  app.waitEngineReady().then(() => {
+    checkUpdate(setupWorkerPath, logger)
+  })
+}

xmcl-electron-app/main/setupWorker.ts

@@ -0,0 +1,8 @@
+import type Drive from 'node-disk-info/dist/classes/drive'
+import { InjectionKey } from '~/app'
+
+export interface SetupWorker {
+  getDiskInfo(): Promise<Drive[]>
+}
+
+export const kSetupWorker: InjectionKey<SetupWorker> = Symbol('kSetupWorker')

xmcl-electron-app/main/setupWorkerEntry.ts

@@ -0,0 +1,4 @@
+import { setHandler } from '@xmcl/runtime/worker/helper'
+import { getDiskInfo } from 'node-disk-info'
+
+setHandler({ getDiskInfo })