From 6c59daafcdf90f7c1ce9512677658dffd199014f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A2=AB=E9=81=97=E5=BF=98=E7=9A=84=E8=AE=B0=E5=BF=86?=
 <32097237+YIWANG-sketch@users.noreply.github.com>
Date: Mon, 21 Oct 2024 08:14:48 +0800
Subject: [PATCH] Create browserDetector.mjs

---
 utils/browserDetector.mjs | 109 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 utils/browserDetector.mjs

diff --git a/utils/browserDetector.mjs b/utils/browserDetector.mjs
new file mode 100644
index 0000000..015fe6c
--- /dev/null
+++ b/utils/browserDetector.mjs
@@ -0,0 +1,109 @@
+import os from 'os';
+import fs from 'fs';
+import path from 'path';
+import { execSync } from 'child_process';
+
+export function detectBrowser(preferredBrowser = 'auto') {
+    const platform = os.platform();
+    let browsers = {
+        'chrome': null,
+        'edge': null
+    };
+
+    if (platform === 'win32') {
+        browsers.chrome = findWindowsBrowser('Chrome');
+        browsers.edge = findWindowsBrowser('Edge');
+    } else if (platform === 'darwin') {
+        browsers.chrome = findMacOSBrowser('Google Chrome');
+        browsers.edge = findMacOSBrowser('Microsoft Edge');
+    } else if (platform === 'linux') {
+        browsers.chrome = findLinuxBrowser('google-chrome');
+        browsers.edge = findLinuxBrowser('microsoft-edge');
+    }
+
+    if (preferredBrowser === 'auto' || preferredBrowser === undefined) {
+        if (browsers.chrome) {
+            return browsers.chrome;
+        } else if (browsers.edge) {
+            return browsers.edge;
+        }
+    } else if (browsers[preferredBrowser]) {
+        console.log(`使用${preferredBrowser === 'chrome' ? 'Chrome' : 'Edge'}浏览器`);
+        return browsers[preferredBrowser];
+    }
+
+    console.error('未找到Chrome或Edge浏览器，请确保已安装其中之一');
+    process.exit(1);
+}
+
+function findWindowsBrowser(browserName) {
+    const regKeys = {
+        'Chrome': ['chrome.exe', 'Google\\Chrome'],
+        'Edge': ['msedge.exe', 'Microsoft\\Edge']
+    };
+    const [exeName, folderName] = regKeys[browserName];
+
+    const regQuery = (key) => {
+        try {
+            return execSync(`reg query "${key}" /ve`).toString().trim().split('\r\n').pop().split('    ').pop();
+        } catch (error) {
+            return null;
+        }
+    };
+
+    let browserPath = regQuery(`HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`) ||
+        regQuery(`HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\${exeName}`);
+
+    if (browserPath && fs.existsSync(browserPath)) {
+        return browserPath;
+    }
+
+    const commonPaths = [
+        `C:\\Program Files\\${browserName}\\Application\\${exeName}`,
+        `C:\\Program Files (x86)\\${browserName}\\Application\\${exeName}`,
+        `C:\\Program Files (x86)\\Microsoft\\${browserName}\\Application\\${exeName}`,
+        `${process.env.LOCALAPPDATA}\\${browserName}\\Application\\${exeName}`,
+        `${process.env.USERPROFILE}\\AppData\\Local\\${browserName}\\Application\\${exeName}`,
+    ];
+
+    const foundPath = commonPaths.find(path => fs.existsSync(path));
+    if (foundPath) {
+        return foundPath;
+    }
+
+    const userAppDataPath = process.env.LOCALAPPDATA || `${process.env.USERPROFILE}\\AppData\\Local`;
+    const appDataPath = path.join(userAppDataPath, folderName, 'Application');
+
+    if (fs.existsSync(appDataPath)) {
+        const files = fs.readdirSync(appDataPath);
+        const exePath = files.find(file => file.toLowerCase() === exeName.toLowerCase());
+        if (exePath) {
+            return path.join(appDataPath, exePath);
+        }
+    }
+
+    return null;
+}
+
+function findMacOSBrowser(browserName) {
+    const paths = [
+        `/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
+        `${os.homedir()}/Applications/${browserName}.app/Contents/MacOS/${browserName}`,
+    ];
+
+    for (const path of paths) {
+        if (fs.existsSync(path)) {
+            return path;
+        }
+    }
+
+    return null;
+}
+
+function findLinuxBrowser(browserName) {
+    try {
+        return execSync(`which ${browserName}`).toString().trim();
+    } catch (error) {
+        return null;
+    }
+}