#!/bin/bash

# Функция для получения температуры процессора
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 addr1=$1
  local addr2=$2
  local speed=$3
  local hex_speed=$(printf "%x" "$speed") # Перевод в шестнадцатеричный формат
  echo "Executing: ipmitool raw 0x2e 0x30 $addr1 $addr2 0x$hex_speed"
  ipmitool raw 0x2e 0x30 "$addr1" "$addr2" "0x$hex_speed"
}

# Обработка аргументов командной строки с новыми значениями по умолчанию
max_temp=80               # Максимальная температура (по умолчанию 80°C)
min_chassis_speed=75      # Минимальная скорость для корпусных вентиляторов (днем)
min_cpu_speed=45          # Минимальная скорость для CPU вентиляторов (днем)
night_start="00:00"       # Время начала ночного режима
night_end="07:00"         # Время окончания ночного режима
night_reduction=15        # Процент снижения скорости ночью (15%)

# Парсинг аргументов
while getopts "t:c:p:s:e:r:" opt; do
  case $opt in
    t) max_temp=$OPTARG ;;
    c) min_chassis_speed=$OPTARG ;;
    p) min_cpu_speed=$OPTARG ;;
    s) night_start=$OPTARG ;;
    e) night_end=$OPTARG ;;
    r) night_reduction=$OPTARG ;;
    \?) echo "Invalid option: -$OPTARG" >&2 ;;
  esac
done

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

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

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

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

# Проверка формата времени
if ! [[ "$night_start" =~ ^[0-9]{2}:[0-9]{2}$ ]]; then
  echo "Error: Invalid night start time format. Use HH:MM."
  exit 1
fi

if ! [[ "$night_end" =~ ^[0-9]{2}:[0-9]{2}$ ]]; then
  echo "Error: Invalid night end time format. Use HH:MM."
  exit 1
fi

# Функция перевода времени в минуты
time_to_minutes() {
  local time=$1
  echo "$((${time%%:*} * 60 + ${time#*:}))"
}

# Функция проверки ночного режима
is_night_mode() {
  local current_minutes=$(date +%H:%M | awk -F: '{print $1*60+$2}')
  local start_minutes=$(time_to_minutes "$night_start")
  local end_minutes=$(time_to_minutes "$night_end")
 
  # Если время начала больше времени окончания (переход через полночь)
  if (( start_minutes > end_minutes )); then
    if (( current_minutes >= start_minutes || current_minutes < end_minutes )); then
      return 0 # Ночной режим
    else
      return 1 # Дневной режим
    fi
  else
    if (( current_minutes >= start_minutes && current_minutes < end_minutes )); then
      return 0 # Ночной режим
    else
      return 1 # Дневной режим
    fi
  fi
}

# Функция для расчета минимальной скорости с учетом ночного режима
calculate_effective_speed() {
  local base_speed=$1
 
  if is_night_mode; then
    # Рассчитываем сниженную скорость
    local reduced_speed=$(( base_speed - (base_speed * night_reduction / 100) ))
   
    # Гарантируем, что скорость не упадет ниже 10%
    if (( reduced_speed < 10 )); then
      reduced_speed=10
    fi
   
    echo $reduced_speed
  else
    echo $base_speed
  fi
}

# Создание таблиц скоростей с учетом текущего режима
create_speed_tables() {
  unset fan_speed_table_chassis
  unset fan_speed_table_cpu
  declare -gA fan_speed_table_chassis
  declare -gA fan_speed_table_cpu

  # Рассчитываем актуальные минимальные скорости
  local effective_chassis_speed=$(calculate_effective_speed $min_chassis_speed)
  local effective_cpu_speed=$(calculate_effective_speed $min_cpu_speed)

  # Выводим текущие минимальные скорости
  echo "Current min speeds: Chassis=$effective_chassis_speed%, CPU=$effective_cpu_speed%"

  # Заполняем таблицы
  temp=30
  while (( temp <= max_temp )); do
    # Формула для корпусных вентиляторов
    chassis_speed=$(( effective_chassis_speed + ( ( temp - 30 ) / 5 ) * 5 ))
    # Ограничение максимальной скорости 100%
    if (( chassis_speed > 100 )); then chassis_speed=100; fi
    fan_speed_table_chassis[$temp]=$chassis_speed
   
    # Формула для CPU вентиляторов
    cpu_speed=$(( effective_cpu_speed + ( ( temp - 30 ) / 5 ) * 5 ))
    # Ограничение максимальной скорости 100%
    if (( cpu_speed > 100 )); then cpu_speed=100; fi
    fan_speed_table_cpu[$temp]=$cpu_speed
   
    (( temp += 5 ))
  done

  # Отладочный вывод таблиц
  echo "Chassis speed table:"
  for t in "${!fan_speed_table_chassis[@]}"; do
    echo "  $t°C -> ${fan_speed_table_chassis[$t]}%"
  done | sort -n
 
  echo "CPU speed table:"
  for t in "${!fan_speed_table_cpu[@]}"; do
    echo "  $t°C -> ${fan_speed_table_cpu[$t]}%"
  done | sort -n
}

# Цикл для постоянного мониторинга температуры
previous_chassis_speed=0
previous_cpu_speed=0
last_mode_check=""
last_table_rebuild=$(date +%s)

# Создать таблицы при запуске
create_speed_tables

while true; do
  # Проверка изменения режима (день/ночь) каждые 60 секунд
  current_time=$(date +%s)
  if (( current_time - last_table_rebuild >= 60 )); then
    if is_night_mode; then
      current_mode="night"
    else
      current_mode="day"
    fi
   
    # Если режим изменился или это первый запуск
    if [[ "$current_mode" != "$last_mode_check" ]]; then
      echo "Switching to $current_mode mode"
      create_speed_tables
      last_mode_check=$current_mode
      # Сбрасываем предыдущие скорости для принудительного обновления
      previous_chassis_speed=0
      previous_cpu_speed=0
    fi
   
    last_table_rebuild=$current_time
  fi

  # Получить температуры CPU
  cpu0_temp=$(get_cpu_temp 0)
  cpu1_temp=$(get_cpu_temp 1)

  # Найти максимальную температуру CPU
  max_cpu_temp=$(echo "$cpu0_temp $cpu1_temp" | awk '{if ($1 > $2) print $1; else print $2}')
  max_cpu_temp_int=${max_cpu_temp%.*}

  # Вывести температуры и текущий режим
  echo "[$current_mode] CPU Temp: CPU0=$cpu0_temp°C, CPU1=$cpu1_temp°C"
  echo "Max CPU Temp: $max_cpu_temp_int°C"

  # Функция для поиска ближайшей температуры в таблице
  find_closest_temp() {
    local current_temp=$1
    closest_temp=0
    closest_diff=999
    for temp in "${!fan_speed_table_chassis[@]}"; do
      diff=$((temp - current_temp))
      if (( diff >= 0 && diff < closest_diff )); then
        closest_temp=$temp
        closest_diff=$diff
      fi
    done
    if (( closest_temp == 0 )); then
      # Если не найдено, берем максимальную температуру из таблицы
      closest_temp=$(printf "%s\n" "${!fan_speed_table_chassis[@]}" | sort -nr | head -1)
    fi
    echo "$closest_temp"
  }

  # Определение температуры для управления
  control_temp=$(find_closest_temp "$max_cpu_temp_int")
 
  # Определение скоростей
  chassis_speed=${fan_speed_table_chassis[$control_temp]}
  cpu_speed=${fan_speed_table_cpu[$control_temp]}

  # Установка скорости корпусных вентиляторов
  if [[ "$chassis_speed" != "$previous_chassis_speed" ]]; then
    echo "Setting chassis fans to $chassis_speed%"
    set_fan_speed "00" "01" "$chassis_speed"
    set_fan_speed "00" "02" "$chassis_speed"
    set_fan_speed "00" "03" "$chassis_speed"
    set_fan_speed "00" "04" "$chassis_speed"
    previous_chassis_speed=$chassis_speed
  fi

  # Установка скорости CPU вентиляторов
  if [[ "$cpu_speed" != "$previous_cpu_speed" ]]; then
    echo "Setting CPU fans to $cpu_speed%"
    set_fan_speed "00" "05" "$cpu_speed"
    set_fan_speed "00" "06" "$cpu_speed"
    previous_cpu_speed=$cpu_speed
  fi

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

[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