import display
import buttons
import mch22
import wifi
import time
import math
import nvs
import urequests as requests

class ButtonInput:
    def __init__(self, a = False, b = False, up = False, down = False):
        self.a = a
        self.b = b
        self.up = up
        self.down = down

class StateWifiConnect:
    connect_timeout = 0
    parent = None

    def __init__(self, parent):
        self.parent = parent

    def enter(self):
        self.parent.drawSplash()
        self.parent.drawWifiStatus()

        display.drawText(0, 60, "Attempting to connect to WiFi of UbaBot.")
        display.drawText(0, 80, "Make sure you are near the machine.")
        display.drawText(0, 100, "It is located in FruBar Island village.")
        display.drawText(0, 160, "Press HOME button to exit this app.")

        display.flush()

        # Start establishing WiFi connection
        wifi.connect("ubabot", "ubabot1337")
        connect_timeout = time.time() + 60

    def act(self, inputs):
        if wifi.status():
            print("WiFi connected")
            return 1 # login state

        if (time.time() >= self.connect_timeout) and (self.connect_timeout > 0):
            print("WiFi timeout")
            self.parent.errorMessage = "WiFi connection timed out!"
            self.parent.errorThrower = 0
            return 3 # error state

        return -1

    def exit(self):
        pass

class StateLogin:
    parent = None
    login_running = False
    login_timeout = 0

    def __init__(self, parent):
        self.parent = parent

    def drawLoginButton(self):
        display.drawText(60, 70, "Please Log In Now", 0x0044FF, "roboto_black22")
        display.drawText(0, 130, "Press A to login to UbaBot.")
        display.drawText(0, 160, "Press HOME button to exit this app.")

    def draw(self):
        self.parent.drawSplash()
        self.parent.drawWifiStatus()

        if self.login_running:
            display.drawText(0, 100, "Logging in to UbaBot.")
            display.drawText(0, 130, "Please wait.")
            display.drawText(0, 160, "Press HOME button to exit this app.")
        else:
            self.drawLoginButton()

        display.flush()

    def enter(self):
        self.login_running = False
        self.login_timeout = 0
        self.draw()

    def performLogin(self):
        try:
            nickname = nvs.nvs_getstr("owner", "nickname")
            if len(nickname) <= 1:
                print("No username set")
                self.parent.errorMessage = "No username set on badge!"
                self.parent.errorThrower = 1
                return 3
            else:
                # TODO nickname should be query-string url-encoded

                login_res = requests.get(self.parent.uba_url + "/badge?name=" + nickname)
                login = login_res.text
                print("got response: \"" + login + "\"")
                if "ok" in login:
                    return 2
                else:
                    print("Other user already logged in")
                    self.parent.errorMessage = login
                    self.parent.errorThrower = 1
                    return 3
        except OSError as e:
            print("Error logging into Ubabot")
            print(e)
            self.parent.errorMessage = "Login Error ({})".format(str(e))
            self.parent.errorThrower = 1
            return 3

        return -1

    def act(self, inputs):
        if inputs.a and not self.login_running:
            self.login_running = True
            self.login_timeout = time.time() + 60
            self.draw()
            return self.performLogin()

        if self.login_running and (time.time() >= self.login_timeout):
            self.login_running = False
            self.draw()

        return -1

    def exit(self):
        pass

class StateStatus:
    parent = None

    def __init__(self, parent):
        self.parent = parent

    def queryAlcoholLevel(self):
        bac = 0.0
        redraw = 0

        try:
            nickname = nvs.nvs_getstr("owner", "nickname")
            if len(nickname) <= 1:
                print("No username set")
                return -1
            else:
                # TODO nickname should be query-string url-encoded

                bac_res = requests.get(self.parent.uba_url + "/bac?name=" + nickname)
                print("got bac response: \"" + bac_res.text + "\"")
                try:
                    bac = float(bac_res.text)
                except ValueError as e:
                    print("Error interpreting BAC")
                    print(e)
                    return -1
        except OSError as e:
            print("Error getting BAC")
            print(e)
            return -1

        return bac

    def drawHelp(self):
        display.drawText(0, 140, "Querying data from UbaBot.")
        display.drawText(0, 160, "Please wait.")
        display.drawText(0, 180, "Press HOME button to exit this app.")

    def drawData(self):
        nickname = nvs.nvs_getstr("owner", "nickname")
        display.drawText(0, 65, "User: " + nickname + "@drinkrobotics.de", 0xFFFFFF, "roboto_regular12")
        display.drawText(0, 90, "Pass: " + nickname + "mch2022", 0xFFFFFF, "roboto_regular12")
        display.drawText(0, 120, "You can order at the machine now!", 0xFFFFFF, "roboto_regular18")

    def drawAlcohol(self):
        pegel = self.bac
        if (pegel >= 0):
            display.drawText(0, 150, "BAC: {:.2f}% / {:.2f}‰".format(pegel, pegel / 10.0), 0x0044FF, "roboto_regular18")

    def drawHint(self):
        display.drawText(0, 180, "Press A to order another drink!", 0xFF0000)

    def enter(self):
        self.parent.drawSplash()
        self.parent.drawWifiStatus()
        self.drawData()
        self.drawHelp()
        display.flush()

        self.bac = self.queryAlcoholLevel()
        self.redraw = time.time() + 10

        self.parent.drawSplash()
        self.parent.drawWifiStatus()
        self.drawData()
        self.drawAlcohol()
        self.drawHint()
        display.flush()

    def act(self, inputs):
        if time.time() >= self.redraw:
            self.bac = self.queryAlcoholLevel()
            self.redraw = time.time() + 10

            self.parent.drawSplash()
            self.parent.drawWifiStatus()
            self.drawData()
            self.drawAlcohol()
            self.drawHint()
            display.flush()

        if inputs.a:
            return 1

        return -1

    def exit(self):
        pass

class StateError:
    parent = None

    def __init__(self, parent):
        self.parent = parent

    def enter(self):
        pass

    def act(self, inputs):
        self.parent.drawSplash()
        self.parent.drawWifiStatus()

        display.drawText(0, 60, "ERROR!", 0xFF0000)

        if (self.parent.errorMessage == None) or (self.parent.errorMessage == ""):
            display.drawText(0, 80, "Connection to UbaBot could not be established!")
        else:
            display.drawText(0, 80, self.parent.errorMessage)

        display.drawText(0, 100, "Please press A or B button to try again.")
        display.drawText(0, 160, "Press HOME button to exit this app.")

        display.flush()

        if inputs.a or inputs.b:
            return self.parent.errorThrower

        return -1

    def exit(self):
        pass

class UbaBot:
    # production server
    uba_url = "http://192.168.0.1:8080"

    # debug / testing server
    #uba_url = "http://192.168.0.108:8000"

    states = []

    current_state = 0
    inputs = ButtonInput()
    errorMessage = ""
    errorThrower = 0

    def reboot(self, pressed):
        if pressed:
            mch22.exit_python()

    def buttonA(self, pressed):
        self.inputs.a = pressed

    def buttonB(self, pressed):
        self.inputs.b = pressed

    def buttonUp(self, pressed):
        self.inputs.up = pressed

    def buttonDown(self, pressed):
        self.inputs.down = pressed

    def main(self):
        self.states = [
            StateWifiConnect(self),
            StateLogin(self),
            StateStatus(self),
            StateError(self)
        ]
        buttons.attach(buttons.BTN_HOME, self.reboot)
        buttons.attach(buttons.BTN_A, self.buttonA)
        buttons.attach(buttons.BTN_B, self.buttonB)
        buttons.attach(buttons.BTN_DOWN, self.buttonDown)
        buttons.attach(buttons.BTN_UP, self.buttonUp)

        while True:
            s = self.states[self.current_state]
            s.enter()

            while True:
                r = s.act(self.inputs)
                self.inputs = ButtonInput()
                if (r >= 0):
                    print("Changing state from " + str(self.current_state) + " to " + str(r))
                    s.exit()
                    self.current_state = r
                    break

    def drawSplash(self):
        display.drawFill(0x000000)
        display.drawText(5, 0, "UbaBot Cocktail Machine", 0xFF0000, "permanentmarker22")
        display.drawText(10, 30, "@ FruBar Island", 0x00FF00, "press_start_2p18")

    def drawWifiStatus(self):
        status = "Disconnected"
        if wifi.status:
            status = "Connected"
        text = "WiFi Status: " + status
        h = display.getTextHeight(text)
        display.drawText(0, display.height() - h, text , 0xFFFFFF)

        #text = self.status_text
        #w = display.getTextWidth(text)
        #h = display.getTextHeight(text)
        #display.drawText(display.width() - w, display.height() - h, text, self.status_color)

uba = UbaBot()
uba.main()