main.ts 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021
  1. // 使用 CommonJS 格式
  2. const { app, BrowserWindow, ipcMain, shell, session, Menu, Tray, nativeImage, webContents } = require('electron');
  3. const { join } = require('path');
  4. const fs = require('fs');
  5. const http = require('http');
  6. const https = require('https');
  7. import { startLocalServices, stopLocalServices, getServiceStatus, LOCAL_NODE_URL, LOCAL_PYTHON_URL, getLogPath } from './local-services';
  8. let mainWindow: typeof BrowserWindow.prototype | null = null;
  9. let tray: typeof Tray.prototype | null = null;
  10. let isQuitting = false;
  11. const VITE_DEV_SERVER_URL = process.env.VITE_DEV_SERVER_URL;
  12. function setupCertificateBypass() {
  13. app.on('certificate-error', (event: Event, _webContents: typeof webContents.prototype, _url: string, _error: string, _certificate: unknown, callback: (isTrusted: boolean) => void) => {
  14. event.preventDefault();
  15. callback(true);
  16. });
  17. }
  18. function setupCorsBypassForApiRequests() {
  19. const ses = session.defaultSession;
  20. if (!ses) return;
  21. ses.webRequest.onHeadersReceived((details: { url: string; responseHeaders?: Record<string, string[] | string> }, callback: (response: { responseHeaders: Record<string, string[] | string> }) => void) => {
  22. const url = String(details.url || '');
  23. const isHttp = url.startsWith('http://') || url.startsWith('https://');
  24. const isApiLike = url.includes('/api/') || url.includes('/uploads/');
  25. if (!isHttp || !isApiLike) {
  26. callback({ responseHeaders: details.responseHeaders || {} });
  27. return;
  28. }
  29. const responseHeaders = { ...(details.responseHeaders || {}) };
  30. // 移除服务端已有的 CORS 头,避免与下面设置的值合并成 "origin1, *" 导致违规
  31. const corsKeys = ['access-control-allow-origin', 'Access-Control-Allow-Origin'];
  32. corsKeys.forEach((k) => delete responseHeaders[k]);
  33. responseHeaders['access-control-allow-origin'] = ['*'];
  34. responseHeaders['access-control-allow-methods'] = ['GET,POST,PUT,PATCH,DELETE,OPTIONS'];
  35. responseHeaders['access-control-allow-headers'] = ['Authorization,Content-Type,X-Requested-With'];
  36. responseHeaders['access-control-expose-headers'] = ['Content-Disposition,Content-Type'];
  37. callback({ responseHeaders });
  38. });
  39. }
  40. function normalizeBaseUrl(url: string): string {
  41. const raw = String(url || '').trim();
  42. if (!raw) return '';
  43. try {
  44. const u = new URL(raw);
  45. return `${u.protocol}//${u.host}`.replace(/\/$/, '');
  46. } catch {
  47. return raw.replace(/\/$/, '');
  48. }
  49. }
  50. function requestJson(url: string, timeoutMs: number): Promise<{ ok: boolean; status?: number; data?: any; error?: string }> {
  51. return new Promise((resolve) => {
  52. const u = new URL(url);
  53. const isHttps = u.protocol === 'https:';
  54. const lib = isHttps ? https : http;
  55. // Windows 上 localhost 常被解析为 ::1,而后端仅监听 127.0.0.1,导致 ECONNREFUSED
  56. const hostname = (u.hostname === 'localhost' || u.hostname === '::1') ? '127.0.0.1' : u.hostname;
  57. const req = lib.request({
  58. method: 'GET',
  59. protocol: u.protocol,
  60. hostname,
  61. port: u.port || (isHttps ? 443 : 80),
  62. path: `${u.pathname}${u.search}`,
  63. headers: {
  64. Accept: 'application/json',
  65. },
  66. timeout: timeoutMs,
  67. rejectUnauthorized: false,
  68. }, (res: any) => {
  69. const chunks: Buffer[] = [];
  70. res.on('data', (c: Buffer) => chunks.push(c));
  71. res.on('end', () => {
  72. const status = Number(res.statusCode || 0);
  73. const rawText = Buffer.concat(chunks).toString('utf-8');
  74. if (status < 200 || status >= 300) {
  75. resolve({ ok: false, status, error: `HTTP ${status}` });
  76. return;
  77. }
  78. try {
  79. const json = rawText ? JSON.parse(rawText) : null;
  80. resolve({ ok: true, status, data: json });
  81. } catch {
  82. resolve({ ok: false, status, error: '响应不是 JSON' });
  83. }
  84. });
  85. });
  86. req.on('timeout', () => {
  87. req.destroy(new Error('timeout'));
  88. });
  89. req.on('error', (err: any) => {
  90. resolve({ ok: false, error: err?.message || '网络错误' });
  91. });
  92. req.end();
  93. });
  94. }
  95. // 获取图标路径
  96. function getIconPath() {
  97. return VITE_DEV_SERVER_URL
  98. ? join(__dirname, '../public/icons/icon-256.png')
  99. : join(__dirname, '../dist/icons/icon-256.png');
  100. }
  101. // 获取托盘图标路径
  102. function getTrayIconPath() {
  103. return VITE_DEV_SERVER_URL
  104. ? join(__dirname, '../public/icons/tray-icon.png')
  105. : join(__dirname, '../dist/icons/tray-icon.png');
  106. }
  107. // 创建托盘图标
  108. function createTrayIcon(): typeof nativeImage.prototype {
  109. const trayIconPath = getTrayIconPath();
  110. return nativeImage.createFromPath(trayIconPath);
  111. }
  112. // 创建系统托盘
  113. function createTray() {
  114. const trayIcon = createTrayIcon();
  115. tray = new Tray(trayIcon);
  116. const contextMenu = Menu.buildFromTemplate([
  117. {
  118. label: '显示主窗口',
  119. click: () => {
  120. if (mainWindow) {
  121. mainWindow.show();
  122. mainWindow.focus();
  123. }
  124. }
  125. },
  126. {
  127. label: '最小化到托盘',
  128. click: () => {
  129. mainWindow?.hide();
  130. }
  131. },
  132. { type: 'separator' },
  133. {
  134. label: '退出',
  135. click: () => {
  136. isQuitting = true;
  137. app.quit();
  138. }
  139. }
  140. ]);
  141. tray.setToolTip('智媒通');
  142. tray.setContextMenu(contextMenu);
  143. // 点击托盘图标显示窗口
  144. tray.on('click', () => {
  145. if (mainWindow) {
  146. if (mainWindow.isVisible()) {
  147. mainWindow.focus();
  148. } else {
  149. mainWindow.show();
  150. mainWindow.focus();
  151. }
  152. }
  153. });
  154. // 双击托盘图标显示窗口
  155. tray.on('double-click', () => {
  156. if (mainWindow) {
  157. mainWindow.show();
  158. mainWindow.focus();
  159. }
  160. });
  161. }
  162. function createWindow() {
  163. // 隐藏默认菜单栏
  164. Menu.setApplicationMenu(null);
  165. const iconPath = getIconPath();
  166. mainWindow = new BrowserWindow({
  167. width: 1400,
  168. height: 900,
  169. minWidth: 1200,
  170. minHeight: 700,
  171. icon: iconPath,
  172. webPreferences: {
  173. preload: join(__dirname, 'preload.js'),
  174. nodeIntegration: false,
  175. contextIsolation: true,
  176. webviewTag: true, // 启用 webview 标签
  177. },
  178. frame: false, // 无边框窗口,自定义标题栏
  179. transparent: false,
  180. backgroundColor: '#f0f2f5',
  181. show: false,
  182. });
  183. // 窗口准备好后再显示,避免白屏
  184. mainWindow.once('ready-to-show', () => {
  185. mainWindow?.show();
  186. setupWindowEvents();
  187. });
  188. // 加载页面
  189. if (VITE_DEV_SERVER_URL) {
  190. mainWindow.loadURL(VITE_DEV_SERVER_URL);
  191. mainWindow.webContents.openDevTools();
  192. } else {
  193. mainWindow.loadFile(join(__dirname, '../dist/index.html'));
  194. }
  195. // 处理外部链接
  196. mainWindow.webContents.setWindowOpenHandler(({ url }: { url: string }) => {
  197. shell.openExternal(url);
  198. return { action: 'deny' };
  199. });
  200. // 关闭按钮默认最小化到托盘
  201. mainWindow.on('close', (event: Event) => {
  202. if (!isQuitting) {
  203. event.preventDefault();
  204. mainWindow?.hide();
  205. // 显示托盘通知(仅首次)
  206. if (tray && !app.isPackaged) {
  207. // 开发模式下可以显示通知
  208. }
  209. }
  210. });
  211. mainWindow.on('closed', () => {
  212. mainWindow = null;
  213. });
  214. }
  215. // 单实例锁定
  216. const gotTheLock = app.requestSingleInstanceLock();
  217. if (!gotTheLock) {
  218. app.quit();
  219. } else {
  220. app.on('second-instance', () => {
  221. if (mainWindow) {
  222. mainWindow.show();
  223. if (mainWindow.isMinimized()) mainWindow.restore();
  224. mainWindow.focus();
  225. }
  226. });
  227. app.whenReady().then(async () => {
  228. // 先创建窗口显示 splash screen
  229. createWindow();
  230. createTray();
  231. // 后台启动本地 Node 和 Python 服务,不阻塞窗口显示
  232. console.log('[Main] 正在后台启动本地服务...');
  233. startLocalServices().then(({ nodeOk, pythonOk }) => {
  234. console.log(`[Main] 本地服务状态: Node=${nodeOk ? '就绪' : '未就绪'}, Python=${pythonOk ? '就绪' : '未就绪'}`);
  235. // 通知渲染进程服务状态变化
  236. mainWindow?.webContents.send('services-status-changed', { nodeOk, pythonOk });
  237. });
  238. // 配置 webview session,允许第三方 cookies 和跨域请求
  239. setupWebviewSessions();
  240. setupCertificateBypass();
  241. setupCorsBypassForApiRequests();
  242. app.on('activate', () => {
  243. if (BrowserWindow.getAllWindows().length === 0) {
  244. createWindow();
  245. } else if (mainWindow) {
  246. mainWindow.show();
  247. }
  248. });
  249. });
  250. }
  251. // 配置 webview sessions
  252. function setupWebviewSessions() {
  253. // 监听新的 webContents 创建
  254. app.on('web-contents-created', (_event: unknown, contents: typeof webContents.prototype) => {
  255. // 为 webview 类型的 webContents 配置
  256. if (contents.getType() === 'webview') {
  257. // 设置 User-Agent(模拟 Chrome 浏览器)
  258. contents.setUserAgent(
  259. 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
  260. );
  261. // 拦截自定义协议链接(如 bitbrowser://)的导航
  262. contents.on('will-navigate', (event: Event, url: string) => {
  263. if (!isAllowedUrl(url)) {
  264. console.log('[WebView] 阻止导航到自定义协议:', url);
  265. event.preventDefault();
  266. }
  267. });
  268. // 拦截新窗口打开(包括自定义协议)
  269. contents.setWindowOpenHandler(({ url }: { url: string }) => {
  270. if (!isAllowedUrl(url)) {
  271. console.log('[WebView] 阻止打开自定义协议窗口:', url);
  272. return { action: 'deny' };
  273. }
  274. // 对于正常的 http/https 链接,在当前 webview 中打开
  275. console.log('[WebView] 拦截新窗口,在当前页面打开:', url);
  276. contents.loadURL(url);
  277. return { action: 'deny' };
  278. });
  279. // 允许所有的权限请求(如摄像头、地理位置等)
  280. contents.session.setPermissionRequestHandler((_webContents: unknown, permission: string, callback: (granted: boolean) => void) => {
  281. // 允许所有权限请求
  282. callback(true);
  283. });
  284. // 配置 webRequest 修改请求头,移除可能暴露 Electron 的特征
  285. contents.session.webRequest.onBeforeSendHeaders((details: { requestHeaders: Record<string, string> }, callback: (response: { requestHeaders: Record<string, string> }) => void) => {
  286. // 移除可能暴露 Electron 的请求头
  287. delete details.requestHeaders['X-DevTools-Emulate-Network-Conditions-Client-Id'];
  288. // 确保有正常的 Origin 和 Referer
  289. if (!details.requestHeaders['Origin'] && !details.requestHeaders['origin']) {
  290. // 不添加 Origin,让浏览器自动处理
  291. }
  292. callback({ requestHeaders: details.requestHeaders });
  293. });
  294. }
  295. });
  296. }
  297. // 检查 URL 是否是允许的协议
  298. function isAllowedUrl(url: string): boolean {
  299. if (!url) return false;
  300. const lowerUrl = url.toLowerCase();
  301. return lowerUrl.startsWith('http://') ||
  302. lowerUrl.startsWith('https://') ||
  303. lowerUrl.startsWith('about:') ||
  304. lowerUrl.startsWith('data:');
  305. }
  306. // 阻止默认的 window-all-closed 行为,保持托盘运行
  307. app.on('window-all-closed', () => {
  308. // 不退出应用,保持托盘运行
  309. // 只有在 isQuitting 为 true 时才真正退出
  310. });
  311. // 应用退出前清理托盘
  312. app.on('before-quit', () => {
  313. isQuitting = true;
  314. });
  315. app.on('quit', () => {
  316. stopLocalServices();
  317. if (tray) {
  318. tray.destroy();
  319. tray = null;
  320. }
  321. });
  322. ipcMain.handle('test-server-connection', async (_event: unknown, args: { url: string }) => {
  323. try {
  324. const baseUrl = normalizeBaseUrl(args?.url);
  325. if (!baseUrl) return { ok: false, error: '未填写服务器地址' };
  326. const result = await requestJson(`${baseUrl}/api/health`, 5000);
  327. if (!result.ok) return { ok: false, error: result.error || '连接失败' };
  328. if (result.data?.status === 'ok') return { ok: true };
  329. return { ok: false, error: '服务器响应异常' };
  330. } catch (e: any) {
  331. return { ok: false, error: e?.message || '连接失败' };
  332. }
  333. });
  334. // Python 服务测试连接(主进程发起,支持 HTTP/HTTPS,无跨域限制)
  335. ipcMain.handle('test-python-service-connection', async (_event: unknown, args: { url: string }) => {
  336. try {
  337. const baseUrl = normalizeBaseUrl(args?.url);
  338. if (!baseUrl) return { ok: false, error: '未填写 Python 服务地址' };
  339. const result = await requestJson(`${baseUrl}/health`, 5000);
  340. if (!result.ok) return { ok: false, error: result.error || '连接失败' };
  341. return { ok: true };
  342. } catch (e: any) {
  343. return { ok: false, error: e?.message || '连接失败' };
  344. }
  345. });
  346. // 本地服务状态查询
  347. ipcMain.handle('get-local-services-status', () => {
  348. return getServiceStatus();
  349. });
  350. ipcMain.handle('get-local-urls', () => {
  351. return { nodeUrl: LOCAL_NODE_URL, pythonUrl: LOCAL_PYTHON_URL };
  352. });
  353. ipcMain.handle('get-service-log', () => {
  354. try {
  355. const logPath = getLogPath();
  356. if (fs.existsSync(logPath)) {
  357. return { path: logPath, content: fs.readFileSync(logPath, 'utf-8') };
  358. }
  359. return { path: logPath, content: '(日志文件不存在)' };
  360. } catch (e: any) {
  361. return { path: '', content: `读取失败: ${e.message}` };
  362. }
  363. });
  364. ipcMain.handle('open-log-file', () => {
  365. try {
  366. const logPath = getLogPath();
  367. if (fs.existsSync(logPath)) {
  368. shell.showItemInFolder(logPath);
  369. }
  370. } catch { /* ignore */ }
  371. });
  372. // IPC 处理
  373. ipcMain.handle('get-app-version', () => {
  374. return app.getVersion();
  375. });
  376. ipcMain.handle('get-platform', () => {
  377. return process.platform;
  378. });
  379. // 窗口控制
  380. ipcMain.on('window-minimize', () => {
  381. mainWindow?.minimize();
  382. });
  383. ipcMain.on('window-maximize', () => {
  384. if (mainWindow?.isMaximized()) {
  385. mainWindow.unmaximize();
  386. } else {
  387. mainWindow?.maximize();
  388. }
  389. });
  390. // 关闭窗口(最小化到托盘)
  391. ipcMain.on('window-close', () => {
  392. mainWindow?.hide();
  393. });
  394. // 真正退出应用
  395. ipcMain.on('app-quit', () => {
  396. isQuitting = true;
  397. app.quit();
  398. });
  399. // 获取窗口最大化状态
  400. ipcMain.handle('window-is-maximized', () => {
  401. return mainWindow?.isMaximized() || false;
  402. });
  403. // 监听窗口最大化/还原事件,通知渲染进程
  404. function setupWindowEvents() {
  405. mainWindow?.on('maximize', () => {
  406. mainWindow?.webContents.send('window-maximized', true);
  407. });
  408. mainWindow?.on('unmaximize', () => {
  409. mainWindow?.webContents.send('window-maximized', false);
  410. });
  411. }
  412. // 弹窗打开平台后台(独立窗口,不嵌入;用于实验,可回归为嵌入)
  413. ipcMain.handle('open-backend-external', async (_event: unknown, payload: { url: string; cookieData?: string; title?: string }) => {
  414. const { url, cookieData, title } = payload || {};
  415. if (!url || typeof url !== 'string') return;
  416. const partition = 'persist:backend-popup-' + Date.now();
  417. const ses = session.fromPartition(partition);
  418. if (cookieData && typeof cookieData === 'string' && cookieData.trim()) {
  419. const raw = cookieData.trim();
  420. let cookiesToSet: Array<{ name: string; value: string; domain?: string; path?: string }> = [];
  421. try {
  422. if (raw.startsWith('[') || raw.startsWith('{')) {
  423. const parsed = JSON.parse(raw);
  424. const arr = Array.isArray(parsed) ? parsed : (parsed?.cookies || []);
  425. cookiesToSet = arr.map((c: { name?: string; value?: string; domain?: string; path?: string }) => ({
  426. name: String(c?.name ?? '').trim(),
  427. value: String(c?.value ?? '').trim(),
  428. domain: c?.domain ? String(c.domain) : undefined,
  429. path: c?.path ? String(c.path) : '/',
  430. })).filter((c: { name: string }) => c.name);
  431. } else {
  432. raw.split(';').forEach((p: string) => {
  433. const idx = p.indexOf('=');
  434. if (idx > 0) {
  435. const name = p.slice(0, idx).trim();
  436. const value = p.slice(idx + 1).trim();
  437. if (name) cookiesToSet.push({ name, value, path: '/' });
  438. }
  439. });
  440. }
  441. } catch (e) {
  442. console.warn('[open-backend-external] 解析 cookie 失败', e);
  443. }
  444. const origin = new URL(url).origin;
  445. const hostname = new URL(url).hostname;
  446. const defaultDomain = hostname.startsWith('www.') ? hostname.slice(4) : hostname;
  447. const domainWithDot = defaultDomain.includes('.') ? '.' + defaultDomain.split('.').slice(-2).join('.') : undefined;
  448. for (const c of cookiesToSet) {
  449. try {
  450. await ses.cookies.set({
  451. url: origin + '/',
  452. name: c.name,
  453. value: c.value,
  454. domain: c.domain || domainWithDot || hostname,
  455. path: c.path || '/',
  456. });
  457. } catch (err) {
  458. console.warn('[open-backend-external] 设置 cookie 失败', c.name, err);
  459. }
  460. }
  461. }
  462. const win = new BrowserWindow({
  463. width: 1280,
  464. height: 800,
  465. title: title || '平台后台',
  466. icon: getIconPath(),
  467. webPreferences: {
  468. session: ses,
  469. nodeIntegration: false,
  470. contextIsolation: true,
  471. },
  472. show: false,
  473. });
  474. win.once('ready-to-show', () => {
  475. win.show();
  476. });
  477. await win.loadURL(url);
  478. return { ok: true };
  479. });
  480. // 获取 webview 的 cookies
  481. ipcMain.handle('get-webview-cookies', async (_event: unknown, partition: string, url: string) => {
  482. try {
  483. const ses = session.fromPartition(partition);
  484. const cookies = await ses.cookies.get({ url });
  485. return cookies;
  486. } catch (error) {
  487. console.error('获取 cookies 失败:', error);
  488. return [];
  489. }
  490. });
  491. // 获取 webview 的全部 cookies(按 partition)
  492. ipcMain.handle('get-webview-all-cookies', async (_event: unknown, partition: string) => {
  493. try {
  494. const ses = session.fromPartition(partition);
  495. return await ses.cookies.get({});
  496. } catch (error) {
  497. console.error('获取全部 cookies 失败:', error);
  498. return [];
  499. }
  500. });
  501. // 清除 webview 的 cookies
  502. ipcMain.handle('clear-webview-cookies', async (_event: unknown, partition: string) => {
  503. try {
  504. const ses = session.fromPartition(partition);
  505. await ses.clearStorageData({ storages: ['cookies'] });
  506. return true;
  507. } catch (error) {
  508. console.error('清除 cookies 失败:', error);
  509. return false;
  510. }
  511. });
  512. // 设置 webview 的 cookies
  513. ipcMain.handle('set-webview-cookies', async (_event: unknown, partition: string, cookies: Electron.CookiesSetDetails[]) => {
  514. try {
  515. if (!Array.isArray(cookies) || cookies.length === 0) {
  516. console.warn(`[Main] set-webview-cookies: cookies 为空, partition=${partition}`);
  517. return false;
  518. }
  519. console.log(`[Main] 设置 webview cookies, partition=${partition}, count=${cookies.length}`);
  520. const ses = session.fromPartition(partition);
  521. // 逐个设置 cookie
  522. let successCount = 0;
  523. for (const cookie of cookies) {
  524. try {
  525. // 确保 Cookie 格式正确
  526. const cookieToSet: Electron.CookiesSetDetails = {
  527. url: cookie.url,
  528. name: cookie.name,
  529. value: cookie.value,
  530. domain: cookie.domain,
  531. path: cookie.path || '/',
  532. };
  533. // 可选字段
  534. if (typeof cookie.expirationDate === 'number' && Number.isFinite(cookie.expirationDate) && cookie.expirationDate > 0) {
  535. cookieToSet.expirationDate = cookie.expirationDate;
  536. }
  537. if (cookie.httpOnly !== undefined) {
  538. cookieToSet.httpOnly = cookie.httpOnly;
  539. }
  540. if (cookie.secure !== undefined) {
  541. cookieToSet.secure = cookie.secure;
  542. }
  543. if (cookie.sameSite) {
  544. cookieToSet.sameSite = cookie.sameSite as 'no_restriction' | 'lax' | 'strict';
  545. }
  546. await ses.cookies.set(cookieToSet);
  547. successCount++;
  548. // 记录关键 Cookie
  549. if (cookie.name === 'BDUSS' || cookie.name === 'STOKEN' || cookie.name === 'sessionid') {
  550. console.log(`[Main] 成功设置关键 Cookie: ${cookie.name}, domain: ${cookie.domain}`);
  551. }
  552. } catch (error) {
  553. console.error(`[Main] 设置 cookie 失败 (${cookie.name}):`, error);
  554. }
  555. }
  556. console.log(`[Main] 成功设置 ${successCount}/${cookies.length} 个 cookies`);
  557. // 验证 Cookie 是否真的设置成功
  558. try {
  559. const setCookies = await ses.cookies.get({ domain: '.baidu.com' });
  560. console.log(`[Main] 验证:当前 session 中有 ${setCookies.length} 个百度 Cookie`);
  561. const keyNames = setCookies.slice(0, 5).map(c => c.name).join(', ');
  562. console.log(`[Main] 关键 Cookie 名称: ${keyNames}`);
  563. } catch (verifyError) {
  564. console.error('[Main] 验证 Cookie 失败:', verifyError);
  565. }
  566. return successCount > 0;
  567. } catch (error) {
  568. console.error('[Main] 设置 cookies 失败:', error);
  569. return false;
  570. }
  571. });
  572. // 截取 webview 页面截图(用于 AI 分析)
  573. ipcMain.handle('capture-webview-page', async (_event: unknown, webContentsId: number) => {
  574. try {
  575. const wc = webContents.fromId(webContentsId);
  576. if (!wc) {
  577. console.error('找不到 webContents:', webContentsId);
  578. return null;
  579. }
  580. const image = await wc.capturePage();
  581. if (!image || image.isEmpty()) {
  582. console.warn('截图为空');
  583. return null;
  584. }
  585. // 转换为 JPEG 格式的 Base64
  586. const buffer = image.toJPEG(80);
  587. return buffer.toString('base64');
  588. } catch (error) {
  589. console.error('截图失败:', error);
  590. return null;
  591. }
  592. });
  593. // 向 webview 发送鼠标点击事件
  594. ipcMain.handle('webview-send-mouse-click', async (_event: unknown, webContentsId: number, x: number, y: number) => {
  595. try {
  596. const wc = webContents.fromId(webContentsId);
  597. if (!wc) {
  598. console.error('找不到 webContents:', webContentsId);
  599. return false;
  600. }
  601. // 发送鼠标移动事件
  602. wc.sendInputEvent({
  603. type: 'mouseMove',
  604. x: Math.round(x),
  605. y: Math.round(y),
  606. });
  607. // 短暂延迟后发送点击事件
  608. await new Promise(resolve => setTimeout(resolve, 50));
  609. // 发送鼠标按下事件
  610. wc.sendInputEvent({
  611. type: 'mouseDown',
  612. x: Math.round(x),
  613. y: Math.round(y),
  614. button: 'left',
  615. clickCount: 1,
  616. });
  617. // 短暂延迟后发送鼠标抬起事件
  618. await new Promise(resolve => setTimeout(resolve, 50));
  619. wc.sendInputEvent({
  620. type: 'mouseUp',
  621. x: Math.round(x),
  622. y: Math.round(y),
  623. button: 'left',
  624. clickCount: 1,
  625. });
  626. console.log(`[webview-send-mouse-click] Clicked at (${x}, ${y})`);
  627. return true;
  628. } catch (error) {
  629. console.error('发送点击事件失败:', error);
  630. return false;
  631. }
  632. });
  633. // 向 webview 发送键盘输入事件
  634. ipcMain.handle('webview-send-text-input', async (_event: unknown, webContentsId: number, text: string) => {
  635. try {
  636. const wc = webContents.fromId(webContentsId);
  637. if (!wc) {
  638. console.error('找不到 webContents:', webContentsId);
  639. return false;
  640. }
  641. // 逐字符输入
  642. for (const char of text) {
  643. wc.sendInputEvent({
  644. type: 'char',
  645. keyCode: char,
  646. });
  647. await new Promise(resolve => setTimeout(resolve, 30));
  648. }
  649. console.log(`[webview-send-text-input] Typed: ${text}`);
  650. return true;
  651. } catch (error) {
  652. console.error('发送输入事件失败:', error);
  653. return false;
  654. }
  655. });
  656. // 获取 webview 页面元素位置
  657. ipcMain.handle('webview-get-element-position', async (_event: unknown, webContentsId: number, selector: string) => {
  658. try {
  659. const wc = webContents.fromId(webContentsId);
  660. if (!wc) {
  661. console.error('找不到 webContents:', webContentsId);
  662. return null;
  663. }
  664. const result = await wc.executeJavaScript(`
  665. (function() {
  666. const el = document.querySelector('${selector.replace(/'/g, "\\'")}');
  667. if (!el) return null;
  668. const rect = el.getBoundingClientRect();
  669. return {
  670. x: rect.left + rect.width / 2,
  671. y: rect.top + rect.height / 2,
  672. width: rect.width,
  673. height: rect.height
  674. };
  675. })()
  676. `);
  677. return result;
  678. } catch (error) {
  679. console.error('获取元素位置失败:', error);
  680. return null;
  681. }
  682. });
  683. // 通过文本内容查找并点击元素
  684. ipcMain.handle('webview-click-by-text', async (_event: unknown, webContentsId: number, text: string) => {
  685. try {
  686. const wc = webContents.fromId(webContentsId);
  687. if (!wc) {
  688. console.error('找不到 webContents:', webContentsId);
  689. return false;
  690. }
  691. // 查找包含指定文本的可点击元素的位置
  692. const position = await wc.executeJavaScript(`
  693. (function() {
  694. const searchText = '${text.replace(/'/g, "\\'")}';
  695. // 查找可点击元素
  696. const clickables = document.querySelectorAll('a, button, [role="button"], [onclick], input[type="submit"], input[type="button"]');
  697. for (const el of clickables) {
  698. if (el.textContent?.includes(searchText) || el.getAttribute('aria-label')?.includes(searchText) || el.getAttribute('title')?.includes(searchText)) {
  699. const rect = el.getBoundingClientRect();
  700. if (rect.width > 0 && rect.height > 0) {
  701. return { x: rect.left + rect.width / 2, y: rect.top + rect.height / 2 };
  702. }
  703. }
  704. }
  705. // 查找所有包含文本的元素
  706. const allElements = document.querySelectorAll('*');
  707. for (const el of allElements) {
  708. const text = el.innerText?.trim();
  709. if (text && text.length < 100 && text.includes(searchText)) {
  710. const rect = el.getBoundingClientRect();
  711. if (rect.width > 0 && rect.height > 0 && rect.width < 500) {
  712. return { x: rect.left + rect.width / 2, y: rect.top + rect.height / 2 };
  713. }
  714. }
  715. }
  716. return null;
  717. })()
  718. `);
  719. if (!position) {
  720. console.warn(`[webview-click-by-text] 未找到包含 "${text}" 的元素`);
  721. return false;
  722. }
  723. // 发送点击事件
  724. wc.sendInputEvent({ type: 'mouseMove', x: Math.round(position.x), y: Math.round(position.y) });
  725. await new Promise(resolve => setTimeout(resolve, 50));
  726. wc.sendInputEvent({ type: 'mouseDown', x: Math.round(position.x), y: Math.round(position.y), button: 'left', clickCount: 1 });
  727. await new Promise(resolve => setTimeout(resolve, 50));
  728. wc.sendInputEvent({ type: 'mouseUp', x: Math.round(position.x), y: Math.round(position.y), button: 'left', clickCount: 1 });
  729. console.log(`[webview-click-by-text] Clicked "${text}" at (${position.x}, ${position.y})`);
  730. return true;
  731. } catch (error) {
  732. console.error('通过文本点击失败:', error);
  733. return false;
  734. }
  735. });
  736. // ========== CDP 网络拦截功能 ==========
  737. // 存储每个 webContents 的网络拦截配置
  738. const networkInterceptors: Map<number, {
  739. patterns: Array<{ match: string, key: string }>;
  740. pendingRequests: Map<string, { url: string, timestamp: number }>;
  741. }> = new Map();
  742. // 清理已销毁的 webContents
  743. app.on('web-contents-destroyed', (_event: unknown, contents: typeof webContents.prototype) => {
  744. const webContentsId = contents.id;
  745. if (networkInterceptors.has(webContentsId)) {
  746. // 清理网络拦截器
  747. try {
  748. contents.debugger.detach();
  749. } catch (e) {
  750. // 忽略错误
  751. }
  752. networkInterceptors.delete(webContentsId);
  753. console.log(`[CDP] 已清理已销毁的 webContents 拦截器: ${webContentsId}`);
  754. }
  755. });
  756. // 启用 CDP 网络拦截
  757. ipcMain.handle('enable-network-intercept', async (_event: unknown, webContentsId: number, patterns: Array<{ match: string, key: string }>) => {
  758. try {
  759. const wc = webContents.fromId(webContentsId);
  760. if (!wc) {
  761. console.error('[CDP] 找不到 webContents:', webContentsId);
  762. return false;
  763. }
  764. // 如果已经有拦截器,先清理
  765. if (networkInterceptors.has(webContentsId)) {
  766. try {
  767. wc.debugger.detach();
  768. } catch (e) {
  769. // 忽略
  770. }
  771. }
  772. // 存储配置
  773. networkInterceptors.set(webContentsId, {
  774. patterns,
  775. pendingRequests: new Map()
  776. });
  777. // 附加调试器
  778. try {
  779. wc.debugger.attach('1.3');
  780. } catch (err: unknown) {
  781. const error = err as Error;
  782. if (!error.message?.includes('Already attached')) {
  783. throw err;
  784. }
  785. }
  786. // 启用网络监听
  787. await wc.debugger.sendCommand('Network.enable');
  788. // 监听网络响应
  789. wc.debugger.on('message', async (_e: unknown, method: string, params: {
  790. requestId?: string;
  791. response?: { url?: string; status?: number; mimeType?: string };
  792. encodedDataLength?: number;
  793. }) => {
  794. const config = networkInterceptors.get(webContentsId);
  795. if (!config) return;
  796. if (method === 'Network.responseReceived') {
  797. const { requestId, response } = params;
  798. if (!requestId || !response?.url) return;
  799. // 调试:打印百家号相关的所有 API 请求
  800. if (response.url.includes('baijiahao.baidu.com')) {
  801. if (response.url.includes('/pcui/') || response.url.includes('/article')) {
  802. console.log(`[CDP DEBUG] 百家号 API: ${response.url}`);
  803. }
  804. }
  805. // 检查是否匹配我们关注的 API
  806. for (const pattern of config.patterns) {
  807. if (response.url.includes(pattern.match)) {
  808. // 记录请求,等待响应完成
  809. config.pendingRequests.set(requestId, {
  810. url: response.url,
  811. timestamp: Date.now()
  812. });
  813. console.log(`[CDP] 匹配到 API: ${pattern.key} - ${response.url}`);
  814. break;
  815. }
  816. }
  817. }
  818. if (method === 'Network.loadingFinished') {
  819. const { requestId } = params;
  820. if (!requestId) return;
  821. const pending = config.pendingRequests.get(requestId);
  822. if (!pending) return;
  823. config.pendingRequests.delete(requestId);
  824. try {
  825. // 获取响应体
  826. const result = await wc.debugger.sendCommand('Network.getResponseBody', { requestId }) as { body: string; base64Encoded: boolean };
  827. let body = result.body;
  828. // 如果是 base64 编码,解码
  829. if (result.base64Encoded) {
  830. body = Buffer.from(body, 'base64').toString('utf8');
  831. }
  832. // 解析 JSON
  833. const data = JSON.parse(body);
  834. // 找到匹配的 key
  835. let matchedKey = '';
  836. for (const pattern of config.patterns) {
  837. if (pending.url.includes(pattern.match)) {
  838. matchedKey = pattern.key;
  839. break;
  840. }
  841. }
  842. if (matchedKey) {
  843. console.log(`[CDP] 获取到响应: ${matchedKey}`, JSON.stringify(data).substring(0, 200));
  844. // 发送到渲染进程
  845. mainWindow?.webContents.send('network-intercept-data', {
  846. webContentsId,
  847. key: matchedKey,
  848. url: pending.url,
  849. data
  850. });
  851. }
  852. } catch (err) {
  853. console.warn(`[CDP] 获取响应体失败:`, err);
  854. }
  855. }
  856. });
  857. console.log(`[CDP] 已启用网络拦截,webContentsId: ${webContentsId}, patterns:`, patterns.map(p => p.key));
  858. return true;
  859. } catch (error) {
  860. console.error('[CDP] 启用网络拦截失败:', error);
  861. return false;
  862. }
  863. });
  864. // 禁用 CDP 网络拦截
  865. ipcMain.handle('disable-network-intercept', async (_event: unknown, webContentsId: number) => {
  866. try {
  867. const wc = webContents.fromId(webContentsId);
  868. if (wc) {
  869. try {
  870. wc.debugger.detach();
  871. } catch (e) {
  872. // 忽略
  873. }
  874. }
  875. const config = networkInterceptors.get(webContentsId);
  876. if (config) {
  877. // 清理待处理请求的 Map
  878. config.pendingRequests.clear();
  879. }
  880. networkInterceptors.delete(webContentsId);
  881. console.log(`[CDP] 已禁用网络拦截,webContentsId: ${webContentsId}`);
  882. return true;
  883. } catch (error) {
  884. console.error('[CDP] 禁用网络拦截失败:', error);
  885. return false;
  886. }
  887. });
  888. // 更新网络拦截的 patterns
  889. ipcMain.handle('update-network-patterns', async (_event: unknown, webContentsId: number, patterns: Array<{ match: string, key: string }>) => {
  890. const config = networkInterceptors.get(webContentsId);
  891. if (config) {
  892. config.patterns = patterns;
  893. console.log(`[CDP] 已更新 patterns,webContentsId: ${webContentsId}`);
  894. return true;
  895. }
  896. return false;
  897. });