2fa-account-tool/main.js

const { app, BrowserWindow, ipcMain, clipboard, Menu } = require('electron');
const path = require('path');
const Database = require('./src/db/database');

// 强制设置用户数据存储位置为项目根目录下的 data 文件夹
// 这样 Electron 的缓存、Localstorage、日志等都不会写到 C 盘
app.setPath('userData', path.join(process.cwd(), 'data'));

let mainWindow;
let db;

function createWindow() {
  // 移除应用菜单栏
  Menu.setApplicationMenu(null);

  mainWindow = new BrowserWindow({
    width: 1200,
    height: 800,
    minWidth: 900,
    minHeight: 600,
    webPreferences: {
      preload: path.join(__dirname, 'preload.js'),
      contextIsolation: true,
      nodeIntegration: false
    },
    backgroundColor: '#0a0a0b',
    titleBarStyle: 'default',
    show: false
  });

  mainWindow.loadFile('src/index.html');

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

  // 开发时打开DevTools
  // mainWindow.webContents.openDevTools();
}

app.whenReady().then(() => {
  db = new Database();
  createWindow();

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

app.on('window-all-closed', () => {
  if (process.platform !== 'darwin') {
    app.quit();
  }
});

// ==================== Vault IPC Handlers ====================
ipcMain.handle('db:getVaults', async () => {
  return db.getVaults();
});

ipcMain.handle('db:addVault', async (event, vault) => {
  return db.addVault(vault);
});

ipcMain.handle('db:updateVault', async (event, id, vault) => {
  return db.updateVault(id, vault);
});

ipcMain.handle('db:deleteVault', async (event, id) => {
  return db.deleteVault(id);
});

// ==================== Account IPC Handlers ====================
ipcMain.handle('db:getAccounts', async (event, page, limit, vaultId) => {
  return db.getAccounts(page, limit, vaultId);
});

ipcMain.handle('db:getAccountCount', async (event, vaultId) => {
  return db.getAccountCount(vaultId);
});

ipcMain.handle('db:addAccount', async (event, account) => {
  return db.addAccount(account);
});

ipcMain.handle('db:updateAccount', async (event, id, account) => {
  return db.updateAccount(id, account);
});

ipcMain.handle('db:deleteAccount', async (event, id) => {
  return db.deleteAccount(id);
});

ipcMain.handle('db:moveAccountToVault', async (event, accountId, vaultId) => {
  return db.moveAccountToVault(accountId, vaultId);
});

ipcMain.handle('db:searchAccounts', async (event, query, vaultId) => {
  return db.searchAccounts(query, vaultId);
});

ipcMain.handle('clipboard:write', async (event, text) => {
  clipboard.writeText(text);
  return true;
});