@pasko70/pibo 3.0.2 → 3.1.0

package/scripts/prepare-browser-use-wrapper.sh

@@ -0,0 +1,255 @@
+#!/bin/sh
+set -e
+
+real_browser_use="$HOME/.pibo/tools/browser-use/.venv/bin/browser-use"
+wrapper_dir="$HOME/.pibo/tools/browser-use/home/bin"
+wrapper_path=$wrapper_dir/browser-use
+
+mkdir -p "$wrapper_dir"
+
+cat > "$wrapper_path" <<'WRAPPER'
+#!/bin/sh
+set -eu
+
+real_browser_use="$HOME/.pibo/tools/browser-use/.venv/bin/browser-use"
+default_profile='PIBo'
+fresh_flag='--fresh-profile'
+python_cmd="$HOME/.pibo/tools/browser-use/.venv/bin/python"
+
+session=${PIBO_BROWSER_USE_SESSION:-default}
+explicit_profile=0
+fresh_profile=0
+starts_browser=0
+show_help=0
+headed=0
+sanitized_args=
+
+append_arg() {
+  if [ "$#" -eq 0 ]; then return; fi
+  if [ -z "$sanitized_args" ]; then
+    sanitized_args=$(printf '%s\n' "$1")
+  else
+    sanitized_args=$(printf '%s\n%s\n' "$sanitized_args" "$1")
+  fi
+}
+
+find_chrome() {
+  if [ -n "${PIBO_BROWSER_USE_CHROME:-}" ]; then
+    printf '%s\n' "$PIBO_BROWSER_USE_CHROME"
+    return
+  fi
+  for candidate in google-chrome chrome chromium chromium-browser /opt/google/chrome/chrome; do
+    if command -v "$candidate" >/dev/null 2>&1; then
+      command -v "$candidate"
+      return
+    fi
+    if [ -x "$candidate" ]; then
+      printf '%s\n' "$candidate"
+      return
+    fi
+  done
+}
+
+resolve_profile_directory() {
+  user_data_dir=$1
+  profile_name=$2
+  if [ -n "${PIBO_BROWSER_USE_CHROME_PROFILE_DIRECTORY:-}" ]; then
+    printf '%s\n' "$PIBO_BROWSER_USE_CHROME_PROFILE_DIRECTORY"
+    return
+  fi
+  "$python_cmd" - "$user_data_dir" "$profile_name" <<'PY'
+import json, sys
+from pathlib import Path
+user_data_dir = Path(sys.argv[1])
+profile_name = sys.argv[2]
+local_state = user_data_dir / "Local State"
+try:
+    data = json.loads(local_state.read_text())
+except Exception:
+    print(profile_name if (user_data_dir / profile_name).is_dir() else "Default")
+    raise SystemExit
+info_cache = data.get("profile", {}).get("info_cache", {})
+if profile_name in info_cache:
+    print(profile_name)
+    raise SystemExit
+profile_name_lower = profile_name.lower()
+for directory, info in info_cache.items():
+    if str(info.get("name", directory)).lower() == profile_name_lower:
+        print(directory)
+        raise SystemExit
+for directory in info_cache:
+    if directory.lower() == profile_name_lower:
+        print(directory)
+        raise SystemExit
+print(profile_name if (user_data_dir / profile_name).is_dir() else "Default")
+PY
+}
+
+find_free_port() {
+  "$python_cmd" - <<'PY'
+import socket
+with socket.socket() as sock:
+    sock.bind(("127.0.0.1", 0))
+    print(sock.getsockname()[1])
+PY
+}
+
+pibo_cdp_is_ready() {
+  "$python_cmd" - "$1" <<'PY'
+import socket, sys
+port = int(sys.argv[1])
+try:
+    with socket.create_connection(("127.0.0.1", port), timeout=0.2) as sock:
+        sock.sendall(b"GET /json/version HTTP/1.1\r\nHost: 127.0.0.1\r\nConnection: close\r\n\r\n")
+        data = sock.recv(128)
+    raise SystemExit(0 if b"200" in data else 1)
+except Exception:
+    raise SystemExit(1)
+PY
+}
+
+persistent_chrome_url_if_alive() {
+  safe_session=$(printf '%s' "$session" | tr -c 'A-Za-z0-9_.-' '_')
+  state_dir=${BROWSER_USE_HOME:-$HOME/.browser-use}/pibo-cdp
+  pid_file=$state_dir/$safe_session.pid
+  port_file=$state_dir/$safe_session.port
+  if [ -s "$pid_file" ] && [ -s "$port_file" ]; then
+    chrome_pid=$(cat "$pid_file" 2>/dev/null || true)
+    chrome_port=$(cat "$port_file" 2>/dev/null || true)
+    if [ -n "$chrome_pid" ] && [ -n "$chrome_port" ] && kill -0 "$chrome_pid" 2>/dev/null && pibo_cdp_is_ready "$chrome_port"; then
+      printf 'http://127.0.0.1:%s\n' "$chrome_port"
+      return 0
+    fi
+  fi
+  return 1
+}
+
+ensure_persistent_chrome() {
+  safe_session=$(printf '%s' "$session" | tr -c 'A-Za-z0-9_.-' '_')
+  state_dir=${BROWSER_USE_HOME:-$HOME/.browser-use}/pibo-cdp
+  mkdir -p "$state_dir"
+  pid_file=$state_dir/$safe_session.pid
+  port_file=$state_dir/$safe_session.port
+  log_file=$state_dir/$safe_session.chrome.log
+
+  if cdp_url=$(persistent_chrome_url_if_alive); then
+    printf '%s\n' "$cdp_url"
+    return
+  fi
+
+  chrome_bin=$(find_chrome)
+  if [ -z "$chrome_bin" ]; then
+    echo "pibo browser-use: could not find Chrome" >&2
+    exit 1
+  fi
+
+  user_data_dir=${PIBO_BROWSER_USE_CHROME_USER_DATA_DIR:-}
+  if [ -z "$user_data_dir" ]; then
+    user_data_dir=${BROWSER_USE_HOME:-$HOME/.browser-use}/chrome-profiles/$default_profile
+  fi
+
+  mkdir -p "$user_data_dir"
+  profile_directory=$(resolve_profile_directory "$user_data_dir" "$default_profile")
+  chrome_port=$(find_free_port)
+  headless_arg=
+  if [ "$headed" -eq 0 ]; then
+    headless_arg="--headless=new"
+  fi
+
+  nohup "$chrome_bin" \
+    --user-data-dir="$user_data_dir" \
+    --profile-directory="$profile_directory" \
+    --remote-debugging-port="$chrome_port" \
+    --no-first-run \
+    --no-default-browser-check \
+    --disable-default-apps \
+    --no-sandbox \
+    --disable-setuid-sandbox \
+    --disable-dev-shm-usage \
+    $headless_arg \
+    about:blank >"$log_file" 2>&1 &
+  chrome_pid=$!
+
+  ready=0
+  if [ "${PIBO_BROWSER_USE_SKIP_CDP_WAIT:-}" = "1" ]; then ready=1; fi
+  attempt=0
+  while [ "$ready" -ne 1 ] && [ "$attempt" -lt 50 ]; do
+    if pibo_cdp_is_ready "$chrome_port"; then ready=1; break; fi
+    if ! kill -0 "$chrome_pid" 2>/dev/null; then break; fi
+    attempt=$((attempt + 1))
+    sleep 0.1
+  done
+
+  if [ "$ready" -ne 1 ]; then
+    echo "pibo browser-use: Chrome did not expose CDP on port $chrome_port." >&2
+    sed -n '1,80p' "$log_file" >&2 2>/dev/null || true
+    exit 1
+  fi
+
+  printf '%s\n' "$chrome_pid" > "$pid_file"
+  printf '%s\n' "$chrome_port" > "$port_file"
+  printf 'pibo browser-use: started Chrome profile "%s" on CDP port %s.\n' "$default_profile" "$chrome_port" >&2
+  printf 'http://127.0.0.1:%s\n' "$chrome_port"
+}
+
+while [ "$#" -gt 0 ]; do
+  arg=$1; shift
+  case "$arg" in
+    "$fresh_flag") fresh_profile=1; continue ;;
+    --help|-h) show_help=1 ;;
+    --headed) headed=1 ;;
+    --profile) explicit_profile=1; append_arg "$arg"; continue ;;
+    --cdp-url) explicit_profile=1; append_arg "$arg"; if [ "$#" -gt 0 ]; then append_arg "$1"; shift; fi; continue ;;
+    --profile=*|--cdp-url=*|--connect|--mcp|--use-cloud|--cloud-profile-id|--cloud-profile-id=*) explicit_profile=1 ;;
+    --session) append_arg "$arg"; if [ "$#" -gt 0 ]; then session=$1; append_arg "$1"; shift; fi; continue ;;
+    --session=*) session=${arg#--session=} ;;
+    -*) ;;
+    open|click|type|input|scroll|back|screenshot|state|switch|close-tab|keys|select|upload|eval|extract|hover|dblclick|rightclick|cookies|wait|get|python) starts_browser=1 ;;
+    close|sessions|install|init|setup|doctor|cloud|profile) ;;
+  esac
+  append_arg "$arg"
+done
+
+if [ "$show_help" -eq 1 ]; then
+  {
+    echo "Pibo browser-use wrapper:"
+    echo "  Default persistent Chrome profile via CDP."
+    echo "  Use $fresh_flag for a temporary browser profile."
+    echo ""
+  } >&2
+fi
+
+set --
+if [ -n "$sanitized_args" ]; then
+  while IFS= read -r item; do
+    set -- "$@" "$item"
+  done <<EOF
+$sanitized_args
+EOF
+fi
+
+session_pid_file=${BROWSER_USE_HOME:-$HOME/.browser-use}/$session.pid
+session_alive=0
+if [ -s "$session_pid_file" ]; then
+  session_pid=$(cat "$session_pid_file" 2>/dev/null || true)
+  if [ -n "$session_pid" ] && kill -0 "$session_pid" 2>/dev/null; then
+    session_alive=1
+  fi
+fi
+
+if [ "$starts_browser" -eq 1 ] && [ "$explicit_profile" -eq 0 ] && [ "$fresh_profile" -eq 0 ]; then
+  if [ "$session_alive" -eq 1 ] && persistent_chrome_url_if_alive >/dev/null; then
+    exec "$real_browser_use" "$@"
+  fi
+  if [ "$session_alive" -eq 1 ]; then
+    "$real_browser_use" --session "$session" close >/dev/null 2>&1 || true
+    sleep 0.2
+  fi
+  cdp_url=$(ensure_persistent_chrome)
+  exec "$real_browser_use" --cdp-url "$cdp_url" "$@"
+fi
+
+exec "$real_browser_use" "$@"
+WRAPPER
+
+chmod +x "$wrapper_path"