[Unit]
Description=Fan Control Service
After=multi-user.target

[Service]
User=root
ExecStart=/srv/fans.sh -t 65 -m 20
Restart=always

[Install]
WantedBy=multi-user.target

#!/bin/bash

# Функция для получения температуры GPU
get_gpu_temp() {
  gpu_temps=$(rocm-smi | grep -Eo "[0-9]+\.[0-9]+°C" | tr '\n' ' ')
  gpu0_temp=$(echo "$gpu_temps" | awk '{print $1}' | sed 's/°C//g')
  gpu1_temp=$(echo "$gpu_temps" | awk '{print $2}' | sed 's/°C//g')
  gpu2_temp=$(echo "$gpu_temps" | awk '{print $3}' | sed 's/°C//g')
  gpu3_temp=$(echo "$gpu_temps" | awk '{print $4}' | sed 's/°C//g')

  # Проверка, что получены числовые значения
  if [[ ! "$gpu0_temp" =~ ^[0-9]+\.[0-9]+$ ]]; then
    gpu0_temp=0
  fi
  if [[ ! "$gpu1_temp" =~ ^[0-9]+\.[0-9]+$ ]]; then
    gpu1_temp=0
  fi
  if [[ ! "$gpu2_temp" =~ ^[0-9]+\.[0-9]+$ ]]; then
    gpu2_temp=0
  fi
  if [[ ! "$gpu3_temp" =~ ^[0-9]+\.[0-9]+$ ]]; then
    gpu3_temp=0
  fi

  echo "$gpu0_temp $gpu1_temp $gpu2_temp $gpu3_temp"
}

# Функция для получения температуры процессора
get_cpu_temp() {
  local id=$1
  if [[ "$id" == "0" ]]; then
    temp=$(sensors | grep 'Package id 0' | awk '{print $4}' | tr -d '+°C')
  else
    temp=$(sensors | grep 'Package id 1' | awk '{print $4}' | tr -d '+°C')
  fi
  echo "$temp"
}

# Функция для установки скорости вентиляторов через ipmi
set_fan_speed() {
  local speed=$1
  local hex_speed=$(printf "%x" "$speed") # Перевод в шестнадцатеричный формат
  ipmitool raw 0x2e 0x30 00 00 "0x$hex_speed"
}

# Обработка аргументов командной строки
max_temp=60
min_speed=30
while getopts "t:m:" opt; do
  case $opt in
    t) max_temp=$OPTARG ;;
    m) min_speed=$OPTARG ;;
    \?) echo "Invalid option: -$OPTARG" >&2 ;;
  esac
done

# Проверка корректности аргументов
if ! [[ "$max_temp" =~ ^[0-9]+$ ]] || (( max_temp < 30 )) || (( max_temp > 80 )); then
  echo "Error: Invalid max temperature. Please enter a number between 30 and 80."
  exit 1
fi

if ! [[ "$min_speed" =~ ^[0-9]+$ ]] || (( min_speed < 20 )) || (( min_speed > 100 )); then
  echo "Error: Invalid min speed. Please enter a number between 20 and 100."
  exit 1
fi

# Создание таблицы соответствия
declare -A fan_speed_table
temp=30
while (( temp <= max_temp )); do
  fan_speed=$(( min_speed + ( ( temp - 30 ) / 5 ) * 5 ))
  fan_speed_table[$temp]=$fan_speed
  (( temp += 5 ))
done

# Цикл для постоянного мониторинга температуры
previous_fan_speed=0
while true; do
  # Получить температуры GPU и CPU
  gpu_temps=$(get_gpu_temp)
  gpu0_temp=$(echo "$gpu_temps" | awk '{print $1}')
  gpu1_temp=$(echo "$gpu_temps" | awk '{print $2}')
  gpu2_temp=$(echo "$gpu_temps" | awk '{print $3}')
  gpu3_temp=$(echo "$gpu_temps" | awk '{print $4}')
  cpu0_temp=$(get_cpu_temp 0)
  cpu1_temp=$(get_cpu_temp 1)

  # Найти максимальную температуру среди всех компонентов
  max_temp=$(echo "$gpu0_temp $gpu1_temp $gpu2_temp $gpu3_temp $cpu0_temp $cpu1_temp" | sed 's/°C//g' | awk '{print ($1 > $2 && $1 > $3 && $1 > $4 && $1 > $5 && $1 > $6) ? $1 : ($2 > $3 && $2 > $4 && $2 > $5 && $2 > $6) ? $2 : ($3 > $4 && $3 > $5 && $3 > $6) ? $3 : ($4 > $5 && $4 > $6) ? $4 : ($5 > $6) ? $5 : $6}')
  max_temp=$(printf "%.0f" $max_temp) 

  # Вывести температуры
  echo "GPU Temp: GPU0=$gpu0_temp°C, GPU1=$gpu1_temp°C"
  echo "CPU Temp: CPU0=$cpu0_temp°C, CPU1=$cpu1_temp°C"
  echo "Max Temp: $max_temp°C"

  # Установить скорость вентиляторов
  closest_temp=0
  closest_diff=999

  for temp in "${!fan_speed_table[@]}"; do
    difference=$(bc <<< "$temp - $max_temp")
    if (( $(bc <<< "$difference >= 0" ) && ( $(bc <<< "$difference < $closest_diff" )) )); then
      closest_temp=$temp
      closest_diff=$difference
    fi
  done

  if (( closest_temp == 0 )); then # Если не найдена подходящая температура
     closest_temp=$(printf "%.0f" "${!fan_speed_table[@]}") # Выбрать максимальную температуру из таблицы
  fi

  fan_speed=${fan_speed_table[$closest_temp]}
    if (( $fan_speed != $previous_fan_speed )); then
      set_fan_speed "$fan_speed"
      previous_fan_speed=$fan_speed
    fi

  # Пауза между измерениями
  sleep 0.5
  done