Poodletooth-iLand/toontown/toon/AvatarPanelBase.py

from pandac.PandaModules import *
from direct.gui.DirectGui import *
from direct.showbase import DirectObject
from otp.avatar import AvatarPanel
from toontown.toonbase import TTLocalizer
from toontown.toontowngui import TTDialog
from toontown.ai import ReportGlobals
IGNORE_SCALE = 0.06
STOP_IGNORE_SCALE = 0.04

class AvatarPanelBase(AvatarPanel.AvatarPanel):

    def __init__(self, avatar, FriendsListPanel = None):
        self.dialog = None
        self.category = None
        AvatarPanel.AvatarPanel.__init__(self, avatar, FriendsListPanel)
        return

    def getIgnoreButtonInfo(self):
        if base.localAvatar.isIgnored(self.avId):
            return (TTLocalizer.AvatarPanelStopIgnoring, self.handleStopIgnoring, STOP_IGNORE_SCALE)
        else:
            return (TTLocalizer.AvatarPanelIgnore, self.handleIgnore, IGNORE_SCALE)

    def handleIgnore(self):
        isAvatarFriend = base.cr.isFriend(self.avatar.doId)
        if isAvatarFriend:
            self.dialog = TTDialog.TTGlobalDialog(
                style=TTDialog.CancelOnly,
                text=TTLocalizer.IgnorePanelAddFriendAvatar % self.avName,
                text_wordwrap=18.5,
                text_scale=0.06,
                cancelButtonText=TTLocalizer.lCancel,
                doneEvent='IgnoreBlocked',
                command=self.freeLocalAvatar)
        else:
            self.dialog = TTDialog.TTGlobalDialog(
                style=TTDialog.TwoChoice,
                text=TTLocalizer.IgnorePanelAddIgnore % self.avName,
                text_wordwrap=18.5,
                text_scale=TTLocalizer.APBdialog,
                okButtonText=TTLocalizer.AvatarPanelIgnore,
                cancelButtonText=TTLocalizer.lCancel,
                doneEvent='IgnoreConfirm',
                command=self.handleIgnoreConfirm)
        DirectLabel(
            parent=self.dialog,
            relief=None,
            pos=(0, TTLocalizer.APBdirectLabelPosY, 0.125),
            text=TTLocalizer.IgnorePanelTitle,
            textMayChange=0,
            text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleStopIgnoring(self):
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.TwoChoice, text=TTLocalizer.IgnorePanelRemoveIgnore % self.avName, text_wordwrap=18.5, text_scale=0.06, okButtonText=TTLocalizer.AvatarPanelStopIgnoring, cancelButtonText=TTLocalizer.lCancel, buttonPadSF=4.0, doneEvent='StopIgnoringConfirm', command=self.handleStopIgnoringConfirm)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, TTLocalizer.APBdirectLabelPosY, 0.15), text=TTLocalizer.IgnorePanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleIgnoreConfirm(self, value):
        if value == -1:
            self.freeLocalAvatar()
            return
        base.localAvatar.addIgnore(self.avId)
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.Acknowledge, text=TTLocalizer.IgnorePanelIgnore % self.avName, text_wordwrap=18.5, text_scale=0.06, topPad=0.1, doneEvent='IgnoreComplete', command=self.handleDoneIgnoring)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, TTLocalizer.APBdirectLabelPosY, 0.15), text=TTLocalizer.IgnorePanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleStopIgnoringConfirm(self, value):
        if value == -1:
            self.freeLocalAvatar()
            return
        base.localAvatar.removeIgnore(self.avId)
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.Acknowledge, text=TTLocalizer.IgnorePanelEndIgnore % self.avName, text_wordwrap=18.5, text_scale=0.06, topPad=0.1, doneEvent='StopIgnoringComplete', command=self.handleDoneIgnoring)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, TTLocalizer.APBdirectLabelPosY, 0.15), text=TTLocalizer.IgnorePanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleDoneIgnoring(self, value):
        self.freeLocalAvatar()

    def handleReport(self):
        if base.localAvatar.isReported(self.avId):
            self.alreadyReported()
        else:
            self.confirmReport()

    def confirmReport(self):
        if base.cr.isFriend(self.avId):
            string = TTLocalizer.ReportPanelBodyFriends
            titlePos = 0.41
        else:
            string = TTLocalizer.ReportPanelBody
            titlePos = 0.35
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.TwoChoice, text=string % self.avName, text_wordwrap=18.5, text_scale=0.06, okButtonText=TTLocalizer.AvatarPanelReport, cancelButtonText=TTLocalizer.lCancel, doneEvent='ReportConfirm', command=self.handleReportConfirm)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, 0, titlePos), text=TTLocalizer.ReportPanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleReportConfirm(self, value):
        self.cleanupDialog()
        if value == 1:
            self.chooseReportCategory()
        else:
            self.requestWalk()

    def alreadyReported(self):
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.Acknowledge, text=TTLocalizer.ReportPanelAlreadyReported % self.avName, text_wordwrap=18.5, text_scale=0.06, topPad=0.1, doneEvent='AlreadyReported', command=self.handleAlreadyReported)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, 0, 0.2), text=TTLocalizer.ReportPanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleAlreadyReported(self, value):
        self.freeLocalAvatar()

    def chooseReportCategory(self):
        self.dialog = TTDialog.TTGlobalDialog(pos=(0, 0, 0.4), style=TTDialog.CancelOnly, text=TTLocalizer.ReportPanelCategoryBody % (self.avName, self.avName), text_wordwrap=18.5, text_scale=0.06, topPad=0.05, midPad=0.75, cancelButtonText=TTLocalizer.lCancel, doneEvent='ReportCategory', command=self.handleReportCategory)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, 0, 0.225), text=TTLocalizer.ReportPanelTitle, textMayChange=0, text_scale=0.08)
        guiButton = loader.loadModel('phase_3/models/gui/quit_button')
        DirectButton(parent=self.dialog, relief=None, image=(guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR')), image_scale=(2.125, 1.0, 1.0), text=TTLocalizer.ReportPanelCategoryLanguage, text_scale=0.06, text_pos=(0, -0.0124), pos=(0, 0, -0.3), command=self.handleReportCategory, extraArgs=[0])
        DirectButton(parent=self.dialog, relief=None, image=(guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR')), image_scale=(2.15, 1.0, 1.0), text=TTLocalizer.ReportPanelCategoryGreening, text_scale=0.06, text_pos=(0, -0.0125), pos=(0, 0, -0.425), command=self.handleReportCategory, extraArgs=[1])
        DirectButton(parent=self.dialog, relief=None, image=(guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR')), image_scale=(2.125, 1.0, 1.0), text=TTLocalizer.ReportPanelCategoryRude, text_scale=0.06, text_pos=(0, -0.0125), pos=(0, 0, -0.55), command=self.handleReportCategory, extraArgs=[2])
        DirectButton(parent=self.dialog, relief=None, image=(guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR')), image_scale=(2.125, 1.0, 1.0), text=TTLocalizer.ReportPanelCategoryName, text_scale=0.06, text_pos=(0, -0.0125), pos=(0, 0, -0.675), command=self.handleReportCategory, extraArgs=[3])
        DirectButton(parent=self.dialog, relief=None, image=(guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR')), image_scale=(2.125, 1.0, 1.0), text=TTLocalizer.ReportPanelCategoryHacking, text_scale=0.06, text_pos=(0, -0.0125), pos=(0, 0, -0.8), command=self.handleReportCategory, extraArgs=[4])
        guiButton.removeNode()
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        self.requestStopped()
        return

    def handleReportCategory(self, value):
        self.cleanupDialog()
        if value >= 0 and ReportGlobals.isValidCategory(value):
            self.category = ReportGlobals.getCategory(value)
            self.confirmReportCategory(value)
        else:
            self.requestWalk()

    def confirmReportCategory(self, category):
        string = TTLocalizer.ReportPanelConfirmations[category]
        string += '\n\n' + TTLocalizer.ReportPanelWarning
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.TwoChoice, text=string % self.avName, text_wordwrap=18.5, text_scale=0.06, topPad=0.1, okButtonText=TTLocalizer.AvatarPanelReport, cancelButtonText=TTLocalizer.lCancel, doneEvent='ReportConfirmCategory', command=self.handleReportCategoryConfirm)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, 0, 0.5), text=TTLocalizer.ReportPanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        return

    def handleReportCategoryConfirm(self, value):
        self.cleanupDialog()
        removed = 0
        
        if value > 0:
            if base.cr.isFriend(self.avId):
                base.cr.removeFriend(self.avId)
                removed = 1
            
            base.cr.reportMgr.sendReport(self.avId, self.category)
            self.reportComplete(removed)
        else:
            self.requestWalk()

    def reportComplete(self, removed):
        string = TTLocalizer.ReportPanelThanks
        titlePos = 0.25
        if removed:
            string += ' ' + TTLocalizer.ReportPanelRemovedFriend % self.avName
            titlePos = 0.3
        self.dialog = TTDialog.TTGlobalDialog(style=TTDialog.Acknowledge, text=string, text_wordwrap=18.5, text_scale=0.06, topPad=0.1, doneEvent='ReportComplete', command=self.handleReportComplete)
        DirectLabel(parent=self.dialog, relief=None, pos=(0, 0, titlePos), text=TTLocalizer.ReportPanelTitle, textMayChange=0, text_scale=0.08)
        self.dialog.show()
        self.__acceptStoppedStateMsg()
        return

    def handleReportComplete(self, value):
        self.freeLocalAvatar()

    def freeLocalAvatar(self, value = None):
        self.cleanupDialog()
        self.requestWalk()

    def cleanupDialog(self):
        if self.dialog:
            base.cr.openAvatarPanels.discard(self)
            self.dialog.ignore('exitingStoppedState')
            self.dialog.cleanup()
            self.dialog = None
        return

    def requestStopped(self):
        base.cr.openAvatarPanels.add(self)
        if not base.cr.playGame.getPlace().fsm.getCurrentState().getName() == 'stickerBook':
            if base.cr.playGame.getPlace().fsm.hasStateNamed('stopped'):
                base.cr.playGame.getPlace().fsm.request('stopped')
            else:
                self.notify.warning('skipping request to stopped in %s' % base.cr.playGame.getPlace())
        else:
            self.cleanup()

    def requestWalk(self):
        if base.cr.playGame.getPlace().fsm.hasStateNamed('finalBattle'):
            base.cr.playGame.getPlace().fsm.request('finalBattle')
        elif base.cr.playGame.getPlace().fsm.hasStateNamed('walk'):
            if base.cr.playGame.getPlace().getState() == 'stopped':
                base.cr.playGame.getPlace().fsm.request('walk')
        else:
            self.notify.warning('skipping request to walk in %s' % base.cr.playGame.getPlace())

    def __acceptStoppedStateMsg(self):
        self.dialog.ignore('exitingStoppedState')
        self.dialog.accept('exitingStoppedState', self.cleanupDialog)