#!/bin/bash
# ~/.config/hypr/scripts/monitor_setup.sh

notify-send -t 3000 "Monitor Setup" "Detecting display configuration..."

LAPTOP="eDP-1"
LID_STATE=$(grep -q "open" /proc/acpi/button/lid/LID/state && echo "open" || echo "closed")
MONITORS=$(hyprctl monitors -j)

if [[ $(echo "$MONITORS" | jq 'length') -eq 0 ]]; then
    notify-send "Monitor Error" "No monitors detected."
    exit 1
fi

EXTERNAL=$(echo "$MONITORS" | jq -r '.[] | select(.name != "'$LAPTOP'") | .name' | head -1)

get_capped_res() {
    local monitor=$1
    local modes=$(echo "$MONITORS" | jq -r ".[] | select(.name == \"$monitor\") | .availableModes[]")

    if [[ -z "$modes" ]]; then
        echo "preferred"
        return
    fi

    local max_pixels=0
    local max_refresh=0
    local best_mode="preferred"

    while IFS= read -r mode; do
        local width=$(echo "$mode" | awk -F'@' '{print $1}' | awk -F'x' '{print $1}')
        local height=$(echo "$mode" | awk -F'@' '{print $1}' | awk -F'x' '{print $2}')
        local refresh=$(echo "$mode" | awk -F'@' '{print $2}' | sed 's/Hz//')

        if (( $(echo "$refresh > 120.1" | bc -l) )); then
            continue
        fi

        local pixels=$((width * height))

        if [[ $pixels -gt $max_pixels || ($pixels -eq $max_pixels && $(echo "$refresh > $max_refresh" | bc -l) -eq 1) ]]; then
            max_pixels=$pixels
            max_refresh=$refresh
            best_mode="${width}x${height}@${refresh}"
        fi
    done <<< "$modes"

    echo "$best_mode"
}

EXT_RES=$(get_capped_res "$EXTERNAL")
LAPTOP_RES=$(get_capped_res "$LAPTOP")

# Case 1: Laptop Only
if [ -z "$EXTERNAL" ]; then
    notify-send -t 3000 "Monitor Setup" "Laptop Mode: $LAPTOP_RES"
    hyprctl keyword monitor "$LAPTOP,$LAPTOP_RES,auto,1"
    
    for ws in {1..5}; do
        hyprctl dispatch workspace "$ws" # Ensure workspace exists
        hyprctl keyword "workspace $ws,monitor:$LAPTOP"
    done
    exit 0
fi

# Case 2: External + Open Lid
if [ "$LID_STATE" = "open" ]; then
    notify-send -t 3000 "Monitor Setup" "Dual Mode\nExternal: $EXT_RES (VRR On)"
    
    hyprctl keyword monitor "$LAPTOP,$LAPTOP_RES,0x0,1"
    hyprctl keyword monitor "$EXTERNAL,$EXT_RES,auto,1,vrr,1"

    sleep 1

    # Workspace 1 & 2 to Big Screen
    for ws in 1 2; do
        hyprctl dispatch workspace "$ws"
        hyprctl keyword "workspace $ws,monitor:$EXTERNAL"
        hyprctl dispatch moveworkspacetomonitor "$ws" "$EXTERNAL"
    done
    # Workspace 3 to Laptop
    hyprctl dispatch workspace 3
    hyprctl keyword "workspace 3,monitor:$LAPTOP"
    hyprctl dispatch moveworkspacetomonitor 3 "$LAPTOP"

# Case 3: External Only (Clamshell)
else
    notify-send -t 3000 "Monitor Setup" "Clamshell Mode\nExternal: $EXT_RES (VRR On)"
    
    hyprctl dispatch dpms off "$LAPTOP"
    hyprctl keyword monitor "$LAPTOP,disable"
    hyprctl keyword monitor "$EXTERNAL,$EXT_RES,0x0,1,vrr,1"

    sleep 1

    for ws in {1..5}; do
        hyprctl dispatch workspace "$ws"
        hyprctl keyword "workspace $ws,monitor:$EXTERNAL"
        hyprctl dispatch moveworkspacetomonitor "$ws" "$EXTERNAL"
    done
fi

hyprctl dispatch workspace 1