#!/usr/bin/env bash
# args_json: {"type":"object","properties":{"output":{"type":"string","description":"Output file path (default: /tmp/browser_capture.png)"},"url_pattern":{"type":"string","description":"URL pattern to match browser tab (default: localhost)"}}}
# ollie:prompt
# ## browser_screencap
#
# Capture a screenshot of a browser window showing a specific URL.
#
# ```
# browser_screencap(output="/tmp/page.png", url_pattern="github.com")
# ```
# ollie:end
set -e

source "${OLLIE_TOOLS_PATH:-$(dirname "$0")}/_lib/args.sh"

OUTPUT=$(arg_get "output" "${TMPDIR:-/tmp}/browser_capture.png")
URL_PATTERN=$(arg_get "url_pattern" "localhost")

case "$(uname -s)" in
    Darwin)
        osascript -e 'tell application "Google Chrome" to activate' 2>/dev/null
        sleep 0.3
        osascript -e "tell application \"Google Chrome\" to set index of (first window whose name contains \"$URL_PATTERN\") to 1" 2>/dev/null
        sleep 0.3
        screencapture -x "$OUTPUT"
        ;;
    Linux)
        if [ -z "$DISPLAY" ]; then
            echo "ERROR: DISPLAY not set" >&2
            exit 1
        fi

        TAB_INFO=$(brotab list | grep -i "$URL_PATTERN" | head -1)
        if [ -z "$TAB_INFO" ]; then
            echo "ERROR: No tab matching '$URL_PATTERN'" >&2
            exit 1
        fi

        TAB_ID=$(echo "$TAB_INFO" | cut -f1)
        TAB_TITLE=$(echo "$TAB_INFO" | cut -f2)
        echo "Found tab: $TAB_ID ($TAB_TITLE)" >&2

        brotab activate "$TAB_ID"
        sleep 0.5

        WIN_ID=$(wmctrl -l | grep -i "$TAB_TITLE" | head -1 | awk '{print $1}')
        if [ -z "$WIN_ID" ]; then
            WIN_ID=$(wmctrl -l | grep -iE "firefox|chrome|chromium" | head -1 | awk '{print $1}')
        fi

        if [ -z "$WIN_ID" ]; then
            echo "ERROR: Could not find browser window" >&2
            exit 1
        fi

        echo "Capturing window: $WIN_ID" >&2
        wmctrl -i -a "$WIN_ID"
        sleep 0.3
        import -window "$WIN_ID" "$OUTPUT"
        ;;
    *)
        echo "ERROR: Unsupported OS: $(uname -s)" >&2
        exit 1
        ;;
esac

if [ -f "$OUTPUT" ]; then
    echo "$OUTPUT"
else
    echo "ERROR: Failed to capture" >&2
    exit 1
fi