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();
  }
});