from pandac.PandaModules import * from direct.task.Task import Task from toontown.toonbase.ToontownGlobals import * from direct.gui.DirectGui import * from pandac.PandaModules import * from direct.showbase import DirectObject from direct.fsm import ClassicFSM, State from direct.fsm import State from direct.directnotify import DirectNotifyGlobal from toontown.toonbase import TTLocalizer from toontown.toon import ToonTeleportPanel from toontown.suit import Suit from toontown.pets import Pet from otp.otpbase import OTPLocalizer from otp.otpbase import OTPGlobals globalFriendInviter = None def showFriendInviter(avId, avName, avDisableName): global globalFriendInviter if globalFriendInviter != None: globalFriendInviter.cleanup() globalFriendInviter = None globalFriendInviter = FriendInviter(avId, avName, avDisableName) return def hideFriendInviter(): global globalFriendInviter if globalFriendInviter != None: globalFriendInviter.cleanup() globalFriendInviter = None return def unloadFriendInviter(): global globalFriendInviter if globalFriendInviter != None: globalFriendInviter.cleanup() globalFriendInviter = None return class FriendInviter(DirectFrame): notify = DirectNotifyGlobal.directNotify.newCategory('FriendInviter') def __init__(self, avId, avName, avDisableName): DirectFrame.__init__(self, pos=(-1.033, 0.1, -0.35), parent=base.a2dTopRight, image_color=GlobalDialogColor, image_scale=(1.0, 1.0, 0.6), text='', text_wordwrap=TTLocalizer.FIdirectFrameWordwrap, text_scale=TTLocalizer.FIdirectFrame, text_pos=TTLocalizer.FIdirectFramePos) self['image'] = DGG.getDefaultDialogGeom() self.avId = avId self.toonName = avName avatar = base.cr.doId2do.get(self.avId) self.avDisableName = avDisableName self.fsm = ClassicFSM.ClassicFSM('FriendInviter', [State.State('off', self.enterOff, self.exitOff), State.State('getNewFriend', self.enterGetNewFriend, self.exitGetNewFriend), State.State('check', self.enterCheck, self.exitCheck), State.State('tooMany', self.enterTooMany, self.exitTooMany), State.State('checkAvailability', self.enterCheckAvailability, self.exitCheckAvailability), State.State('notAvailable', self.enterNotAvailable, self.exitNotAvailable), State.State('notAcceptingFriends', self.enterNotAcceptingFriends, self.exitNotAcceptingFriends), State.State('wentAway', self.enterWentAway, self.exitWentAway), State.State('already', self.enterAlready, self.exitAlready), State.State('askingCog', self.enterAskingCog, self.exitAskingCog), State.State('askingPet', self.enterAskingPet, self.exitAskingPet), State.State('endFriendship', self.enterEndFriendship, self.exitEndFriendship), State.State('friendsNoMore', self.enterFriendsNoMore, self.exitFriendsNoMore), State.State('self', self.enterSelf, self.exitSelf), State.State('ignored', self.enterIgnored, self.exitIgnored), State.State('asking', self.enterAsking, self.exitAsking), State.State('yes', self.enterYes, self.exitYes), State.State('no', self.enterNo, self.exitNo), State.State('otherTooMany', self.enterOtherTooMany, self.exitOtherTooMany), State.State('maybe', self.enterMaybe, self.exitMaybe), State.State('down', self.enterDown, self.exitDown), State.State('cancel', self.enterCancel, self.exitCancel)], 'off', 'off') self.context = None from toontown.toon import ToonAvatarDetailPanel ToonTeleportPanel.hideTeleportPanel() ToonAvatarDetailPanel.hideAvatarDetail() buttons = loader.loadModel('phase_3/models/gui/dialog_box_buttons_gui') gui = loader.loadModel('phase_3.5/models/gui/avatar_panel_gui') self.bOk = DirectButton(self, image=(buttons.find('**/ChtBx_OKBtn_UP'), buttons.find('**/ChtBx_OKBtn_DN'), buttons.find('**/ChtBx_OKBtn_Rllvr')), relief=None, text=OTPLocalizer.FriendInviterOK, text_scale=0.05, text_pos=(0.0, -0.1), pos=(0.0, 0.0, -0.1), command=self.__handleOk) self.bOk.hide() self.bCancel = DirectButton(self, image=(buttons.find('**/CloseBtn_UP'), buttons.find('**/CloseBtn_DN'), buttons.find('**/CloseBtn_Rllvr')), relief=None, text=OTPLocalizer.FriendInviterCancel, text_scale=0.05, text_pos=(0.0, -0.1), pos=TTLocalizer.FIbCancelPos, command=self.__handleCancel) self.bCancel.hide() self.bStop = DirectButton(self, image=(gui.find('**/Ignore_Btn_UP'), gui.find('**/Ignore_Btn_DN'), gui.find('**/Ignore_Btn_RLVR')), relief=None, text=OTPLocalizer.FriendInviterStopBeingFriends, text_align=TextNode.ALeft, text_scale=TTLocalizer.FIbStop, text_pos=TTLocalizer.FIbStopTextPos, pos=TTLocalizer.FIbStopPos, command=self.__handleStop) self.bStop.hide() self.bYes = DirectButton(self, image=(buttons.find('**/ChtBx_OKBtn_UP'), buttons.find('**/ChtBx_OKBtn_DN'), buttons.find('**/ChtBx_OKBtn_Rllvr')), relief=None, text=OTPLocalizer.FriendInviterYes, text_scale=0.05, text_pos=(0.0, -0.1), pos=TTLocalizer.FIbYesPos, command=self.__handleYes) self.bYes.hide() self.bNo = DirectButton(self, image=(buttons.find('**/CloseBtn_UP'), buttons.find('**/CloseBtn_DN'), buttons.find('**/CloseBtn_Rllvr')), relief=None, text=OTPLocalizer.FriendInviterNo, text_scale=0.05, text_pos=(0.0, -0.1), pos=(0.15, 0.0, -0.1), command=self.__handleNo) self.bNo.hide() buttons.removeNode() gui.removeNode() self.fsm.enterInitialState() if self.avId == None: self.fsm.request('getNewFriend') else: self.fsm.request('check') return def cleanup(self): self.fsm.request('cancel') del self.fsm self.destroy() def getName(self): name = self.toonName if name == None: name = TTLocalizer.FriendInviterThatToon return name def enterOff(self): pass def exitOff(self): pass def enterGetNewFriend(self): self['text'] = TTLocalizer.FriendInviterClickToon % len(base.localAvatar.friendsList) self.bCancel.show() self.accept('clickedNametag', self.__handleClickedNametag) def exitGetNewFriend(self): self.bCancel.hide() self.ignore('clickedNametag') def __handleClickedNametag(self, avatar): self.avId = avatar.doId self.toonName = avatar.getName() self.avDisableName = avatar.uniqueName('disable') self.fsm.request('check') def enterCheck(self): myId = base.localAvatar.doId self.accept(self.avDisableName, self.__handleDisableAvatar) if self.avId == myId: self.fsm.request('self') elif base.cr.isFriend(self.avId): self.fsm.request('already') else: if len(base.localAvatar.friendsList) >= MaxFriends: self.fsm.request('tooMany') else: self.fsm.request('checkAvailability') def exitCheck(self): self.ignore(self.avDisableName) def enterTooMany(self): text = OTPLocalizer.FriendInviterToonTooMany name = self.toonName self['text'] = text % name self.bCancel.show() self.bCancel.setPos(0.0, 0.0, -0.16) def exitTooMany(self): self.bCancel.hide() def enterCheckAvailability(self): self.accept(self.avDisableName, self.__handleDisableAvatar) if base.localAvatar.hasPet() and base.localAvatar.getPetId() == self.avId: self.fsm.request('askingPet') return if self.avId not in base.cr.doId2do: self.fsm.request('wentAway') return else: avatar = base.cr.doId2do.get(self.avId) if isinstance(avatar, Suit.Suit): self.fsm.request('askingCog') return if isinstance(avatar, Pet.Pet): self.fsm.request('askingPet') return if not base.cr.friendManager: self.notify.warning('No FriendManager available.') self.fsm.request('down') return base.cr.friendManager.up_friendQuery(self.avId) self['text'] = OTPLocalizer.FriendInviterCheckAvailability % self.toonName self.accept('friendResponse', self.__friendResponse) self.bCancel.show() self.accept('friendConsidering', self.__friendConsidering) def exitCheckAvailability(self): self.ignore(self.avDisableName) self.ignore('friendConsidering') self.ignore('friendResponse') self.bCancel.hide() def enterNotAvailable(self): self['text'] = OTPLocalizer.FriendInviterNotAvailable % self.getName() self.context = None self.bOk.show() return def exitNotAvailable(self): self.bOk.hide() def enterNotAcceptingFriends(self): self['text'] = OTPLocalizer.FriendInviterFriendSaidNoNewFriends % self.getName() self.context = None self.bOk.show() return def exitNotAcceptingFriends(self): self.bOk.hide() def enterWentAway(self): self['text'] = OTPLocalizer.FriendInviterWentAway % self.getName() if self.context != None: base.cr.friendManager.up_cancelFriendQuery(self.context) self.context = None self.bOk.show() return def exitWentAway(self): self.bOk.hide() def enterAlready(self): self['text'] = TTLocalizer.FriendInviterToonAlready % self.getName() self.bStop['text'] = TTLocalizer.FriendInviterStopBeingToonFriends self.context = None self.bStop.show() self.bCancel.show() return def exitAlready(self): self['text'] = '' self.bStop.hide() self.bCancel.hide() def enterAskingCog(self): self['text'] = OTPLocalizer.FriendInviterAskingCog % self.getName() taskMgr.doMethodLater(2.0, self.cogReplies, 'cogFriendship') self.bCancel.show() def exitAskingCog(self): taskMgr.remove('cogFriendship') self.bCancel.hide() def cogReplies(self, task): self.fsm.request('no') return Task.done def enterAskingPet(self): if base.localAvatar.hasPet() and base.localAvatar.getPetId() == self.avId: self['text'] = OTPLocalizer.FriendInviterAskingMyPet % self.getName() else: self['text'] = OTPLocalizer.FriendInviterAskingPet % self.getName() self.context = None self.bOk.show() def exitAskingPet(self): self.bOk.hide() def enterEndFriendship(self): self['text'] = TTLocalizer.FriendInviterEndFriendshipToon % self.getName() self.context = None self.bYes.show() self.bNo.show() return def exitEndFriendship(self): self.bYes.hide() self.bNo.hide() def enterFriendsNoMore(self): base.cr.removeFriend(self.avId) self['text'] = OTPLocalizer.FriendInviterFriendsNoMore % self.getName() self.bOk.show() if self.avId not in base.cr.doId2do: messenger.send(self.avDisableName) def exitFriendsNoMore(self): self.bOk.hide() def enterSelf(self): self['text'] = OTPLocalizer.FriendInviterSelf self.context = None self.bOk.show() return def exitSelf(self): self.bOk.hide() def enterIgnored(self): self['text'] = OTPLocalizer.FriendInviterIgnored % self.toonName self.context = None self.bOk.show() return def exitIgnored(self): self.bOk.hide() def enterAsking(self): self.accept(self.avDisableName, self.__handleDisableAvatar) self['text'] = OTPLocalizer.FriendInviterAsking % self.toonName self.accept('friendResponse', self.__friendResponse) self.bCancel.show() def exitAsking(self): self.ignore(self.avDisableName) self.ignore('friendResponse') self.bCancel.hide() def enterYes(self): self['text'] = OTPLocalizer.FriendInviterFriendSaidYes % self.toonName self.context = None self.bOk.show() return def exitYes(self): self.bOk.hide() def enterNo(self): self['text'] = OTPLocalizer.FriendInviterFriendSaidNo % self.toonName self.context = None self.bOk.show() return def exitNo(self): self.bOk.hide() def enterOtherTooMany(self): self['text'] = OTPLocalizer.FriendInviterOtherTooMany % self.toonName self.context = None self.bOk.show() return def exitOtherTooMany(self): self.bOk.hide() def enterMaybe(self): self['text'] = OTPLocalizer.FriendInviterMaybe % self.toonName self.context = None self.bOk.show() return def exitMaybe(self): self.bOk.hide() def enterDown(self): self['text'] = OTPLocalizer.FriendInviterDown self.context = None self.bOk.show() return def exitDown(self): self.bOk.hide() def enterCancel(self): if self.context != None: base.cr.friendManager.up_cancelFriendQuery(self.context) self.context = None self.fsm.request('off') return def exitCancel(self): pass def __handleOk(self): if base.config.GetBool('want-qa-regression', 0): self.notify.info('QA-REGRESSION: MAKEAFRIENDSHIP: Make a friendship') unloadFriendInviter() def __handleCancel(self): unloadFriendInviter() def __handleStop(self): if base.config.GetBool('want-qa-regression', 0): self.notify.info('QA-REGRESSION: BREAKAFRIENDSHIP: Break a friendship') self.fsm.request('endFriendship') def __handleYes(self): if self.fsm.getCurrentState().getName() == 'endFriendship': self.fsm.request('friendsNoMore') else: unloadFriendInviter() def __handleNo(self): unloadFriendInviter() def __handleList(self): messenger.send('openFriendsList') def __friendConsidering(self, yesNoAlready, context): if yesNoAlready == 1: self.context = context self.fsm.request('asking') elif yesNoAlready == 0: self.fsm.request('notAvailable') elif yesNoAlready == 2: self.fsm.request('already') elif yesNoAlready == 3: self.fsm.request('self') elif yesNoAlready == 4: self.fsm.request('ignored') elif yesNoAlready == 6: self.fsm.request('notAcceptingFriends') elif yesNoAlready == 10: self.fsm.request('no') elif yesNoAlready == 13: self.fsm.request('otherTooMany') else: self.notify.warning('Got unexpected response to friendConsidering: %s' % yesNoAlready) self.fsm.request('maybe') def __friendResponse(self, yesNoMaybe, context): if self.context != context: self.notify.warning('Unexpected change of context from %s to %s.' % (self.context, context)) self.context = context if yesNoMaybe == 1: self.fsm.request('yes') elif yesNoMaybe == 0: self.fsm.request('no') elif yesNoMaybe == 3: self.fsm.request('otherTooMany') else: self.notify.warning('Got unexpected response to friendResponse: %s' % yesNoMaybe) self.fsm.request('maybe') def __handleDisableAvatar(self): self.fsm.request('wentAway')