#!/bin/bash

# ♥©♥ hana, mod annelin

# disabled
# exit 0

user=ann
display=":0.0"
usbdevices=""
brightness="/sys/class/backlight/intel_backlight/brightness"
cpus="2 3 4 5 6 7"
wlan="wlp0s20f3"
ps_blacklist="systemd zsh ssh audacious pulseaudio gajim polkit- dbus- xfce4-power-manager xfce4-notifyd i3lock"


function whichusb(){
 for X in /sys/bus/usb/devices/*; do
    if [ "${1:0:4}" == "$(cat "$X/idVendor" 2>/dev/null)" -a "${1:5}" == "$(cat "$X/idProduct" 2>/dev/null)" ]
    then
        echo "$X"
    fi
 done
}

case $1 in

    sleep)

            # disable xfce sleep md #
            #sudo -u ${user} DISPLAY=${display} xfconf-query -c xfce4-power-manager -p /xfce4-power-manager/inactivity-on-battery -s 14

            # usb #
            for usbd in $usbdevices; do
              # echo "unbind $usbd `whichusb $usbd` "
              echo $(whichusb $usbd | sed 's/\/.*\///g') > /sys/bus/usb/drivers/usb/unbind
            done

            # wifi #
            iwconfig $wlan txpower 10

            # backlight #
            cat $brightness > /tmp/fakesleep.brightness
            echo 0 > $brightness
            sudo -u ${user} DISPLAY=${display} xset dpms force off

            # sleep user procs #
            for pid in $(ps -u ${user} -o pid,cmd | grep -vE "`echo $ps_blacklist|sed 's/ /|/g'`" | awk '{ print $1 }') ; do
                echo $pid >> /tmp/fakesleep.pid
                kill -STOP $pid
            done


            # sleep Xorg #
            # kill -STOP $(pidof X)

            # sleep cores #
            for i in $cpus ; do echo 0 > /sys/devices/system/cpu/cpu${i}/online; done

            # completed #
            # echo -n 7 on >> /proc/acpi/ibm/led
            echo "`date` stopping completed for `hostname`"  >> /tmp/fakesleep.log ;;

    resume)

            if [ ! -f /tmp/fakesleep.pid ] ; then exit ; fi

            # cpu #
            for i in $cpus ; do echo 1 > /sys/devices/system/cpu/cpu${i}/online; done

            # ps resume #
            # kill -CONT $(pidof X)
            for i in $(cat /tmp/fakesleep.pid) ; do
                 kill -CONT $i
            done

            # disable xfce sleep md #
            # sudo -u ${user} DISPLAY=${display} xfconf-query -c xfce4-power-manager -p /xfce4-power-manager/inactivity-on-battery -s 60

            for usbd in $usbdevices; do
                echo $(whichusb $usbd | sed 's/\/.*\///g') > /sys/bus/usb/drivers/usb/bind
            done

            # wifi #
            iwconfig $wlan txpower 24

            # disp #
            sleep 1s
            sudo -u ${user} DISPLAY=${display} xset dpms force on
            cat /tmp/fakesleep.brightness > $brightness


            # remove tmp data #
            rm /tmp/fakesleep.pid
            rm /tmp/fakesleep.brightness


            # completed #
            # echo -n 7 off >> /proc/acpi/ibm/led
            echo "`date` resume completed for `hostname`"  >> /tmp/fakesleep.log ;;
esac