116 lines
4.0 KiB
Bash
Executable File
116 lines
4.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 工作手机 Agent 连接顺序：① NAS 局域网 ② 本机 Docker ③ NAS 外网 frp
|
||
# 用法: bash sdk/scripts/nas_primary_agent_setup.sh [device_id]
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
SDK_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"
|
||
DEVICE_ID="${1:-xgfe65eimrrofyws}"
|
||
SDK_PORT="${WORKPHONE_SDK_PORT:-8899}"
|
||
NAS_FRP_HOST="${WORKPHONE_NAS_FRP_HOST:-open.quwanzhi.com}"
|
||
ADB_PORT="${ANDROID_ADB_SERVER_PORT:-5038}"
|
||
ADB=(adb -P "${ADB_PORT}")
|
||
export ANDROID_ADB_SERVER_PORT="${ADB_PORT}"
|
||
|
||
pick_mac_lan_ip() {
|
||
local ip
|
||
for ip in $(ifconfig 2>/dev/null | awk '/inet /{print $2}' | grep -v '^127\.'); do
|
||
[[ "$ip" == 198.18.* ]] && continue
|
||
[[ "$ip" == 192.168.110.* ]] && { echo "$ip"; return; }
|
||
done
|
||
for ip in $(ifconfig 2>/dev/null | awk '/inet /{print $2}' | grep -v '^127\.'); do
|
||
[[ "$ip" == 198.18.* ]] && continue
|
||
[[ "$ip" == 192.168.* ]] && { echo "$ip"; return; }
|
||
done
|
||
echo "127.0.0.1"
|
||
}
|
||
|
||
health_ok() {
|
||
curl -sf --max-time 3 "http://$1:${SDK_PORT}/health" >/dev/null 2>&1
|
||
}
|
||
|
||
MAC_IP="$(pick_mac_lan_ip)"
|
||
|
||
# ① NAS 局域网 → ② 本机 Docker → ③ NAS 外网
|
||
PRIMARY_HOST=""
|
||
PRIMARY_TIER="nas_lan"
|
||
for h in 192.168.110.101 192.168.1.201; do
|
||
if health_ok "$h"; then
|
||
PRIMARY_HOST="$h"
|
||
break
|
||
fi
|
||
done
|
||
if [[ -z "${PRIMARY_HOST}" && "${MAC_IP}" != "127.0.0.1" ]] && health_ok "${MAC_IP}"; then
|
||
PRIMARY_HOST="${MAC_IP}"
|
||
PRIMARY_TIER="mac_docker"
|
||
fi
|
||
if [[ -z "${PRIMARY_HOST}" ]] && health_ok "${NAS_FRP_HOST}"; then
|
||
PRIMARY_HOST="${NAS_FRP_HOST}"
|
||
PRIMARY_TIER="nas_frp"
|
||
fi
|
||
if [[ -z "${PRIMARY_HOST}" ]]; then
|
||
PRIMARY_HOST="192.168.110.101"
|
||
echo "[nas-primary] 警告: ①②③ 均不可达，server_url 仍写 NAS 局域网 ${PRIMARY_HOST}（Agent 将按序探测）"
|
||
fi
|
||
|
||
PRIMARY_WS="ws://${PRIMARY_HOST}:${SDK_PORT}/ws/device"
|
||
CONFIG_TMP="$(mktemp)"
|
||
export DEVICE_ID PRIMARY_WS SDK_PORT MAC_IP NAS_FRP_HOST CONFIG_TMP
|
||
python3 - <<'PY'
|
||
import json, os
|
||
|
||
port = os.environ["SDK_PORT"]
|
||
mac = os.environ["MAC_IP"]
|
||
frp = os.environ["NAS_FRP_HOST"]
|
||
primary = os.environ["PRIMARY_WS"]
|
||
|
||
# 固定回退顺序：NAS 内网 → 本机 Docker → NAS 外网
|
||
ordered = [
|
||
f"ws://192.168.110.101:{port}/ws/device",
|
||
f"ws://192.168.1.201:{port}/ws/device",
|
||
f"ws://{mac}:{port}/ws/device",
|
||
f"ws://{frp}:{port}/ws/device",
|
||
]
|
||
pubs = []
|
||
seen = {primary}
|
||
for u in ordered:
|
||
if u not in seen:
|
||
seen.add(u)
|
||
pubs.append(u)
|
||
|
||
cfg = {
|
||
"device_id": os.environ["DEVICE_ID"],
|
||
"server_url": primary,
|
||
"public_servers": pubs,
|
||
"heartbeat_interval": 10,
|
||
"project_id": "cunkebao",
|
||
"connection_priority": ["nas_lan", "mac_docker", "nas_frp"],
|
||
}
|
||
with open(os.environ["CONFIG_TMP"], "w", encoding="utf-8") as f:
|
||
json.dump(cfg, f, ensure_ascii=False, indent=2)
|
||
PY
|
||
|
||
echo "[nas-primary] 连接顺序: ①NAS局域网 ②本机Docker ③NAS外网(${NAS_FRP_HOST})"
|
||
echo "[nas-primary] 当前主服 tier=${PRIMARY_TIER} host=${PRIMARY_HOST}"
|
||
echo "[nas-primary] server_url: ${PRIMARY_WS}/${DEVICE_ID}"
|
||
cat "${CONFIG_TMP}"
|
||
|
||
REMOTE_CFG="/data/data/com.termux/files/home/workphone/agent/config.json"
|
||
SERIAL="$("${ADB[@]}" devices 2>/dev/null | awk '/device$/{print $1; exit}')"
|
||
PUBLIC_LIST="ws://192.168.110.101:${SDK_PORT}/ws/device,ws://192.168.1.201:${SDK_PORT}/ws/device,ws://${MAC_IP}:${SDK_PORT}/ws/device,ws://${NAS_FRP_HOST}:${SDK_PORT}/ws/device"
|
||
|
||
if [[ -n "${SERIAL}" ]]; then
|
||
echo "[nas-primary] 推送 config → ${SERIAL}"
|
||
"${ADB[@]}" -s "${SERIAL}" push "${CONFIG_TMP}" "${REMOTE_CFG}" 2>/dev/null || \
|
||
"${ADB[@]}" -s "${SERIAL}" push "${CONFIG_TMP}" /sdcard/workphone_config.json
|
||
echo "[nas-primary] Termux:"
|
||
echo " export WP_SERVER_URL=${PRIMARY_WS}"
|
||
echo " export WP_PUBLIC_SERVERS=${PUBLIC_LIST}"
|
||
echo " export WP_AUTO_DISCOVER=1"
|
||
echo " cd ~/workphone/agent && python agent.py -d ${DEVICE_ID}"
|
||
else
|
||
echo "[nas-primary] 未检测到 adb device → ${SDK_ROOT}/agent/config.nas-active.json"
|
||
cp "${CONFIG_TMP}" "${SDK_ROOT}/agent/config.nas-active.json"
|
||
fi
|
||
rm -f "${CONFIG_TMP}"
|