local name="$1"

local keys="$2"

local delay="${3:-0.5}"

echo "📸 Capturing: $name"

# Focus the window

niri msg action focus-window --id "$WINDOW_ID"

sleep 0.2

# Inject keystrokes if provided

if [ -n "$keys" ]; then

echo " Typing: $keys"

wtype "$keys"

sleep "$delay"

fi

# Screenshot

niri msg action screenshot-window --path "$OUTPUT_DIR/${name}.png"

echo " ✅ Saved: $OUTPUT_DIR/${name}.png"

# Clear any existing input with Ctrl+U, then Escape to close popups

wtype -M ctrl -k u

sleep 0.1

wtype -k Escape

sleep 0.2