import machine 
import joystick
from menu import *   # bad habit but makes our menu definition nice
from vos_state import vos_state
import vectoros
import colors
import gc 
import random


# run the sketch demo
def runsketch(arg):
    vos_state.show_menu=False     # get the menu of the way
    vectoros.launch_task('sketch')
    return EXIT

def planets(arg):
    vos_state.show_menu=False     # get the menu of the way
    vectoros.launch_task('planets')  # launch
    return EXIT

def menu_custom(the_menu):
    if the_menu.level==1:
        if machine.Pin(22).value():
            the_menu.current[2]=[" Sound Off",toggle_sound,0]
        else:
            the_menu.current[2]=[" Sound On",toggle_sound,1]

def toggle_sound(arg):
    ## toggle sound here
    machine.Pin(22, machine.Pin.OUT).toggle()
    return CONT


# the main vector scope demo
def run_lissajous(arg):
    vos_state.show_menu=False
    vectoros.launch_task('lissajous')  
    # we never come back, vectorscope
    return EXIT

def reboot(arg):
    if arg==False:
        vectoros.reset()
    else:
        vectoros.soft_reset()

# handle slots
def abcd(key):
    if vos_state.show_menu:
        vos_debug.debug_print(vos_debug.DEBUG_LEVEL_INFO,f"Menu key {key}")
        kdict={ keyleds.KEY_A: 'A', keyleds.KEY_B: 'B', keyleds.KEY_C: 'C', keyleds.KEY_D: 'D'}
        await vectoros.launch_vecslot("slot"+kdict[key])
    

# I really didn't want this to be async but it seems like do_menu must have an await
# and run rarely returns when you have a lot going on
async def vos_main():
# you do NOT have to use with here
# but if you don't you have to worry about the menu controller's joystick instance going out of scope yourself
# or just make everything global -- the menu is smart enough to not listen to events when it is not active
# note: m_back and m_exit were imported from menu
    ## Start with sound off
    machine.Pin(22, machine.Pin.OUT).toggle()
    
    screen=vectoros.get_screen()
    splashes = ["splash_x.jpg", "splash_7.jpg", "splash_wrencher.jpg"]
    screen.jpg(random.choice(splashes))
    await asyncio.sleep_ms(1000)

    

    while True: # since this is the main menu, we don't really every quit
        print("creating slotkey")
        slotkey=keyboardcb.KeyboardCB(abcd,keyleds.KEY_ABCD)
        with Menu(clear_after=True,fg_color=colors.PHOSPHOR_DARK,bg_color=colors.PHOSPHOR_BG,
                  cursor_bg=colors.PHOSPHOR_BG, cursor_fg=colors.PHOSPHOR_BRIGHT) as amenu:  
            ## name in menu, command to run, return value?
            submenu=[["  Planets", planets, 0],["  Sketch",runsketch,0],["  Back",m_exit,None]]
            mainmenu=[[" Lissajous", run_lissajous,None],
                      [" Demos", SUBMENU, submenu] ,
                      [" Sound", toggle_sound, None],
                      [" Reboot",reboot,False],
                      ]

# comment next line for default font
            amenu.set_font("*")   # set default vector font
            amenu.set_callback(menu_custom)
            await amenu.do_menu(mainmenu)
        vos_debug.debug_print(vos_debug.DEBUG_LEVEL_INFO,f"Menu waiting {vos_state.show_menu}")
        while vos_state.show_menu==False:   # wait until we have to be seen again
            await asyncio.sleep_ms(0)
    

def main():
    asyncio.run(vos_main())
    # this never runs 

if __name__=="__main__":
    import vectoros
    vectoros.run()