#!/usr/bin/env bash

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")
echo "enabled:"
echo $enabled_outputs
disabled_outputs=$(jq -c '.outputs[] | select(.enabled == false)' "$PROFILE")
echo "disabled:"
echo $disabled_outputs
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
    while IFS= read -r out; do
        id=$(echo "$out" | jq -r '.id')
        name=$(echo "$out" | jq -r '.name')
        enabled=$(echo "$out" | jq -r '.enabled')

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

enable_outputs $enabled_outputs
enable_outputs $disabled_outputs

function load_profile_to_outputs {
    local outputs=$1
    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=$(echo "$out" | jq -r '.currentModeId')
        priority=$(echo "$out" | jq -r '.priority')

        # Mode (Resolution + refresh)
        kscreen-doctor "output.$id.mode.$mode"

        # Position
        kscreen-doctor "output.$id.position.$posx,$posy"

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

        # Primary / Not Primary
        echo $priority
        if [ $priority -eq 1 ]; then
            kscreen-doctor "output.$id.primary"
        fi

    done <<< "$outputs"
}

load_profile_to_outputs $outputs

#########################
# 2. Restore clone groups
#########################

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
            kscreen-doctor "output.$o.clone.$primary"
        done
    done <<< "$clone_groups"
}

restore_clone_groups $clone_groups

echo "Display configuration restored."