#!/usr/bin/python3
# source: https://github.com/clockworkpi/DevTerm/tree/main/Code/devterm_fan_daemon_cpi_a06

import glob
import os
import sys
import getopt
import subprocess
import time

cpus = []
mid_freq = 0
max_freq = 0

# Replace fan-on temperature w/ 55C, rather than 70C which doesn't seem to be enough to avoid overheating
MAX_TEMP = 55000
ONCE_TIME = 30


def init_fan_gpio():
    os.popen("gpio mode 41 out")


def fan_on():
    init_fan_gpio()
    os.popen("gpio write 41 1")
    time.sleep(ONCE_TIME)


def fan_off():
    init_fan_gpio()
    os.popen("gpio write 41 0")


def isDigit(x):
    try:
        float(x)
        return True
    except ValueError:
        return False


def cpu_infos():
    global cpus
    global mid_freq
    global max_freq

    cpus = glob.glob('/sys/devices/system/cpu/cpu[0-9]')
    cpus.sort()
# /sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies
# /sys/devices/system/cpu/cpu0/cpufreq/scaling_governor

    scaling_available_freq = open(os.path.join(
        cpus[0], "cpufreq/scaling_available_frequencies"), "r").read()
    scaling_available_freq = scaling_available_freq.strip("\n")
    scaling_available_freqs = scaling_available_freq.split(" ")

    for var in scaling_available_freqs:
        if isDigit(var):
            if(int(var) > 1000000):
                if(mid_freq == 0):
                    mid_freq = int(var)
            max_freq = int(var)


def set_gov(gov):
    global cpus
    for var in cpus:
        gov_f = os.path.join(var, "cpufreq/scaling_governor")
        # print(gov_f)
        subprocess.run("echo %s | sudo tee  %s" % (gov, gov_f), shell=True)


def set_performance(scale):
    global cpus
    global mid_freq
    global max_freq

    freq = mid_freq
    if scale == "mid":
        freq = mid_freq
    elif scale == "max":
        freq = max_freq

    for var in cpus:
        _f = os.path.join(var, "cpufreq/scaling_max_freq")
        # print(_f)
        subprocess.run("echo %s | sudo tee  %s" % (freq, _f), shell=True)


def fan_loop():
    while True:
        temps = glob.glob('/sys/class/thermal/thermal_zone[0-9]/')
        temps.sort()
        for var in temps:
            _f = os.path.join(var, "temp")
            #print( open(_f).read().strip("\n") )
            _t = open(_f).read().strip("\n")
            if isDigit(_t):
                if int(_t) > MAX_TEMP:
                    fan_on()
                    fan_off()

        time.sleep(5)


def main(argv):
    global cpus
    scale = 'mid'
    gov = 'powersave'
    try:
        opts, args = getopt.getopt(argv, "hs:g:", ["scale=", "governor="])
    except getopt.GetoptError:
        print('test.py -s <scale> -g <governor>')
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print('test.py -s <scale>')
            sys.exit()
        elif opt in ("-s", "--scale"):
            scale = arg
        elif opt in ("-g", "--governor"):
            gov = arg

    print('Scale is ', scale, "Gov is ", gov)

    init_fan_gpio()
    cpu_infos()
    set_gov(gov)
    set_performance(scale)
    fan_loop()


if __name__ == "__main__":
    main(sys.argv[1:])