#!/bin/bash
# Screenshot tool for Claude Code.
# Usage: bash tools/screenshot.sh <url> [filename]
# Screenshots are saved to Documentation/screenshots/
#
# The snap-based Chromium can only write to $HOME, so we
# capture there and move to the project directory.

URL="${1:?Usage: screenshot.sh <url> [filename]}"
FILENAME="${2:-screenshot_$(date +%Y%m%d_%H%M%S).png}"
[[ "$FILENAME" != *.png ]] && FILENAME="${FILENAME}.png"

OUTPUT_DIR="/var/www/html/searchmercials_v2/Documentation/screenshots"
TEMP_FILE="$HOME/__claude_screenshot_temp.png"

mkdir -p "$OUTPUT_DIR"

cd "$HOME" && chromium-browser \
    --headless \
    --disable-gpu \
    --no-sandbox \
    --disable-software-rasterizer \
    --screenshot="__claude_screenshot_temp.png" \
    --window-size=1280,900 \
    "$URL" 2>/dev/null

if [ -f "$TEMP_FILE" ]; then
    cp "$TEMP_FILE" "$OUTPUT_DIR/$FILENAME"
    rm "$TEMP_FILE"
    echo "$OUTPUT_DIR/$FILENAME"
else
    echo "ERROR: Screenshot failed" >&2
    exit 1
fi