AndroidRemoteController/nodejs/adb/screenshot.js

const { spawn } = require('child_process')
const { execSync } = require('child_process')
const path = require('path')
const fs = require('fs')

const configPath = process.env.STATIC_ROOT
  ? path.join(path.dirname(process.env.STATIC_ROOT), 'config.js')
  : path.join(__dirname, '..', '..', 'config.js')
const projectRoot = path.dirname(path.resolve(configPath))
const config = fs.existsSync(configPath) ? require(configPath) : {}
const adbPath = config.adbPath?.path
  ? (path.isAbsolute(config.adbPath.path) ? config.adbPath.path : path.resolve(projectRoot, config.adbPath.path))
  : path.join(projectRoot, 'lib', 'scrcpy-adb', process.platform === 'win32' ? 'adb.exe' : 'adb')
const scrcpyDir = path.dirname(adbPath)
const scrcpyPath = path.join(scrcpyDir, 'scrcpy.exe')
const staticRoot = process.env.STATIC_ROOT ? path.resolve(process.env.STATIC_ROOT) : path.join(projectRoot, 'static')
const pidFile = path.join(staticRoot, 'scrcpy-pid.json')

const action = process.argv[2]
const ipArg = process.argv[3]
const portArg = process.argv[4]
// 支持两种传参：1) ip, port 分开  2) ip:port 合并（无默认 port，只用传入值）
let ip, port
if (portArg !== undefined && portArg !== '') {
  ip = ipArg
  port = portArg
} else if (ipArg && String(ipArg).includes(':')) {
  const parts = String(ipArg).split(':')
  ip = parts[0] || ''
  port = parts[1] || ''
} else {
  ip = ipArg || ''
  port = portArg || ''
}

/** 终止指定 PID 的进程 */
function killPidIfRunning(pid) {
  spawn('taskkill.exe', ['/F', '/PID', pid.toString()], {
    stdio: 'ignore',
    detached: true
  }).unref()
}

/** 从 tasklist 获取 scrcpy.exe 的 PID（用 bat start 启动后需轮询得到 pid） */
function getScrcpyPid() {
  const output = execSync('tasklist /FI "IMAGENAME eq scrcpy.exe" /FO CSV', { encoding: 'utf-8' })
  const lines = output.split('\n').filter(line => line.includes('scrcpy.exe'))
  if (lines.length === 0) return null
  const pidMatch = lines[0].match(/"(\d+)"/)
  return pidMatch ? parseInt(pidMatch[1]) : null
}

/** 判断是否为 IP（或 IP:port），用于无线设备选择器 */
function isDeviceIp(val) {
  return val && !/^\d+$/.test(val) && /[\d.]/.test(val)
}

/** 启动 scrcpy：先 adb connect（若有 IP），再用 scrcpy-noconsole.bat（start "" scrcpy.exe %*）无控制台启动 */
function startScrcpy() {
  if (fs.existsSync(pidFile)) {
    const { pid } = JSON.parse(fs.readFileSync(pidFile, 'utf-8'))
    killPidIfRunning(pid)
  }

  let deviceSelector = ''
  if (ip && port && isDeviceIp(ip)) {
    deviceSelector = `${ip}:${port}`
    execSync(`"${adbPath}" connect ${deviceSelector}`, { encoding: 'utf-8', cwd: scrcpyDir })
  }

  const serial = deviceSelector
  const args = serial
    ? ['--pause-on-exit=if-error', '-s', serial]
    : ['-e', '--pause-on-exit=if-error']
  const argsStr = args.join(' ')
  const vbsPath = path.join(scrcpyDir, 'scrcpy-noconsole.vbs')
  execSync(`wscript "${vbsPath.replace(/"/g, '""')}" ${argsStr}`, { cwd: scrcpyDir })

  const sleep = (ms) => { const t = Date.now(); while (Date.now() - t < ms) {} }
  sleep(1500)
  const runningPid = getScrcpyPid()
  if (!runningPid) {
    console.log(JSON.stringify({ success: false, error: 'scrcpy did not appear in tasklist' }))
    process.exit(1)
  }
  fs.writeFileSync(pidFile, JSON.stringify({ pid: runningPid }), 'utf-8')
  console.log(JSON.stringify({ success: true, pid: runningPid }))
  process.exit(0)
}

/** 停止 scrcpy：有 pid 则 taskkill 并删文件，返回 action: 'stop'；无 pid 文件返回 action: 'none'，供前端区分是否继续执行 start */
function stopScrcpy() {
  if (!fs.existsSync(pidFile)) {
    console.log(JSON.stringify({ success: true, action: 'none' }))
    process.exit(0)
    return
  }
  const pid = JSON.parse(fs.readFileSync(pidFile, 'utf-8')).pid
  killPidIfRunning(pid)
  fs.unlinkSync(pidFile)
  console.log(JSON.stringify({ success: true, action: 'stop' }))
  process.exit(0)
}

switch (action) {
  case 'start':
    startScrcpy()
    break
  case 'stop':
    stopScrcpy()
    break
  default:
    console.log(JSON.stringify({ 
      success: false, 
      error: 'Usage: node screenshot.js [start|stop] [ip] [port]' 
    }))
    process.exit(1)
}