JiGuang/jg/electron/main.mjs

import { app, BrowserWindow, ipcMain, dialog } from "electron";
import { fileURLToPath } from "url";
import path from "path";
import { execFile } from "child_process";
import fs from "fs";
import Papa from "./libs/papaparse/papaparse.js";
process.env["ELECTRON_DISABLE_SECURITY_WARNINGS"] = "true";

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

let mainWindow = null;

function createWindow() {
  mainWindow = new BrowserWindow({
    width: 1400,
    height: 900,
    minWidth: 800,
    minHeight: 600,
    webPreferences: {
      preload: path.join(__dirname, "./preload.mjs"),
      contextIsolation: true,
      nodeIntegration: false,
      allowRunningInsecureContent: false,
    },
    icon: path.join(__dirname, "../src/assets/images/login/bg.png"), // 开发环境的图标
  });
  //环境判断引入不同的页面
  if (process.env.NODE_ENV === "development") {
    mainWindow.loadURL("http://localhost:5173");
    mainWindow.webContents.openDevTools();
  } else {
    //生产环境将引入打包好的文件
    mainWindow.loadFile(path.join(__dirname, "../dist/index.html"));
    mainWindow.webContents.openDevTools();
  }
  // **🔹 监听渲染进程调用 Python EXE**
  ipcMain.handle("run-python-exe", async (event, apiName, params = {}) => {
    return new Promise((resolve, reject) => {
      let pythonExePath = null;
      if (process.env.NODE_ENV === "development") {
        // **开发环境**
        pythonExePath = path.join(__dirname, "../serves/dist/api_test.exe");
      } else {
        //这里需要区分开发环境还是生产环境 生产环境用到这个路径
        pythonExePath = path.join(
          process.resourcesPath, // `app.asar.unpacked` 的默认路径
          "app.asar.unpacked",
          "serves",
          "dist",
          "api_test.exe"
        );
      }
      // **修改 Python EXE 的路径**

      if (!fs.existsSync(pythonExePath)) {
        reject(`Python EXE 不存在: ${pythonExePath}`);
        return;
      }
      // **参数列表：API 名称 + JSON 格式参数**
      // const args = [apiName, JSON.stringify(params)];
      // ✅ **解决中文乱码**：使用 Base64 进行参数传输
      const args = [
        apiName,
        Buffer.from(JSON.stringify(params)).toString("base64"),
      ];
      const res = execFile(
        pythonExePath,
        args,
        { encoding: "utf8" },
        (error, stdout, stderr) => {
          if (error) {
            console.error("❌ Python EXE 运行失败:", error);
            reject(`Error: ${error.message}`);
            return;
          }
          if (stderr) {
            console.warn("⚠️ Python EXE 输出警告:", stderr);
          }

          // **尝试解析 JSON**
          try {
            console.log("✅ Python EXE 输出:", stdout);
            resolve(JSON.parse(stdout)); // 返回 JSON 数据
          } catch (parseError) {
            console.warn("⚠️ Python EXE 返回的不是 JSON:", stdout);
            resolve(stdout.trim()); // 返回原始文本
          }
        }
      );
      // **✅ 实时查看 Python print 的输出**
      res.stdout.on("data", (data) => {
        console.log("🐍 Python stdout:", data.toString());
      });

      res.stderr.on("data", (data) => {
        console.error("🐍 Python stderr:", data.toString());
      });
    });
  });
  ipcMain.handle("get-install-path", async () => {
    return process.execPath; // 获取可执行文件路径
  });
  // 监听前端请求，读取 CSV 文件
  ipcMain.handle("read-csv", async (event, filePath) => {
    try {
      const csvData = fs.readFileSync(filePath, "utf-8");
      const parsedData = Papa.parse(csvData, {
        header: true,
        skipEmptyLines: true, // ← 这个也可以帮助跳过完全空的行
        transformHeader: (header) => header.trim(), // ← 这个也可以帮助跳过完全空的行
      });
      // 过滤掉所有字段都为空的行
      const cleanData = parsedData.data.filter((row) =>
        Object.values(row).some((val) => val && val.trim() !== "")
      );

      return cleanData; // 返回解析后的数据
    } catch (error) {
      console.error("读取 CSV 失败:", error);
      return { error: error.message };
    }
  });

  // 监听 get-file-path 事件 监听渲染进程请求文件路径
  ipcMain.handle("get-file-path", async () => {
    try {
      const { filePaths } = await dialog.showOpenDialog({
        properties: ["openFile", "multiSelections"], // ✅ 允许选择多个文件
      });
      if (filePaths.length > 0) {
        console.log("用户选择的文件路径:", filePaths[0]);
        return filePaths[0]; // 返回文件路径
      } else {
        console.log("用户取消了选择");
        return null; // 用户没有选择文件
      }
    } catch (error) {
      console.error("获取文件路径失败:", error);
      return null;
    }
  });
  mainWindow.on("close", (event) => {
    event.preventDefault(); // ✅ 阻止默认关闭行为

    const choice = dialog.showMessageBoxSync(mainWindow, {
      type: "question",
      buttons: ["取消", "退出"],
      defaultId: 0, // 默认选中 "取消"
      title: "确认退出",
      message: "退出程序正在分析中的数据会丢失，确定要退出程序吗？",
    });

    if (choice === 1) {
      // ✅ 直接移除监听器，允许窗口关闭
      mainWindow.removeAllListeners("close");
      mainWindow.close();
    }
  });

  mainWindow.on("closed", () => {
    mainWindow = null; // ✅ 窗口关闭后清空引用
  });
}

app.whenReady().then(createWindow);

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

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

export { createWindow };