"""
uv-belichter-software - The main program run at the Pi Picos startup
Copyright (C) 2024  Benjamin Burkhardt

This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""


import utils
from lcdMenu import lcdMenu
from WelcomeScreen import WelcomeScreen
from time import sleep
from gc import collect  # garbage collector for better memory performance

config = utils.Config()
btn_mapping = {"ok_btn": config.PIN_IN_BTN_1, "next_btn": config.PIN_IN_BTN_2}  # the btn mapping for all lcdMenus

# extra functions to access the garbage collector
def manual():
    config.LCD.clear()
    set_value = config.PIN_OUT_RELAIS.value()
    config.LCD.putstr(f"---- MANUAL ----    State: {set_value}    ")
    while True:
        if set_value != config.PIN_IN_SWITCH.value():
            config.PIN_OUT_RELAIS.value(config.PIN_IN_SWITCH.value())
            set_value = config.PIN_OUT_RELAIS.value()
            config.LCD.putstr(f"---- MANUAL ----    State: {set_value}    ")
        if config.PIN_IN_BTN_1.value() == 1 or config.PIN_IN_BTN_2.value() == 1:
            return True # exit on press of these buttons
def timer():
    # display WIP
    config.LCD.clear()
    config.LCD.putstr("      Still     work-in-progress")
    sleep(3)
    return True  # disable the "Quitting" message from lcdMenu
def uv_on():
    config.RELAIS.value(1)
    config.LCD.clear()
    config.LCD.putstr("------ UV ------   turned on    ")
    sleep(1)
    return True  # disable the "Quitting" message from lcdMenu
def uv_off():
    config.RELAIS.value(0)
    config.LCD.clear()
    config.LCD.putstr("------ UV ------   turned off   ")
    sleep(1)
    return True  # disable the "Quitting" message from lcdMenu
def lcd_big_hello():
    import lcd_big_hello as lbh
    lbh.run()
    del lbh
    collect()
    return True
def input_tests():
    import input_tests as input_tests
    input_tests.run(serial_output=False)
    collect()
    return True
def settings():
    # display WIP
    config.LCD.clear()
    config.LCD.putstr("      Still     work-in-progress")
    sleep(3)
    return True
def run_demo_menu():
    demo_menu = lcdMenu(config.LCD, btn_mapping, scroll_direction=True, cycle=True, hide_menu_name=False, name="DEMOS")
    demo_programs = [("LCD Demo", lcd_big_hello),
                     ("Input tests", input_tests),
                     ("Exit", demo_menu.stop)]
    demo_menu.setup(demo_programs)  # give it the callback list
    ret = demo_menu.run()
    del demo_menu, demo_programs
    collect()
    return ret

if config.STARTUP_WELCOME_SHOW:
    ws = WelcomeScreen(config.LCD,
                       interrupt_pins=[config.PIN_IN_BTN_1, config.PIN_IN_BTN_2, config.PIN_IN_SWITCH],
                       subtitle=config.STARTUP_PROJECT_NAME,
                       starting_msg=config.STARTUP_MESSAGE_STARTING,
                       started_msg=config.STARTUP_MESSAGE_FINISHED)
    ws.show(cycles=config.STARTUP_WELCOME_CYCLES)
    del ws
    collect()

    
# create the menus
main_menu = lcdMenu(config.LCD, btn_mapping, scroll_direction=True, cycle=True, hide_menu_name=False, name="PROGRAMS")
main_programs = [("Timer", timer),
                 ("Manual", manual),
                 ("UV off", uv_off),
                 ("UV on", uv_on),
                 ("Demos", run_demo_menu),
                 ("Settings", settings)]
main_menu.setup(main_programs)  # give it the callback list

# and run the main menu (will be an endless loop)
main_menu.run()