cvpilot-tool/apps/desktop/electron/main.ts

import {
  app,
  BrowserWindow,
  Tray,
  Menu,
  globalShortcut,
  nativeImage,
} from "electron";
// import { createRequire } from "node:module";
import { fileURLToPath } from "node:url";
import path from "node:path";
// import registerIpcMainHandlers from "./ipcMainHandlers";
import PythonManager from "./core/PythonManager";
import { createDatabase } from "./core/db";
import { getMachineId } from "./core/auth";

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

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

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;

let win: BrowserWindow | null;
let tray: Tray | null = null;
let pythonManager: PythonManager | null;
const theme: "dark" | "light" = "light";

const themeTitleBarStyles = {
  dark: { color: "rgb(32,32,32)", symbolColor: "#fff" },
  light: {},
};

function createWindow() {
  win = new BrowserWindow({
    width: 1280,
    height: 800,
    show: false, // 先隐藏。等待渲染完成，防止闪烁
    icon: path.join(process.env.VITE_PUBLIC, "AI.png"),
    // frame: false,
    titleBarStyle: "hidden", // customButtonsOnHover || hidden || hiddenInset
    titleBarOverlay: { height: 36, ...themeTitleBarStyles[theme] }, // 渲染进程发消息动态改变这个
    webPreferences: {
      preload: path.join(__dirname, "preload.mjs"),
      nodeIntegration: true,
    },
  });

  // Test active push message to Renderer-process.
  win.webContents.on("did-finish-load", () => {
    win?.webContents.send("main-process-message", {
      platform: process.platform === "darwin" ? "macos" : "windows",
      theme,
    });
  });

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

  // pythonManager = new PythonManager(win, "http://127.0.0.1:15001", 3000);
  // registerIpcMainHandlers(win, pythonManager);

  // 如果有传递路径，通过IPC传递给渲染进程
  if (process.argv.length >= 2) {
    const folderPath = process.argv[1]; // 第二个参数是文件或文件夹路径
    console.log("Opened folder path:", folderPath);

    // 你可以将路径发送到渲染进程
    win.webContents.send("open-folder", folderPath);
  }
}

function createTray() {
  if (tray) tray.destroy();
  const iconPath = path.join(process.env.VITE_PUBLIC, "AI.png"); // 使用 PNG 图标
  const icon = nativeImage
    .createFromPath(iconPath)
    .resize({ width: 20, height: 20 });
  tray = new Tray(icon);

  const contextMenu = Menu.buildFromTemplate([
    {
      label: "Show App",
      click: function () {
        if (win) {
          win.show();
        }
      },
    },
    {
      label: "Quit",
      click: function () {
        app.quit();
      },
    },
  ]);

  tray.setToolTip("My Electron App");
  tray.setContextMenu(contextMenu);

  tray.on("click", () => {
    if (win) {
      win.isVisible() ? win.hide() : win.show();
    }
  });
}

function registerGlobalShortcuts() {
  // 注册全局快捷键 'CommandOrControl+Shift+S' 来显示应用窗口
  globalShortcut.register("Option+N", () => {
    if (win) {
      win.isVisible() ? win.hide() : win.show();
    }
  });
}

app.on("window-all-closed", () => {
  if (process.platform !== "darwin") {
    app.quit();
    win = null;
  }
});

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

app.on("before-quit", () => {
  if (pythonManager?.flaskProcess) pythonManager?.stopFlask();
});

app.whenReady().then(() => {
  getMachineId();
  createWindow();
  createTray();
  registerGlobalShortcuts();
  createDatabase({ name: "cvpilot.json" });
  console.log(path.join(app.getPath("userData")));

  // 设置 Dock 图标
  if (process.platform === "darwin") {
    const dockIconPath = path.join(process.env.VITE_PUBLIC, "girl.png");
    const dockIcon = nativeImage.createFromPath(dockIconPath);
    app.dock.setIcon(dockIcon);
  }
});

// 注销全局快捷键，当应用退出时
app.on("will-quit", () => {
  globalShortcut.unregisterAll();
});