from direct.gui.DirectGui import *
from pandac.PandaModules import *
from direct.directnotify import DirectNotifyGlobal
from toontown.toonbase import ToontownGlobals
from toontown.toonbase import TTLocalizer
from direct.task import Task
from toontown.estate import GardenGlobals
from direct.interval.IntervalGlobal import *
from toontown.estate import SpecialsPhoto
USE_SCROLLING_BEAN_BOX = False
JELLY_BEAN_PICKER_HAS_EMPTY_BOX = False
CAN_CHANGE_BEAN_COLOR = True
FORCE_LEFT_TO_RIGHT = True
ONLY_ONE_SPIFFY_BOX_CAN_BE_CLICKED = True
DO_PICKER_INTERVAL = False
PICKER_ALWAYS_UP = True

def loadJellyBean(parent, beanIndex):
    gui = loader.loadModel('phase_5.5/models/estate/jellyBean')
    newBean = gui.instanceTo(parent)
    parent.setScale(0.09)
    colors = GardenGlobals.BeanColors[beanIndex]
    parent.setColorScale(colors[0] / 255.0 * 1.0, colors[1] / 255.0 * 1.0, colors[2] / 255.0 * 1.0, 1)


class GenericBoxScrollList(DirectScrolledList):

    def __init__(self, parent, items, **kw):
        gui = loader.loadModel('phase_3.5/models/gui/friendslist_gui')
        self.forceHeight = 1
        optiondefs = (('parent', parent, None),
         ('relief', None, None),
         ('incButton_image', (gui.find('**/FndsLst_ScrollUp'),
           gui.find('**/FndsLst_ScrollDN'),
           gui.find('**/FndsLst_ScrollUp_Rllvr'),
           gui.find('**/FndsLst_ScrollUp')), None),
         ('incButton_relief', None, None),
         ('incButton_scale', (1.3, 1.3, -1.3), None),
         ('incButton_pos', (0, 0, -0.525), None),
         ('incButton_image3_color', Vec4(0.8, 0.8, 0.8, 0.5), None),
         ('decButton_image', (gui.find('**/FndsLst_ScrollUp'),
           gui.find('**/FndsLst_ScrollDN'),
           gui.find('**/FndsLst_ScrollUp_Rllvr'),
           gui.find('**/FndsLst_ScrollUp')), None),
         ('decButton_relief', None, None),
         ('decButton_scale', (1.3, 1.3, 1.3), None),
         ('decButton_pos', (0, 0, 0.525), None),
         ('decButton_image3_color', Vec4(0.8, 0.8, 0.8, 0.5), None),
         ('numItemsVisible', 1, None),
         ('items', items, None),
         ('scrollSpeed', 1.0, None),
         ('forceHeight', 1, None))
        gui.removeNode()
        self.defineoptions(kw, optiondefs)
        DirectScrolledList.__init__(self, parent, forceHeight=self.forceHeight)
        self.initialiseoptions(GenericBoxScrollList)
        return None


class BoxItem(NodePath):

    def getHeight(self):
        return 0.05


JellyBeanPickerEndPos = (0, 0, 0.1)
JellyBeanPickerScale = (1.1, 1.0, 0.13)
if JELLY_BEAN_PICKER_HAS_EMPTY_BOX:
    JellyBeanPickerGeomScale = (1.0, 1.0, 1.0)
else:
    JellyBeanPickerGeomScale = (0.9, 1.0, 1.0)
JellyBeanPickerScaleInverse = (1.0 / JellyBeanPickerScale[0], 1.0 / JellyBeanPickerScale[1], 1.0 / JellyBeanPickerScale[2])

class JellyBeanPicker(DirectFrame):
    notify = DirectNotifyGlobal.directNotify.newCategory('PlantingGUI')

    def __init__(self, parent, callBack, boxPosition, **kw):
        self.index = 0
        self.selectedIndex = 0
        self.callBack = callBack
        self.boxPosition = boxPosition
        DirectFrame.__init__(self, relief=None, state='normal', geom=DGG.getDefaultDialogGeom(), geom_color=(0.8, 0.8, 0.8, 1.0), geom_scale=JellyBeanPickerGeomScale, scale=JellyBeanPickerScale, frameSize=(-1, 1, -1, 1), pos=(0, 0, 0))
        self.initialiseoptions(JellyBeanPicker)
        self.jellyBeanBoxList = []
        self.createJellyBeanBoxes()
        return

    def jellyBeanBoxClicked(self, beanIndex):
        if JELLY_BEAN_PICKER_HAS_EMPTY_BOX:
            self.callBack(self.boxPosition, beanIndex)
        else:
            self.callBack(self.boxPosition, beanIndex + 1)

    def createJellyBeanBox(self, beanIndex, xPos, zPos):
        geomColor = (1, 1, 1, 1)
        state = DGG.NORMAL
        command = self.jellyBeanBoxClicked
        newBox = DirectButton(parent=self, pos=(xPos, 0, zPos), geom=DGG.getDefaultDialogGeom(), geom_scale=(0.1, 1.0, 0.1), geom_color=geomColor, scale=JellyBeanPickerScaleInverse, relief=None, state=state, command=command, extraArgs=[beanIndex], text='', text_pos=(0.0, 0.1), text_scale=0.07, text_fg=Vec4(0, 0, 0, 0), text1_fg=Vec4(0, 0, 0, 0), text2_fg=Vec4(0, 0, 0, 1), text3_fg=Vec4(0, 0, 0, 0))
        if JELLY_BEAN_PICKER_HAS_EMPTY_BOX:
            if beanIndex:
                beanParent = newBox.attachNewNode('bean_%d' % (beanIndex - 1))
                loadJellyBean(beanParent, beanIndex - 1)
        else:
            beanParent = newBox.attachNewNode('bean_%d' % beanIndex)
            loadJellyBean(beanParent, beanIndex)
        self.jellyBeanBoxList.append(newBox)
        return

    def setColorText(self):
        for beanIndex in xrange(len(self.jellyBeanBoxList)):
            if JELLY_BEAN_PICKER_HAS_EMPTY_BOX:
                if beanIndex:
                    box = self.jellyBeanBoxList[beanIndex]
                    box['text'] = TTLocalizer.BeanColorWords[beanIndex - 1]
            else:
                box = self.jellyBeanBoxList[beanIndex]
                box['text'] = TTLocalizer.BeanColorWords[beanIndex]

    def createJellyBeanBoxes(self):
        zCoord = 0
        xIncrement = 0.095
        xPos = 0
        maxBoxes = len(GardenGlobals.BeanColors)
        if JELLY_BEAN_PICKER_HAS_EMPTY_BOX:
            maxBoxes += 1
        startingXCoord = -0.1 * float(maxBoxes) / 2.0 + 0.075
        for activeBox in xrange(maxBoxes):
            xPos = xIncrement * activeBox + startingXCoord
            self.createJellyBeanBox(activeBox, xPos, zCoord)


class SpiffyBeanBox(DirectButton):

    def __init__(self, parent, index, **kw):
        self.boxIndex = index
        self.selectedIndex = 0
        optiondefs = ()
        self.defineoptions(kw, optiondefs)
        DirectButton.__init__(self, parent=parent)
        self.initialiseoptions(SpiffyBeanBox)
        self.selectedBean = self.attachNewNode('selectedBean')

    def getSelectedIndex(self):
        if hasattr(self, 'selectedIndex'):
            return self.selectedIndex
        return 0

    def setSelectedIndex(self, newIndex):
        self.selectedIndex = newIndex
        self.selectedBean.removeNode()
        self.selectedBean = self.attachNewNode('selectedBean')
        if newIndex:
            newIndex -= 1
            loadJellyBean(self.selectedBean, newIndex)


class PlantingGUI(DirectFrame):
    notify = DirectNotifyGlobal.directNotify.newCategory('PlantingGUI')

    def __init__(self, doneEvent, specialBoxActive = False):
        if specialBoxActive:
            instructions = TTLocalizer.GardeningChooseBeansItem
            instructionsPos = (0, 0.4)
        else:
            instructions = TTLocalizer.GardeningChooseBeans
            instructionsPos = (0, 0.35)
        DirectFrame.__init__(self, relief=None, state='normal', geom=DGG.getDefaultDialogGeom(), geom_color=ToontownGlobals.GlobalDialogColor, geom_scale=(1.5, 1.0, 1.0), frameSize=(-1, 1, -1, 1), pos=(0, 0, 0), text=instructions, text_wordwrap=20, text_scale=0.08, text_pos=instructionsPos)
        self.initialiseoptions(PlantingGUI)
        self.doneEvent = doneEvent
        buttons = loader.loadModel('phase_3/models/gui/dialog_box_buttons_gui')
        okImageList = (buttons.find('**/ChtBx_OKBtn_UP'), buttons.find('**/ChtBx_OKBtn_DN'), buttons.find('**/ChtBx_OKBtn_Rllvr'))
        cancelImageList = (buttons.find('**/CloseBtn_UP'), buttons.find('**/CloseBtn_DN'), buttons.find('**/CloseBtn_Rllvr'))
        resetImageList = (buttons.find('**/CloseBtn_UP'), buttons.find('**/CloseBtn_DN'), buttons.find('**/CloseBtn_Rllvr'))
        self.cancelButton = DirectButton(parent=self, relief=None, image=cancelImageList, pos=(-0.3, 0, -0.35), text=TTLocalizer.PlantingGuiCancel, text_scale=0.06, text_pos=(0, -0.1), command=self.__cancel)
        self.okButton = DirectButton(parent=self, relief=None, image=okImageList, pos=(0.3, 0, -0.35), text=TTLocalizer.PlantingGuiOk, text_scale=0.06, text_pos=(0, -0.1), command=self.__doPlant)
        self.resetButton = DirectButton(parent=self, relief=None, image=resetImageList, pos=(0.0, 0, -0.35), text=TTLocalizer.PlantingGuiReset, text_scale=0.06, text_pos=(0, -0.1), command=self.__reset)
        buttons.removeNode()
        self.availableBoxes = base.localAvatar.getBoxCapability()
        self.maxBoxes = GardenGlobals.getNumberOfShovelBoxes()
        self.activeBoxesList = []
        self.specialBox = None
        self.specialBoxActive = specialBoxActive
        self.boxList = []
        self.jellyBeanPicker = None
        self.jellyBeanPickerInterval = None
        self.createBoxes()
        guiItems = loader.loadModel('phase_5.5/models/gui/catalog_gui')
        self.beanBank = DirectLabel(self, relief=None, image=guiItems.find('**/bean_bank'), text=str(base.localAvatar.getMoney() + base.localAvatar.getBankMoney()), text_align=TextNode.ARight, text_scale=0.11, text_fg=(0.95, 0.95, 0, 1), text_shadow=(0, 0, 0, 1), text_pos=(0.75, -0.81), text_font=ToontownGlobals.getSignFont(), pos=(-0.85, 0, 0.2), scale=0.5)
        self.matchBoxesToAvailableMoney()
        if PICKER_ALWAYS_UP:
            self.spiffyBeanBoxClicked(0)
        return

    def destroy(self):
        if self.boxList:
            for box in self.boxList:
                box.destroy()

        self.boxList = []
        DirectFrame.destroy(self)
        self.doneEvent = None
        if self.jellyBeanPickerInterval:
            self.jellyBeanPickerInterval.finish()
        self.jellyBeanPickerInterval = None
        if self.jellyBeanPicker:
            self.jellyBeanPicker.destroy()
        self.jellyBeanPicker = None
        if hasattr(self, 'specialPhotoList') and self.specialPhotoList:
            for photo in self.specialPhotoList:
                photo.destroy()

            self.specialPhotoList = []
        return

    def __cancel(self):
        messenger.send(self.doneEvent, [0, '', -1])
        messenger.send('wakeup')

    def __reset(self):
        if self.jellyBeanPicker:
            self.jellyBeanPicker.destroy()
        for box in self.boxList:
            box.setSelectedIndex(0)

        self.beanBank['text'] = str(base.localAvatar.getMoney() + base.localAvatar.getBankMoney())
        self.matchBoxesToAvailableMoney()
        if PICKER_ALWAYS_UP:
            self.spiffyBeanBoxClicked(0)
        messenger.send('wakeup')

    def getRecipeStr(self):
        retval = ''
        for box in self.boxList:
            beanIndex = box.getSelectedIndex()
            if beanIndex:
                beanIndex -= 1
                beanLetter = GardenGlobals.BeanColorLetters[beanIndex]
                retval += beanLetter

        return retval

    def __doPlant(self):
        recipeStr = self.getRecipeStr()
        selectedSpecial = self.specialButton.getSelectedIndex()
        selectedSpecial -= 1
        messenger.send(self.doneEvent, [1, recipeStr, selectedSpecial])
        messenger.send('wakeup')

    def createBoxes(self):
        zCoord = -0.15
        xIncrement = 0.1
        xPos = 0
        startingXCoord = -0.1 * float(self.maxBoxes) / 2.0 + 0.05
        for activeBox in xrange(self.availableBoxes):
            xPos = xIncrement * activeBox + startingXCoord
            if USE_SCROLLING_BEAN_BOX:
                self.createScrollingBeanBox(activeBox, xPos, zCoord, True)
            else:
                self.createSpiffyBeanBox(activeBox, xPos, zCoord, True)

        for disabledBox in xrange(self.availableBoxes, self.maxBoxes):
            xPos = xIncrement * disabledBox + startingXCoord
            if USE_SCROLLING_BEAN_BOX:
                self.createScrollingBeanBox(disabledBox, xPos, zCoord, False)
            else:
                self.createSpiffyBeanBox(disabledBox, xPos, zCoord, False)

        xPos += xIncrement * 2
        self.createSpecialBox(xPos, zCoord)

    def createSpecialBox(self, xPos, zPos):
        geomColor = (1, 1, 1, 1)
        if not self.specialBoxActive:
            geomColor = (0.5, 0.5, 0.5, 1)
        geomScaleX = 0.2
        geomScaleZ = 0.2
        self.specialButtonFrame = DirectFrame(parent=self, pos=(xPos, 0, zPos), geom=DGG.getDefaultDialogGeom(), geom_scale=(geomScaleX, 1.0, geomScaleZ), geom_color=geomColor, relief=None)
        items = []
        if self.specialBoxActive:
            gardenSpecials = base.localAvatar.getGardenSpecials()
            tempItem = BoxItem(self.attachNewNode('blankSpecial'))
            items.append(tempItem)
            self.specialPhotoList = []
            for item in gardenSpecials:
                tempItem = BoxItem(self.specialButtonFrame.attachNewNode('temp1'))
                specialsPhoto = SpecialsPhoto.SpecialsPhoto(item[0], parent=tempItem)
                specialsPhoto.setBackBounds(-geomScaleX / 2.0, geomScaleX / 2.0, -geomScaleZ / 2.0, geomScaleZ / 2.0)
                specialsPhoto.setBackColor(1.0, 1.0, 1.0, 1.0)
                items.append(tempItem)
                self.specialPhotoList.append(specialsPhoto)

        self.specialButton = GenericBoxScrollList(self.specialButtonFrame, items, incButton_pos=(0, 0, -0.135), incButton_scale=(0.75, 1.0, -1.0), decButton_pos=(0, 0, 0.135), decButton_scale=(0.75, 1.0, 1.0), command=self.photoSpecialChanged)
        return

    def photoSpecialChanged(self):
        if not hasattr(self, 'specialButton'):
            return
        messenger.send('wakeup')
        selectedSpecial = self.specialButton.getSelectedIndex()
        if selectedSpecial == 0:
            for photo in self.specialPhotoList:
                photo.hide()

        else:
            selectedSpecial -= 1
            self.specialPhotoList[selectedSpecial].show()

    def showFirstSpecial(self):
        if len(self.specialButton['items']) > 1:
            self.specialButton.scrollTo(self.specialButton.getSelectedIndex() + 1)

    def createScrollingBeanBox(self, index, xPos, zPos, active):
        geomColor = (1, 1, 1, 1)
        if not active:
            geomColor = (0.5, 0.5, 0.5, 1)
        boxFrame = DirectFrame(parent=self, pos=(xPos, 0, zPos), geom=DGG.getDefaultDialogGeom(), geom_scale=(0.1, 1.0, 0.1), geom_color=geomColor, relief=None)
        items = []
        if active:
            tempItem = BoxItem(self.attachNewNode('emptyBean'))
            items.append(tempItem)
            for curBean in xrange(len(GardenGlobals.BeanColors)):
                tempItem = BoxItem(self.attachNewNode('bean-%d-%d' % (index, curBean)))
                loadJellyBean(tempItem, curBean)
                items.append(tempItem)

        box = GenericBoxScrollList(boxFrame, items, incButton_pos=(0, 0, -0.07), incButton_scale=(0.4, 1.0, -1.0), decButton_pos=(0, 0, 0.065), decButton_scale=(0.4, 1.0, 1.0))
        self.boxList.append(box)
        return

    def spiffyBeanBoxClicked(self, index):
        if self.jellyBeanPicker:
            self.jellyBeanPicker.destroy()
        outOfMoney = int(self.beanBank['text']) <= 0
        if self.boxList[index].getSelectedIndex() == 0 and outOfMoney:
            return
        if not CAN_CHANGE_BEAN_COLOR and self.boxList[index].getSelectedIndex():
            return
        self.jellyBeanPicker = JellyBeanPicker(self, self.selectedNewBeanColor, index)
        self.jellyBeanPicker.setPos(*JellyBeanPickerEndPos)
        if DO_PICKER_INTERVAL:
            boxPos = self.boxList[index].getPos()
            self.jellyBeanPicker.setPos(boxPos)
            self.jellyBeanPickerInterval = Sequence(Parallel(self.jellyBeanPicker.posInterval(duration=0.3, pos=VBase3(*JellyBeanPickerEndPos)), self.jellyBeanPicker.scaleInterval(duration=0.3, startScale=VBase3(JellyBeanPickerScale[0] / 10.0, JellyBeanPickerScale[1], JellyBeanPickerScale[2]), scale=VBase3(*JellyBeanPickerScale))), Func(self.jellyBeanPicker.setColorText))
            self.jellyBeanPickerInterval.start()
        else:
            self.jellyBeanPicker.setColorText()

    def matchBoxesToAvailableMoney(self):
        outOfMoney = int(self.beanBank['text']) <= 0
        if outOfMoney:
            for box in self.boxList:
                if box.getSelectedIndex() == 0:
                    box['state'] = DGG.DISABLED
                    box.setState()
                    box.setColorScale(0.5, 0.5, 0.5, 1)

        else:
            for box in self.boxList:
                box['state'] = DGG.NORMAL
                box.setState()
                box.setColorScale(1, 1, 1, 1)

        if FORCE_LEFT_TO_RIGHT:
            boxIndexToEnable = len(self.getRecipeStr())
            for i in xrange(0, boxIndexToEnable):
                box = self.boxList[i]
                if ONLY_ONE_SPIFFY_BOX_CAN_BE_CLICKED:
                    box['state'] = DGG.DISABLED
                    box.setState()
                    box.setColorScale(0.875, 0.875, 0.875, 1)

            if boxIndexToEnable < self.maxBoxes:
                box = self.boxList[boxIndexToEnable]
                if boxIndexToEnable >= self.availableBoxes:
                    box['state'] = DGG.DISABLED
                    box.setState()
                    box.setColorScale(0.5, 0.5, 0.5, 1)
            for i in xrange(boxIndexToEnable + 1, len(self.boxList)):
                box = self.boxList[i]
                box['state'] = DGG.DISABLED
                box.setState()
                box.setColorScale(0.5, 0.5, 0.5, 1)

    def selectedNewBeanColor(self, boxPosition, beanIndex):
        self.boxList[boxPosition].setSelectedIndex(beanIndex)
        if self.jellyBeanPicker:
            self.jellyBeanPicker.destroy()
            self.jellyBeanPicker = None
        cost = len(self.getRecipeStr())
        newMoney = base.localAvatar.getMoney() + base.localAvatar.getBankMoney() - cost
        self.beanBank['text'] = str(newMoney)
        self.matchBoxesToAvailableMoney()
        messenger.send('wakeup')
        if PICKER_ALWAYS_UP:
            numBoxesFilled = len(self.getRecipeStr())
            if numBoxesFilled < self.availableBoxes:
                self.spiffyBeanBoxClicked(numBoxesFilled)
        return

    def createSpiffyBeanBox(self, index, xPos, zPos, active):
        geomColor = (1, 1, 1, 1)
        state = DGG.NORMAL
        command = self.spiffyBeanBoxClicked
        if not active:
            geomColor = (0.5, 0.5, 0.5, 1)
            command = None
            state = DGG.DISABLED
        newBox = SpiffyBeanBox(index=index, parent=self, pos=(xPos, 0, zPos), geom=DGG.getDefaultDialogGeom(), geom_scale=(0.1, 1.0, 0.1), geom_color=geomColor, relief=None, state=state, command=command, extraArgs=[index])
        self.boxList.append(newBox)
        return