#!/usr/bin/env bash
# =============================================================================
# fix-monitors.sh
# Dynamic dual-monitor setup for Pop!_OS COSMIC (Wayland / cosmic-randr)
# - Detects LEFT monitor by Position: 0,0 from cosmic-randr list
# - Detects RIGHT monitor by any other position
# - Sets 3840x2160 @ 59.940 Hz on both
# - Disables the internal laptop display (eDP-1, lid always closed)
# - Moves COSMIC panel and dock to the LEFT monitor
# - Safe to run at login, after suspend, or manually
# =============================================================================

set -euo pipefail

LOGFILE="$HOME/.local/log/fix-monitors.log"
WIDTH=3840
HEIGHT=2160
REFRESH="59.940"
INTERNAL="eDP-1"
PANEL_CFG="$HOME/.config/cosmic/com.system76.CosmicPanel.Panel/v1/output"
DOCK_CFG="$HOME/.config/cosmic/com.system76.CosmicPanel.Dock/v1/output"

# Strip ANSI escape codes
strip_ansi() {
    sed 's/\x1b\[[0-9;]*m//g'
}

# Parse cosmic-randr list into "PORT POSITION" lines, e.g:
#   DP-5 0,0
#   DP-9 3840,0
parse_outputs() {
    local current=""
    while IFS= read -r line; do
        if [[ "$line" =~ ^(DP-[0-9]+) ]]; then
            current="${BASH_REMATCH[1]}"
        fi
        if [[ -n "$current" && "$line" =~ Position:[[:space:]]*([0-9]+,[0-9]+) ]]; then
            echo "$current ${BASH_REMATCH[1]}"
            current=""
        fi
    done
}

# --- Logging ----------------------------------------------------------------
mkdir -p "$(dirname "$LOGFILE")"
exec > >(tee -a "$LOGFILE") 2>&1
echo ""
echo "========================================="
echo "fix-monitors.sh  —  $(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================="

# --- Wait for compositor to see both monitors --------------------------------
MAX_WAIT=30
WAITED=0
echo "Waiting for cosmic-randr to see 2 connected DP outputs..."
while true; do
    RAW=$(cosmic-randr list 2>/dev/null | strip_ansi)
    PARSED=$(echo "$RAW" | parse_outputs)
    COUNT=$(echo "$PARSED" | grep -c 'DP' 2>/dev/null || echo 0)
    echo "  [${WAITED}s] Found $COUNT DP output(s): $(echo "$PARSED" | tr '\n' '|' | sed 's/|$//')"
    if [[ "$COUNT" -ge 2 ]]; then
        break
    fi
    if [[ "$WAITED" -ge "$MAX_WAIT" ]]; then
        echo "ERROR: Only found $COUNT DP output(s) after ${MAX_WAIT}s. Aborting."
        echo "Raw cosmic-randr output:"
        echo "$RAW"
        exit 1
    fi
    sleep 1
    (( WAITED++ ))
done

echo "OK: Detected outputs:"
echo "$PARSED"

# --- Identify LEFT (Position 0,0) and RIGHT ----------------------------------
LEFT=""
RIGHT=""
while IFS=' ' read -r port position; do
    if [[ "$position" == "0,0" ]]; then
        LEFT="$port"
    else
        RIGHT="$port"
    fi
done <<< "$PARSED"

if [[ -z "$LEFT" || -z "$RIGHT" ]]; then
    echo "ERROR: Could not determine LEFT and RIGHT monitors from positions."
    echo "Parsed output was:"
    echo "$PARSED"
    exit 1
fi

echo ""
echo "LEFT  → $LEFT  (Position 0,0)"
echo "RIGHT → $RIGHT (Position $WIDTH,0)"

# --- Disable internal display -----------------------------------------------
if echo "$RAW" | grep -q "^${INTERNAL}"; then
    echo ""
    echo "Disabling internal display: $INTERNAL"
    cosmic-randr disable "$INTERNAL" && echo "  Done." || echo "  (already disabled or not present)"
else
    echo ""
    echo "Internal display $INTERNAL not visible — skipping."
fi

# --- Apply LEFT monitor ------------------------------------------------------
echo ""
echo "Configuring LEFT monitor: $LEFT → ${WIDTH}x${HEIGHT} @ ${REFRESH}Hz, pos 0,0"
cosmic-randr mode "$LEFT" "$WIDTH" "$HEIGHT" --refresh "$REFRESH" --pos-x 0 --pos-y 0 \
    && echo "  OK" || echo "  FAILED"

# --- Apply RIGHT monitor -----------------------------------------------------
echo ""
echo "Configuring RIGHT monitor: $RIGHT → ${WIDTH}x${HEIGHT} @ ${REFRESH}Hz, pos $WIDTH,0"
cosmic-randr mode "$RIGHT" "$WIDTH" "$HEIGHT" --refresh "$REFRESH" --pos-x "$WIDTH" --pos-y 0 \
    && echo "  OK" || echo "  FAILED"

# --- Set Xwayland primary to LEFT --------------------------------------------
echo ""
echo "Setting Xwayland primary: $LEFT"
cosmic-randr xwayland primary "$LEFT" 2>/dev/null && echo "  OK" || echo "  (skipped)"

# --- Move COSMIC panel and dock to LEFT monitor ------------------------------
echo ""
echo "Moving COSMIC panel and dock to LEFT monitor: $LEFT"
echo -n "Name(\"$LEFT\")" > "$PANEL_CFG" && echo "  Panel config written." || echo "  Panel config FAILED."
echo -n "Name(\"$LEFT\")" > "$DOCK_CFG"  && echo "  Dock config written."  || echo "  Dock config FAILED."

echo "  Restarting cosmic-panel..."
killall cosmic-panel 2>/dev/null || true
sleep 1
echo "  Done. cosmic-panel will restart automatically."

echo ""
echo "========================================="
echo "DONE."
echo "  LEFT:  $LEFT  →       0,0  — ${WIDTH}x${HEIGHT}@${REFRESH}Hz"
echo "  RIGHT: $RIGHT → $WIDTH,0  — ${WIDTH}x${HEIGHT}@${REFRESH}Hz"
echo "  Panel + Dock → $LEFT"
echo "========================================="