From 831dbadc74b774c296e42201c4fe0d1a5e50eaad Mon Sep 17 00:00:00 2001
From: bossgeekgo <bossgeekgo>
Date: Tue, 20 Feb 2024 01:35:17 +0800
Subject: [PATCH] make DependenciesSetupProgressIndicatorDialog support execute
 more than 1 tasks (pervious support download puppeteer only)

---
 .../check-and-download-puppeteer.ts           | 25 +++---
 packages/ui/src/main/window/mainWindow.ts     |  5 +-
 .../index.vue                                 | 88 +++++++++++++------
 .../operations.ts                             |  5 +-
 .../GeekAutoStartChatWithBoss.vue             |  4 +-
 .../renderer/src/page/Configuration/index.vue |  4 +-
 6 files changed, 89 insertions(+), 42 deletions(-)

diff --git a/packages/ui/src/main/flow/CHECK_AND_DOWNLOAD_DEPENDENCIES/check-and-download-puppeteer.ts b/packages/ui/src/main/flow/CHECK_AND_DOWNLOAD_DEPENDENCIES/check-and-download-puppeteer.ts
index 89dfe64..9821ddc 100644
--- a/packages/ui/src/main/flow/CHECK_AND_DOWNLOAD_DEPENDENCIES/check-and-download-puppeteer.ts
+++ b/packages/ui/src/main/flow/CHECK_AND_DOWNLOAD_DEPENDENCIES/check-and-download-puppeteer.ts
@@ -12,10 +12,8 @@ const cacheDir = path.join(
 )
 
 const getPuppeteerManagerModule = async () => {
-  const runtimeDependencies = await import(
-    path.join(os.homedir(), '.bossgeekgo', 'external-node-runtime-dependencies/index.mjs')
-  )
-  return runtimeDependencies.puppeteerManager
+  const puppeteerManager = await import('@puppeteer/browsers')
+  return puppeteerManager
 }
 
 export const getExpectPuppeteerExecutablePath = async () => {
@@ -29,14 +27,21 @@ export const getExpectPuppeteerExecutablePath = async () => {
 }
 
 export const checkPuppeteerExecutable = async () => {
-  const executablePath = await getExpectPuppeteerExecutablePath()
-  return fs.existsSync(executablePath)
+  try {
+    const executablePath = await getExpectPuppeteerExecutablePath()
+    return fs.existsSync(executablePath)
+  } catch {
+    // should limit [ERR_MODULE_NOT_FOUND]
+    return false
+  }
 }
 
-const checkAndDownloadPuppeteer = async (options: {
-  downloadProgressCallback?: (downloadedBytes: number, totalBytes: number) => void,
-  confirmContinuePromise?: Promise<void>
-} = {}) => {
+const checkAndDownloadPuppeteer = async (
+  options: {
+    downloadProgressCallback?: (downloadedBytes: number, totalBytes: number) => void
+    confirmContinuePromise?: Promise<void>
+  } = {}
+) => {
   const puppeteerManager = await getPuppeteerManagerModule()
   let installedBrowser: InstalledBrowser
   if (!(await checkPuppeteerExecutable())) {
diff --git a/packages/ui/src/main/window/mainWindow.ts b/packages/ui/src/main/window/mainWindow.ts
index 214f64d..1ae17f3 100644
--- a/packages/ui/src/main/window/mainWindow.ts
+++ b/packages/ui/src/main/window/mainWindow.ts
@@ -130,7 +130,10 @@ export function createMainWindow(): void {
   })
 
   ipcMain.handle('check-dependencies', async () => {
-    return await checkPuppeteerExecutable()
+    const [puppeteerExecutableAvailable] = await Promise.all([checkPuppeteerExecutable()])
+    return {
+      puppeteerExecutableAvailable
+    }
   })
 
   let subProcessOfCheckAndDownloadDependencies: ChildProcess | null = null
diff --git a/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/index.vue b/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/index.vue
index 9ed3eae..e54a9bf 100644
--- a/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/index.vue
+++ b/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/index.vue
@@ -1,46 +1,84 @@
 <template>
-  <el-dialog v-bind="$attrs" @open="percentage = 0">
-    <div>Downloading the necessary dependencies...</div>
-    <el-progress :percentage="percentage" :format="(n) => `${n.toFixed(1)}%`" />
+  <el-dialog v-bind="$attrs" @open="handleDialogOpen">
+    <template v-if="!copiedDependenciesStatus.puppeteerExecutableAvailable">
+      <div>Downloading Dedicate Browser</div>
+      <el-progress :percentage="browserDownloadPercentage" :format="(n) => `${n.toFixed(1)}%`" />
+    </template>
   </el-dialog>
 </template>
 
 <script lang="ts" setup>
-import { ref, onUnmounted, onMounted } from 'vue'
-import { ElMessageBox } from 'element-plus';
+import { ref, onUnmounted, PropType } from 'vue'
+import { ElMessageBox } from 'element-plus'
 
 const props = defineProps({
-  dispose: Function
+  dispose: Function,
+  dependenciesStatus: {
+    type: Object as PropType<Record<string, boolean>>,
+    default: () => ({})
+  }
 })
 
-const percentage = ref(0)
-const handleProgress = (ev, { downloadedBytes, totalBytes }) => {
-  percentage.value = (downloadedBytes / totalBytes) * 100
+// shallow copy
+const copiedDependenciesStatus = {
+  ...props.dependenciesStatus
 }
-electron.ipcRenderer.on('PUPPETEER_DOWNLOAD_PROGRESS', handleProgress)
+
+const handleDialogOpen = () => {
+  browserDownloadPercentage.value = 0
+}
+
+const browserDownloadPercentage = ref(0)
+const handleBrowserDownloadProgress = (ev, { downloadedBytes, totalBytes }) => {
+  browserDownloadPercentage.value = (downloadedBytes / totalBytes) * 100
+}
+electron.ipcRenderer.on('PUPPETEER_DOWNLOAD_PROGRESS', handleBrowserDownloadProgress)
 onUnmounted(() =>
-  electron.ipcRenderer.removeListener('PUPPETEER_DOWNLOAD_PROGRESS', handleProgress)
+  electron.ipcRenderer.removeListener('PUPPETEER_DOWNLOAD_PROGRESS', handleBrowserDownloadProgress)
 )
 const downloadProcessExitCode = ref(0)
 
-const processDownload = async () => {
+const processDownloadBrowser = async () => {
   downloadProcessExitCode.value = 0
-  percentage.value = 0
+  browserDownloadPercentage.value = 0
   try {
     await electron.ipcRenderer.invoke('setup-dependencies')
-    props.dispose?.()
-  } catch(err) {
+    browserDownloadPercentage.value = 100
+  } catch (err) {
     downloadProcessExitCode.value = 1
-
-    ElMessageBox.confirm('Encounter error while setup dependencies. Retry?')
-    .then(() => {
-      processDownload()
-    })
-    .catch(() => {
-      // FIXME: should exit app here
-      props.dispose?.()
-    })
+    throw err
   }
 }
-processDownload()
+
+const promiseList: Array<Promise<void>> = []
+const processTasks = async () => {
+  if (!copiedDependenciesStatus.puppeteerExecutableAvailable) {
+    const p = processDownloadBrowser()
+    promiseList.push(p)
+    p.then(() => {
+      copiedDependenciesStatus.puppeteerExecutableAvailable = true
+    }).finally(() => {
+      const idx = promiseList.indexOf(p)
+      idx >= 0 && promiseList.splice(idx, 1)
+    })
+  }
+
+  while (promiseList.length) {
+    try {
+      await promiseList[0]
+    } catch {
+      ElMessageBox.confirm('Encounter error while setup dependencies. Retry?')
+        .then(() => {
+          processTasks()
+        })
+        .catch(() => {
+          // FIXME: should exit app here
+          promiseList.length = 0
+          props.dispose?.()
+        })
+    }
+  }
+}
+
+processTasks()
 </script>
diff --git a/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/operations.ts b/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/operations.ts
index 3463948..23e49c1 100644
--- a/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/operations.ts
+++ b/packages/ui/src/renderer/src/features/DependenciesSetupProgressIndicatorDialog/operations.ts
@@ -2,7 +2,7 @@ import { createApp } from 'vue'
 import ElementPlus from 'element-plus'
 import DependenciesSetupProgressIndicatorDialog from './index.vue'
 
-export const mountGlobalDialog = () => {
+export const mountGlobalDialog = (dependenciesStatus: Record<string, boolean>) => {
   const containerElId = `elForDependenciesSetupProgressIndicatorDialog`
 
   if (document.getElementById(containerElId)) {
@@ -27,7 +27,8 @@ export const mountGlobalDialog = () => {
     onClosed() {
       dispose()
     },
-    dispose
+    dispose,
+    dependenciesStatus
   }).use(ElementPlus)
   app.mount(containerEl)
 
diff --git a/packages/ui/src/renderer/src/page/Configuration/GeekAutoStartChatWithBoss.vue b/packages/ui/src/renderer/src/page/Configuration/GeekAutoStartChatWithBoss.vue
index 49eeca7..53de824 100644
--- a/packages/ui/src/renderer/src/page/Configuration/GeekAutoStartChatWithBoss.vue
+++ b/packages/ui/src/renderer/src/page/Configuration/GeekAutoStartChatWithBoss.vue
@@ -95,8 +95,8 @@ const handleSubmit = async () => {
         message: `Some dependencies might be corrupt. I'm trying to check and fix them.`
       })
       const checkDependenciesResult = await electron.ipcRenderer.invoke('check-dependencies')
-      if (!checkDependenciesResult) {
-        mountDependenciesSetupProgressIndicatorDialog()
+      if (Object.values(checkDependenciesResult).includes(false)) {
+        mountDependenciesSetupProgressIndicatorDialog(checkDependenciesResult)
         // TODO: should continue interrupted task
       }
     }
diff --git a/packages/ui/src/renderer/src/page/Configuration/index.vue b/packages/ui/src/renderer/src/page/Configuration/index.vue
index 877235a..57b767a 100644
--- a/packages/ui/src/renderer/src/page/Configuration/index.vue
+++ b/packages/ui/src/renderer/src/page/Configuration/index.vue
@@ -15,8 +15,8 @@ onUnmounted(() => {
 })
 ;(async () => {
   const checkDependenciesResult = await electron.ipcRenderer.invoke('check-dependencies')
-  if (!checkDependenciesResult) {
-    mountDependenciesSetupProgressIndicatorDialog()
+  if (Object.values(checkDependenciesResult).includes(false)) {
+    mountDependenciesSetupProgressIndicatorDialog(checkDependenciesResult)
   }
 })()
 </script>