Web Access

#!/usr/bin/env bash
# web-access 共享工具函数，供 ensure-browser.sh 和 close-browser.sh source

# 返回 macos / linux / windows
get_os() {
  case "$(uname -s)" in
    Darwin)              echo "macos" ;;
    Linux)               echo "linux" ;;
    MINGW*|MSYS*|CYGWIN*) echo "windows" ;;
    *)                   echo "linux" ;;
  esac
}

# 跨平台 Chrome 路径探测
find_chrome() {
  # macOS
  local mac_path="/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
  [ -f "$mac_path" ] && echo "$mac_path" && return

  # Linux
  for p in \
    "$(command -v google-chrome 2>/dev/null)" \
    "$(command -v google-chrome-stable 2>/dev/null)" \
    "$(command -v chromium 2>/dev/null)" \
    "$(command -v chromium-browser 2>/dev/null)" \
    /usr/bin/google-chrome /usr/bin/google-chrome-stable \
    /usr/bin/chromium /usr/bin/chromium-browser /snap/bin/chromium; do
    [ -f "$p" ] && echo "$p" && return
  done

  # Windows（Git Bash）
  if [ "$(get_os)" = "windows" ]; then
    local local_app programfiles
    local_app=$(cygpath "$LOCALAPPDATA" 2>/dev/null)
    programfiles=$(cygpath "$PROGRAMFILES" 2>/dev/null)
    for p in \
      "$local_app/Google/Chrome/Application/chrome.exe" \
      "$programfiles/Google/Chrome/Application/chrome.exe"; do
      [ -f "$p" ] && echo "$p" && return
    done
  fi

  echo ""
}

# 按 OS 适当关闭进程
kill_pid() {
  local pid=$1
  case "$(get_os)" in
    macos)
      osascript -e "tell application \"System Events\" to tell (first process whose unix id is ${pid}) to quit" 2>/dev/null \
        || kill "$pid" 2>/dev/null
      ;;
    windows)
      taskkill /PID "$pid" /F 2>/dev/null
      ;;
    *)
      kill "$pid" 2>/dev/null
      ;;
  esac
}

#!/usr/bin/env bash
# 环境探测 - 输出依赖状态供 AI 判断和处理

UTILS_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=_utils.sh
source "$UTILS_DIR/_utils.sh"

echo "OS:            $(uname -s) $(uname -m)"

CHROME_PATH=$(find_chrome)
echo "chrome:        ${CHROME_PATH:-missing}"
echo "node:          $(command -v node 2>/dev/null && node --version 2>/dev/null || echo 'missing')"
echo "npm:           $(command -v npm 2>/dev/null || echo 'missing')"
echo "agent-browser: $(command -v agent-browser 2>/dev/null || echo 'missing  → npm install -g agent-browser')"

#!/usr/bin/env bash
# 优雅关闭 agent-browser Chrome 实例（跨平台）
# 用法：bash close-browser.sh [PORT]  # 默认 9222

UTILS_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=_utils.sh
source "$UTILS_DIR/_utils.sh"

CDP_PORT=${1:-9222}
if [ "$CDP_PORT" = "9222" ]; then
  PROFILE_DIR="$HOME/.claude/browser-profile"
else
  PROFILE_DIR="$HOME/.claude/browser-profile-${CDP_PORT}"
fi
SNAPSHOT_DIR="$HOME/.claude/browser-profile-snapshot"

# 1. CDP Browser.close —— 协议层关闭，跨平台，Chrome 自己正常退出
# 用 Python 发 WebSocket 帧，不依赖外部包
python3 - "${CDP_PORT}" <<'PYEOF' 2>/dev/null
import json, socket, base64, urllib.request, struct, sys

try:
    port = int(sys.argv[1]) if len(sys.argv) > 1 else 9222
    data = json.loads(urllib.request.urlopen(f'http://localhost:{port}/json/version', timeout=3).read())
    ws_url = data['webSocketDebuggerUrl']
    path = ws_url[len(f'ws://localhost:{port}'):]

    s = socket.socket()
    s.settimeout(5)
    s.connect(('localhost', port))
    key = base64.b64encode(b'claude-web-access!!').decode()
    s.send(f'GET {path} HTTP/1.1\r\nHost: localhost:{port}\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: {key}\r\nSec-WebSocket-Version: 13\r\n\r\n'.encode())
    s.recv(4096)

    msg = json.dumps({"id": 1, "method": "Browser.close"}).encode()
    mask = bytes([0, 0, 0, 0])
    masked = bytes(b ^ mask[i % 4] for i, b in enumerate(msg))
    frame = bytes([0x81, 0x80 | len(msg)]) + mask + masked
    s.send(frame)
    s.close()
    print("Browser closed")
except Exception as e:
    print(f"CDP close failed: {e}", flush=True)
    exit(1)
PYEOF

# 2. 兜底：CDP 失败时按 OS 强制退出
if [ $? -ne 0 ]; then
  case "$(get_os)" in
    macos)
      OUR_PID=$(ps aux | grep "Google Chrome" | grep -- "--user-data-dir=${PROFILE_DIR}" | grep -v grep | awk '{print $2}' | head -1)
      [ -n "$OUR_PID" ] && kill_pid "$OUR_PID" && echo "Browser closed (osascript)"
      ;;
    linux)
      OUR_PID=$(ps aux | grep "google-chrome\|chromium" | grep -- "--user-data-dir=${PROFILE_DIR}" | grep -v grep | awk '{print $2}' | head -1)
      [ -n "$OUR_PID" ] && kill_pid "$OUR_PID" && echo "Browser closed (SIGTERM)"
      ;;
    windows)
      # wmic 已在 Windows 11+ 移除，改用 Get-CimInstance
      OUR_PID=$(powershell.exe -NoProfile -Command \
        "Get-CimInstance Win32_Process -Filter 'Name=\"chrome.exe\"' | Where-Object { \$_.CommandLine -match 'browser-profile' } | Select-Object -First 1 -ExpandProperty ProcessId" \
        2>/dev/null | tr -d '\r\n ')
      [ -n "$OUR_PID" ] && kill_pid "$OUR_PID" && echo "Browser closed (taskkill)"
      ;;
    *)
      echo "Unknown OS — please close the browser manually"
      ;;
  esac
fi

# 3. 关闭 9222 后自动更新 snapshot，供非 9222 端口克隆登录态
if [ "$CDP_PORT" = "9222" ] && [ -d "$PROFILE_DIR" ]; then
  mkdir -p "$SNAPSHOT_DIR"
  if command -v rsync >/dev/null 2>&1; then
    rsync -a --delete \
      --exclude="SingletonLock" --exclude="SingletonCookie" --exclude="SingletonSocket" \
      --exclude="Default/Cache/" --exclude="Default/Code Cache/" --exclude="Default/GPUCache/" \
      --exclude="Default/Service Worker/CacheStorage/" \
      --exclude="ShaderCache/" --exclude="GrShaderCache/" --exclude="*.lock" \
      "$PROFILE_DIR/" "$SNAPSHOT_DIR/" 2>/dev/null
    echo "Profile snapshot updated"
  elif [ "$(get_os)" = "windows" ]; then
    # Git Bash 默认无 rsync，改用 robocopy（Windows 内置）
    PROFILE_WIN=$(cygpath -w "$PROFILE_DIR" 2>/dev/null || echo "$PROFILE_DIR")
    SNAPSHOT_WIN=$(cygpath -w "$SNAPSHOT_DIR" 2>/dev/null || echo "$SNAPSHOT_DIR")
    powershell.exe -NoProfile -Command \
      "robocopy '$PROFILE_WIN' '$SNAPSHOT_WIN' /MIR /XD Cache 'Code Cache' GPUCache ShaderCache GrShaderCache /XF '*.lock' SingletonLock SingletonCookie SingletonSocket | Out-Null" 2>/dev/null
    echo "Profile snapshot updated (robocopy)"
  fi
fi

# 4. 清理 agent-browser session daemon（如果存在）
SESSION_PID=$(cat "$HOME/.agent-browser/port-${CDP_PORT}.pid" 2>/dev/null)
if [ -n "$SESSION_PID" ] && kill -0 "$SESSION_PID" 2>/dev/null; then
  kill "$SESSION_PID" 2>/dev/null
fi
rm -f "$HOME/.agent-browser/port-${CDP_PORT}.pid" \
      "$HOME/.agent-browser/port-${CDP_PORT}.sock"

#!/usr/bin/env bash
# Chrome CDP 生命周期管理（始终 headed 模式）
# 用法：bash ensure-browser.sh [PORT]
# PORT 由主 agent 分配（默认 9222；并行时传入 9223、9224 等）

UTILS_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=_utils.sh
source "$UTILS_DIR/_utils.sh"

CHROME=$(find_chrome)
if [ -z "$CHROME" ]; then
  echo "ERROR: Chrome not found. Please install Google Chrome." >&2
  exit 1
fi

CDP_PORT=${1:-9222}
SNAPSHOT_DIR="$HOME/.claude/browser-profile-snapshot"
OS=$(get_os)

# Profile 路径（9222 保持历史路径兼容）
if [ "$CDP_PORT" = "9222" ]; then
  PROFILE_DIR="$HOME/.claude/browser-profile"
else
  PROFILE_DIR="$HOME/.claude/browser-profile-${CDP_PORT}"
fi

# Windows 下 Chrome 需要 Windows 风格路径
if [ "$OS" = "windows" ]; then
  PROFILE_DIR_CHROME=$(cygpath -w "$PROFILE_DIR")
else
  PROFILE_DIR_CHROME="$PROFILE_DIR"
fi

# 检测 CDP 端口是否已就绪
check_ready() {
  curl -s "http://localhost:${CDP_PORT}/json/version" >/dev/null 2>&1
}

# 检测当前运行的 Chrome 是否使用了正确的 profile
check_profile() {
  if [ "$OS" = "windows" ]; then
    local profile_pattern
    if [ "$CDP_PORT" = "9222" ]; then
      # 9222 的目录名是 browser-profile（无后缀），用 [^-] 避免误匹配 browser-profile-9223 等
      profile_pattern="browser-profile[^-]"
    else
      profile_pattern="browser-profile-${CDP_PORT}"
    fi
    wmic process where "name='chrome.exe'" get commandline 2>/dev/null \
      | grep -qE "$profile_pattern" 2>/dev/null
  else
    ps aux | grep -E "Google Chrome|google-chrome|chromium" \
      | grep -- "--user-data-dir=${PROFILE_DIR}" \
      | grep -v grep >/dev/null 2>&1
  fi
}

if check_ready; then
  if check_profile; then
    AGENT_BROWSER_SESSION="port-${CDP_PORT}" agent-browser connect "${CDP_PORT}" >/dev/null 2>&1 || true
    echo "Browser ready on port ${CDP_PORT}"
    exit 0
  else
    echo "ERROR: Port ${CDP_PORT} is in use by another process." >&2
    exit 1
  fi
fi

# 非主端口：每次从 snapshot 克隆最新登录态（覆盖已有 profile）
if [ "$CDP_PORT" != "9222" ]; then
  if [ -d "$SNAPSHOT_DIR" ]; then
    mkdir -p "$PROFILE_DIR"
    rsync -a --delete \
      --exclude="SingletonLock" --exclude="SingletonCookie" --exclude="SingletonSocket" \
      --exclude="Default/Cache/" --exclude="Default/Code Cache/" --exclude="Default/GPUCache/" \
      --exclude="Default/Service Worker/CacheStorage/" \
      --exclude="ShaderCache/" --exclude="GrShaderCache/" --exclude="*.lock" \
      "$SNAPSHOT_DIR/" "$PROFILE_DIR/" 2>/dev/null
    echo "Profile cloned from snapshot" >&2
  else
    # 全新环境，用空 profile 启动
    mkdir -p "$PROFILE_DIR"
    echo "INFO: No snapshot available, starting with fresh profile" >&2
  fi
fi

# 如果有我们自己的 Chrome 残留（有 profile 但没监听端口），清理掉
if [ "$OS" = "windows" ]; then
  OUR_PID=$(wmic process where "name='chrome.exe'" get commandline,processid 2>/dev/null \
    | grep "browser-profile" | grep -oE '[0-9]+$' | head -1 | tr -d ' \r\n')
else
  OUR_PID=$(ps aux | grep -E "Google Chrome|google-chrome|chromium" \
    | grep -- "--user-data-dir=${PROFILE_DIR}" \
    | grep -v grep | awk '{print $2}' | head -1)
fi

if [ -n "$OUR_PID" ]; then
  kill_pid "$OUR_PID"
  sleep 1
fi

# 启动 Chrome（后台，始终 headed）
"$CHROME" \
  "--remote-debugging-port=${CDP_PORT}" \
  "--user-data-dir=${PROFILE_DIR_CHROME}" \
  "--no-first-run" \
  "--no-default-browser-check" \
  "--exclude-switches=enable-automation" \
  "--disable-infobars" \
  >/dev/null 2>&1 &

# 等待 CDP 就绪（最多 15 秒）
for i in $(seq 1 30); do
  if check_ready; then
    AGENT_BROWSER_SESSION="port-${CDP_PORT}" agent-browser connect "${CDP_PORT}" >/dev/null 2>&1 || true
    echo "Browser ready on port ${CDP_PORT}"
    exit 0
  fi
  sleep 0.5
done

echo "ERROR: Browser failed to start within 15 seconds" >&2
exit 1