AndroidRemoteController/nodejs/enable-wirless-connect.js

#!/usr/bin/env node
const { execSync } = require('child_process')
const path = require('path')

// 根目录 = 与根目录 config.js 同目录
const configPath = process.env.STATIC_ROOT
  ? path.join(path.dirname(process.env.STATIC_ROOT), 'config.js')
  : path.join(__dirname, '..', 'config.js')
const PROJECT_ROOT = path.dirname(path.resolve(configPath))
const TCPIP_PORT = 5555

/** 从配置解析并返回 ADB 可执行文件路径 */
function getAdbPath() {
  const fs = require('fs')
  if (!fs.existsSync(configPath)) return path.join(PROJECT_ROOT, 'lib', 'scrcpy-adb', process.platform === 'win32' ? 'adb.exe' : 'adb')
  const config = require(configPath)
  const p = config.adbPath?.path
  return p ? (path.isAbsolute(p) ? p : path.resolve(PROJECT_ROOT, p)) : path.join(PROJECT_ROOT, 'lib', 'scrcpy-adb', process.platform === 'win32' ? 'adb.exe' : 'adb')
}

/** 返回当前通过 USB 连接的设备 ID 列表（仅 status 为 device 且非 IP:port，排除无线设备） */
function getConnectedDeviceIds(adbPath) {
  const out = execSync(`"${adbPath}" devices`, { encoding: 'utf-8' })
  return out
    .split('\n')
    .filter((line) => line.trim() && !line.startsWith('List') && line.includes('\tdevice'))
    .map((line) => line.trim().split('\t')[0])
    .filter((id) => id && !id.includes(':'))
}

/** 操作一：在 USB 设备上开启无线调试（settings adb_wifi_enabled 1） */
function enableWirelessSetting(adbPath, deviceId) {
  execSync(`"${adbPath}" -s ${deviceId} shell settings put global adb_wifi_enabled 1`, { encoding: 'utf-8' })
}

/** 操作二：在 USB 设备上激活 5555 端口（tcpip），用于无线连接 */
function enableTcpipPort(adbPath, deviceId, port) {
  return execSync(`"${adbPath}" -s ${deviceId} tcpip ${port}`, { encoding: 'utf-8' }).trim()
}

/** 从 adb shell 输出中匹配 inet 后的 IPv4（支持 inet 192.168.1.1/24 格式），可排除回环 */
function parseInetFromOutput(out, skipLoopback = false) {
  const re = /inet\s+(\d+\.\d+\.\d+\.\d+)/g
  const ips = []
  let m
  while ((m = re.exec(out || '')) !== null) ips.push(m[1])
  if (ips.length === 0) return null
  if (skipLoopback) {
    const nonLoop = ips.find((ip) => ip !== '127.0.0.1')
    return nonLoop || null
  }
  return ips[0]
}

/** 安全拼 adb -s 参数（deviceId 含空格时需引号） */
function adbShellCmd(adbPath, deviceId, shellCmd) {
  const id = deviceId.indexOf(' ') >= 0 ? `"${deviceId}"` : deviceId
  return `"${adbPath}" -s ${id} shell ${shellCmd}`
}

/** 获取设备当前 WiFi IP：多接口 + 多命令兼容不同机型 */
function getDeviceIp(adbPath, deviceId) {
  const run = (shellCmd) => {
    try {
      return execSync(adbShellCmd(adbPath, deviceId, shellCmd), { encoding: 'utf-8' })
    } catch (e) {
      return ''
    }
  }
  // 1) wlan0 inet（最常见）
  let out = run('ip -4 addr show wlan0')
  let ip = parseInetFromOutput(out)
  if (ip) return ip
  // 2) eth0（部分机型/平板）
  out = run('ip -4 addr show eth0')
  ip = parseInetFromOutput(out)
  if (ip) return ip
  // 3) ip route get：src 或 from
  out = run('ip route get 1.1.1.1')
  let m = out.match(/\bsrc\s+(\d+\.\d+\.\d+\.\d+)\b/)
  if (m) return m[1]
  m = out.match(/\bfrom\s+(\d+\.\d+\.\d+\.\d+)\b/)
  if (m) return m[1]
  // 4) getprop
  for (const prop of ['dhcp.wlan0.ipaddress', 'net.wlan0.ipaddress', 'dhcp.eth0.ipaddress']) {
    out = run('getprop ' + prop)
    ip = (out || '').trim()
    if (/^\d+\.\d+\.\d+\.\d+$/.test(ip)) return ip
  }
  // 5) ip -4 addr show 全量，取第一个非回环 IP（避免 lo 的 127.0.0.1）
  out = run('ip -4 addr show')
  return parseInetFromOutput(out, true)
}

/** 主流程：取 USB 设备，先取 IP（与 getip.js 一致，此时 USB 稳定），再开无线与 tcpip，最后输出 */
function run() {
  const adbPath = getAdbPath()
  const devices = getConnectedDeviceIds(adbPath)
  if (devices.length === 0) {
    process.stderr.write('No devices found. Please connect a device via USB.\n')
    process.exit(1)
  }
  const deviceId = devices[0]
  // 先取 IP（在 tcpip 之前，USB 稳定时取，与 bat-tool/getip 行为一致）
  const ip = getDeviceIp(adbPath, deviceId)
  enableWirelessSetting(adbPath, deviceId)
  const tcpipOut = enableTcpipPort(adbPath, deviceId, TCPIP_PORT)
  process.stdout.write(tcpipOut + '\n')
  if (ip) {
    process.stdout.write('DEVICE_IP:' + ip + '\n')
  }
  process.exit(0)
}

run()