#!/usr/bin/env zsh

# === CONFIG ===
VIRTUAL_MONITOR="HEADLESS-2"
VIRTUAL_WORKSPACE=9
REAL_MONITOR="eDP-1"

# === CLEANUP FUNCTION ===
cleanup() {
    echo "\n[wayvnc] Cleaning up..."
    hyprctl dispatch moveworkspacetomonitor $VIRTUAL_WORKSPACE $REAL_MONITOR
    hyprctl dispatch focusmonitor "$REAL_MONITOR"
    pkill wayvnc
    echo "[wayvnc] Done."
    exit 0
}

# === Trap Exit for Cleanup ===
trap cleanup INT TERM EXIT

# === Check if HEADLESS-2 is already active, create it if not ===
if ! hyprctl monitors | grep -q "$VIRTUAL_MONITOR"; then
    echo "[wayvnc] Creating $VIRTUAL_MONITOR dynamically..."
    hyprctl output create headless
    sleep 0.5
fi

# === Assign workspace and activate it ===
echo "[wayvnc] Moving workspace $VIRTUAL_WORKSPACE to $VIRTUAL_MONITOR..."
hyprctl dispatch moveworkspacetomonitor $VIRTUAL_WORKSPACE $VIRTUAL_MONITOR
sleep 0.2
hyprctl dispatch workspace $VIRTUAL_WORKSPACE
sleep 0.2

# === Return focus to your real monitor so you don't get stuck on the headless one ===
hyprctl dispatch focusmonitor "$REAL_MONITOR"

# === Reloading Hyprland for avoid bugs ===
hyprctl reload

# === Start WayVNC ===
echo "[wayvnc] Starting WayVNC on $VIRTUAL_MONITOR..."
wayvnc 0.0.0.0 5900 -r "$VIRTUAL_MONITOR"