index.js

import { app, BrowserWindow, Menu, globalShortcut, ipcMain } from "electron";
import { dirname, join } from "path";
import { fileURLToPath } from "url";
import {
  initWebsocket,
  closeWebsocket,
  emit,
  userList,
} from "./utils/webSocket.js";
import { getTime } from "./src/utils/index.js";

const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
const createWindow = () => {
  const win = new BrowserWindow({
    width: 1100,
    height: 900,
    webPreferences: {
      // 启用上下文隔离，这是推荐的安全实践
      contextIsolation: true,
      // 禁用远程模块，也是出于安全考虑
      enableRemoteModule: false,
      // 加载 preload 脚本
      preload: join(__dirname, "preload.cjs"),
      // 允许在 preload 脚本中使用 nodeIntegration
      nodeIntegration: false,
      // 允许在 preload 脚本中访问 contextBridge API
      contextIsolation: true,
    },
    icon: __dirname + "/public/bd.jpg",
  });
  const menuTemplate = [];
  const menu = Menu.buildFromTemplate(menuTemplate);
  Menu.setApplicationMenu(menu);
  win.loadFile("./dist/index.html");
};

// app.whenReady() ready 事件的专用监听
app.whenReady().then(() => {
  app.on("activate", () => {
    if (BrowserWindow.getAllWindows().length === 0) createWindow();
  });
  createWindow();
  const window = BrowserWindow.getFocusedWindow();
  ipcMain.on("server-ws", (_event, value) => {
    if (value) {
      initWebsocket(value);
    } else {
      closeWebsocket();
    }
  });

  ipcMain.on("send-message", (_event, value) => {
    const { key, message, host } = value;
    const sendObj = {
      type: 0,
      time: getTime(),
      message,
      host,
    };
    const user = userList.get(key);
    if (user) {
      user.connection.send(message);
      serverSendCallBack(sendObj, window);
    }
    if (!key && !user) {
      userList.forEach((value, key) => {
        value.connection.send(message);
      });
      serverSendCallBack(sendObj, window);
    }
  });

  globalShortcut.register("CommandOrControl+k", () => {
    // 获取当前窗口
    BrowserWindow.getFocusedWindow().webContents.openDevTools();
  });

  emit.on("message", (v) => {
    sendView(v, window);
  });

  emit.on("addServerList", (v) => {
    addServerList(v, window);
  });
});

// 关闭所有窗口时退出应用
app.on("window-all-closed", () => {
  if (process.platform !== "darwin") app.quit();
});

const serverSendCallBack = (request, window) => {
  if (window) {
    window.webContents.send("ws-message", {
      serverSendCallBack: JSON.stringify(request),
    });
  }
};

const sendView = (request, window) => {
  if (window) {
    window.webContents.send("ws-message", {
      message: JSON.stringify(request),
    });
  }
};

const addServerList = (request, window) => {
  if (window) {
    window.webContents.send("ws-message", {
      addServerList: JSON.stringify(request),
    });
  }
};