2fa-account-tool/main.js

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

// Force userData path to project root/data folder
// Prevents Electron cache/localStorage/logs from writing to C drive
app.setPath('userData', path.join(process.cwd(), 'data'));

let mainWindow;
let db;
let backupInterval;

// Auto backup feature
function startAutoBackup() {
  const backupDir = path.join(process.cwd(), 'DB');
  const dbPath = path.join(process.cwd(), 'accounts.db');

  // Ensure backup directory exists
  if (!fs.existsSync(backupDir)) {
    fs.mkdirSync(backupDir, { recursive: true });
  }

  // Backup every 30 minutes
  backupInterval = setInterval(() => {
    try {
      if (fs.existsSync(dbPath)) {
        const now = new Date();
        const timestamp = now.getFullYear().toString() +
          (now.getMonth() + 1).toString().padStart(2, '0') +
          now.getDate().toString().padStart(2, '0') + '_' +
          now.getHours().toString().padStart(2, '0') +
          now.getMinutes().toString().padStart(2, '0') +
          now.getSeconds().toString().padStart(2, '0');

        const backupPath = path.join(backupDir, `accounts_${timestamp}.db`);
        fs.copyFileSync(dbPath, backupPath);
        console.log(`[Backup] Auto backup success: ${backupPath}`);

        // Clean old backups, keep last 10
        cleanOldBackups(backupDir, 10);
      }
    } catch (e) {
      console.error('[Backup] Backup failed:', e);
    }
  }, 30 * 60 * 1000); // 30 minutes

  console.log('[Backup] Auto backup started, interval 30 minutes');
}

// Clean old backups
function cleanOldBackups(backupDir, keepCount) {
  try {
    const files = fs.readdirSync(backupDir)
      .filter(f => f.startsWith('accounts_') && f.endsWith('.db'))
      .map(f => ({ name: f, path: path.join(backupDir, f), time: fs.statSync(path.join(backupDir, f)).mtime }))
      .sort((a, b) => b.time - a.time);

    if (files.length > keepCount) {
      files.slice(keepCount).forEach(f => {
        fs.unlinkSync(f.path);
        console.log(`[Backup] Deleted old backup: ${f.name}`);
      });
    }
  } catch (e) {
    console.error('[Backup] Clean old backups failed:', e);
  }
}

function createWindow() {
  // Remove application menu bar
  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,
      sandbox: false
    },
    backgroundColor: '#0a0a0b',
    titleBarStyle: 'default',
    show: false
  });

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

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

  // DevTools: opt-in only (prevents auto-opening on startup)
  if (process.env.ELECTRON_DEVTOOLS === '1') {
    mainWindow.webContents.openDevTools();
  }
}

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

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

app.on('window-all-closed', () => {
  if (backupInterval) {
    clearInterval(backupInterval);
  }
  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;
});

// Screen capture feature
ipcMain.handle('screen:capture', async () => {
  try {
    const primaryDisplay = screen.getPrimaryDisplay();
    const { width, height } = primaryDisplay.size;
    const scaleFactor = primaryDisplay.scaleFactor;

    const sources = await desktopCapturer.getSources({
      types: ['screen'],
      thumbnailSize: { width: width * scaleFactor, height: height * scaleFactor }
    });

    if (sources.length > 0) {
      const thumbnail = sources[0].thumbnail;
      return {
        success: true,
        image: thumbnail.toDataURL(),
        width: thumbnail.getSize().width,
        height: thumbnail.getSize().height
      };
    }
    return { success: false, error: 'Failed to capture screen' };
  } catch (e) {
    console.error('Screen capture failed:', e);
    return { success: false, error: e.message };
  }
});