137 lines
4.5 KiB
TypeScript
137 lines
4.5 KiB
TypeScript
import { app, BrowserWindow } from "electron";
|
||
import path from "path";
|
||
import fs from "fs";
|
||
import Module from "module";
|
||
|
||
// 默认端口配置
|
||
const defaultPort = 10588;
|
||
|
||
/**
|
||
* 将 extraResources 中的 data 目录复制到用户数据目录（跳过已存在的文件，保留用户修改）
|
||
*/
|
||
function initializeData(): void {
|
||
const srcDir = path.join(process.resourcesPath, "data");
|
||
const destDir = path.join(app.getPath("userData"), "data");
|
||
if (fs.existsSync(destDir)) return;
|
||
copyDirRecursive(srcDir, destDir);
|
||
}
|
||
|
||
function copyDirRecursive(src: string, dest: string): void {
|
||
if (!fs.existsSync(src)) return;
|
||
if (!fs.existsSync(dest)) fs.mkdirSync(dest, { recursive: true });
|
||
for (const entry of fs.readdirSync(src, { withFileTypes: true })) {
|
||
const srcPath = path.join(src, entry.name);
|
||
const destPath = path.join(dest, entry.name);
|
||
if (entry.isDirectory()) {
|
||
copyDirRecursive(srcPath, destPath);
|
||
} else if (!fs.existsSync(destPath)) {
|
||
fs.copyFileSync(srcPath, destPath);
|
||
}
|
||
}
|
||
}
|
||
|
||
//获取全部依赖路径，优先从 unpacked 加载原生模块，其他模块从 asar 加载
|
||
function getNodeModulesPaths(): string[] {
|
||
const paths: string[] = [];
|
||
if (app.isPackaged) {
|
||
// external 依赖（原生模块）在 unpacked 目录
|
||
const unpackedNodeModules = path.join(process.resourcesPath, "app.asar.unpacked", "node_modules");
|
||
if (fs.existsSync(unpackedNodeModules)) {
|
||
paths.push(unpackedNodeModules);
|
||
}
|
||
// 普通依赖在 asar 内
|
||
const asarNodeModules = path.join(process.resourcesPath, "app.asar", "node_modules");
|
||
paths.push(asarNodeModules);
|
||
} else {
|
||
paths.push(path.join(process.cwd(), "node_modules"));
|
||
}
|
||
return paths;
|
||
}
|
||
|
||
//动态加载
|
||
function requireWithCustomPaths(modulePath: string): any {
|
||
const appNodeModulesPaths = getNodeModulesPaths();
|
||
// 保存原始方法
|
||
const originalNodeModulePaths = (Module as any)._nodeModulePaths;
|
||
// 临时修改模块路径解析
|
||
(Module as any)._nodeModulePaths = function (from: string): string[] {
|
||
const paths = originalNodeModulePaths.call(this, from);
|
||
// 将主程序的 node_modules 添加到前面
|
||
for (let i = appNodeModulesPaths.length - 1; i >= 0; i--) {
|
||
const p = appNodeModulesPaths[i];
|
||
if (!paths.includes(p)) {
|
||
paths.unshift(p);
|
||
}
|
||
}
|
||
return paths;
|
||
};
|
||
try {
|
||
// 清除缓存确保加载最新
|
||
delete require.cache[require.resolve(modulePath)];
|
||
return require(modulePath);
|
||
} finally {
|
||
// 恢复原始方法
|
||
(Module as any)._nodeModulePaths = originalNodeModulePaths;
|
||
}
|
||
}
|
||
|
||
function createMainWindow(port: any): void {
|
||
const win = new BrowserWindow({
|
||
width: 900,
|
||
height: 600,
|
||
show: true,
|
||
autoHideMenuBar: true,
|
||
});
|
||
win.webContents.on("did-start-loading", () => {
|
||
void win.webContents.executeJavaScript(`window.$electron = true; window.$port = ${port};`);
|
||
});
|
||
const isDev = process.env.NODE_ENV === "dev" || !app.isPackaged;
|
||
if (process.env.VITE_DEV) {
|
||
void win.loadURL("http://localhost:50188");
|
||
} else {
|
||
const htmlPath = isDev ? path.join(process.cwd(), "data", "web", "index.html") : path.join(app.getPath("userData"), "data", "web", "index.html");
|
||
void win.loadFile(htmlPath);
|
||
}
|
||
}
|
||
|
||
let closeServeFn: (() => Promise<void>) | undefined;
|
||
|
||
app.whenReady().then(async () => {
|
||
try {
|
||
let servePath: string;
|
||
if (app.isPackaged) {
|
||
// 生产环境：从 extraResources 初始化数据到用户目录，然后从用户目录加载后端服务
|
||
initializeData();
|
||
servePath = path.join(app.getPath("userData"), "data", "serve", "app.js");
|
||
} else {
|
||
// 开发环境：直接加载源码（tsx 通过 -r tsx 注册了 require 钩子）
|
||
servePath = path.join(process.cwd(), "src", "app.ts");
|
||
}
|
||
// 使用自定义路径加载模块
|
||
const mod = requireWithCustomPaths(servePath);
|
||
closeServeFn = mod.closeServe;
|
||
const port = await mod.default(true);
|
||
console.log("%c Line:112 🍇 port", "background:#2eafb0", port);
|
||
createMainWindow(port);
|
||
} catch (err) {
|
||
console.error("[服务启动失败]:", err);
|
||
// 如果服务启动失败，使用默认端口创建窗口
|
||
createMainWindow(defaultPort);
|
||
}
|
||
});
|
||
|
||
app.on("window-all-closed", () => {
|
||
if (process.platform !== "darwin") app.quit();
|
||
});
|
||
|
||
app.on("activate", () => {
|
||
if (BrowserWindow.getAllWindows().length === 0) {
|
||
// 重新激活时使用默认端口
|
||
createMainWindow(defaultPort);
|
||
}
|
||
});
|
||
|
||
app.on("before-quit", async (event) => {
|
||
if (closeServeFn) await closeServeFn();
|
||
});
|