AutoAndroidController/main.js

import { app, BrowserWindow, session, ipcMain } from 'electron';
import { fileURLToPath } from 'url';
import path from 'path';
import { exec } from 'child_process';
import { promisify } from 'util';

const execAsync = promisify(exec);

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const isDev = process.env.NODE_ENV === 'development' || !app.isPackaged;

// 设置内容安全策略（CSP），防止 XSS 攻击
function setContentSecurityPolicy() {
  session.defaultSession.webRequest.onHeadersReceived((details, callback) => {
    const csp = isDev
      ? "default-src 'self'; script-src 'self' 'unsafe-inline' http://localhost:*; style-src 'self' 'unsafe-inline'; connect-src 'self' http://localhost:* ws://localhost:*; img-src 'self' data: https: blob:; font-src 'self' data:; worker-src 'self' blob:;"
      : "default-src 'self'; script-src 'self'; style-src 'self' 'unsafe-inline'; connect-src 'self'; img-src 'self' data: https:; font-src 'self' data:;";
    
    const responseHeaders = Object.assign({}, details.responseHeaders);
    responseHeaders['Content-Security-Policy'] = [csp];
    
    callback({ responseHeaders });
  });
}

// 创建主窗口，根据环境加载不同内容源
function createWindow() {
  const mainWindow = new BrowserWindow({
    width: 1200,
    height: 800,
    webPreferences: {
      preload: path.join(__dirname, 'preload.cjs'),
      nodeIntegration: false,
      contextIsolation: true
    }
  });

  if (isDev) {
    mainWindow.loadURL('http://localhost:5173');
    mainWindow.webContents.openDevTools();
  } else {
    mainWindow.loadFile(path.join(__dirname, 'dist/index.html'));
  }
}

// 获取已连接的 ADB 设备列表
async function getADBDevices() {
  try {
    const { stdout } = await execAsync('adb devices');
    const lines = stdout.split('\n').slice(1);
    const devices = [];
    
    for (const line of lines) {
      const parts = line.trim().split(/\s+/);
      if (parts.length >= 2 && parts[0] && parts[1] === 'device') {
        devices.push({
          id: parts[0],
          status: parts[1]
        });
      }
    }
    return devices;
  } catch (error) {
    console.error('获取设备列表失败:', error);
    return [];
  }
}

// IPC 处理程序：获取 ADB 设备列表
ipcMain.handle('get-adb-devices', async () => {
  return await getADBDevices();
});

// IPC 处理程序：连接 ADB 设备
ipcMain.handle('connect-adb-device', async (event, ipPort) => {
  try {
    await execAsync(`adb connect ${ipPort}`);
    return { success: true };
  } catch (error) {
    console.error('连接设备失败:', error);
    return { success: false, error: error.message };
  }
});

// IPC 处理程序：获取设备分辨率
ipcMain.handle('get-device-resolution', async (event, ipPort) => {
  if (!ipPort) {
    return { success: false, error: '缺少设备 ID' };
  }
  try {
    // 使用 wm size 命令获取设备分辨率
    const { stdout } = await execAsync(`adb -s ${ipPort} shell wm size`);
    // 输出格式通常是: "Physical size: 1080x2400" 或 "1080x2400"
    const match = stdout.match(/(\d+)x(\d+)/);
    if (match) {
      return {
        success: true,
        width: parseInt(match[1], 10),
        height: parseInt(match[2], 10)
      };
    }
    // 如果解析失败，返回默认值
    return { success: true, width: 1280, height: 2400 };
  } catch (error) {
    console.error('获取设备分辨率失败:', error);
    // 返回默认值
    return { success: true, width: 1280, height: 2400 };
  }
});

// IPC 处理程序：抓取设备截屏（返回 base64 PNG）
ipcMain.handle('capture-screenshot', async (event, ipPort) => {
  if (!ipPort) {
    return { success: false, error: '缺少设备 ID' };
  }
  try {
    const { stdout } = await execAsync(`adb -s ${ipPort} exec-out screencap -p`, {
      encoding: 'buffer',
      maxBuffer: 25 * 1024 * 1024,
    });
    return { success: true, data: stdout.toString('base64') };
  } catch (error) {
    console.error('截屏失败:', error);
    return { success: false, error: error.message };
  }
});

// IPC 处理程序：发送 tap 事件到设备
ipcMain.handle('send-tap', async (event, ipPort, x, y) => {
  if (!ipPort) {
    return { success: false, error: '缺少设备 ID' };
  }
  if (typeof x !== 'number' || typeof y !== 'number') {
    return { success: false, error: '坐标必须是数字' };
  }
  try {
    const command = `adb -s ${ipPort} shell input tap ${x} ${y}`;
    await execAsync(command, {
      timeout: 5000,
      maxBuffer: 1024 * 1024
    });
    return { success: true };
  } catch (error) {
    console.error('Tap 失败:', error.message);
    return { success: false, error: error.message };
  }
});

// IPC 处理程序：发送 swipe 事件到设备
ipcMain.handle('send-swipe', async (event, ipPort, x1, y1, x2, y2, duration = 300) => {
  if (!ipPort) {
    return { success: false, error: '缺少设备 ID' };
  }
  if (typeof x1 !== 'number' || typeof y1 !== 'number' || typeof x2 !== 'number' || typeof y2 !== 'number') {
    return { success: false, error: '坐标必须是数字' };
  }
  try {
    const command = `adb -s ${ipPort} shell input swipe ${x1} ${y1} ${x2} ${y2} ${duration}`;
    await execAsync(command, {
      timeout: 5000,
      maxBuffer: 1024 * 1024
    });
    return { success: true };
  } catch (error) {
    console.error('Swipe 失败:', error.message);
    return { success: false, error: error.message };
  }
});

// 应用启动逻辑：设置 CSP、创建窗口、监听激活事件
app.whenReady().then(() => {
  setContentSecurityPolicy();
  createWindow();

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

// 应用关闭逻辑：macOS 保持运行，其他平台退出
app.on('window-all-closed', () => {
  if (process.platform !== 'darwin') {
    app.quit();
  }
});