# ==================================================
# Warmwassersteuerung
# ==================================================
#
# Komponenten:
#   - 4 Zeilen LCD
#   - 2 Ventile
#   - 2 Buttons
#   - 1 DS18B20 I2C Temperatursensor
#
#   Copyright (C) 2023, 3x3cut0r
#
#   Veröffentlicht unter der MIT Lizenz.

# ==================================================
# imports
# ==================================================

# micro python imports
import gc  # https://docs.micropython.org/en/latest/library/gc.html
import time  # https://docs.micropython.org/en/latest/library/time.html
import uasyncio as asyncio  # https://docs.micropython.org/en/latest/library/asyncio.html

# custom imports
from config import config
from lcd import init_lcd
from led import init_led
from relay import init_relays
from webserver import run_webserver
from wifi import connect_wifi, check_wifi_isactivated, check_wifi_isconnected
from functions import (
    categorize_temp_change,
    adjust_update_time_based_on_temp_category,
    update_temp,
    print_nominal_temp,
    set_relay,
    open_relays,
    check_buttons,
    update_timer,
    wait_start,
)


async def main():
    print(f"INFO: main()")

    # ========================================
    # setup
    # ========================================
    print(f"--------------------------------")
    print(f"INFO: main setup()")
    print(f"--------------------------------")

    # init led
    init_led()

    # init lcd
    init_lcd()

    # init relays
    init_relays()

    # connect wifi
    while check_wifi_isactivated() and not check_wifi_isconnected():
        connect_wifi()

    # update temp
    update_temp()
    config.set_value(
        "tempAtLastMeasurement", config.get_float_value("current_temp", -127.0)
    )

    # print nominal temp
    print_nominal_temp()

    # wait start
    wait_start(config.get_int_value("delay_before_start_1"))
    set_relay(
        config.get_int_value("RELAY_CLOSE_PIN", 15),
        config.get_int_value("init_relay_time", 2000),
    )  # open relay initial
    wait_start(config.get_int_value("delay_before_start_2"))

    previous_millis = 0
    interval = config.get_int_value("interval")
    update_time = config.get_int_value("update_time")

    # ==================================================
    # main loop
    # ==================================================
    print(f"--------------------------------")
    print(f"INFO: main loop()")
    print(f"--------------------------------")

    while True:
        current_millis = time.ticks_ms()

        # adjust temp category
        if current_millis - config.get_int_value(
            "temp_last_measurement_time"
        ) >= config.get_int_value("temp_sampling_interval"):
            update_temp()
            temp_change = config.get_float_value(
                "current_temp", -127.0
            ) - config.get_float_value("temp_last_measurement")

            # categorize temp change
            categorize_temp_change(temp_change)

            # update last measurement temp
            config.set_value(
                "temp_last_measurement", config.get_float_value("current_temp", -127.0)
            )

            # update last measurement temp time
            config.set_value("temp_last_measurement_time", current_millis)

        # main
        if current_millis - previous_millis > interval:
            # update time
            if update_time >= 0:
                update_timer(update_time)

                # print mem alloc
                print("INFO: mem_alloc(): {} Bytes".format(gc.mem_alloc()))

                # update temp on temp update interval
                if update_time % config.get_int_value("temp_update_interval") == 0:
                    update_temp()

                update_time -= 1

            # check buttons
            check_buttons()

            if update_time == 0:
                # open relay
                open_relays()

                # set and adjust update_time based on temp category
                update_time = adjust_update_time_based_on_temp_category()

                # create config backup
                config.create_config_backup()

                # connect wifi
                if not check_wifi_isconnected():
                    connect_wifi()

            # update previous millis
            previous_millis = current_millis
            config.set_value("previous_millis", previous_millis)

        await asyncio.sleep(0.1)


if __name__ == "__main__":
    print(f"INFO: __main__")

    # create asyncio event loop
    loop = asyncio.get_event_loop()

    # run webserver() as task
    loop.create_task(run_webserver())

    # run main() as task
    loop.create_task(main())

    # run event loop forever
    loop.run_forever()