#!/bin/bash

key="$1"
if [ -z $key ]
then
    echo "Displays:"
    displays=$(xrandr -q | grep -w "connected" | awk '{print $1}')
    a=0; for m in $displays; do a=$((a+1)); echo "  ${a}) ${m}"; done
    echo
    echo "Choose Display Setup:
        1) Desktop Mode (3 screens)
        2) Desktop Mode (No laptop)
        3) Bed Mode
        "
    read var
else
    var=$key
fi

if [ $var -eq 1 ]
then
    #desktop mode
    echo "Switching to Desktop Mode"; sleep 2;
    xrandr --output DVI-I-1-1 --auto --primary
    xrandr --output eDP-1 --auto --left-of DVI-I-1-1
    xrandr --output DVI-I-2-2 --auto --left-of eDP-1
    #
    xrandr --output HDMI-1 --off
elif [ $var -eq 2 ]
then
    #desktop mode (no laptop)
    echo "Switching to Desktop Mode (no laptop)"; sleep 2;
    xrandr --output DVI-I-1-1 --auto --primary
    xrandr --output DVI-I-2-2 --auto --left-of DVI-I-1-1
    #
    #xrandr --output eDP-1 --off
    xrandr --output HDMI-1 --off
elif [ $var -eq 3 ]
then
    #bed
    echo "Switching to Bed Mode"; sleep 2;
    xrandr --output HDMI-1 --auto --primary --transform 1.06,0,-45,0,1.06,-30,0,0,1
    #
    xrandr --output DVI-I-1-1 --off
    xrandr --output DVI-I-2-2 --off
    xrandr --output eDP-1 --off
else
    echo "Invalid option"; sleep 5;
fi

sleep 3
exit