AutoAndroidController/main-js/adb/device-manager.js

import { ipcMain, BrowserWindow } from 'electron';
import { exec } from 'child_process';
import { promisify } from 'util';
import { getCachedAdbPath, loadConfig } from '../config.js';

const execAsync = promisify(exec);

// 获取主窗口引用（用于实时推送设备发现事件）
function getMainWindow() {
  const windows = BrowserWindow.getAllWindows();
  return windows.length > 0 ? windows[0] : null;
}

// 获取 ADB 设备列表（每次都重新扫描网络）
export async function getADBDevices() {
  try {
    // 每次都重新扫描网络设备，不记录已连接的设备
    return await scanNetworkDevices(null);
  } catch (error) {
    console.error('获取设备列表失败:', error);
    return [];
  }
}

// 网络扫描：按顺序扫描 192.168.0、192.168.1、192.168.2 三个网段，尝试连接设备（实时推送结果）
export async function scanNetworkDevices(event) {
  const adbPath = getCachedAdbPath();
  const port = 5555;
  const maxConcurrent = 50; // 限制并发数，避免过载
  const connectTimeout = 1500; // 连接超时时间（毫秒）
  
  const foundDevices = new Set(); // 使用 Set 避免重复
  
  // 定义要扫描的网段（按顺序：0.0 -> 1.0 -> 2.0）
  const networkSegments = ['192.168.0', '192.168.1', '192.168.2'];
  
  // 推送设备发现的辅助函数
  const pushDevice = (ipPort) => {
    if (!foundDevices.has(ipPort)) {
      foundDevices.add(ipPort);
      const device = {
        id: ipPort,
        status: 'device'
      };
      // 实时发送发现的设备
      if (event && event.sender) {
        event.sender.send('device-found', device);
      } else {
        const mainWindow = getMainWindow();
        if (mainWindow) {
          mainWindow.webContents.send('device-found', device);
        }
      }
      console.log('发现设备:', ipPort);
    }
  };
  
  // 扫描单个 IP 地址的函数
  const scanIP = async (ip) => {
    const ipPort = `${ip}:${port}`;
    try {
      // 先尝试连接设备
      await execAsync(`${adbPath} connect ${ipPort}`, {
        timeout: connectTimeout,
        maxBuffer: 1024 * 1024
      });
      
      // 连接后稍等片刻，让设备注册到 ADB 服务器
      await new Promise(resolve => setTimeout(resolve, 500));
    } catch (error) {
      // 连接失败是正常的，继续检查设备列表
    }
    
    // 调用 adb devices 命令获取所有设备列表，检查是否包含端口 5555 的设备
    try {
      const { stdout } = await execAsync(`${adbPath} devices`, {
        timeout: 2000,
        maxBuffer: 1024 * 1024
      });
      
      // 检查输出中是否包含该 IP:PORT（匹配端口 5555）
      // 使用正则表达式匹配 IP:5555 格式
      const devicePattern = new RegExp(`${ip.replace(/\./g, '\\.')}:${port}\\s+device`, 'i');
      if (devicePattern.test(stdout)) {
        // 发现设备，立即推送显示（不等待整批完成）
        pushDevice(ipPort);
        return ipPort;
      }
      // 如果没有匹配到，说明没有设备
    } catch (checkError) {
      // 检查失败，忽略
    }
    return null;
  };
  
  // 按顺序扫描每个网段
  for (const baseIP of networkSegments) {
    console.log(`开始扫描网段: ${baseIP}.x`);
    
    // 生成当前网段的 IP 地址列表
    const ipList = [];
    for (let i = 1; i <= 255; i++) {
      ipList.push(`${baseIP}.${i}`);
    }
    
    // 分批并发扫描当前网段，扫描到一个显示一个
    for (let i = 0; i < ipList.length; i += maxConcurrent) {
      const batch = ipList.slice(i, i + maxConcurrent);
      const promises = batch.map(ip => scanIP(ip));
      
      // 并发执行当前批次，但不等待所有完成才推送
      // 每个 promise 一旦发现设备就会立即推送
      await Promise.all(promises);
    }
    
    console.log(`完成扫描网段: ${baseIP}.x`);
  }
  
  // 返回所有发现的设备
  return Array.from(foundDevices).map(ipPort => ({
    id: ipPort,
    status: 'device'
  }));
}

// 连接 ADB 设备
export async function connectDevice(ipPort) {
  try {
    const adbPath = getCachedAdbPath();
    await execAsync(`${adbPath} connect ${ipPort}`);
    return { success: true };
  } catch (error) {
    console.error('连接设备失败:', error);
    return { success: false, error: error.message };
  }
}

// 注册 IPC 处理器
export function registerIpcHandlers() {
  // IPC 处理程序：获取 ADB 路径配置
  ipcMain.handle('get-adb-path-config', async () => {
    const config = loadConfig();
    return config ? config['adb-path'] : null;
  });

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

  // IPC 处理程序：扫描网络设备（支持实时推送）
  ipcMain.handle('scan-adb-devices', async (event) => {
    try {
      const devices = await scanNetworkDevices(event);
      return devices;
    } catch (error) {
      console.error('网络扫描失败:', error);
      return [];
    }
  });

  // IPC 处理程序：连接 ADB 设备
  ipcMain.handle('connect-adb-device', async (event, ipPort) => {
    return await connectDevice(ipPort);
  });
}