#!/bin/sh -e

. ./common-script.sh

# Function to check bluetoothctl is installed
setup_xrandr() {
    echo "Install xrandr if not already installed..."
    if ! command_exists xrandr; then
        case ${PACKAGER} in
        pacman)
            sudo "${PACKAGER}" -S --noconfirm xorg-xrandr
            ;;
        apt-get)
            sudo "${PACKAGER}" install -y x11-xserver-utils
            ;;
        *)
            sudo "${PACKAGER}" install -y xorg-x11-server-utils
            ;;
        esac
    else
        echo "xrandr is already installed."
    fi
}

# Function to display colored text
colored_echo() {
    local color=$1
    local text=$2
    case $color in
    red) echo -e "\033[31m$text\033[0m" ;;
    green) echo -e "\033[32m$text\033[0m" ;;
    yellow) echo -e "\033[33m$text\033[0m" ;;
    blue) echo -e "\033[34m$text\033[0m" ;;
    *) echo "$text" ;;
    esac
}

# Function to check the display server type
check_display_server() {
    if [ "$XDG_SESSION_TYPE" = "wayland" ]; then
        colored_echo "red" "You are using Wayland."
        colored_echo "red" "This script is designed for X11. It may not work correctly on Wayland."

        if ! confirm_action "Do you want to continue anyway?"; then
            echo "Exiting script."
            exit 1
        fi

    elif [ "$XDG_SESSION_TYPE" = "x11" ]; then
        colored_echo "green" "You are using X11 (X-server)."
        return 0
    else
        colored_echo "red" "Unable to determine the display server type."
        exit 1
    fi
}

# Function to execute xrandr commands and handle errors
execute_command() {
    local command="$1"
    echo "Executing: $command"
    eval "$command" 2>&1 | tee /tmp/xrandr.log | tail -n 20
    if [ $? -ne 0 ]; then
        echo "An error occurred while executing the command. Check /tmp/xrandr.log for details."
    fi
}

# Function to detect connected monitors
detect_connected_monitors() {
    xrandr_output=$(xrandr)
    echo "$xrandr_output" | grep " connected" | awk '{print $1}'
}

# Function to get resolutions for a monitor
get_unique_resolutions() {
    local monitor="$1"
    xrandr_output=$(xrandr)
    echo "$xrandr_output" | grep -A 10 "$monitor connected" | grep -oP '\d+x\d+' | sort -u
}

# Function to prompt for confirmation
confirm_action() {
    local action="$1"
    echo "$action"
    read -p "Are you sure? (y/n): " confirm
    if [[ "$confirm" =~ ^[Yy]$ ]]; then
        return 0
    else
        return 1
    fi
}

checkEnv
setup_xrandr
check_display_server