main.js

/**
 * EXE 图标替换工具 & PNG↔ICO 转换器 — Electron 主进程
 */
const { app, BrowserWindow, ipcMain, dialog, Menu } = require('electron');
const path = require('path');
const fs = require('fs');
const os = require('os');
const {
  NtExecutable,
  NtExecutableResource,
  Data,
  Resource,
} = require('resedit');

let mainWindow = null;

// ============================================================
// 窗口创建
// ============================================================
function createWindow() {
  mainWindow = new BrowserWindow({
    width: 1000,
    height: 720,
    minWidth: 800,
    minHeight: 600,
    title: '图标工具箱 v1.10.1 — PNG↔ICO & EXE图标替换',
    icon: path.join(__dirname, 'app-icon.ico'),
    backgroundColor: '#0f172a',
    show: false,
    webPreferences: {
      preload: path.join(__dirname, 'preload.js'),
      nodeIntegration: false,
      contextIsolation: true,
      sandbox: false,
    },
  });

  const menuTemplate = [
    {
      label: '文件',
      submenu: [
        { label: '打开 EXE...', click: () => mainWindow.webContents.send('menu-open-exe') },
        { label: '打开 ICO...', click: () => mainWindow.webContents.send('menu-open-ico') },
        { type: 'separator' },
        { role: 'quit', label: '退出' },
      ],
    },
    {
      label: '视图',
      submenu: [
        { role: 'reload', label: '刷新' },
        { role: 'toggleDevTools', label: '开发者工具' },
        { type: 'separator' },
        { role: 'resetZoom', label: '重置缩放' },
        { role: 'zoomIn', label: '放大' },
        { role: 'zoomOut', label: '缩小' },
      ],
    },
  ];
  Menu.setApplicationMenu(Menu.buildFromTemplate(menuTemplate));

  mainWindow.loadFile('index.html');

  mainWindow.once('ready-to-show', () => {
    mainWindow.show();
  });

  mainWindow.on('closed', () => {
    mainWindow = null;
  });
}

app.whenReady().then(createWindow);

app.on('window-all-closed', () => {
  app.quit();
});

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

// ============================================================
// 辅助：写临时文件 + 弹出保存对话框
// ============================================================
async function saveViaDialog(buf, defaultName, filters) {
  const tmpDir = os.tmpdir();
  const tmpPath = path.join(tmpDir, defaultName);
  fs.writeFileSync(tmpPath, buf);

  const result = await dialog.showSaveDialog(mainWindow, {
    defaultPath: defaultName,
    filters: filters || [],
  });

  if (!result.canceled && result.filePath) {
    try {
      fs.copyFileSync(tmpPath, result.filePath);
      try { fs.unlinkSync(tmpPath); } catch {}
      return { success: true, savedPath: result.filePath };
    } catch (err) {
      return { success: false, error: '保存文件失败: ' + err.message };
    }
  }

  try { fs.unlinkSync(tmpPath); } catch {}
  return { success: false, cancelled: true };
}

// ============================================================
// IPC: 替换图标（使用文件路径，避免大文件 IPC 传输）
// ============================================================
ipcMain.handle('replace-icon', async (_event, { exePath, icoPath }) => {
  // 直接用 Buffer（.buffer 是 pool-backed 含垃圾数据，会破坏 ICO 解析）
  const exeBuf = fs.readFileSync(exePath);
  const icoBuf = fs.readFileSync(icoPath);

  // 解析 EXE
  const ntExecutable = NtExecutable.from(exeBuf, { ignoreCert: true });
  const resource = NtExecutableResource.from(ntExecutable);

  // 解析 ICO
  const iconFile = Data.IconFile.from(icoBuf);
  if (!iconFile.icons || iconFile.icons.length === 0) {
    throw new Error('ICO 文件无效：未找到图标数据。请确认拖入的是 .ico 图标文件（非 PNG 直接改名）');
  }
  const iconItems = iconFile.icons.map((item) => item.data);

  // 查找并替换所有图标组
  const groupIconType = 14;
  const iconGroupIds = resource.entries
    .filter((e) => e.type === groupIconType)
    .map((e) => e.id);

  if (iconGroupIds.length === 0) {
    Resource.IconGroupEntry.replaceIconsForResource(
      resource.entries, 1, 0, iconItems
    );
  } else {
    for (const gid of iconGroupIds) {
      Resource.IconGroupEntry.replaceIconsForResource(
        resource.entries, gid, 0, iconItems
      );
    }
  }

  // 写回
  resource.outputResource(ntExecutable);
  const resultBuffer = Buffer.from(ntExecutable.generate());

  // 弹出保存对话框
  const outName = path.basename(exePath, '.exe') + '_new.exe';
  return await saveViaDialog(resultBuffer, outName, [
    { name: '可执行文件', extensions: ['exe'] },
  ]);
});

// ============================================================
// IPC: 提取图标（使用文件路径，避免大文件 IPC 传输）
// ============================================================
ipcMain.handle('extract-icon', async (_event, { exePath }) => {
  const exeBuf = fs.readFileSync(exePath);

  const ntExecutable = NtExecutable.from(exeBuf, { ignoreCert: true });
  const resource = NtExecutableResource.from(ntExecutable);

  const groupIconType = 14;
  const iconGroups = resource.entries.filter((e) => e.type === groupIconType);

  if (iconGroups.length === 0) {
    throw new Error('该 EXE 文件中没有图标资源');
  }

  const numericIds = iconGroups
    .filter((e) => typeof e.id === 'number')
    .map((e) => e.id);
  const mainGroupId = numericIds.length > 0
    ? Math.min(...numericIds)
    : iconGroups[0].id;

  const entries = Resource.IconGroupEntry.fromEntries(resource.entries);
  const targetGroup = entries.find((g) => g.id === mainGroupId);

  if (!targetGroup) {
    throw new Error('无法读取图标组数据');
  }

  const iconItems = targetGroup.getIconItemsFromEntries(resource.entries);

  const icoFile = new Data.IconFile();
  icoFile.icons = iconItems.map((item) => ({ data: item }));

  const resultBuffer = Buffer.from(icoFile.generate());

  const outName = path.basename(exePath, '.exe') + '_icon.ico';
  return await saveViaDialog(resultBuffer, outName, [
    { name: '图标文件', extensions: ['ico'] },
  ]);
});