ele-HeiXiu/electron/main.ts

import { app, BrowserWindow, ipcMain, dialog } from 'electron';
import { createRequire } from 'node:module';
import { fileURLToPath } from 'node:url';
import path from 'node:path';

import { getPlatform, isMacOS } from './main/utils/platform';
import { getWindowIconPath } from './main/utils/logo';
import { buildWindowTitle, parseEdition } from '../shared/constants/app';
import { loadEnvFile } from './main/load-env';
import { initUpdater, registerUpdateIpcHandlers } from './main/updater';
import { setupAppMenu } from './main/menu';
import { initLogger, flushLogger, logger } from './main/logger';
import { registerLogIpcHandlers } from './main/log-ipc';
import { registerSafeStorageIpcHandlers } from './main/safe-storage-ipc';
import { BIDIRECTIONAL } from '../shared/constants/ipc-channels';

createRequire(import.meta.url);
const __dirname = path.dirname(fileURLToPath(import.meta.url));

process.env.APP_ROOT = path.join(__dirname, '..');

// 加载 .env 文件中的环境变量到 process.env
// 必须在任何读取 process.env.UPDATE_API_URL 的模块之前调用
loadEnvFile();

export const VITE_DEV_SERVER_URL = process.env['VITE_DEV_SERVER_URL'];
export const MAIN_DIST = path.join(process.env.APP_ROOT, 'dist-electron');
export const RENDERER_DIST = path.join(process.env.APP_ROOT, 'dist');

process.env.VITE_PUBLIC = VITE_DEV_SERVER_URL
   ? path.join(process.env.APP_ROOT, 'public')
   : RENDERER_DIST;

// ---------- 平台 & 版本信息 ----------
const currentPlatform = getPlatform();
const currentEdition = parseEdition(process.env.EDITION);
const WINDOW_TITLE = buildWindowTitle(currentPlatform, currentEdition);

// ============================================================
// 全局未捕获异常（主进程兜底记录）
// ============================================================

process.on('uncaughtException', (error) => {
   try {
      logger.error('app', 'Uncaught exception (main process)', error);
   } catch {
      /* logger 自身异常 — 最后防线 */
   }
});

process.on('unhandledRejection', (reason) => {
   try {
      const error = reason instanceof Error ? reason : new Error(String(reason));
      logger.error('app', 'Unhandled rejection (main process)', error);
   } catch {
      /* logger 自身异常 — 最后防线 */
   }
});

// ============================================================
// 窗口引用
// ============================================================

let mainWindow: BrowserWindow | null = null;

// ============================================================
// 主窗口
// ============================================================

function createMainWindow(): BrowserWindow {
   const iconPath = getWindowIconPath(app.getAppPath());

   mainWindow = new BrowserWindow({
      title: WINDOW_TITLE,
      icon: iconPath,
      width: 1280,
      height: 800,
      minWidth: 960,
      minHeight: 600,
      show: false,
      ...(isMacOS() ? { titleBarStyle: 'hiddenInset' as const } : {}),
      webPreferences: {
         preload: path.join(__dirname, 'preload.mjs'),
         contextIsolation: true,
         nodeIntegration: false,
      },
   });

   mainWindow.setTitle(WINDOW_TITLE);

   if (VITE_DEV_SERVER_URL) {
      mainWindow.webContents.openDevTools();
   }

   // macOS 全屏事件
   if (isMacOS()) {
      mainWindow.on('enter-full-screen', () =>
         mainWindow?.webContents.send('window-fullscreen-changed', true),
      );
      mainWindow.on('leave-full-screen', () =>
         mainWindow?.webContents.send('window-fullscreen-changed', false),
      );
   }

   mainWindow.webContents.on('did-finish-load', () => {
      mainWindow?.webContents.send('main-process-message', new Date().toLocaleString());
      mainWindow?.webContents.send('platform-info', {
         platform: currentPlatform,
         edition: currentEdition,
      });
   });

   // ready-to-show 后显示，避免白屏闪烁
   mainWindow.once('ready-to-show', () => {
      mainWindow?.show();
   });

   if (VITE_DEV_SERVER_URL) {
      mainWindow.loadURL(VITE_DEV_SERVER_URL);
   } else {
      mainWindow.loadFile(path.join(RENDERER_DIST, 'index.html'));
   }

   return mainWindow;
}

// ============================================================
// 应用生命周期
// ============================================================

app.on('window-all-closed', () => {
   if (!isMacOS()) {
      app.quit();
   }
});

app.on('activate', () => {
   if (BrowserWindow.getAllWindows().length === 0) {
      createMainWindow();
   }
});

app.on('before-quit', () => {
   flushLogger();
});

app.whenReady().then(() => {
   initLogger();
   registerLogIpcHandlers();
   registerSafeStorageIpcHandlers();
   setupAppMenu();
   app.setName(WINDOW_TITLE);
   registerUpdateIpcHandlers();

   // 文件对话框 IPC 处理（供渲染进程选择文件夹）
   ipcMain.handle(BIDIRECTIONAL.FILE_DIALOG, async (_event, options: Electron.OpenDialogOptions) => {
      if (!mainWindow) return { canceled: true, filePaths: [] };
      return dialog.showOpenDialog(mainWindow, {
         title: options?.title || '选择文件夹',
         defaultPath: options?.defaultPath || app.getPath('home'),
         properties: options?.properties || ['openDirectory'],
      });
   });

   // 始终打开主窗口
   // 登录/注册由渲染进程内的 Modal 弹层处理，不再新开窗口
   const win = createMainWindow();
   initUpdater(win);
});