#!/usr/bin/env bash

set -euo pipefail

if [[ $EUID -ne 0 ]]; then
   echo "[RaveOS-GPU] Error: Root permissions required." >&2
   exit 1
fi

log() {
  echo "[RaveOS-GPU] $1"
}

optimize_amd() {
  log "Optimizing AMD GPU acceleration without P-state locking..."
  for card in /sys/class/drm/card*/device; do
    if [ -f "$card/power_dpm_force_performance_level" ]; then
      echo "auto" > "$card/power_dpm_force_performance_level" 2>/dev/null || true
    fi
  done
}

optimize_nvidia() {
  if command -v nvidia-smi &>/dev/null; then
    log "Optimizing NVIDIA GPU (Adaptive PowerMizer mode)..."
    nvidia-smi --auto-boost-default=ENABLED 2>/dev/null || true
    
    if command -v nvidia-settings &>/dev/null && [ -n "${DISPLAY:-}" ]; then
      nvidia-settings -a "[gpu:0]/GpuPowerMizerMode=0" 2>/dev/null || true
    fi
  fi
}

main() {
  log "Starting safe GPU optimization..."
  
  if lspci | grep -iE "vga|3d|display" | grep -i "amd" &>/dev/null; then
    optimize_amd
  fi
  
  if lspci | grep -iE "vga|3d|display" | grep -i "nvidia" &>/dev/null; then
    optimize_nvidia
  fi
  
  log "GPU Optimization complete. Dynamic P-states and ASPM preserved."
}

main "$@"