#!/bin/bash
# throttlectl, a user-discrection power management utility

[[ -z "$@" ]] && throttle_help

throttle_help() {
  [[ ! "${1}" == "help" ]] && \
    printf "Invalid input: ${@}\n"
  echo -e "\nSystem-wide power management:\n"\
          "${0} {{cut,full}} - system-wide runtime powersaving\n"\
          "${0} {check} - inspect runtime powersaving\n"\
          "\nPer-component power management:\n" \
          "${0} sys {cut,full} - System devices\n"\
          "${0} turbo {on,off} - Intel P-State Turbo\n"\
          "${0} ht {on,off} - Hyperthreads\n"\
          "${0} gov {cut,full} - CPU Governor\n"\
          "${0} gpu {cut,full} - GPU runtime powersaving (only Nvidia ATM)\n"\
          "${0} backlight {0,1,2} - Backlight brightness (laptops only?)\n" \
          "\nAll options are stackable and take {{cut,powersave,start},{full,performace,stop},{check,status}}\n"
  exit
}

config_help() {
  echo -e "\nConfiguration files not found:\n"\
          "Performance profile: /etc/throttle.d/performance\n"\
          "Powersave profile: /etc/throttle.d/powersave\n"\
          "See example profiles provided with the throttlectl pkgbuild:\n"\
          "https://github.com/quequotion/pantheon-qq/tree/master/EXTRAS/throttlectl"
  exit
}

throttle_ht() {
  [[ -f /sys/devices/system/cpu/smt/control ]] && \
    printf "${HYPERTHREADS}" > /sys/devices/system/cpu/smt/control &
}

check_ht() {
  [[ -f /sys/devices/system/cpu/smt/control ]] && \
    echo -e "\nHyperthreads\n/sys/devices/system/cpu/smt/\n"\
            "control\t$(cat /sys/devices/system/cpu/smt/control)"
}

throttle_gov() {
  for i in /sys/bus/cpu/drivers/processor/cpu*/cpufreq/scaling_governor; do \
    [[ "$(printf ${i} | tr -cd [:digit:])" -ge "${CORES}" && \
       "$(cat /sys/devices/system/cpu/cpu$(printf ${i} | tr -cd [:digit:])/online)" == 0 ]] && \
      continue ; printf "${GOVERNOR}" > "${i}" & done &
}

check_gov() {
  printf "\nCPU Governor\n/sys/bus/cpu/drivers/processor/cpu*/cpufreq/scaling_governor\n"
  for i in /sys/bus/cpu/drivers/processor/cpu*/cpufreq/scaling_governor; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/bus/cpu/drivers/processor/||g; s|/.*||g; N;s|\n|\t|'
}

throttle_turbo() {
  [[ -f /sys/devices/system/cpu/intel_pstate/no_turbo ]] && \
    printf "${XPSTURBO}" > /sys/devices/system/cpu/intel_pstate/no_turbo &
}

check_turbo() {
  [[ -f /sys/devices/system/cpu/intel_pstate/no_turbo ]] && \
    echo -e "\nDisable Intel P-State Turbo\n/sys/devices/system/cpu/intel_pstate/\n"\
            "no_turbo\t$(cat /sys/devices/system/cpu/intel_pstate/no_turbo)"
}

throttle_backlight() {
  [[ -f /sys/class/backlight/acpi_video0 ]] && \
    for i in /sys/class/backlight/acpi_video*/brightness; do \
      printf "${BACKLIGHT}" > "${i}" & done &
}

check_backlight() {
  [[ -f /sys/class/backlight/acpi_video0 ]] && \
    printf "\nBacklight\n/sys/class/backlight/\n" && \
      for i in /sys/class/backlight/acpi_video*/brightness; do \
        printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/class/backlight/||g; s|/.*||g; N;s|\n|\t|'
}

throttle_nvpm() {
  [[ -z "$(command -v nvidia-settings)" ]] && \
    printf "nvidia-settings not found; only nvidia currently supported" || \
    DISPLAY=":0.0" nvidia-settings -a [gpu:0]/GPUPowerMizerMode="${NVPM}" > /dev/null &
}

check_nvpm() {
  [[ -n "$(command -v nvidia-settings)" ]] && \
    printf "\nNvidia PowerMizer\n " && \
      DISPLAY=":0.0" nvidia-settings -q [gpu:0]/GPUPowerMizerMode | grep "Attribute" | sed 's|.*\[||g;s|\]):||g;s| |\t|g;s|\.$||g'
}

throttle_system() {
  # ENERGY_PERF_BIAS
  [[ -n "$(command -v x86_energy_perf_policy)" ]] && \
    x86_energy_perf_policy "${ENERGYPERF}" &

  # Virtual Memory (Swap)
  [[ "$(sysctl -n vm.laptop_mode)" == "${LAPTOP}" ]] || \
    printf "${LAPTOP}" > /proc/sys/vm/laptop_mode &
  [[ "$(sysctl -n vm.dirty_writeback_centisecs)" == "${DIRTYWBC}" ]] || \
    printf "${DIRTYWBC}" > /proc/sys/vm/dirty_writeback_centisecs &
  [[ "$(sysctl -n vm.dirty_expire_centisecs)" == "${DIRTYWBC}" ]] || \
    printf "${DIRTYWBC}" > /proc/sys/vm/dirty_expire_centisecs &

  # NMI watchdog
  [[ -f /proc/sys/kernel/watchdog ]] && \
      printf "${NMIDOG}" > /proc/sys/kernel/watchdog &

  # SATA link power management
  for i in /sys/class/scsi_host/host*/link_power_management_policy; do [[ -f "${i}" ]] && \
    printf "${POLICY}" > "${i}" & done &

  # Hard drives
  [[ -n "$(command -v hdparm)" ]] && \
    hdparm -qB "${APM}" -qM "${AAM}" /dev/[hs]d[a-z] 2> /dev/null &

  # Block device autosuspend
  for i in /sys/block/*/device/power/autosuspend_delay_ms; do \
    printf -- "${DISKAUTOSUSPEND}" > "${i}" & done &

  # Runtime power management for devices (with exceptions for devices by VID:PID)
  for i in /sys/{class/*,bus/*/devices}/*/power/control; do \
    [[ "${CONTROL}" == "auto" ]] && \
      [[ -n "$(printf ${i} | grep msr[${CORES}-$((${CORES} * 2 - 1))])" || \
         -n "$(printf ${i} | grep machinecheck[${CORES}-$((${CORES} * 2 - 1))])" || \
         -n "$(ls $(printf ${i} | sed 's|power.*||g')/*/ | grep 046D\:C52B)" ]] && \
        continue ; printf "${CONTROL}" > "${i}" & done &

  # USB Autosuspend (with exceptions for devices by VID:PID)
  for i in /sys/bus/usb/devices/*/power/autosuspend{,_delay_ms}; do \
    [[ "${AUTOSUSPEND}" == "1" ]] && \
      [[ -n "$(ls $(printf ${i} | sed 's|power.*||g')/*/ | grep 046D\:C52B)" ]] && \
        continue ; printf -- "${AUTOSUSPEND}" > "${i}" & done &

  # Powersaving for modules
  for i in /sys/module/*/parameters/power_save; do \
    printf "${POWERSAVE}" > "${i}" & done &
  for i in /sys/module/*/parameters/power_save_controller; do \
    printf "${CONTROLLER}" > "${i}" & done &

  # Network powersaving
  for i in /sys/class/net/wl*; do \
    [[ -n "$(command -v iw)" ]] && \
      iw dev "$(printf ${i} | sed 's/^.*wl/wl/')" set power_save "${WLPOWERSAVE}" 2> /dev/null & done &
  for i in /sys/class/net/e*; do \
    [[ -n "$(command -v ethtool)" ]] && \
      ethtool -s "$(printf ${i} | sed 's/^.*e/e/')" wol "${WOLA}" 2> /dev/null & done &
  for i in /sys/class/net/*/device/power/wakeup; do \
    printf "${WOLB}" > "${i}" & done &

  # LEDs
  for i in /sys/class/leds/*/brightness; do \
    printf "${LEDBRIGHT}" > "${i}" & done &

  # Save state for frontend
  printf "$SYSTEM" > /etc/throttle.d/systemstate &
}

check_system() {
  [[ -n "$(command -v x86_energy_perf_policy)" ]] && \
    printf "\nEnergy Perf Bias\n" && \
    x86_energy_perf_policy -r #SUID

  echo -e "\nVirtual Memory (Swap)\n/proc/sys/vm/\n"\
          "laptop_mode\t\t\t$(cat /proc/sys/vm/laptop_mode)\n"\
          "dirty_writeback_centisecs\t$(cat /proc/sys/vm/dirty_writeback_centisecs)\n"\
          "dirty_expire_centisecs\t\t$(cat /proc/sys/vm/dirty_expire_centisecs)"

  [[ -f /proc/sys/kernel/watchdog ]] && \
    echo -e "\nNMI watchdog\n"\
            "/proc/sys/kernel/watchdog\t$(cat /proc/sys/kernel/watchdog)"

  printf "\nSATA link power management\n/sys/class/scsi_host/host*/link_power_management_policy\n"
  for i in /sys/class/scsi_host/host*/link_power_management_policy; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/class/scsi_host/||g; s|/.*||g; N;s|\n|\t|'

  [[ -n "$(command -v hdparm)" ]] && \
    printf "\nHard drives\n/dev/[hs]d[a-z]\n" && \
    hdparm -B -M /dev/[hs]d[a-z] 2> /dev/null | sed 'N;s|\n||g; s|/dev/||g; s|\t| |g;' #SUID

  printf "\nBlock device autosuspend\n/sys/block/*/device/power/autosuspend_delay_ms\n"
  for i in /sys/block/*/device/power/autosuspend_delay_ms; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/block/||g; s|/device/.*||g; s|/.*||g; N;s|\n|\t|'

  printf "\nRuntime power management\n/sys/{class/*,bus/*/devices}/*/power/control\n"
  for i in /sys/{class/*,bus/*/devices}/*/power/control; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/class/||g; s|/sys/bus/||g; s|/devices||g; s|/power/control||g; N;s|\n|\t|'

  printf "\nUSB Autosuspend\n/sys/bus/usb/devices/*/power/autosuspend{,_delay_ms}\n"
  for i in /sys/bus/usb/devices/*/power/autosuspend{,_delay_ms}; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/bus/usb/devices/||g; s|/.*||g; N;s|\n|\t|'

  printf "\nPower saving for modules\n/sys/module/*/parameters/power_save{,_controller}\n"
  for i in /sys/module/*/parameters/power_save{,_controller}; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/module/||g; s|/parameters/.*||g; s|/.*||g; N;s|\n|\t|'

  printf "\nNetwork device powersaving\n";
  [[ -n "$(command -v iw)" ]] && \
    for i in /sys/class/net/wl*; do \
      printf " ${i}\n$(iw dev $(printf ${i} | sed 's/^.*wl/wl/') get power_save)\n"; done | sed 's|^\t|  |g;s|/sys/class/net/||g; s|/.*||g'

  printf "\nWake on {W,}LAN\n/sys/class/net/{wl*,e*,*/device/power/wakeup}\n";
  [[ -n "$(command -v ethtool)" ]] && \
    for i in /sys/class/net/e*; do \
      printf " ${i}\n$(ethtool $(printf ${i} | sed 's/^.*e/e/') | grep Wake-on)\n"; done | sed 's|^\t|  |g;s|/sys/class/net/||g; s|/.*||g'
  for i in /sys/class/net/*/device/power/wakeup; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/class/net/||g; s|/device/power/wakeup||g; s|/.*||g; N;s|\n|\t|'

  printf "\nLEDs\n/sys/class/leds/*/brightness\n"
  for i in /sys/class/leds/*/brightness; do \
    printf " ${i}\n$(cat ${i})\n"; done | sed 's|/sys/class/leds/||g; s|/brightness||g; s|/.*||g; N;s|\n|\t|'
}

while (( "$#" )); do
  case "$1" in
    sys)
      case "${2}" in
        cut|off|powersave|start)
          [[ -f /etc/throttle.d/powersave ]] && . /etc/throttle.d/powersave || config_help; SYSTEM="0"
        ;;
        full|on|performance|stop)
          [[ -f /etc/throttle.d/performance ]] && . /etc/throttle.d/performance || config_help; SYSTEM="1"
        ;;
        check|status)
          check_system
        ;;
      esac
      [[ -n "$SYSTEM" ]] && throttle_system
      shift "2"
      unset ENERGYPERF LAPTOP DIRTYWBC NMIDOG POLICY APM AAM CONTROL AUTOSUSPEND POWERSAVE CONTROLLER \
            WLPOWERSAVE WOLA WOLB LEDBRIGHT SYSTEM
    ;;
    turbo)
      [[ -z "${XPSTURBO}" ]] && \
        case "${2}" in \
          cut|off|powersave|start) XPSTURBO="1" ;; \
          full|on|performance|stop) XPSTURBO="0" ;; \
          check|status) check_turbo ;; esac
      [[ -n "${XPSTURBO}" ]] && throttle_turbo
      shift "2"
      unset XPSTURBO
    ;;
    ht)
      [[ -z "${HYPERTHREADS}" ]] && \
        case "${2}" in \
          cut|off|powersave|start) HYPERTHREADS="off" ;; \
          full|on|performance|stop) HYPERTHREADS="on" ;; \
          check|status) check_ht ;; esac
      [[ -n "${HYPERTHREADS}" ]] && throttle_ht
      shift "2"
      unset HYPERTHREADS
    ;;
    gov|cpu)
      [[ -z "${GOVERNOR}" ]] && \
        case "${2}" in \
          cut|off|powersave|start) GOVERNOR="powersave" ;; \
          full|on|performance|stop) GOVERNOR="performance" ;; \
          check|status) check_gov ;; esac
      [[ -n "${GOVERNOR}" ]] && throttle_gov
      shift "2"
      unset GOVERNOR
    ;;
    gpu)
      [[ -z "${NVPM}" ]] && \
        case "${2}" in \
          cut|off|powersave|start) NVPM="0" ;; \
          full|on|performance|stop) NVPM="1" ;; \
          check|status) check_nvpm ;; esac
      [[ -n "${NVPM}" ]] && throttle_nvpm
      shift "2"
      unset NVPM
    ;;
    backlight)
      [[ -z "${BACKLIGHT}" ]] && \
        case "${2}" in \
          cut|off|0|powersave|start) BACKLIGHT="0" ;; \
          full|on|1|performance|stop) BACKLIGHT="1" ;; \
          2) BACKLIGHT="2" ;; \
          check|status) check_backlight ;; esac
      [[ -n "${BACKLIGHT}" ]] && throttle_backlight
      shift "2"
      unset BACKLIGHT
    ;;
    full|on|performance|start|cut|off|powersave|stop)
      case "${1}" in
        cut|off|powersave|start)
          [[ -f /etc/throttle.d/powersave ]] && . /etc/throttle.d/powersave || config_help; SYSTEM="1"
        ;;
        full|on|performance|stop)
          [[ -f /etc/throttle.d/performance ]] && . /etc/throttle.d/performance || config_help; SYSTEM="0"
        ;;
      esac
      throttle_ht
      throttle_gov
      throttle_turbo
      throttle_backlight
      throttle_nvpm
      throttle_system
      shift
      unset GOVERNOR XPSTURBO HYPERTHREADS ENERGYPERF BACKLIGHT LAPTOP DIRTYWBC NMIDOG \
            POLICY APM AAM CONTROL AUTOSUSPEND POWERSAVE CONTROLLER \
            WLPOWERSAVE WOLA WOLB LEDBRIGHT NVPM SYSTEM
    ;;
    check|status)
      check_ht
      check_gov
      check_turbo
      check_backlight
      check_nvpm
      check_system
      shift
    ;;
    help|*)
      throttle_help
    ;;
  esac

  [[ -z "${1}" ]] || wait # Hey, let's stop! :) *stops* ... YAY! =D

done