899 lines
33 KiB
JavaScript
899 lines
33 KiB
JavaScript
const { app, globalShortcut, BrowserWindow, net, dialog, ipcMain } = require('electron');
|
||
const { startSSE } = require('./js/sse-server');
|
||
const { createBurstBroadcaster } = require('./js/burst-broadcast');
|
||
const mq = require('./js/rabbitmq-service');
|
||
const axios = require('axios');
|
||
const os = require('os');
|
||
const fsp = require('node:fs/promises')
|
||
const { fileURLToPath } = require('node:url')
|
||
|
||
|
||
const path = require('path');
|
||
const USER_ROOT = path.resolve(os.homedir());
|
||
const { autoUpdater } = require('electron-updater')
|
||
const fs = require('fs')
|
||
const log = require('electron-log')
|
||
Object.assign(console, log.functions)
|
||
// 存储 SSE 服务器的引用
|
||
let sseServer = null;
|
||
|
||
|
||
// ✅ 本地 HTTPS 白名单（按你的实际地址改端口）
|
||
const LOCAL_HTTPS_WHITELIST = [
|
||
'https://192.168.1.218:34567',
|
||
'https://192.168.1.231:34567',
|
||
'https://127.0.0.1:34567',
|
||
'https://localhost:34567'
|
||
]
|
||
|
||
let userData = { tenantId: null, userId: null }
|
||
const { exec, spawn, execFile } = require('child_process'); // 如果你用 exec 启动 scrcpy，保留此行
|
||
|
||
// app.commandLine.appendSwitch('remote-debugging-port', '9222'); //远程控制台端口F12
|
||
|
||
app.commandLine.appendSwitch('enable-precise-memory-info')
|
||
app.commandLine.appendSwitch('js-flags', '--expose-gc --max-old-space-size=4096') // 以 MB 为单位，建议设置 2048-4096
|
||
|
||
app.commandLine.appendSwitch('enable-experimental-web-platform-features');
|
||
app.commandLine.appendSwitch('enable-features', 'WebCodecs,MediaStreamInsertableStreams');
|
||
// === 更新门闩：更新进行时，禁止跳转业务页面 ===
|
||
let updateInProgress = false; // 有更新或正在下载 -> true
|
||
let updateDownloaded = false; // 已下载完成（通常会随即重启安装）
|
||
let pendingNavs = []; // 被延迟的跳转请求（函数队列）
|
||
let installingNow = false; // 正在静默安装（用户点击“立即重启安装”之后）
|
||
|
||
|
||
// --- 持久化兼容模式开关（仅影响本机） ---
|
||
const compatFile = path.join(app.getPath('userData'), 'compat.json');
|
||
function readCompat() { try { return JSON.parse(fs.readFileSync(compatFile, 'utf8')); } catch { return {}; } }
|
||
function writeCompat(obj) { try { fs.writeFileSync(compatFile, JSON.stringify(obj, null, 2)); } catch { } }
|
||
|
||
// 支持三种触发：环境变量、命令行、持久化标记
|
||
const SAFE_MODE =
|
||
process.env.YOLO_SAFE_MODE === '1' ||
|
||
process.argv.includes('--safe-mode') ||
|
||
readCompat().safeMode === true;
|
||
|
||
// （可选）允许命令行清除持久化标记：--clear-safe-mode
|
||
if (process.argv.includes('--clear-safe-mode')) {
|
||
const c = readCompat(); delete c.safeMode; writeCompat(c);
|
||
console.log('[兼容性] 已清除安全模式标志');
|
||
}
|
||
|
||
// === 兼容模式等价于 bat 里的那些开关（不含日志） ===
|
||
if (SAFE_MODE) {
|
||
app.disableHardwareAcceleration(); // --disable-gpu
|
||
app.commandLine.appendSwitch('disable-gpu');
|
||
app.commandLine.appendSwitch('in-process-gpu'); // --in-process-gpu
|
||
app.commandLine.appendSwitch('disable-direct-composition'); // --disable-direct-composition
|
||
// 如需更稳再关沙箱：默认开着更安全；若你确实要关，则解除注释
|
||
// app.commandLine.appendSwitch('no-sandbox'); // --no-sandbox
|
||
console.log('[安全模式] 已启用');
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
function flushPendingNavs() {
|
||
const fns = pendingNavs.slice();
|
||
pendingNavs = [];
|
||
for (const fn of fns) { try { fn(); } catch (e) { console.warn('[导航] 延迟运行错误:', e); } }
|
||
}
|
||
|
||
function safeLoadURL(win, url, onFail) {
|
||
if (!win || win.isDestroyed()) return;
|
||
win.loadURL(url).catch(err => {
|
||
console.warn('[加载url 失败]', err);
|
||
onFail?.(err);
|
||
});
|
||
}
|
||
// 进程异常处理
|
||
process.on('uncaughtException', (error) => {
|
||
console.error('uncaughtException未捕获异常:', error);
|
||
});
|
||
process.on('unhandledRejection', (reason, p) => {
|
||
console.error('[unhandledRejection未捕获异常]', reason);
|
||
});
|
||
|
||
// 渲染崩溃 → 持久化 safeMode=true（下次自动以兼容模式启动）
|
||
app.on('web-contents-created', (_, contents) => {
|
||
contents.on('render-process-gone', (_event, details) => {
|
||
console.error('⚠️ 渲染崩溃:', details);
|
||
if (['launch-failed', 'abnorm al-exit'].includes(details.reason)) {
|
||
const c = readCompat(); c.safeMode = true; writeCompat(c);
|
||
console.warn('[兼容性] 已设置安全模式为true（已持久化），下次启动时将自动使用安全模式');
|
||
|
||
}
|
||
});
|
||
});
|
||
//（可选）更广义的子进程异常也触发降级
|
||
app.on('child-process-gone', (_e, details) => {
|
||
if ((details.type === 'GPU' || details.type === 'Renderer') &&
|
||
['launch-failed', 'abnormal-exit'].includes(details.reason)) {
|
||
const c = readCompat(); c.safeMode = true; writeCompat(c);
|
||
console.warn('[兼容性] 已设置安全模式为true（已持久化），下次启动时将自动使用安全模式');
|
||
|
||
}
|
||
});
|
||
//本地https 证书跳过校验
|
||
app.on('certificate-error', (event, webContents, url, error, certificate, callback) => {
|
||
// console.warn('[certificate-error]', url, error)
|
||
|
||
// 只放行你自己的本地 HTTPS
|
||
if (LOCAL_HTTPS_WHITELIST.some(prefix => url.startsWith(prefix))) {
|
||
event.preventDefault() // 告诉 Electron：我来处理这个错误
|
||
callback(true) // ✅ 忽略证书错误，照样加载
|
||
} else {
|
||
// 其他地址还是正常校验证书，防止误放行
|
||
callback(false)
|
||
}
|
||
})
|
||
|
||
|
||
app.on('ready', () => {
|
||
console.log('[App Ready] 当前版本:', app.getVersion());
|
||
|
||
// 配置日志记录（可选）
|
||
const log = require('electron-log');
|
||
autoUpdater.logger = log;
|
||
autoUpdater.logger.transports.file.level = 'info';
|
||
|
||
// 启用静默更新（关键设置）
|
||
autoUpdater.autoDownload = true; // 自动下载更新
|
||
autoUpdater.autoInstallOnAppQuit = false; // 我们手动调用安装（静默）
|
||
// 监听事件（用于控制流程或调试，可保留日志）
|
||
autoUpdater.on('checking-for-update', () => {
|
||
console.log('[autoUpdater] 正在检查更新...');
|
||
});
|
||
|
||
autoUpdater.on('update-available', async (info) => {
|
||
updateInProgress = true; // ⛔ 开始阻塞导航
|
||
console.log(`[autoUpdater] 检测到新版本: ${info.version}，开始自动下载...`);
|
||
await killIOSAI()
|
||
broadcast('update:available', info);
|
||
});
|
||
let lastSend = 0;
|
||
autoUpdater.on('update-not-available', () => {
|
||
console.log('[autoUpdater] 当前已是最新版本');
|
||
updateInProgress = false; // ✅ 解锁
|
||
flushPendingNavs(); // 立刻执行被延迟的导航
|
||
});
|
||
|
||
autoUpdater.on('download-progress', (p) => {
|
||
updateInProgress = true; // ⛔ 持续阻塞
|
||
// 节流 150ms，避免刷屏
|
||
const now = Date.now();
|
||
if (now - lastSend > 150) {
|
||
lastSend = now;
|
||
broadcast('update:progress', {
|
||
percent: p.percent, // 0~100
|
||
bytesPerSecond: p.bytesPerSecond, // B/s
|
||
transferred: p.transferred, // 已下载字节
|
||
total: p.total // 总字节
|
||
});
|
||
}
|
||
|
||
// 维持你原来的任务栏进度
|
||
const win = BrowserWindow.getAllWindows()[0];
|
||
if (win) win.setProgressBar(p.percent / 100);
|
||
});
|
||
|
||
autoUpdater.on('update-downloaded', (info) => {
|
||
updateInProgress = true; // 已下载，通常马上安装并退出
|
||
updateDownloaded = true;
|
||
console.log('[autoUpdater] 更新下载完成');
|
||
broadcast('update:downloaded', info);
|
||
|
||
// 如果想“用户点按钮再安装”，注释掉下一行，交给渲染进程触发
|
||
// autoUpdater.quitAndInstall(false, true);
|
||
});
|
||
|
||
autoUpdater.on('error', (err) => {
|
||
console.error('[autoUpdater] 错误:', err);
|
||
broadcast('update:error', { message: String(err) });
|
||
updateInProgress = false; // ✅ 出错也解锁（按需可保持阻塞）
|
||
flushPendingNavs();
|
||
});
|
||
// 可选：提供渲染端调用的安装入口
|
||
|
||
|
||
// ✅ 渲染端主动触发静默安装（不显示 UI）
|
||
ipcMain.handle('update:quitAndInstallNow', async () => {
|
||
try {
|
||
installingNow = true; // 进入“静默安装中”阶段
|
||
// 安装前做一次业务清理：但避免误杀自身或阻塞
|
||
await killIOSAI();
|
||
} catch (e) {
|
||
console.warn('[更新] 预安装清理警告:', e);
|
||
|
||
}
|
||
// electron-updater 实际上传给 NSIS /S 静默参数，第二个 true 表示强制重启后运行新版本
|
||
autoUpdater.quitAndInstall(false, true);
|
||
});
|
||
ipcMain.handle('update:checkNow', () => {
|
||
return autoUpdater.checkForUpdates();
|
||
});
|
||
|
||
// 启动检查更新（不弹窗）
|
||
autoUpdater.checkForUpdates();
|
||
});
|
||
|
||
// 应用程序生命周期管理
|
||
app.whenReady().then(() => {
|
||
setInterval(dumpAllMem, 5000); //循环查看内存情况
|
||
|
||
|
||
// 再创建窗口
|
||
createWindow();
|
||
|
||
// macOS 特有：激活时若无打开窗口再创建
|
||
app.on('activate', () => {
|
||
if (BrowserWindow.getAllWindows().length === 0) createWindow();
|
||
});
|
||
|
||
|
||
// globalShortcut.register('F12', () => toggleDevtools(BrowserWindow.getFocusedWindow()));
|
||
globalShortcut.register('Control+Shift+i', () => toggleDevtools(BrowserWindow.getFocusedWindow()));
|
||
// globalShortcut.register('Control+Alt+m', () => toggleDevtools(BrowserWindow.getFocusedWindow()));
|
||
});
|
||
app.on('will-quit', () => globalShortcut.unregisterAll());
|
||
|
||
|
||
// 仅在“不是更新安装中”时，做你的退出清理
|
||
app.on('before-quit', async () => {
|
||
if (!installingNow) {
|
||
await killIOSAI();
|
||
} else {
|
||
console.log('[退出前] 跳过 killIOSAI 因为 installingNow = true');
|
||
|
||
}
|
||
});
|
||
function toggleDevtools(win) {
|
||
if (!win) return;
|
||
const wc = win.webContents;
|
||
if (wc.isDevToolsOpened()) wc.closeDevTools();
|
||
else wc.openDevTools({ mode: 'detach' }); // 独立窗口
|
||
}
|
||
|
||
// 窗口全部关闭时，退出或保留后台进程
|
||
app.on('window-all-closed', async () => {
|
||
await killIOSAI();
|
||
if (process.platform !== 'darwin') {
|
||
app.quit();
|
||
}
|
||
});
|
||
// 再加几道保险（进程异常/信号也关）
|
||
process.on('exit', killIOSAI);
|
||
process.on('SIGINT', killIOSAI);
|
||
process.on('SIGTERM', killIOSAI);
|
||
|
||
function toMB(v) {
|
||
if (!v || v <= 0) return 0;
|
||
// 某些平台字段是 bytes、也有历史 Electron 返回 KB 的情况，做兜底
|
||
// 这里按优先：bytes->MB，其次 KB->MB，最后假定已经是MB
|
||
// const bytesToMB = v / (1024 * 1024);
|
||
// if (bytesToMB > 1) return Number(bytesToMB.toFixed(1)); // 看起来像 bytes
|
||
const kbToMB = v / 1024;
|
||
if (kbToMB > 1) return Number(kbToMB.toFixed(1)); // 看起来像 KB
|
||
return Number(v.toFixed(1)); // 当作 MB
|
||
}
|
||
|
||
function dumpAllMem() {
|
||
try {
|
||
const metrics = app.getAppMetrics();
|
||
|
||
const report = metrics.map(m => {
|
||
const mem = m.memory || {};
|
||
const workingSetMB = toMB(mem.workingSetSize ?? mem.workingSet ?? 0);
|
||
const privateMB = toMB(mem.privateBytes ?? mem.private ?? 0);
|
||
const sharedMB = toMB(mem.shared ?? 0);
|
||
return { pid: m.pid, type: m.type, workingSetMB, privateMB, sharedMB };
|
||
});
|
||
|
||
// console.log('>Tab workingSetMB=', report[3].workingSetMB);
|
||
|
||
// // 渲染进程类型在不同版本可能是 "Renderer" 或 "Tab"
|
||
// const isRenderer = (t) => t === 'Renderer' || t === 'Tab';
|
||
// const renderers = report.filter(r => isRenderer(r.type));
|
||
// const gpu = report.filter(r => r.type.toLowerCase().includes('gpu'));
|
||
|
||
// const sum = (arr, k) => arr.reduce((a, b) => a + (b[k] || 0), 0).toFixed(1);
|
||
// console.log('> [MEM] SUM Renderer workingSetMB=', sum(renderers, 'workingSetMB'));
|
||
// if (gpu.length) console.log('> [MEM] GPU=', gpu);
|
||
} catch (e) {
|
||
console.warn('获取应用指标错误:', e);
|
||
|
||
}
|
||
}
|
||
|
||
// 在函数外定义计数器（或者放在函数内部，用闭包封装）
|
||
let consumeCount = 0;
|
||
async function setupMQConsumerAndPublisher(emitMessage, tenantId) {
|
||
const queueNames = [
|
||
`q.tenant.${tenantId}`,
|
||
`b.tenant.${tenantId}`, // 新增队列
|
||
];
|
||
|
||
for (const qName of queueNames) {
|
||
await mq.startConsumer(
|
||
qName,
|
||
async (msg) => {
|
||
const payload = msg.json ?? msg.text; // 原始业务数据
|
||
consumeCount++; // 所有队列共用计数器
|
||
|
||
// 标记来源类型：普通队列 / burst 队列
|
||
const isBurstQueue = qName.startsWith('b.');
|
||
const meta = isBurstQueue ? 2 : 1;
|
||
|
||
console.log(
|
||
`[MQ消费] [${qName}]`,
|
||
payload?.hostsId,
|
||
payload?.country,
|
||
'共' + consumeCount + '条数据'
|
||
);
|
||
|
||
// ⚠️ 关键：在原有 payload 的基础上，增加 _mqMeta 字段
|
||
// 这样你之前前端用 payload.hostsId / payload.country 的地方完全不用改
|
||
const wrapped = {
|
||
...payload,
|
||
_mqMeta: meta
|
||
};
|
||
|
||
// 广播到前端
|
||
emitMessage(wrapped);
|
||
// 成功返回会在 mq 客户端内部自动 ack
|
||
},
|
||
{ prefetch: 1, requeueOnError: false, durable: true, assertQueue: true }
|
||
);
|
||
}
|
||
|
||
// 供渲染进程发送消息到队列（保持原来的 q.tenant.* 不变）
|
||
ipcMain.removeHandler('mq-send');
|
||
ipcMain.handle('mq-send', async (_event, user) => {
|
||
console.log('消息已发送', user);
|
||
await mq.publishToQueue(`q.tenant.${tenantId}`, user);
|
||
return { ok: true };
|
||
});
|
||
}
|
||
|
||
|
||
|
||
/**
|
||
* 创建 BrowserWindow，并在加载 http://localhost:8080/ 失败时启动定时检测，成功后自动 reload
|
||
*/
|
||
function createWindow() {
|
||
const appVersion = app.getVersion(); // 获取 package.json 中的 version
|
||
const win = new BrowserWindow({
|
||
width: 1920,
|
||
height: 1080,
|
||
title: `YOLO（AI助手ios） v${appVersion}`, // 设置窗口标题
|
||
frame: true,
|
||
fullscreenable: false,
|
||
webPreferences: {
|
||
preload: path.join(__dirname, 'js', 'preload.js'),
|
||
contextIsolation: true,
|
||
nodeIntegration: false,
|
||
|
||
// 推荐：默认 true；只有 SAFE_MODE 才 false（等价 --no-sandbox）
|
||
sandbox: SAFE_MODE ? false : true,
|
||
|
||
webSecurity: true,
|
||
backgroundThrottling: false,
|
||
enableRemoteModule: false,
|
||
offscreen: false,
|
||
|
||
// 兼容模式更稳：关实验特性
|
||
experimentalFeatures: SAFE_MODE ? false : true,
|
||
|
||
autoplayPolicy: 'no-user-gesture-required',
|
||
devTools: true,
|
||
}
|
||
});
|
||
|
||
win.setMenu(null); // 禁用菜单栏F12
|
||
win.maximize();
|
||
|
||
|
||
// ✅ 把渲染进程的 console.* 打进 main 的日志
|
||
win.webContents.on('console-message', (event, level, message, line, sourceId) => {
|
||
// level: 0=log,1=warn,2=error,3=info,4=debug
|
||
if (level == 1) {
|
||
console.log(
|
||
`[页面日志] ${message}`
|
||
);
|
||
}
|
||
|
||
});
|
||
|
||
|
||
// 自动判断环境使用不同的页面地址
|
||
const isProd = app.isPackaged;
|
||
// const targetURL = isProd ? 'https://iosaitest.yolozs.com' : 'http://192.168.1.128:8080';
|
||
const targetURL = isProd ? 'https://iosai.yolozs.com' : 'http://192.168.1.128:8080';
|
||
console.log('[页面加载] 使用地址:', targetURL);
|
||
|
||
let retryIntervalId = null;
|
||
|
||
// 启动定时重试逻辑
|
||
function scheduleRetry() {
|
||
const intervalMs = 5000; // 每 5 秒检测一次，可根据需求调整
|
||
console.log(`[Auto-Reconnect] 启动定时检测，每 ${intervalMs / 1000}s 重新检测服务`);
|
||
}
|
||
// 统一的“尝试跳转”函数：更新中 -> 延迟；否则直接跳
|
||
const tryNavigate = (reasonTag = '') => {
|
||
const ts = Date.now() // 或者用前端版本号
|
||
const go = () => safeLoadURL(win, `${targetURL}?t=${ts}`, () => {
|
||
if (!retryIntervalId) scheduleRetry();
|
||
});
|
||
|
||
if (updateInProgress) {
|
||
console.log(`[Nav] 阻塞跳转（${reasonTag}）：正在更新中，等待解锁`);
|
||
pendingNavs.push(go); // 等待 autoUpdater 事件解锁后 flush 执行
|
||
return;
|
||
}
|
||
go();
|
||
};
|
||
|
||
// 监听前端请求文件选择
|
||
ipcMain.handle('select-apk-file', async () => {
|
||
const result = await dialog.showOpenDialog({
|
||
title: '选择 APK 文件',
|
||
filters: [{ name: 'APK', extensions: ['apk'] }],
|
||
properties: ['openFile']
|
||
})
|
||
|
||
if (result.canceled || result.filePaths.length === 0) {
|
||
return null
|
||
}
|
||
|
||
return result.filePaths[0]
|
||
})
|
||
// 监听前端请求文件选择
|
||
ipcMain.handle('select-file', async () => {
|
||
const result = await dialog.showOpenDialog({
|
||
title: '选择文件',
|
||
// filters: [{ name: 'APK', extensions: ['apk'] }],
|
||
properties: ['openFile']
|
||
})
|
||
|
||
if (result.canceled || result.filePaths.length === 0) {
|
||
return null
|
||
}
|
||
|
||
return result.filePaths[0]
|
||
})
|
||
|
||
// 接收渲染进程请求执行 GC
|
||
ipcMain.handle('manual-gc', () => {
|
||
if (global.gc) {
|
||
global.gc();
|
||
console.log('🧹 手动触发 GC 成功');
|
||
} else {
|
||
console.warn('⚠️ global.gc 不存在，请确认是否添加了 --expose-gc');
|
||
}
|
||
});
|
||
|
||
|
||
// 小工具：异步 sleep
|
||
const sleep = (ms) => new Promise(r => setTimeout(r, ms));
|
||
|
||
/**
|
||
* 前端调用：一直检测，直到检测到 iproxy.exe 才返回
|
||
* 用法：
|
||
* const res = await window.electron.ipcRenderer.invoke('iproxy:waitUntilRunning', { intervalMs: 2000, exeName: 'iproxy.exe' })
|
||
* if (res.running) { /* 决定是否跳转 *\/ }
|
||
*/
|
||
ipcMain.handle('isiproxy', async (_event, opts = {}) => {
|
||
startIOSAIExecutable(); //启动iosAi服务
|
||
const { intervalMs = 2000, exeName = 'iproxy.exe', maxWaitMs = 0 } = opts;
|
||
|
||
// 非 Windows 平台直接认为通过（按需改）
|
||
if (process.platform !== 'win32') return { running: true, platform: process.platform };
|
||
|
||
const start = Date.now();
|
||
while (true) {
|
||
const ok = await isProcessRunningWin(exeName);
|
||
if (ok) return { running: true };
|
||
if (maxWaitMs > 0 && (Date.now() - start) >= maxWaitMs) {
|
||
return { running: false, timeout: true };
|
||
}
|
||
await sleep(intervalMs);
|
||
}
|
||
});
|
||
|
||
|
||
|
||
// 仅检查固定路径：C:\Users\Administrator\IOSAI\aiConfig.json
|
||
ipcMain.handle('file-exists', async () => {
|
||
try {
|
||
// ★ 固定路径（如需改用户，改这里即可）
|
||
const fixedPath = `${USER_ROOT}\\IOSAI\\aiConfig.json`;
|
||
|
||
// 用 stat 判断是否存在 & 获取信息
|
||
const stat = await fsp.stat(fixedPath).catch(err => {
|
||
if (err && err.code === 'ENOENT') return null; // 不存在
|
||
throw err; // 其他错误抛出
|
||
});
|
||
|
||
if (!stat) {
|
||
return { ok: true, exists: false, path: fixedPath };
|
||
}
|
||
|
||
return {
|
||
ok: true,
|
||
exists: true,
|
||
path: fixedPath,
|
||
isFile: stat.isFile(),
|
||
isDirectory: stat.isDirectory(),
|
||
size: stat.size,
|
||
mtimeMs: stat.mtimeMs
|
||
};
|
||
} catch (e) {
|
||
return { ok: false, error: e.message || String(e) };
|
||
}
|
||
});
|
||
|
||
|
||
|
||
|
||
// // 判断文件存不存在
|
||
// ipcMain.handle('file-exists', async (_evt, targetPath, baseDir /* 可选 */) => {
|
||
// try {
|
||
// const full = normalizePath(targetPath, baseDir)
|
||
// const stat = await fsp.stat(full).catch(err => {
|
||
// if (err && err.code === 'ENOENT') return null
|
||
// throw err
|
||
// })
|
||
// if (!stat) {
|
||
// return { ok: true, exists: false, path: full }
|
||
// }
|
||
// return {
|
||
// ok: true,
|
||
// exists: true,
|
||
// path: full,
|
||
// isFile: stat.isFile(),
|
||
// isDirectory: stat.isDirectory(),
|
||
// size: stat.size,
|
||
// mtimeMs: stat.mtimeMs
|
||
// }
|
||
// } catch (e) {
|
||
// return { ok: false, error: e.message || String(e) }
|
||
// }
|
||
// })
|
||
|
||
// 监听加载失败事件
|
||
win.webContents.on('did-fail-load', (event, errorCode, errorDescription, validatedURL, isMainFrame) => {
|
||
if (isMainFrame) {
|
||
console.warn(`[did-fail-load] 加载失败(${errorCode}): ${errorDescription}，URL: ${validatedURL}`);
|
||
if (!retryIntervalId) {
|
||
scheduleRetry();
|
||
}
|
||
}
|
||
});
|
||
|
||
|
||
|
||
|
||
// 监听加载完成事件：一旦成功加载，清除重试定时器
|
||
win.webContents.on('did-finish-load', () => {
|
||
const currentURL = win.webContents.getURL();
|
||
console.log(`[did-finish-load] 加载完成，当前 URL: ${currentURL}`)
|
||
|
||
if (retryIntervalId) {
|
||
clearInterval(retryIntervalId);
|
||
retryIntervalId = null;
|
||
}
|
||
|
||
});
|
||
|
||
// === 首次加载：先上“加载页” ===
|
||
const loadingFile = path.join(__dirname, 'waiting.html');
|
||
const hasLoading = fs.existsSync(loadingFile);
|
||
|
||
if (process.platform === 'win32') {
|
||
if (hasLoading) {
|
||
win.loadFile(loadingFile).catch(err => {
|
||
console.warn('[加载页 loadFile 失败，退回内联HTML]', err);
|
||
// 兜底：内联一个极简页面
|
||
win.loadURL('data:text/html;charset=utf-8,' + encodeURIComponent('<h3>正在等待 iproxy.exe 启动…</h3>'));
|
||
});
|
||
} else {
|
||
// 如果没有把加载页文件放到指定位置，给个兜底
|
||
win.loadURL('data:text/html;charset=utf-8,' + encodeURIComponent('<h3>正在等待 iproxy.exe 启动…</h3>'));
|
||
}
|
||
// ✅ 不再自动检测 iproxy，3 秒后直接进入业务页面
|
||
setTimeout(() => {
|
||
tryNavigate('delay-3s-autogo');
|
||
}, 3000);
|
||
// // 开始检测 iproxy，检测到后再跳转 targetURL
|
||
// waitForIproxyAndNavigate(win, targetURL, {
|
||
// intervalMs: 2000,
|
||
// timeoutMs: 999999,
|
||
// exeName: 'iproxy.exe'
|
||
// }, tryNavigate);
|
||
} else {
|
||
// 非 Windows 平台直接进页面
|
||
tryNavigate('non-win-first');
|
||
}
|
||
|
||
|
||
//启动sse
|
||
sseServer = startSSE();
|
||
const emitMessage = createBurstBroadcaster(sseServer.broadcast, {
|
||
event: 'message', // 你前端监 n听的事件名
|
||
idleMs: 10_000,
|
||
startPayload: 'start',
|
||
startOnFirst: true // 如果不想程序启动后的第一条先发 start，就设为 false
|
||
});
|
||
|
||
// 开启mq
|
||
ipcMain.handle('start-mq', async (event, tentId, id) => {
|
||
userData.tenantId = tentId;
|
||
userData.userId = id;
|
||
//启动mq
|
||
setupMQConsumerAndPublisher(emitMessage, tentId)
|
||
|
||
});
|
||
// 可选：开发阶段打开 DevTools F12
|
||
// win.webContents.openDevTools();
|
||
|
||
return win;
|
||
}
|
||
|
||
|
||
|
||
// —— 启动 iOSAI.exe ——
|
||
// 兼容开发/打包路径：优先同级目录，其次 resources，再到源码目录
|
||
function resolveIOSAIPath() {
|
||
const candidates = [
|
||
// 打包后：和应用同级目录（app.exe 同目录）
|
||
path.join(path.dirname(process.execPath), 'iOSAI', 'IOSAI.exe'),
|
||
// 打包后：在 resources 目录（需要用 electron-builder 的 extraResources 拷贝进去）
|
||
path.join(process.resourcesPath || '', 'iOSAI', 'IOSAI.exe'),
|
||
// 开发环境：以当前文件目录为基准
|
||
path.join(__dirname, 'iOSAI', 'IOSAI.exe'),
|
||
// 兜底：以当前工作目录为基准
|
||
path.join(process.cwd(), 'iOSAI', 'IOSAI.exe'),
|
||
];
|
||
for (const p of candidates) {
|
||
try {
|
||
if (fs.existsSync(p)) return p;
|
||
} catch (_) { }
|
||
}
|
||
return null;
|
||
}
|
||
|
||
let iosAIProcess = null; // 如果你后面想控制/关闭它，用得到；“只要打开就行”也可以不管
|
||
|
||
// function startIOSAIExecutable() {
|
||
// if (process.platform !== 'win32') {
|
||
// console.warn('[IOSAI] 当前不是 Windows，跳过启动 IOSAI.exe');
|
||
// return;
|
||
// }
|
||
// const exePath = resolveIOSAIPath();
|
||
// if (!exePath) {
|
||
// console.error('[IOSAI] 未找到 IOSAI.exe，请确认路径：根目录/iOSAI/IOSAI.exe（或已打包到 resources）');
|
||
// return;
|
||
// }
|
||
|
||
// const exeDir = path.dirname(exePath);
|
||
// const exeFile = path.basename(exePath);
|
||
|
||
// // 打印确认：应为 ...\IOSAI.exe
|
||
// console.log('[IOSAI] 计划启动文件：', exePath);
|
||
// console.log('[IOSAI] 文件存在：', fs.existsSync(exePath));
|
||
|
||
// // 注意：start 的第一个带引号参数是窗口标题，这里用 "" 占位
|
||
// // /D 指定工作目录；最后只传文件名，避免整条路径再被拆分
|
||
// // const cmd = `start "" /D "${exeDir}" "${exeFile}"`;
|
||
// const params = 'iosai';
|
||
|
||
// const cmd = `start "" /D "${exeDir}" "${exeFile}" ${params}`;
|
||
// try {
|
||
// exec(cmd, { cwd: exeDir, windowsHide: false }, (err) => {
|
||
// if (err) {
|
||
// console.error('[IOSAI] 启动失败：', err);
|
||
// } else {
|
||
// console.log('[IOSAI] 启动命令已执行（独立控制台窗口中运行）');
|
||
// }
|
||
// });
|
||
// } catch (e) {
|
||
// console.error('[IOSAI] 启动异常：', e);
|
||
// }
|
||
// }
|
||
|
||
// 启动 IOSAI.exe（管理员权限）
|
||
// 检查当前 Electron 是否已是管理员（高完整性）
|
||
function isElevated(cb) {
|
||
// 查询当前进程完整性级别：High 表示管理员
|
||
exec('whoami /groups', { windowsHide: true }, (err, stdout) => {
|
||
if (err) return cb(false);
|
||
cb(/High Mandatory Level/i.test(stdout));
|
||
});
|
||
}
|
||
|
||
function psEscapeSingleQuotes(s) {
|
||
return s.replace(/'/g, "''");
|
||
}
|
||
|
||
function startIOSAIExecutable() {
|
||
if (process.platform !== 'win32') {
|
||
console.warn('[IOSAI] 当前不是 Windows，跳过启动 IOSAI.exe');
|
||
return;
|
||
}
|
||
|
||
const exePath = resolveIOSAIPath();
|
||
if (!exePath || !fs.existsSync(exePath)) {
|
||
console.error('[IOSAI] 未找到 IOSAI.exe：', exePath);
|
||
return;
|
||
}
|
||
const exeDir = path.dirname(exePath);
|
||
const args = ['iosai'];
|
||
|
||
isElevated((elev) => {
|
||
console.log('[IOSAI] 当前 Electron 是否管理员：', elev);
|
||
|
||
if (elev) {
|
||
const exeFile = path.basename(exePath);
|
||
|
||
// 已经是管理员：直接普通方式启动即可
|
||
const cmd = `start "" /D "${exeDir}" "${exeFile}" ${args[0]}`;
|
||
exec(cmd, { cwd: exeDir, windowsHide: false }, (err) => {
|
||
console.log(err)
|
||
if (err) console.error('[IOSAI] 启动失败：', err);
|
||
else console.log('[IOSAI] 已启动（父进程已是管理员，无需 UAC 弹窗）');
|
||
});
|
||
return;
|
||
}
|
||
|
||
// 未提权 → 用 PowerShell 提权
|
||
const exePathPS = psEscapeSingleQuotes(exePath);
|
||
const exeDirPS = psEscapeSingleQuotes(exeDir);
|
||
const argListPS = `@(${args.map(a => `'${psEscapeSingleQuotes(String(a))}'`).join(',')})`;
|
||
|
||
const psCommand = `Start-Process -FilePath '${exePathPS}' -ArgumentList ${argListPS} -WorkingDirectory '${exeDirPS}' -Verb RunAs -WindowStyle Normal -PassThru`;
|
||
console.log('[IOSAI] 提权启动命令(PS)：', psCommand);
|
||
|
||
execFile(
|
||
'powershell.exe',
|
||
['-NoProfile', '-ExecutionPolicy', 'Bypass', '-Command', psCommand],
|
||
{ windowsHide: true },
|
||
(err) => {
|
||
if (err) {
|
||
console.error('[IOSAI] 提权启动失败：', err);
|
||
} else {
|
||
console.log('[IOSAI] 已请求管理员启动（按系统策略可能不弹 UAC）');
|
||
// 可选：用 tasklist 简单确认是否起来了
|
||
setTimeout(() => {
|
||
exec('tasklist /FI "IMAGENAME eq IOSAI.exe"', { windowsHide: true }, (e2, out) => {
|
||
console.log('[IOSAI] 进程检查：\n', out);
|
||
});
|
||
}, 1200);
|
||
}
|
||
}
|
||
);
|
||
});
|
||
}
|
||
|
||
|
||
async function killIOSAI() {
|
||
console.log('尝试关闭IOSAI', userData);
|
||
// console.log('axios', axios);
|
||
|
||
await axios.post('https://crawlclient.api.yolozs.com/api/user/aiChat-logout', { userId: userData.userId, tenantId: userData.tenantId }, {
|
||
headers: {
|
||
'Content-Type': 'application/json', // 设置请求头
|
||
'vvtoken': userData.tokenValue
|
||
}
|
||
}).then(response => {
|
||
console.log("发送登出请求成功")
|
||
}).catch(error => {
|
||
console.log("发送登出请求错误", error)
|
||
}).finally(error => {
|
||
// console.log("发送")
|
||
})
|
||
try {
|
||
|
||
if (iosAIProcess && iosAIProcess.pid) {
|
||
// 按 PID 杀，并带上 /T 杀掉它可能拉起的子进程
|
||
exec(`taskkill /PID ${iosAIProcess.pid} /T /F`, (err) => {
|
||
if (err) {
|
||
console.warn('[IOSAI] 按 PID 关闭失败，改用按进程名：', err.message);
|
||
// 兜底：按进程名杀（会杀掉系统里所有 IOSAI.exe，若允许同名多实例请注意）
|
||
exec('taskkill /IM IOSAI.exe /F');
|
||
}
|
||
});
|
||
} else {
|
||
// 没有 PID（比如程序二次拉起后原 PID 退出），直接按进程名兜底
|
||
exec('taskkill /IM IOSAI.exe /F');
|
||
exec('taskkill /IM iproxy.exe /F');
|
||
exec('taskkill /IM pythonw.exe /F');
|
||
exec('taskkill /IM ios.exe /F');
|
||
}
|
||
} catch (_) { }
|
||
iosAIProcess = null;
|
||
}
|
||
|
||
const broadcast = (channel, payload) => {
|
||
BrowserWindow.getAllWindows().forEach(w => {
|
||
if (!w.isDestroyed()) w.webContents.send(channel, payload);
|
||
});
|
||
};
|
||
|
||
|
||
// —— 进程检测：Windows 使用 tasklist 精确匹配 ——
|
||
// 返回 Promise<boolean>
|
||
function isProcessRunningWin(exeName) {
|
||
return new Promise((resolve) => {
|
||
if (process.platform !== 'win32') return resolve(true); // 非 Win 平台直接放行
|
||
// /FI 过滤进程名；/FO CSV 便于稳定解析；/NH 去表头
|
||
const cmd = `tasklist /FI "IMAGENAME eq ${exeName}" /FO CSV /NH`;
|
||
exec(cmd, { windowsHide: true }, (err, stdout) => {
|
||
if (err || !stdout) return resolve(false);
|
||
// 匹配形如："iproxy.exe","xxxx","Console",...
|
||
const found = stdout.toLowerCase().includes(`"${exeName.toLowerCase()}"`);
|
||
resolve(found);
|
||
});
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 等待 iproxy.exe 启动后跳转到目标 URL
|
||
* @param {BrowserWindow} win
|
||
* @param {string} targetURL
|
||
* @param {object} options
|
||
*/
|
||
function waitForIproxyAndNavigate(win, targetURL, options = {}, tryNavigate) {
|
||
const {
|
||
intervalMs = 2000, // 轮询间隔
|
||
timeoutMs = 999999, // 超时提示（不强制跳转）
|
||
exeName = 'iproxy.exe'
|
||
} = options;
|
||
|
||
let elapsed = 0;
|
||
// 加个不确定进度（任务栏）提示
|
||
win.setProgressBar(2, { mode: 'indeterminate' });
|
||
|
||
const timer = setInterval(async () => {
|
||
const ok = await isProcessRunningWin(exeName);
|
||
if (ok) {
|
||
clearInterval(timer);
|
||
win.setProgressBar(-1);
|
||
console.log(`[iproxy] 检测到 ${exeName} 已运行，进入页面：${targetURL}`);
|
||
// 检测到后再真正加载业务页面
|
||
tryNavigate('no-waiting-file')
|
||
return;
|
||
}
|
||
elapsed += intervalMs;
|
||
if (elapsed >= timeoutMs) {
|
||
// 仅提示，不强制退出。你也可以在此弹窗或修改窗口标题
|
||
console.warn(`[iproxy] 等待 ${exeName} 超时 ${timeoutMs / 1000}s，仍未检测到进程。`);
|
||
win.setTitle('YOLO（AI助手ios） - 正在等待 iproxy.exe 启动…（可检查连接/杀软/权限）');
|
||
// 到时间后依旧保持加载页 + 继续轮询（如果想停止轮询，把 clearInterval 放这里）
|
||
// clearInterval(timer);
|
||
}
|
||
}, intervalMs);
|
||
}
|
||
|
||
|
||
/**
|
||
* 将传入路径标准化：
|
||
* - 支持 file:// URL
|
||
* - 相对路径会按 baseDir 解析（不传 baseDir 则按当前进程工作目录）
|
||
*/
|
||
function normalizePath(targetPath, baseDir) {
|
||
if (typeof targetPath !== 'string' || !targetPath.trim()) {
|
||
throw new Error('无效的路径参数')
|
||
}
|
||
// 支持 file://
|
||
if (targetPath.startsWith('file://')) {
|
||
return fileURLToPath(new URL(targetPath))
|
||
}
|
||
// 相对路径 → 解析到 baseDir
|
||
if (!path.isAbsolute(targetPath)) {
|
||
const base = baseDir && typeof baseDir === 'string' ? baseDir : process.cwd()
|
||
return path.resolve(base, targetPath)
|
||
}
|
||
return path.resolve(targetPath)
|
||
} |