Poodletooth-iLand/toontown/estate/DistributedClosetAI.py

from toontown.estate.DistributedFurnitureItemAI import DistributedFurnitureItemAI
from toontown.toon.ToonDNA import ToonDNA
from direct.distributed.ClockDelta import globalClockDelta
import ClosetGlobals


class DistributedClosetAI(DistributedFurnitureItemAI):
    notify = directNotify.newCategory('DistributedClosetAI')

    def __init__(self, air, furnitureMgr, itemType):
        DistributedFurnitureItemAI.__init__(self, air, furnitureMgr, itemType)

        self.avId = None
        self.customerDNA = None
        self.topList = []
        self.botList = []
        self.gender = 'm'
        self.removedBottoms = []
        self.removedTops = []

    def generate(self):
        if self.furnitureMgr.ownerId:
            owner = self.air.doId2do.get(self.furnitureMgr.ownerId)
            if owner:
                self.topList = owner.clothesTopsList
                self.botList = owner.clothesBottomsList
                self.gender = owner.dna.gender
            else:
                self.air.dbInterface.queryObject(self.air.dbId, self.furnitureMgr.ownerId, self.__gotOwner)

    def __gotOwner(self, dclass, fields):
        if dclass != self.air.dclassesByName['DistributedToonAI']:
            return
        self.botList = fields['setClothesBottomsList'][0]
        self.topList = fields['setClothesTopsList'][0]
        dna = ToonDNA(str=fields['setDNAString'][0])
        self.gender = dna.gender

    def getOwnerId(self):
        return self.furnitureMgr.ownerId

    def __verifyAvatarInMyZone(self, av):
        return av.getLocation() == self.getLocation()

    def enterAvatar(self):
        avId = self.air.getAvatarIdFromSender()
        if self.avId:
            if self.avId == avId:
                self.air.writeServerEvent('suspicious', avId, 'Tried to use closet twice!')
            self.sendUpdateToAvatarId(avId, 'freeAvatar', [])
            return
        av = self.air.doId2do.get(avId)
        if not av:
            self.air.writeServerEvent('suspicious', avId, 'Not in same shard as closet!')
            return
        if not self.__verifyAvatarInMyZone(av):
            self.air.writeServerEvent('suspicious', avId, 'Not in same zone as closet!')
            return
        self.customerDNA = av.dna
        self.avId = avId
        self.d_setState(ClosetGlobals.OPEN, avId, self.furnitureMgr.ownerId, self.gender, self.topList, self.botList)

    def removeItem(self, item, topOrBottom):
        avId = self.air.getAvatarIdFromSender()
        if avId != self.furnitureMgr.ownerId:
            self.air.writeServerEvent('suspicious', avId, 'Tried to remove item from someone else\'s closet!')
            return
        if avId != self.avId:
            self.air.writeServerEvent('suspicious', avId, 'Tried to remove item while not interacting with closet!')
            return
        av = self.air.doId2do.get(avId)
        if not av:
            self.air.writeServerEvent('suspicious', avId, 'Tried to interact with a closet from another shard!')
            return
        tempDna = ToonDNA()
        if not tempDna.isValidNetString(item):
            self.air.writeServerEvent('suspicious', avId, 'Sent an invalid DNA string!')
            return
        tempDna.makeFromNetString(item)
        if topOrBottom == ClosetGlobals.SHIRT:
            self.removedTops.append([tempDna.topTex, tempDna.topTexColor, tempDna.sleeveTex, tempDna.sleeveTexColor])
        elif topOrBottom == ClosetGlobals.SHORTS:
            self.removedBottoms.append([tempDna.botTex, tempDna.botTexColor])
        else:
            self.air.writeServerEvent('suspicious', avId, 'Set an invalid topOrBottom value!')
            return

    def __checkValidDNAChange(self, av, testDNA):
        if testDNA.head != av.dna.head:
            return False
        if testDNA.torso != av.dna.torso:
            if av.dna.gender == 'm':
                return False
            elif testDNA.torso[0] != av.dna.torso[0]:
                return False
        if testDNA.legs != av.dna.legs:
            return False
        if testDNA.gender != av.dna.gender:
            return False
        if testDNA.armColor != av.dna.armColor:
            return False
        if testDNA.gloveColor != av.dna.gloveColor:
            return False
        if testDNA.legColor != av.dna.legColor:
            return False
        if testDNA.headColor != av.dna.headColor:
            return False
        return True

    def setDNA(self, dnaString, finished, whichItem):
        avId = self.air.getAvatarIdFromSender()
        if avId != self.avId:
            self.air.writeServerEvent('suspicious', avId, 'Tried to set DNA from closet while not using it!')
            return
        av = self.air.doId2do.get(avId)
        if not av:
            self.air.writeServerEvent('suspicious', avId, 'Interacted with a closet from another shard!')
            return
        if not self.__verifyAvatarInMyZone(av):
            self.air.writeServerEvent('suspicious', avId, 'Tried to setDNA while in another zone!')
            return
        testDna = ToonDNA()
        if not testDna.isValidNetString(dnaString):
            self.air.writeServerEvent('suspicious', avId, 'Tried to set invalid DNA at a closet!')
            return
        if not finished:
            testDna.makeFromNetString(dnaString)
            if not self.__checkValidDNAChange(av, testDna):
                self.air.writeServerEvent('suspicious', avId, 'Tried to change their DNA temporarily!')
                return
            self.sendUpdate('setCustomerDNA', [avId, dnaString])
            return
        elif finished == 1:
            self.d_setMovie(ClosetGlobals.CLOSET_MOVIE_COMPLETE, avId, globalClockDelta.getRealNetworkTime())
            self.resetMovie()
            self.d_setState(ClosetGlobals.CLOSED, 0, self.furnitureMgr.ownerId, self.gender, self.topList, self.botList)
            av.b_setDNAString(self.customerDNA.makeNetString())
            self.removedBottoms = []
            self.removedTops = []
            self.customerDNA = None
            self.avId = None
        elif finished == 2:
            if avId != self.furnitureMgr.ownerId:
                self.air.writeServerEvent('suspicious', avId, 'Tried to set their clothes from somebody else\'s closet!')
                return
            testDna.makeFromNetString(dnaString)
            if whichItem & ClosetGlobals.SHIRT:
                success = av.replaceItemInClothesTopsList(testDna.topTex, testDna.topTexColor, testDna.sleeveTex, testDna.sleeveTexColor, self.customerDNA.topTex, self.customerDNA.topTexColor, self.customerDNA.sleeveTex, self.customerDNA.sleeveTexColor)
                if success:
                    self.customerDNA.topTex = testDna.topTex
                    self.customerDNA.topTexColor = testDna.topTexColor
                    self.customerDNA.sleeveTex = testDna.sleeveTex
                    self.customerDNA.sleeveTexColor = testDna.sleeveTexColor
                else:
                    self.air.writeServerEvent('suspicious', avId, 'Tried to set their shirt to a shirt they don\'t own!')
            if whichItem & ClosetGlobals.SHORTS:
                success = av.replaceItemInClothesBottomsList(testDna.botTex, testDna.botTexColor, self.customerDNA.botTex, self.customerDNA.botTexColor)
                if success:
                    self.customerDNA.botTex = testDna.botTex
                    self.customerDNA.botTexColor = testDna.botTexColor
                    if self.customerDNA.torso != testDna.torso:
                        if self.customerDNA.gender == 'm':
                            self.air.writeServerEvent('suspicious', avId, 'Tried to change their torso size!')
                            return
                        elif self.customerDNA.torso[0] != testDna.torso[0]:
                            self.air.writeServerEvent('suspicious', avId, 'Tried to change their torso size!')
                            return
                    self.customerDNA.torso = testDna.torso
                else:
                    self.air.writeServerEvent('suspicious', avId, 'Tried to set their shorts to a pair they don\'t own!')
            for bottom in self.removedBottoms:
                botTex, botTexColor = bottom
                success = av.removeItemInClothesBottomsList(botTex, botTexColor)
                if not success:
                    self.air.writeServerEvent('suspicious', avId, 'Tried to remove a bottom they didn\'t have!')
            for top in self.removedTops:
                topTex, topTexColor, sleeveTex, sleeveTexColor = top
                success = av.removeItemInClothesTopsList(topTex, topTexColor, sleeveTex, sleeveTexColor)
                if not success:
                    self.air.writeServerEvent('suspicious', avId, 'Tried to remove a top they didn\'t have!')
            av.b_setDNAString(self.customerDNA.makeNetString())
            av.b_setClothesTopsList(av.getClothesTopsList())
            av.b_setClothesBottomsList(av.getClothesBottomsList())
            self.topList = av.getClothesTopsList()
            self.botList = av.getClothesBottomsList()
            self.removedBottoms = []
            self.removedTops = []
            self.d_setMovie(ClosetGlobals.CLOSET_MOVIE_COMPLETE, avId, globalClockDelta.getRealNetworkTime())
            self.resetMovie()
            self.d_setState(ClosetGlobals.CLOSED, 0, self.furnitureMgr.ownerId, self.gender, self.topList, self.botList)
            self.customerDNA = None
            self.avId = None

    def setState(self, todo0, todo1, todo2, todo3, todo4, todo5):
        pass

    def d_setState(self, mode, avId, ownerId, gender, topList, botList):
        self.sendUpdate('setState', [mode, avId, ownerId, gender, topList, botList])

    def d_setMovie(self, movie, avId, time):
        self.sendUpdate('setMovie', [movie, avId, time])

    def resetMovie(self):
        taskMgr.doMethodLater(1, self.d_setMovie, 'resetMovie-%d' % self.getDoId(), extraArgs=[ClosetGlobals.CLOSET_MOVIE_CLEAR, 0, globalClockDelta.getRealNetworkTime()])

    def setMovie(self, todo0, todo1, todo2):
        pass

    def setCustomerDNA(self, todo0, todo1):
        pass