Load display

load_display.sh

# http://stackoverflow.com/a/1162345/971529
load_display() {
    if [ -z "$1" ]; then
        PID=$(pgrep -n -u $USER 'gnome-session|xfdesktop|xterm|kodi|lxsession|mate-session')
    else
        PID=$(pgrep -n -u $USER -f "$1")
    fi
    FALLBACK=$(pgrep -n 'unity-greeter')
    if [ -n "$PID" ]; then
        export DISPLAY=$(cat /proc/$PID/environ | strings | awk 'BEGIN{FS="=";}  $1=="DISPLAY" {print $2; exit}')
        echo "DISPLAY set to $DISPLAY"
    elif [ -n "$FALLBACK" ]; then
        export DISPLAY=$(sudo cat /proc/$FALLBACK/environ | strings | awk 'BEGIN{FS="=";}  $1=="DISPLAY" {print $2; exit}')
        echo "Using Login Screen. DISPLAY set to $DISPLAY"
    else
        echo "Could not set DISPLAY"
    fi
    unset PID
}
# needs to be a function since it loads it into the current environment