from direct.interval.IntervalGlobal import *
from panda3d.core import *
from toontown.toonbase import TTLocalizer
from toontown.toonbase import ToontownBattleGlobals
from toontown.battle import BattleParticles
from direct.particles import Particles
import random
TTBG = ToontownBattleGlobals
TTL = TTLocalizer
EFFECT_RADIUS = 30
RESISTANCE_TOONUP = 0
RESISTANCE_RESTOCK = 1
RESISTANCE_MONEY = 2
resistanceMenu = [RESISTANCE_TOONUP, RESISTANCE_RESTOCK, RESISTANCE_MONEY]
resistanceDict = {RESISTANCE_TOONUP: {'menuName': TTL.ResistanceToonupMenu,
                     'itemText': TTL.ResistanceToonupItem,
                     'chatText': TTL.ResistanceToonupChat,
                     'values': [10,
                                20,
                                40,
                                80,
                                -1],
                     'items': [0,
                               1,
                               2,
                               3,
                               4]},
 RESISTANCE_MONEY: {'menuName': TTL.ResistanceMoneyMenu,
                    'itemText': TTL.ResistanceMoneyItem,
                    'chatText': TTL.ResistanceMoneyChat,
                    'values': [100,
                               200,
                               350,
                               600],
                    'items': [0,
                              1,
                              2,
                              3]},
 RESISTANCE_RESTOCK: {'menuName': TTL.ResistanceRestockMenu,
                      'itemText': TTL.ResistanceRestockItem,
                      'chatText': TTL.ResistanceRestockChat,
                      'values': [TTBG.HEAL_TRACK,
                                 TTBG.TRAP_TRACK,
                                 TTBG.LURE_TRACK,
                                 TTBG.SOUND_TRACK,
                                 TTBG.THROW_TRACK,
                                 TTBG.SQUIRT_TRACK,
                                 TTBG.DROP_TRACK,
                                 -1],
                      'extra': [TTL.MovieNPCSOSHeal,
                                TTL.MovieNPCSOSTrap,
                                TTL.MovieNPCSOSLure,
                                TTL.MovieNPCSOSSound,
                                TTL.MovieNPCSOSThrow,
                                TTL.MovieNPCSOSSquirt,
                                TTL.MovieNPCSOSDrop,
                                TTL.MovieNPCSOSAll],
                      'items': [0,
                                1,
                                2,
                                3,
                                4,
                                5,
                                6,
                                7]}}

def encodeId(menuIndex, itemIndex):
    textId = menuIndex * 100
    textId += resistanceDict[menuIndex]['items'][itemIndex]
    return textId


def decodeId(textId):
    menuIndex = int(textId / 100)
    itemIndex = textId % 100
    return (menuIndex, itemIndex)


def validateId(textId):
    if textId < 0:
        return 0
    menuIndex, itemIndex = decodeId(textId)
    if menuIndex not in resistanceDict:
        return 0
    if itemIndex >= len(resistanceDict[menuIndex]['values']):
        return 0
    return 1


def getItems(menuIndex):
    return resistanceDict[menuIndex]['items']


def getMenuName(textId):
    menuIndex, itemIndex = decodeId(textId)
    return resistanceDict[menuIndex]['menuName']


def getItemText(textId):
    menuIndex, itemIndex = decodeId(textId)
    value = resistanceDict[menuIndex]['values'][itemIndex]
    text = resistanceDict[menuIndex]['itemText']
    if menuIndex is RESISTANCE_TOONUP:
        if value == -1:
            value = TTL.ResistanceToonupItemMax
    elif menuIndex is RESISTANCE_RESTOCK:
        value = resistanceDict[menuIndex]['extra'][itemIndex]
    return text % str(value)


def getChatText(textId):
    menuIndex, itemIndex = decodeId(textId)
    return resistanceDict[menuIndex]['chatText']


def getItemValue(textId):
    menuIndex, itemIndex = decodeId(textId)
    return resistanceDict[menuIndex]['values'][itemIndex]


def getRandomId():
    menuIndex = random.choice(resistanceMenu)
    itemIndex = random.choice(getItems(menuIndex))
    return encodeId(menuIndex, itemIndex)


def doEffect(textId, speakingToon, nearbyToons):
    menuIndex, itemIndex = decodeId(textId)
    itemValue = getItemValue(textId)
    if menuIndex == RESISTANCE_TOONUP:
        effect = BattleParticles.loadParticleFile('resistanceEffectSparkle.ptf')
        fadeColor = VBase4(1, 0.5, 1, 1)
    elif menuIndex == RESISTANCE_MONEY:
        effect = BattleParticles.loadParticleFile('resistanceEffectBean.ptf')
        bean = loader.loadModel('phase_4/models/props/jellybean4.bam')
        bean = bean.find('**/jellybean')
        colors = {'particles-1': (1, 1, 0, 1),
         'particles-2': (1, 0, 0, 1),
         'particles-3': (0, 1, 0, 1),
         'particles-4': (0, 0, 1, 1),
         'particles-5': (1, 0, 1, 1)}
        for name, color in list(colors.items()):
            node = bean.copyTo(NodePath())
            node.setColorScale(*color)
            p = effect.getParticlesNamed(name)
            p.renderer.setGeomNode(node.node())

        fadeColor = VBase4(0, 1, 0, 1)
    elif menuIndex == RESISTANCE_RESTOCK:
        effect = BattleParticles.loadParticleFile('resistanceEffectSprite.ptf')
        invModel = loader.loadModel('phase_3.5/models/gui/inventory_icons')
        invModel.setScale(4)
        invModel.flattenLight()
        icons = []
        if itemValue != -1:
            for item in range(6):
                iconName = ToontownBattleGlobals.AvPropsNew[itemValue][item]
                icons.append(invModel.find('**/%s' % iconName))

        else:
            tracks = list(range(7))
            random.shuffle(tracks)
            for i in range(6):
                track = tracks[i]
                item = random.randint(0, 5)
                iconName = ToontownBattleGlobals.AvPropsNew[track][item]
                icons.append(invModel.find('**/%s' % iconName))

        iconDict = {'particles-1': icons[0],
         'particles-2': icons[1],
         'particles-3': icons[2],
         'particles-4': icons[3],
         'particles-5': icons[4],
         'particles-6': icons[5]}
        for name, icon in list(iconDict.items()):
            p = effect.getParticlesNamed(name)
            p.renderer.setFromNode(icon)

        fadeColor = VBase4(0, 0, 1, 1)
    else:
        return
    recolorToons = Parallel()
    for toonId in nearbyToons:
        toon = base.cr.doId2do.get(toonId)
        if toon and not toon.ghostMode:
            i = Sequence(toon.doToonColorScale(fadeColor, 0.3), toon.doToonColorScale(toon.defaultColorScale, 0.3), Func(toon.restoreDefaultColorScale))
            recolorToons.append(i)

    i = Parallel(ParticleInterval(effect, speakingToon, worldRelative=0, duration=3, cleanup=True), Sequence(Wait(0.2), recolorToons), autoFinish=1)
    i.start()