gpu-mode-switcher.sh

#!/usr/bin/env bash

# Usage:
#   chmod +x gpu-mode-switcher.sh
#   ./gpu-mode-switcher.sh

set -u

RUNTIME_STATUS_PATH="/sys/bus/pci/devices/0000:01:00.0/power/runtime_status"
NVIDIA_POWER_PATH="/proc/driver/nvidia/gpus/0000:01:00.0/power"

has_cmd() {
  command -v "$1" >/dev/null 2>&1
}

run_with_sudo() {
  if [[ "${EUID}" -eq 0 ]]; then
    "$@"
  else
    sudo "$@"
  fi
}

confirm() {
  local prompt="$1"
  local answer
  read -r -p "${prompt} [y/N]: " answer
  [[ "${answer}" =~ ^[Yy]$ ]]
}

require_cmd() {
  local cmd="$1"
  if ! has_cmd "${cmd}"; then
    echo "Missing command: ${cmd}"
    return 1
  fi
}

post_switch_notice() {
  echo
  echo "Mode change applied. A reboot is required."
  echo
}

switch_to_hybrid() {
  require_cmd envycontrol || return
  if run_with_sudo envycontrol -s hybrid --rtd3 3; then
    post_switch_notice
  fi
}

switch_to_integrated() {
  require_cmd envycontrol || return
  if run_with_sudo envycontrol -s integrated; then
    post_switch_notice
  fi
}

switch_to_nvidia() {
  require_cmd envycontrol || return
  if run_with_sudo envycontrol -s nvidia; then
    post_switch_notice
  fi
}

run_initramfs_update() {
  require_cmd update-initramfs || return
  run_with_sudo update-initramfs -u
}

reboot_now() {
  if confirm "Reboot now?"; then
    run_with_sudo reboot
  fi
}

show_verification_checks() {
  echo
  echo "Checking dGPU runtime status:"
  if [[ -r "${RUNTIME_STATUS_PATH}" ]]; then
    cat "${RUNTIME_STATUS_PATH}"
  else
    echo "Cannot read ${RUNTIME_STATUS_PATH}"
  fi

  echo
  echo "Checking NVIDIA RTD3 info:"
  if [[ -r "${NVIDIA_POWER_PATH}" ]]; then
    cat "${NVIDIA_POWER_PATH}"
  else
    echo "Cannot read ${NVIDIA_POWER_PATH}"
  fi

  echo
  echo "Checking TLP battery thresholds:"
  if has_cmd tlp-stat; then
    run_with_sudo tlp-stat -b
  else
    echo "Missing command: tlp-stat"
  fi
  echo
}

show_renderer_checks() {
  echo
  echo "prime-run renderer check:"
  if has_cmd prime-run && has_cmd glxinfo; then
    prime-run glxinfo | grep "OpenGL renderer" || echo "Renderer string not found"
  else
    echo "Missing command(s): prime-run and/or glxinfo"
  fi

  echo
  echo "Default renderer check:"
  if has_cmd glxinfo; then
    glxinfo | grep "OpenGL renderer" || echo "Renderer string not found"
  else
    echo "Missing command: glxinfo"
  fi
  echo
}

optional_actions_menu() {
  local choice
  while true; do
    cat <<'EOF'
Optional actions:
1) update-initramfs -u
2) Reboot now
3) Run post-reboot verification checks
4) Run renderer checks (prime-run / default)
5) Back
EOF
    read -r -p "Select option: " choice
    case "${choice}" in
      1) run_initramfs_update ;;
      2) reboot_now ;;
      3) show_verification_checks ;;
      4) show_renderer_checks ;;
      5) return ;;
      *) echo "Invalid option." ;;
    esac
  done
}

startup_checks() {
  echo "Quick dependency check:"
  has_cmd envycontrol || echo "- Missing: envycontrol"
  has_cmd update-initramfs || echo "- Missing: update-initramfs"
  has_cmd tlp-stat || echo "- Missing (optional): tlp-stat"
  has_cmd glxinfo || echo "- Missing (optional): glxinfo"
  has_cmd prime-run || echo "- Missing (optional): prime-run"
  echo
}

main_menu() {
  local choice
  while true; do
    cat <<'EOF'
GPU mode switcher (envycontrol)
1) Hybrid (envycontrol -s hybrid --rtd3 3)
2) Integrated (envycontrol -s integrated)
3) NVIDIA (envycontrol -s nvidia)
4) Optional actions
5) Exit
EOF
    read -r -p "Select option: " choice
    case "${choice}" in
      1) switch_to_hybrid ;;
      2) switch_to_integrated ;;
      3) switch_to_nvidia ;;
      4) optional_actions_menu ;;
      5) exit 0 ;;
      *) echo "Invalid option." ;;
    esac
  done
}

startup_checks
main_menu