142 lines
3.8 KiB
JavaScript
142 lines
3.8 KiB
JavaScript
const { app, BrowserWindow, ipcMain } = require("electron");
|
||
const path = require("path");
|
||
const https = require("https");
|
||
const http = require("http");
|
||
|
||
let mainWindow;
|
||
|
||
function createWindow() {
|
||
mainWindow = new BrowserWindow({
|
||
width: 1080,
|
||
height: 1920,
|
||
webPreferences: {
|
||
preload: path.join(__dirname, "preload.js"),
|
||
nodeIntegration: false,
|
||
contextIsolation: true,
|
||
},
|
||
});
|
||
|
||
// 开发环境下加载 Vite 开发服务器地址
|
||
// 生产环境下加载打包后的 index.html
|
||
const isDev = !app.isPackaged;
|
||
|
||
if (isDev) {
|
||
mainWindow.loadURL("http://localhost:5173");
|
||
// 打开开发者工具
|
||
mainWindow.webContents.openDevTools();
|
||
} else {
|
||
mainWindow.loadFile(path.join(__dirname, "../dist/index.html"));
|
||
}
|
||
}
|
||
|
||
// 处理PDF获取请求（绕过CORS）
|
||
ipcMain.handle("fetch-pdf", async (event, pdfUrl) => {
|
||
return new Promise((resolve, reject) => {
|
||
const protocol = pdfUrl.startsWith("https") ? https : http;
|
||
|
||
protocol
|
||
.get(pdfUrl, (response) => {
|
||
// 处理重定向
|
||
if (response.statusCode === 301 || response.statusCode === 302) {
|
||
const redirectUrl = response.headers.location;
|
||
protocol
|
||
.get(redirectUrl, (redirectResponse) => {
|
||
const chunks = [];
|
||
redirectResponse.on("data", (chunk) => chunks.push(chunk));
|
||
redirectResponse.on("end", () => {
|
||
const buffer = Buffer.concat(chunks);
|
||
resolve({
|
||
success: true,
|
||
data: buffer.toString("base64"),
|
||
});
|
||
});
|
||
redirectResponse.on("error", (error) => {
|
||
reject({ success: false, error: error.message });
|
||
});
|
||
})
|
||
.on("error", (error) => {
|
||
reject({ success: false, error: error.message });
|
||
});
|
||
} else {
|
||
const chunks = [];
|
||
response.on("data", (chunk) => chunks.push(chunk));
|
||
response.on("end", () => {
|
||
const buffer = Buffer.concat(chunks);
|
||
resolve({
|
||
success: true,
|
||
data: buffer.toString("base64"),
|
||
});
|
||
});
|
||
response.on("error", (error) => {
|
||
reject({ success: false, error: error.message });
|
||
});
|
||
}
|
||
})
|
||
.on("error", (error) => {
|
||
reject({ success: false, error: error.message });
|
||
});
|
||
});
|
||
});
|
||
|
||
// 处理PDF打印请求
|
||
ipcMain.handle("print-pdf", async (event, pdfUrl) => {
|
||
try {
|
||
// 创建一个隐藏的窗口用于加载PDF
|
||
const printWindow = new BrowserWindow({
|
||
width: 800,
|
||
height: 600,
|
||
show: false,
|
||
webPreferences: {
|
||
nodeIntegration: false,
|
||
contextIsolation: true,
|
||
},
|
||
});
|
||
|
||
// 加载PDF
|
||
await printWindow.loadURL(pdfUrl);
|
||
|
||
// 等待页面加载完成
|
||
await new Promise((resolve) => {
|
||
printWindow.webContents.on("did-finish-load", resolve);
|
||
});
|
||
|
||
// 静默打印（直接调用系统打印对话框）
|
||
printWindow.webContents.print(
|
||
{
|
||
silent: false, // 显示打印对话框
|
||
printBackground: true,
|
||
margins: {
|
||
marginType: "none",
|
||
},
|
||
},
|
||
(success, errorType) => {
|
||
printWindow.close();
|
||
if (!success) {
|
||
console.error("Print failed:", errorType);
|
||
}
|
||
}
|
||
);
|
||
|
||
return { success: true };
|
||
} catch (error) {
|
||
console.error("Print error:", error);
|
||
return { success: false, error: error.message };
|
||
}
|
||
});
|
||
|
||
app.whenReady().then(() => {
|
||
createWindow();
|
||
|
||
app.on("activate", () => {
|
||
if (BrowserWindow.getAllWindows().length === 0) {
|
||
createWindow();
|
||
}
|
||
});
|
||
});
|
||
|
||
app.on("window-all-closed", () => {
|
||
if (process.platform !== "darwin") {
|
||
app.quit();
|
||
}
|
||
});
|