multi-platform-media-manage/client/electron/main.ts

// 使用 CommonJS 格式
const { app, BrowserWindow, ipcMain, shell, session, Menu, Tray, nativeImage } = require('electron');
const { join } = require('path');
const fs = require('fs');

let mainWindow: typeof BrowserWindow.prototype | null = null;
let tray: typeof Tray.prototype | null = null;
let isQuitting = false;

const VITE_DEV_SERVER_URL = process.env.VITE_DEV_SERVER_URL;

// 获取图标路径
function getIconPath() {
  return VITE_DEV_SERVER_URL 
    ? join(__dirname, '../public/favicon.svg')
    : join(__dirname, '../dist/favicon.svg');
}

// 创建托盘图标
function createTrayIcon(): typeof nativeImage.prototype {
  // 创建一个简单的蓝色图标作为托盘图标
  // 由于 SVG 在某些系统上可能不支持，这里使用 data URL 创建图标
  const iconSize = 16;
  const canvas = `
    <svg xmlns="http://www.w3.org/2000/svg" width="${iconSize}" height="${iconSize}" viewBox="0 0 512 512">
      <rect x="32" y="32" width="448" height="448" rx="96" fill="#4f8cff"/>
      <circle cx="256" cy="256" r="100" fill="#fff"/>
      <path d="M 228 190 L 228 322 L 330 256 Z" fill="#4f8cff"/>
    </svg>
  `;
  
  const dataUrl = `data:image/svg+xml;base64,${Buffer.from(canvas).toString('base64')}`;
  return nativeImage.createFromDataURL(dataUrl);
}

// 创建系统托盘
function createTray() {
  const trayIcon = createTrayIcon();
  tray = new Tray(trayIcon);
  
  const contextMenu = Menu.buildFromTemplate([
    {
      label: '显示主窗口',
      click: () => {
        if (mainWindow) {
          mainWindow.show();
          mainWindow.focus();
        }
      }
    },
    {
      label: '最小化到托盘',
      click: () => {
        mainWindow?.hide();
      }
    },
    { type: 'separator' },
    {
      label: '退出',
      click: () => {
        isQuitting = true;
        app.quit();
      }
    }
  ]);
  
  tray.setToolTip('多平台媒体管理系统');
  tray.setContextMenu(contextMenu);
  
  // 点击托盘图标显示窗口
  tray.on('click', () => {
    if (mainWindow) {
      if (mainWindow.isVisible()) {
        mainWindow.focus();
      } else {
        mainWindow.show();
        mainWindow.focus();
      }
    }
  });
  
  // 双击托盘图标显示窗口
  tray.on('double-click', () => {
    if (mainWindow) {
      mainWindow.show();
      mainWindow.focus();
    }
  });
}

function createWindow() {
  // 隐藏默认菜单栏
  Menu.setApplicationMenu(null);
  
  const iconPath = getIconPath();

  mainWindow = new BrowserWindow({
    width: 1400,
    height: 900,
    minWidth: 1200,
    minHeight: 700,
    icon: iconPath,
    webPreferences: {
      preload: join(__dirname, 'preload.js'),
      nodeIntegration: false,
      contextIsolation: true,
      webviewTag: true, // 启用 webview 标签
    },
    frame: false, // 无边框窗口，自定义标题栏
    transparent: false,
    backgroundColor: '#f0f2f5',
    show: false,
  });

  // 窗口准备好后再显示，避免白屏
  mainWindow.once('ready-to-show', () => {
    mainWindow?.show();
    setupWindowEvents();
  });

  // 加载页面
  if (VITE_DEV_SERVER_URL) {
    mainWindow.loadURL(VITE_DEV_SERVER_URL);
    mainWindow.webContents.openDevTools();
  } else {
    mainWindow.loadFile(join(__dirname, '../dist/index.html'));
  }

  // 处理外部链接
  mainWindow.webContents.setWindowOpenHandler(({ url }: { url: string }) => {
    shell.openExternal(url);
    return { action: 'deny' };
  });

  // 关闭按钮默认最小化到托盘
  mainWindow.on('close', (event: Event) => {
    if (!isQuitting) {
      event.preventDefault();
      mainWindow?.hide();
      
      // 显示托盘通知（仅首次）
      if (tray && !app.isPackaged) {
        // 开发模式下可以显示通知
      }
    }
  });

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

// 单实例锁定
const gotTheLock = app.requestSingleInstanceLock();

if (!gotTheLock) {
  app.quit();
} else {
  app.on('second-instance', () => {
    if (mainWindow) {
      mainWindow.show();
      if (mainWindow.isMinimized()) mainWindow.restore();
      mainWindow.focus();
    }
  });

  app.whenReady().then(() => {
    createTray();
    createWindow();

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

// 阻止默认的 window-all-closed 行为，保持托盘运行
app.on('window-all-closed', () => {
  // 不退出应用，保持托盘运行
  // 只有在 isQuitting 为 true 时才真正退出
});

// 应用退出前清理托盘
app.on('before-quit', () => {
  isQuitting = true;
});

app.on('quit', () => {
  if (tray) {
    tray.destroy();
    tray = null;
  }
});

// IPC 处理
ipcMain.handle('get-app-version', () => {
  return app.getVersion();
});

ipcMain.handle('get-platform', () => {
  return process.platform;
});

// 窗口控制
ipcMain.on('window-minimize', () => {
  mainWindow?.minimize();
});

ipcMain.on('window-maximize', () => {
  if (mainWindow?.isMaximized()) {
    mainWindow.unmaximize();
  } else {
    mainWindow?.maximize();
  }
});

// 关闭窗口（最小化到托盘）
ipcMain.on('window-close', () => {
  mainWindow?.hide();
});

// 真正退出应用
ipcMain.on('app-quit', () => {
  isQuitting = true;
  app.quit();
});

// 获取窗口最大化状态
ipcMain.handle('window-is-maximized', () => {
  return mainWindow?.isMaximized() || false;
});

// 监听窗口最大化/还原事件，通知渲染进程
function setupWindowEvents() {
  mainWindow?.on('maximize', () => {
    mainWindow?.webContents.send('window-maximized', true);
  });
  mainWindow?.on('unmaximize', () => {
    mainWindow?.webContents.send('window-maximized', false);
  });
}

// 获取 webview 的 cookies
ipcMain.handle('get-webview-cookies', async (_event: unknown, partition: string, url: string) => {
  try {
    const ses = session.fromPartition(partition);
    const cookies = await ses.cookies.get({ url });
    return cookies;
  } catch (error) {
    console.error('获取 cookies 失败:', error);
    return [];
  }
});

// 清除 webview 的 cookies
ipcMain.handle('clear-webview-cookies', async (_event: unknown, partition: string) => {
  try {
    const ses = session.fromPartition(partition);
    await ses.clearStorageData({ storages: ['cookies'] });
    return true;
  } catch (error) {
    console.error('清除 cookies 失败:', error);
    return false;
  }
});