# -*- coding: utf-8 -*-
import wx
import wx.lib.buttons
import wx.lib.platebtn
from plot import *
from database import *
from settings import *
import flask as fla
if settings.enableRFID:
    import rfid

class MainWindow(wx.Frame):

    def __init__(self, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)
        self.panelStart = PanelStart(self)
        self.panelDrinks = PanelDrinks(self)
        self.panelUsers = PanelUsers(self)
        self.panelThanks = PanelThanks(self)
        self.panelSorry = PanelSorry(self)
        self.panelRFID = PanelRFID(self)

        if settings.enableRFID:
            self.rfid = rfid.RFID(self.on_rfid)

        self.user = User()
        self.drinkl = str()
        self.active = 0

        self.init_ui()

    def init_ui(self):
        self.SetSize((480, 320))
        self.SetTitle('Baroness Control')

        self.active = 0
        self.switchPanels()

        self.Show()

    def onStart(self, e):
        self.active = 2
        self.switchPanels()

    def delayExit(self, e=None):
        wx.CallLater(5000, self.onExit)

    def getUser(self):
        return self.user.id
    #    return self.user

    def getDrink(self):
        return self.drink

    def onExit(self, e=None):
        self.active = 0
        self.switchPanels()

    def onUser(self, e):
        longn = e.GetEventObject().GetLabelText()
        for user in get_users():
            if user.longname == longn:
                self.user = user
        if self.user.isblack:
            self.active = 4 #Sorry Bro
        else:
            self.active = 1 #Drinks
        self.switchPanels()

    def on_rfid(self, rfidid):
        if self.active != 0:
            return
        self.user = get_user_by_rfid(rfidid)
        if self.user is None:
            self.panelRFID.label_1.SetLabel(rfidid)
            self.active = 5 #TODO: Screen showing RFID ID
        elif self.user.isblack:
            self.active = 4 #Sorry Bro
        else:
            self.active = 1 #Drinks
        self.switchPanels()

    def onProduct(self, e):
        self.active = 3
        print self.user.longname + ' consumes'
        self.drinkl = e.GetEventObject().GetLabelText()
        drink = get_product_by_name(self.drinkl.split('\n')[0]).id
        with app.app_context():
            for i in range(0, int(self.panelDrinks.GetAmount())):
                add_consume(self.user.name, drink)
        plot_all_thread(self.user)
        self.switchPanels()

    def switchPanels(self):
        self.panelStart.Hide()
        self.panelDrinks.Hide()
        self.panelUsers.Hide()
        self.panelThanks.Hide()
        self.panelSorry.Hide()
        self.panelRFID.Hide()
        if self.active == 0:
            self.panelStart.Show()
        elif self.active == 1:
            if not settings.onlyOneDrink:
                self.panelDrinks.l_amount.SetLabel("%02d" % 1)
            self.panelDrinks.l_user.SetLabel(self.user.longname)
            self.panelDrinks.Show()
        elif self.active == 2:
            self.panelUsers.Show()
        elif self.active == 3:
            self.panelThanks.label_1.SetLabel(self.user.longname + "\n" + "%02d x " % int(self.panelDrinks.GetAmount()) + self.drinkl.split('\n')[0])
            self.panelThanks.label_1.Wrap(340)
            try:
                self.panelThanks.bitmap_2.SetBitmap(wx.Bitmap("./app/static/product_%s.png" % self.drinkl.split('\n')[0], wx.BITMAP_TYPE_ANY))
            except:
                print "no picture for drink:", self.drinkl.split('\n')
            self.panelThanks.Show()
            self.delayExit()
        elif self.active == 4:
            self.panelSorry.label_1.SetLabel(self.user.longname)
            self.panelSorry.Show()
        elif self.active == 5:
            self.panelRFID.Show()

class PanelStart (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
        #panel = wx.Panel(self, -1)
        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/start.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
        if not settings.hideGuiList:
            self.Bind(wx.EVT_LEFT_DOWN, parent.onStart)
            self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onStart)

    def on_quit(self, e):
        print "close"
        self.Destroy()


class PanelThanks (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
        #panel = wx.Panel(self, -1)
        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/thanks.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
        self.bitmap_2 = wx.StaticBitmap(self, wx.ID_ANY, wx.NullBitmap, pos=(10, 10))

        self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(120, 50), size=(340, 100))
        self.label_1.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        self.label_1.SetForegroundColour("white")

        #self.l_product = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(120, 100), size=(340, 100))
        #self.l_product.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        #self.l_product.SetForegroundColour("white")

    def on_quit(self, e):
        print "close"
        self.Destroy()


class PanelSorry (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
        #panel = wx.Panel(self, -1)
        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/sorry.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
        self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onExit)
        self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(100,100))
        self.label_1.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))

    def on_quit(self, e):
        print "close"
        self.Destroy()


class PanelRFID (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
        #panel = wx.Panel(self, -1)
        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/rfid.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
        self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onExit)
        self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(100,100))
        self.label_1.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))

    def on_quit(self, e):
        print "close"
        self.Destroy()


class PanelDrinks (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))

        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/products.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))

        products = get_products()
        buttonids = []
        i = 0
        for product in products:
            if product.isshown:
                #480x320
                self.but = wx.Button(self, id=wx.ID_ANY, label=product.name + u"\n" + u"%0.2f" % product.price, pos=(0+i*120, 0), size=(120, 120))
                self.but.SetBackgroundColour((255-(i*20 % 40), (160+(i*50 % 100)), 0))
                #self.but.SetForegroundColour("#006699")
                self.but.SetFont(wx.Font(23, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
                self.but.Bind(wx.EVT_LEFT_DOWN, parent.onProduct, id=self.but.Id)
                i += 1

        if not settings.onlyOneDrink:
            self.b_less = wx.Button(self, id = wx.ID_ANY, label=u"-", pos=(0,240), size=(120, 80))
            self.b_less.SetFont(wx.Font(60, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
            self.b_less.Bind(wx.EVT_LEFT_DOWN, self.onLess, id=self.b_less.Id)

            self.l_amount = wx.StaticText(self, wx.ID_ANY, "%2d" % 0, pos=(137, 245), style=wx.ALIGN_CENTER)
            self.l_amount.SetFont(wx.Font(50, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Sans"))

            self.b_more = wx.Button(self, id = wx.ID_ANY, label=u"+", pos=(240,240), size=(120, 80))
            self.b_more.SetFont(wx.Font(60, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
            self.b_more.Bind(wx.EVT_LEFT_DOWN, self.onMore, id=self.b_more.Id)

        self.b_exit = wx.Button(self, id = wx.ID_ANY, label=u"x", pos=(360,240), size=(120, 80))
        self.b_exit.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        self.b_exit.Bind(wx.EVT_LEFT_DOWN, parent.onExit, id=self.b_exit.Id)

        self.l_user = wx.StaticText(self, wx.ID_ANY, "todo", pos=(20,170), style=wx.ALIGN_CENTER)
        self.l_user.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))

        self.Bind(wx.EVT_CLOSE, self.on_quit)
        #self.Show(True)

    def onMore(self,e, id=-1):
        self.amount = int(self.l_amount.GetLabelText()) + 1
        if self.amount <= settings.drinkLimit:
            self.l_amount.SetLabel("%02d" % self.amount)
            self.Layout()

    def onLess(self,e, id=-1):
        self.amount = int(self.l_amount.GetLabelText()) - 1
        if self.amount > 0:
            self.l_amount.SetLabel( "%02d" % self.amount)
            self.Layout()

    def GetAmount(self):
        return int(self.l_amount.GetLabelText())

    def on_quit(self, e):
        print "close"
        self.Destroy()


class PanelUsers (wx.Panel):

    def __init__(self, parent):
        wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))

        self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/users.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))

        users = get_users()
        names = list()
        for user in users:
            if user.isshown:
                names.append(user.longname)

        self.but_names = list()
        i = 0
        for name in names:
            #480x320
            #self.but = wx.Button (self, id=wx.ID_ANY, label=name, pos=(0,0+i*80), size=(400, 80))
            self.but = wx.lib.platebtn.PlateButton(self, label=name, pos=(3, 3+i*80),
                                                   style=wx.BU_EXACTFIT | wx.lib.platebtn.PB_STYLE_SQUARE)
            self.but.SetSize((392, 74))

            self.but.SetBackgroundColour((255-(i*20 % 40), (160+(i*50 % 100)), 0))
            #self.but.SetForegroundColour("#006699")
            self.but.SetPressColor(wx.Color(255,255,255,0))
            self.but.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
            self.but.Bind(wx.EVT_LEFT_DOWN, parent.onUser, id=self.but.Id)
            self.but_names.append(self.but)
            i += 1

        self.b_up = wx.Button(self, id=wx.ID_ANY, label=u"\u25B2", pos=(400, 0), size=(80, 80))
        self.b_up.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        self.b_up.Bind(wx.EVT_LEFT_DOWN, self.on_up, id=self.b_up.Id)

        self.b_down = wx.Button(self, id=wx.ID_ANY, label=u"\u25BC", pos=(400, 240), size=(80, 80))
        self.b_down.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        self.b_down.Bind(wx.EVT_LEFT_DOWN, self.on_down, id=self.b_down.Id)

        self.b_exit = wx.Button(self, id=wx.ID_ANY, label="X", pos=(400, 120), size=(80, 80))
        self.b_exit.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
        self.b_exit.Bind(wx.EVT_LEFT_DOWN, parent.onExit, id=self.b_exit.Id)
        self.Bind(wx.EVT_CLOSE, self.on_quit)
        #self.Show(True)

    #This is bad programming :)
    def on_down(self,e , id=-1):
        if self.but_names[len(self.but_names)-1].GetPosition()[1] < 320:
            return
        for button in self.but_names:
            button.SetPosition((button.GetPosition()[0], button.GetPosition()[1]-320))
        self.Layout()

    def on_up(self,e , id=-1):
        if self.but_names[0].GetPosition()[1] >= 0:
            return
        for button in self.but_names:
            button.SetPosition((button.GetPosition()[0], button.GetPosition()[1]+320))
        self.Layout()

    def on_button_press(self,e, id=-1):
        print e.Id
        print id
        #if id == self.b_down.Id:

    def trap(self, e):
        e.Skip()
        return

    def on_quit(self, e):
        print "close"
        self.Destroy()