from untitled import ui
from untitled.screens import shop, work


def options():
    while True:
        match ui.select("Please choose an option:", ["Save", "Save and quit", "Back"]):
            case "Back":
                return "back"
            case "Save":
                return "save"
            case "Save and quit":
                return "savequit"


def go_to(save):
    match ui.select("Where do you want to go?", ["The shop", "Work", "Back"]):
        case "The shop":
            shop.shop(save)
        case "Work":
            work.work(save)