#!/usr/bin/env bash

VERBOSE=1

function log {
    if [ $VERBOSE -eq 1 ]; then
        echo "$@"
    fi
}
set -e

# Make sure a config file was provided
if [ $# -lt 1 ]; then
    echo "Usage: $0 <config file>"
    exit
fi

PROFILE=$1

if [[ ! -f "$PROFILE" ]]; then
    echo "Profile file not found: $PROFILE" >&2
    exit 1
fi

if ! command -v jq &>/dev/null; then
    echo "jq not installed!" >&2
    exit 1
fi

if ! command -v kscreen-doctor &>/dev/null; then
    echo "kscreen-doctor not found!" >&2
    exit 1
fi

# Extract outputs list
enabled_outputs=$(jq -c '.outputs[] | select(.enabled == true)' "$PROFILE")
disabled_outputs=$(jq -c '.outputs[] | select(.enabled == false)' "$PROFILE")
outputs=$(jq -c '.outputs[]' "$PROFILE")

# Restore enabled/disabled starting with the enabled monitors
# Starting with a disabled monitor might not work if it was
# previously the only enabled monitor

function enable_outputs {
    local outputs=$1

    # If empty or only whitespace, return early
    [[ -z "$outputs" ]] && return    

    while IFS= read -r out; do
        id=$(echo "$out" | jq -r '.id')
        log "[VAR] id:" $id
        name=$(echo "$out" | jq -r '.name')
        log "[VAR] name:" $name
        enabled=$(echo "$out" | jq -r '.enabled')
        log "[VAR] enabled:" $enabled

        # Enable/disable
        if [[ "$enabled" == "true" ]]; then
            CMD="kscreen-doctor output.$name.enable"
            log "[CMD]" $CMD
            $CMD
        else
            CMD="kscreen-doctor output.$name.disable"
            log "[CMD]" $CMD
            $CMD
        fi
    done <<< "$outputs"
}

log "Enabling enabled outputs..."
enable_outputs "$enabled_outputs"
log "Disabling disabled outputs..."
enable_outputs "$disabled_outputs"

function load_profile_to_outputs {
    local outputs=$1

    # If empty or only whitespace, return early
    [[ -z "$outputs" ]] && return

    while IFS= read -r out; do
        id=$(echo "$out" | jq -r '.id')
        name=$(echo "$out" | jq -r '.name')
        posx=$(echo "$out" | jq -r '.pos.x')
        posy=$(echo "$out" | jq -r '.pos.y')
        rotation=$(echo "$out" | jq -r '.rotation')
        scale=$(echo "$out" | jq -r '.scale')
        mode_id=$(echo "$out" | jq -r '.currentModeId')
        # mode_name=$(echo "$out" | jq -r ".modes[] | select(.id == \"$mode_id\") | .name")
        # log "[VAR] mode_name:" $mode_name
        refresh_rate=$(echo "$out" | jq -r ".modes[] | select(.id == \"$mode_id\") | .refreshRate")
        height=$(echo "$out" | jq -r ".modes[] | select(.id == \"$mode_id\") | .size.height")
        width=$(echo "$out" | jq -r ".modes[] | select(.id == \"$mode_id\") | .size.width")
        mode="${width}x${height}@${refresh_rate}"
        replication_source_id=$(echo "$out" | jq -r ".replicationSource")
        log "[VAR] replication_source_id: $replication_source_id"
        if [ $replication_source_id -neq 0 ]; then
            replication_source_name=$(echo "$outputs.[] | select(.id == \"$replication_source_id\" | .name)" )
            log "[VAR] replication_source_name: $replication_source_name"
            CMD="kscreen-doctor output.$"
        done
        log "[VAR] mode: $mode" 
        priority=$(echo "$out" | jq -r '.priority')

        # 

        # Mode (Resolution + refresh)
        # CMD="kscreen-doctor output.$name.mode.$mode"
        CMD="kscreen-doctor output.$name.mode.$mode"
        log "[CMD]" $CMD
        $CMD

        # Position
        CMD="kscreen-doctor output.$name.position.$posx,$posy"
        log "[CMD]" $CMD
        $CMD

        # Scale
        CMD="kscreen-doctor output.$name.scale.$scale"
        log "[CMD]" $CMD
        $CMD
        
        # Rotation (map from JSON names to kscreen-doctor options)
        CMD=""
        case "$rotation" in
            "1")    CMD="kscreen-doctor output.$name.rotation.normal" ;;
            "2")    CMD="kscreen-doctor output.$name.rotation.left" ;;
            "4")    CMD="kscreen-doctor output.$name.rotation.inverted" ;;
            "8")    CMD="kscreen-doctor output.$name.rotation.right" ;;
        esac
        log "[CMD]" $CMD
        $CMD

        # Primary / Not Primary
        if [ $priority -eq 1 ]; then
            CMD="kscreen-doctor output.$name.primary"
            log "[CMD]" $CMD
            $CMD
        fi

    done <<< "$outputs"
}

load_profile_to_outputs "$outputs"

clone_groups=$(jq -c '.clones[]?' "$PROFILE")

function restore_clone_groups {
    local clone_groups=$1
    while IFS= read -r clone; do
        primary=$(echo "$clone" | jq -r '.[0]')
        others=$(echo "$clone" | jq -r '.[]' | tail -n +2)

        for o in $others; do
            CMD="kscreen-doctor output.$o.clone.$primary"
            log "[CMD]" $CMD
            $CMD

        done
    done <<< "$clone_groups"
}

restore_clone_groups "$clone_groups"

echo "Display configuration restored."