mu-agent-bridge 1.0.71 → 1.0.78

package/plugins/zcode/dist/src-ts/plugins/desktop-launch.js

@@ -0,0 +1,238 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findWindowsExecutable = findWindowsExecutable;
+exports.findExecutableInTree = findExecutableInTree;
+exports.launchAndWait = launchAndWait;
+exports.launchManagedAndWait = launchManagedAndWait;
+exports.restartOwnedAndWait = restartOwnedAndWait;
+exports.terminateWindowsDesktop = terminateWindowsDesktop;
+exports.terminateOwnedProcess = terminateOwnedProcess;
+exports.findChatgptExecutable = findChatgptExecutable;
+exports.waitFor = waitFor;
+const node_child_process_1 = require("node:child_process");
+const promises_1 = require("node:fs/promises");
+const node_fs_1 = require("node:fs");
+const node_path_1 = require("node:path");
+const node_util_1 = require("node:util");
+const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
+const versionDirectory = /^(?:app[-_ ]?)?v?\d+(?:[.-]\d+)+$/i;
+/** 先从 Windows 注册信息和开始菜单解析，失败时再查常见安装目录。 */
+async function findWindowsExecutable(fileName, configured, run = execFileAsync) {
+    if (configured?.trim()) {
+        const configuredPath = (0, node_path_1.resolve)(configured);
+        if (fileName.toLowerCase() === "doubao.exe") {
+            let root = (0, node_path_1.dirname)(configuredPath);
+            while (/^(?:app|launcher|doubao_update)$/i.test((0, node_path_1.basename)(root)) || versionDirectory.test((0, node_path_1.basename)(root)))
+                root = (0, node_path_1.dirname)(root);
+            const updated = (0, node_path_1.join)(root, "Doubao_update", fileName);
+            if ((0, node_fs_1.existsSync)(updated))
+                return updated;
+            const latest = await findExecutableInTree(root, fileName, /^/, 3);
+            if (latest)
+                return latest;
+        }
+        // 千问升级会保留或删除旧版本目录，从同一安装位置重新定位新版。
+        if (fileName.toLowerCase() === "qwenworkcn.exe" && /^v?\d+(?:[.-]\d+)+$/i.test((0, node_path_1.basename)((0, node_path_1.dirname)(configuredPath)))) {
+            const latest = await findExecutableInTree((0, node_path_1.dirname)((0, node_path_1.dirname)(configuredPath)), fileName, /^/, 1);
+            if (latest)
+                return latest;
+        }
+        if ((0, node_fs_1.existsSync)(configuredPath))
+            return configuredPath;
+    }
+    if (process.platform !== "win32")
+        return undefined;
+    const registered = await findRegisteredWindowsExecutable(fileName, run);
+    if (registered)
+        return registered;
+    const pattern = /doubao|豆包|qwen|qwwork|千问|deepseek|harness|dsh|zcode/i;
+    const roots = [...new Set([
+            process.env.DSH_INSTALL_DIR,
+            "D:\\Programs\\DeepSeek Harness",
+            "D:\\DSH",
+            process.env.LOCALAPPDATA && (0, node_path_1.join)(process.env.LOCALAPPDATA, "Programs"),
+            process.env.LOCALAPPDATA,
+            process.env.ProgramFiles,
+            process.env["ProgramFiles(x86)"],
+            ...Array.from({ length: 24 }, (_, index) => `${String.fromCharCode(67 + index)}:\\`),
+        ].filter((value) => typeof value === "string" && (0, node_fs_1.existsSync)(value)))];
+    for (const root of roots) {
+        const found = await findExecutableInTree(root, fileName, pattern, 6);
+        if (found)
+            return found;
+    }
+    return undefined;
+}
+async function findRegisteredWindowsExecutable(fileName, run) {
+    const target = fileName.replace(/'/g, "''");
+    const script = `$ErrorActionPreference='SilentlyContinue';[Console]::OutputEncoding=[Text.Encoding]::UTF8;$target='${target}'
+function Emit-Path($value){if(-not $value){return};$value=[Environment]::ExpandEnvironmentVariables(([string]$value).Trim());if($value.StartsWith('"')){$end=$value.IndexOf('"',1);if($end -gt 1){$value=$value.Substring(1,$end-1)}}elseif($value -match '^(\S+?\.exe)(?:\s|$)'){$value=$Matches[1]};if($value -match '(?i)\.exe$'){[Console]::WriteLine($value)}}
+$keys=@("HKCU:\\Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\$target","HKLM:\\Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\$target","HKLM:\\Software\\WOW6432Node\\Microsoft\\Windows\\CurrentVersion\\App Paths\\$target","HKCU:\\Software\\Classes\\Applications\\$target\\shell\\open\\command","HKLM:\\Software\\Classes\\Applications\\$target\\shell\\open\\command")
+foreach($key in $keys){$item=Get-Item $key -ErrorAction SilentlyContinue;if($item){Emit-Path $item.GetValue('')}}
+$dirs=@();if($env:APPDATA){$dirs+=[IO.Path]::Combine($env:APPDATA,'Microsoft\\Windows\\Start Menu\\Programs')};if($env:ProgramData){$dirs+=[IO.Path]::Combine($env:ProgramData,'Microsoft\\Windows\\Start Menu\\Programs')};$shell=New-Object -ComObject WScript.Shell
+foreach($dir in $dirs){Get-ChildItem $dir -Filter '*.lnk' -File -Recurse -ErrorAction SilentlyContinue|ForEach-Object{$link=$shell.CreateShortcut($_.FullName);if($link.TargetPath -and [IO.Path]::GetFileName($link.TargetPath) -ieq $target){Emit-Path $link.TargetPath}}}`;
+    try {
+        const { stdout } = await run("powershell.exe", ["-NoProfile", "-NonInteractive", "-EncodedCommand", Buffer.from(script, "utf16le").toString("base64")], { timeout: 10_000, windowsHide: true });
+        for (const candidate of stdout.split(/\r?\n/).map(value => value.trim()).filter(Boolean)) {
+            const found = await resolveRegisteredCandidate(fileName, candidate);
+            if (found)
+                return found;
+        }
+    }
+    catch { /* 注册表或开始菜单不可读时继续使用目录兜底 */ }
+    return undefined;
+}
+async function resolveRegisteredCandidate(fileName, candidate) {
+    const candidatePath = (0, node_path_1.resolve)(candidate.replace(/^"|"$/g, ""));
+    if (fileName.toLowerCase() === "doubao.exe" && (0, node_path_1.basename)((0, node_path_1.dirname)(candidatePath)).toLowerCase() === "application") {
+        const client = (0, node_path_1.join)((0, node_path_1.dirname)(candidatePath), "app", fileName);
+        if ((0, node_fs_1.existsSync)(client))
+            return client;
+    }
+    if ((0, node_fs_1.existsSync)(candidatePath) && (0, node_path_1.basename)(candidatePath).toLowerCase() === fileName.toLowerCase())
+        return candidatePath;
+    return findExecutableInTree((0, node_path_1.dirname)(candidatePath), fileName, /^/, 3);
+}
+async function findExecutableInTree(directory, fileName, pattern, depth) {
+    return search(directory, fileName, pattern, depth, pattern.test(directory));
+}
+async function search(directory, fileName, pattern, depth, insideProduct) {
+    try {
+        const entries = await (0, promises_1.readdir)(directory, { withFileTypes: true });
+        // 豆包上层文件是启动器，优先查找新版目录或 app 中的实际程序。
+        if (insideProduct && depth > 0 && fileName.toLowerCase() === "doubao.exe") {
+            const versions = entries.filter(entry => entry.isDirectory() && (versionDirectory.test(entry.name) || /^(?:app|doubao_update)$/i.test(entry.name)));
+            versions.sort((left, right) => {
+                const rank = (name) => /^doubao_update$/i.test(name) ? 2 : /^app$/i.test(name) ? 0 : 1;
+                return rank(right.name) - rank(left.name) || right.name.replace(/^\D+/, "").localeCompare(left.name.replace(/^\D+/, ""), "en", { numeric: true });
+            });
+            for (const entry of versions) {
+                const found = await search((0, node_path_1.join)(directory, entry.name), fileName, pattern, depth - 1, true);
+                if (found)
+                    return found;
+            }
+        }
+        for (const entry of entries)
+            if (entry.isFile() && insideProduct && entry.name.toLowerCase() === fileName.toLowerCase() && !(fileName.toLowerCase() === "doubao.exe" && /^(?:launcher|doubao)$/i.test((0, node_path_1.basename)(directory))))
+                return (0, node_path_1.join)(directory, entry.name);
+        if (depth <= 0)
+            return undefined;
+        // 按数字比较版本目录，避免 1.2.5 排在 1.2.10 之前。
+        entries.sort((left, right) => right.name.localeCompare(left.name, "en", { numeric: true }));
+        for (const entry of entries)
+            if (entry.isDirectory() && !entry.name.startsWith(".") && (insideProduct || pattern.test(entry.name) || /^(?:Program Files(?: \(x86\))?|Programs)$/i.test(entry.name))) {
+                const found = await search((0, node_path_1.join)(directory, entry.name), fileName, pattern, depth - 1, insideProduct || pattern.test(entry.name));
+                if (found)
+                    return found;
+            }
+    }
+    catch { /* 无权限或目录在扫描期间消失时继续查找 */ }
+    return undefined;
+}
+async function launchAndWait(executable, args, connect, timeoutMs, launch = (file, argv) => (0, node_child_process_1.spawn)(file, argv, { detached: true, stdio: "ignore", windowsHide: false })) {
+    const child = launch(executable, args);
+    child.unref();
+    return waitFor(connect, timeoutMs, 250);
+}
+async function launchManagedAndWait(executable, args, profileDirectory, connect, timeoutMs, onProcess, launch = (file, argv) => (0, node_child_process_1.spawn)(file, argv, { detached: true, stdio: "ignore", windowsHide: false })) {
+    if (profileDirectory)
+        await (0, promises_1.mkdir)(profileDirectory, { recursive: true });
+    const child = launch(executable, profileDirectory ? [...args, `--user-data-dir=${profileDirectory}`] : args);
+    // 先等待 spawn，避免异步 error 终止 daemon 或在启动失败后继续轮询。
+    const events = child;
+    const started = typeof events.once === "function" ? new Promise((resolve, reject) => {
+        events.once("error", reject);
+        events.once("spawn", resolve);
+    }) : Promise.resolve();
+    child.unref();
+    onProcess(child);
+    try {
+        await started;
+        return await waitFor(connect, timeoutMs, 250);
+    }
+    catch (error) {
+        await terminateOwnedProcess(child).catch(() => undefined);
+        onProcess(undefined);
+        throw error;
+    }
+}
+async function restartOwnedAndWait(child, executable, args, profileDirectory, connect, timeoutMs, onProcess) {
+    await terminateOwnedProcess(child);
+    onProcess(undefined);
+    return launchManagedAndWait(executable, args, profileDirectory, connect, timeoutMs, onProcess);
+}
+/** 仅在用户确认重启后清理同一用户、同一安装路径的桌面进程。 */
+async function terminateWindowsDesktop(executable) {
+    if (process.platform !== "win32")
+        return [];
+    const script = `
+$ErrorActionPreference = 'Stop'
+$target = '${(0, node_path_1.resolve)(executable).replace(/'/g, "''")}'
+$sid = [Security.Principal.WindowsIdentity]::GetCurrent().User.Value
+$processes = @(Get-CimInstance Win32_Process | Where-Object { $_.ExecutablePath -ieq $target } | Where-Object { (Invoke-CimMethod -InputObject $_ -MethodName GetOwnerSid).Sid -eq $sid })
+$ids = @($processes.ProcessId)
+$roots = @($processes | Where-Object { $ids -notcontains $_.ParentProcessId })
+foreach ($item in $roots) {
+  $current = Get-CimInstance Win32_Process -Filter "ProcessId=$($item.ProcessId)"
+  if (-not $current -or $current.CreationDate -ne $item.CreationDate) { continue }
+  & taskkill.exe /PID $item.ProcessId /T /F | Out-Null
+  if ($LASTEXITCODE -ne 0 -and (Get-Process -Id $item.ProcessId -ErrorAction SilentlyContinue)) { throw "无法结束桌面进程 $($item.ProcessId)" }
+  $remaining = Get-Process -Id $item.ProcessId -ErrorAction SilentlyContinue
+  if ($remaining -and -not $remaining.WaitForExit(10000)) { throw "等待桌面进程退出超时 $($item.ProcessId)" }
+}
+ConvertTo-Json -Compress -InputObject @($roots | ForEach-Object { [int]$_.ProcessId })
+`;
+    const { stdout } = await execFileAsync("powershell.exe", ["-NoProfile", "-NonInteractive", "-EncodedCommand", Buffer.from(script, "utf16le").toString("base64")], { timeout: 30_000, windowsHide: true });
+    return JSON.parse(stdout.trim());
+}
+async function terminateOwnedProcess(child, platform = process.platform, run = async (file, args) => execFileAsync(file, args, { timeout: 10_000 })) {
+    const pid = child?.pid;
+    if (!Number.isSafeInteger(pid) || !pid || child?.exitCode !== null && child?.exitCode !== undefined || child?.signalCode)
+        return;
+    if (platform === "win32") {
+        try {
+            await run("taskkill.exe", ["/PID", String(pid), "/T", "/F"]);
+        }
+        catch (error) {
+            if (!child.exitCode && !child.signalCode && error.code !== "ESRCH")
+                throw error;
+        }
+    }
+    else {
+        try {
+            process.kill(pid, "SIGTERM");
+        }
+        catch (error) {
+            if (error.code !== "ESRCH")
+                throw error;
+        }
+    }
+}
+async function findChatgptExecutable(configured) {
+    if (configured?.trim() && (0, node_fs_1.existsSync)((0, node_path_1.resolve)(configured)))
+        return (0, node_path_1.resolve)(configured);
+    if (process.platform !== "win32")
+        return undefined;
+    const script = "$candidates = foreach ($package in (Get-AppxPackage | Where-Object { $_.Name -like 'OpenAI.Codex*' })) { $appDir = Join-Path $package.InstallLocation 'app'; $path = Join-Path $appDir 'ChatGPT.exe'; if (-not (Test-Path -LiteralPath $path)) { $path = Get-ChildItem -LiteralPath $appDir -Filter 'ChatGPT*.exe' -File -ErrorAction SilentlyContinue | Select-Object -First 1 -ExpandProperty FullName }; if ($path) { [pscustomobject]@{ Version = [version]$package.Version; Executable = $path } } }; $candidate = $candidates | Sort-Object Version -Descending | Select-Object -First 1; if ($candidate) { $candidate.Executable }";
+    try {
+        const { stdout } = await execFileAsync("powershell.exe", ["-NoProfile", "-NonInteractive", "-Command", script], { timeout: 10_000 });
+        return stdout.trim().split(/\r?\n/).at(-1)?.trim() || undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+async function waitFor(action, timeoutMs, intervalMs = 500) {
+    const deadline = Date.now() + timeoutMs;
+    let last;
+    while (Date.now() < deadline) {
+        try {
+            return await action();
+        }
+        catch (error) {
+            last = error;
+            await new Promise((resolve) => setTimeout(resolve, intervalMs));
+        }
+    }
+    throw last instanceof Error ? last : new Error("桌面程序启动超时");
+}