feat(skill/playwright): unify shared-session CLI flow for web skills

2026-03-04 12:19:57 +08:00
parent aa72b570e1
commit 8fc5c6e128
12 changed files with 411 additions and 236 deletions
@@ -0,0 +1,148 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+if [[ $# -lt 1 ]]; then
+  echo "Usage: tools/pw <playwright-cli-command> [args...]" >&2
+  exit 1
+fi
+
+for arg in "$@"; do
+  if [[ "$arg" == "--session" || "$arg" == "--session="* ]]; then
+    echo "Do not pass --session directly. Use PLAYWRIGHT_SHARED_SESSION." >&2
+    exit 2
+  fi
+done
+
+CODEX_HOME="${CODEX_HOME:-$HOME/.codex}"
+PWCLI="${PWCLI:-$CODEX_HOME/skills/playwright/scripts/playwright_cli.sh}"
+SESSION="${PLAYWRIGHT_SHARED_SESSION:-codex-shared}"
+LOCK_TIMEOUT="${PLAYWRIGHT_SHARED_LOCK_TIMEOUT:-120}"
+LOCK_DIR="${PLAYWRIGHT_SHARED_LOCK_DIR:-/tmp/pw-shared-session.lock}"
+INIT_MODE="${PLAYWRIGHT_SHARED_INIT_MODE:-headed}"
+
+if ! command -v npx >/dev/null 2>&1; then
+  echo "npx is required." >&2
+  exit 1
+fi
+if [[ ! -x "$PWCLI" ]]; then
+  echo "Playwright wrapper not found or not executable: $PWCLI" >&2
+  exit 1
+fi
+
+acquire_lock() {
+  local start_ts now lock_pid lock_mtime
+  start_ts="$(date +%s)"
+  while ! mkdir "$LOCK_DIR" 2>/dev/null; do
+    lock_pid=""
+    if [[ -f "$LOCK_DIR/pid" ]]; then
+      lock_pid="$(cat "$LOCK_DIR/pid" 2>/dev/null || true)"
+    fi
+    if [[ -z "$lock_pid" ]]; then
+      lock_mtime="$(stat -f %m "$LOCK_DIR" 2>/dev/null || echo 0)"
+      now="$(date +%s)"
+      if (( now - lock_mtime >= 5 )); then
+        rm -rf "$LOCK_DIR" 2>/dev/null || true
+        continue
+      fi
+    fi
+    if [[ -n "$lock_pid" ]] && ! kill -0 "$lock_pid" 2>/dev/null; then
+      rm -rf "$LOCK_DIR" 2>/dev/null || true
+      continue
+    fi
+    now="$(date +%s)"
+    if (( now - start_ts >= LOCK_TIMEOUT )); then
+      echo "Timeout waiting for Playwright shared-session lock: $LOCK_DIR" >&2
+      exit 1
+    fi
+    sleep 1
+  done
+  printf "%s\n" "$$" > "$LOCK_DIR/pid"
+  trap 'rm -rf "$LOCK_DIR" 2>/dev/null || true' EXIT INT TERM
+}
+
+run_pw() {
+  "$PWCLI" --session "$SESSION" "$@"
+}
+
+is_missing_session_error() {
+  local text="$1"
+  if echo "$text" | rg -qi "session.*not found|no browser|not open|closed"; then
+    return 0
+  fi
+  return 1
+}
+
+acquire_lock
+
+cmd=("$@")
+verb="${cmd[0]}"
+
+if [[ "$verb" == "open" ]]; then
+  url=""
+  for (( i = 1; i < ${#cmd[@]}; i++ )); do
+    arg="${cmd[$i]}"
+    if [[ "$arg" != -* ]]; then
+      url="$arg"
+      break
+    fi
+  done
+
+  if [[ -n "$url" ]]; then
+    set +e
+    out="$(run_pw snapshot 2>&1)"
+    code=$?
+    set -e
+    if [[ $code -eq 0 ]]; then
+      run_pw goto "$url"
+      run_pw snapshot
+      exit 0
+    fi
+    if ! is_missing_session_error "$out"; then
+      echo "$out" >&2
+      exit $code
+    fi
+  fi
+fi
+
+if [[ "$verb" != "open" ]]; then
+  set +e
+  out="$(run_pw "${cmd[@]}" 2>&1)"
+  code=$?
+  set -e
+  if [[ $code -eq 0 ]]; then
+    echo "$out"
+    exit 0
+  fi
+  if is_missing_session_error "$out"; then
+    set +e
+    if [[ "$INIT_MODE" == "headless" ]]; then
+      run_pw open about:blank >/dev/null 2>&1
+      init_code=$?
+    elif [[ "$INIT_MODE" == "headed" ]]; then
+      run_pw open about:blank --headed >/dev/null 2>&1
+      init_code=$?
+      if [[ $init_code -ne 0 ]]; then
+        run_pw open about:blank >/dev/null 2>&1
+        init_code=$?
+      fi
+    else
+      run_pw open about:blank --headed >/dev/null 2>&1
+      init_code=$?
+      if [[ $init_code -ne 0 ]]; then
+        run_pw open about:blank >/dev/null 2>&1
+        init_code=$?
+      fi
+    fi
+    set -e
+    if [[ $init_code -ne 0 ]]; then
+      echo "Failed to initialize shared Playwright session." >&2
+      exit 1
+    fi
+    run_pw "${cmd[@]}"
+    exit 0
+  fi
+  echo "$out" >&2
+  exit $code
+fi
+
+run_pw "${cmd[@]}"