mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-26 13:12:30 -06:00
367 lines
No EOL
13 KiB
Python
367 lines
No EOL
13 KiB
Python
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.fsm import ClassicFSM
|
|
from direct.fsm import State
|
|
from direct.gui.DirectGui import *
|
|
from direct.showbase import DirectObject
|
|
from pandac.PandaModules import *
|
|
from otp.otpbase import OTPLocalizer
|
|
from toontown.chat.ChatGlobals import *
|
|
|
|
|
|
ChatEvent = 'ChatEvent'
|
|
NormalChatEvent = 'NormalChatEvent'
|
|
SCChatEvent = 'SCChatEvent'
|
|
SCCustomChatEvent = 'SCCustomChatEvent'
|
|
SCEmoteChatEvent = 'SCEmoteChatEvent'
|
|
OnScreen = 0
|
|
OffScreen = 1
|
|
Thought = 2
|
|
ThoughtPrefix = '.'
|
|
|
|
def isThought(message):
|
|
if len(message) == 0:
|
|
return 0
|
|
elif message.find(ThoughtPrefix, 0, len(ThoughtPrefix)) >= 0:
|
|
return 1
|
|
else:
|
|
return 0
|
|
|
|
|
|
def removeThoughtPrefix(message):
|
|
if isThought(message):
|
|
return message[len(ThoughtPrefix):]
|
|
else:
|
|
return message
|
|
|
|
|
|
class ChatManager(DirectObject.DirectObject):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('ChatManager')
|
|
|
|
def __init__(self, cr, localAvatar):
|
|
self.cr = cr
|
|
self.localAvatar = localAvatar
|
|
self.wantBackgroundFocus = 1
|
|
self.__scObscured = 0
|
|
self.__normalObscured = 0
|
|
self.noSecretChatAtAll = None
|
|
self.noSecretChatAtAllAndNoWhitelist = None
|
|
self.fsm = ClassicFSM.ClassicFSM('chatManager', [State.State('off', self.enterOff, self.exitOff),
|
|
State.State('mainMenu', self.enterMainMenu, self.exitMainMenu),
|
|
State.State('speedChat', self.enterSpeedChat, self.exitSpeedChat),
|
|
State.State('normalChat', self.enterNormalChat, self.exitNormalChat),
|
|
State.State('whisper', self.enterWhisper, self.exitWhisper),
|
|
State.State('whisperChat', self.enterWhisperChat, self.exitWhisperChat),
|
|
State.State('whisperChatPlayer', self.enterWhisperChatPlayer, self.exitWhisperChatPlayer),
|
|
State.State('whisperSpeedChat', self.enterWhisperSpeedChat, self.exitWhisperSpeedChat),
|
|
State.State('whisperSpeedChatPlayer', self.enterWhisperSpeedChatPlayer, self.exitWhisperSpeedChatPlayer),
|
|
State.State('noSecretChatAtAll', self.enterNoSecretChatAtAll, self.exitNoSecretChatAtAll),
|
|
State.State('noSecretChatAtAllAndNoWhitelist', self.enterNoSecretChatAtAllAndNoWhitelist, self.exitNoSecretChatAtAllAndNoWhitelist),
|
|
State.State('otherDialog', self.enterOtherDialog, self.exitOtherDialog),
|
|
State.State('whiteListOpenChat', self.enterWhiteListOpenChat, self.exitWhiteListOpenChat),
|
|
State.State('whiteListAvatarChat', self.enterWhiteListAvatarChat, self.exitWhiteListAvatarChat),
|
|
State.State('whiteListPlayerChat', self.enterWhiteListPlayerChat, self.exitWhiteListPlayerChat)], 'off', 'off')
|
|
self.fsm.enterInitialState()
|
|
return
|
|
|
|
def delete(self):
|
|
self.ignoreAll()
|
|
del self.fsm
|
|
if hasattr(self.chatInputNormal, 'destroy'):
|
|
self.chatInputNormal.destroy()
|
|
self.chatInputNormal.delete()
|
|
del self.chatInputNormal
|
|
self.chatInputSpeedChat.delete()
|
|
del self.chatInputSpeedChat
|
|
if self.noSecretChatAtAll:
|
|
self.noSecretChatAtAll.destroy()
|
|
self.noSecretChatAtAll = None
|
|
if self.noSecretChatAtAllAndNoWhitelist:
|
|
self.noSecretChatAtAllAndNoWhitelist.destroy()
|
|
self.noSecretChatAtAllAndNoWhitelist = None
|
|
del self.localAvatar
|
|
del self.cr
|
|
return
|
|
|
|
def obscure(self, normal, sc):
|
|
self.__scObscured = sc
|
|
if self.__scObscured:
|
|
self.scButton.hide()
|
|
self.__normalObscured = normal
|
|
if self.__normalObscured:
|
|
self.normalButton.hide()
|
|
|
|
def isObscured(self):
|
|
return (self.__normalObscured, self.__scObscured)
|
|
|
|
def stop(self):
|
|
self.fsm.request('off')
|
|
self.ignoreAll()
|
|
|
|
def start(self):
|
|
self.fsm.request('mainMenu')
|
|
|
|
def announceChat(self):
|
|
messenger.send(ChatEvent)
|
|
|
|
def announceSCChat(self):
|
|
messenger.send(SCChatEvent)
|
|
self.announceChat()
|
|
|
|
def sendChatString(self, message):
|
|
chatFlags = CFSpeech | CFTimeout
|
|
if base.cr.wantSwitchboardHacks:
|
|
from otp.switchboard import badwordpy
|
|
badwordpy.init('', '')
|
|
message = badwordpy.scrub(message)
|
|
if isThought(message):
|
|
message = removeThoughtPrefix(message)
|
|
chatFlags = CFThought
|
|
messenger.send(NormalChatEvent)
|
|
self.announceChat()
|
|
|
|
def sendWhisperString(self, message, whisperAvatarId):
|
|
pass
|
|
|
|
def sendSCChatMessage(self, msgIndex):
|
|
base.talkAssistant.sendOpenSpeedChat(1, msgIndex)
|
|
|
|
def sendSCWhisperMessage(self, msgIndex, whisperAvatarId, toPlayer):
|
|
if toPlayer:
|
|
base.talkAssistant.sendPlayerWhisperSpeedChat(1, msgIndex, whisperAvatarId)
|
|
else:
|
|
base.talkAssistant.sendAvatarWhisperSpeedChat(1, msgIndex, whisperAvatarId)
|
|
|
|
def sendSCCustomChatMessage(self, msgIndex):
|
|
base.talkAssistant.sendOpenSpeedChat(3, msgIndex)
|
|
|
|
def sendSCCustomWhisperMessage(self, msgIndex, whisperAvatarId, toPlayer):
|
|
if toPlayer:
|
|
base.talkAssistant.sendPlayerWhisperSpeedChat(3, msgIndex, whisperAvatarId)
|
|
else:
|
|
base.talkAssistant.sendAvatarWhisperSpeedChat(3, msgIndex, whisperAvatarId)
|
|
|
|
def sendSCEmoteChatMessage(self, emoteId):
|
|
base.talkAssistant.sendOpenSpeedChat(2, emoteId)
|
|
|
|
def sendSCEmoteWhisperMessage(self, emoteId, whisperAvatarId, toPlayer):
|
|
if toPlayer:
|
|
base.talkAssistant.sendPlayerWhisperSpeedChat(2, emoteId, whisperAvatarId)
|
|
else:
|
|
base.talkAssistant.sendAvatarWhisperSpeedChat(2, emoteId, whisperAvatarId)
|
|
|
|
def enterOff(self):
|
|
self.scButton.hide()
|
|
self.normalButton.hide()
|
|
self.ignoreAll()
|
|
|
|
def exitOff(self):
|
|
pass
|
|
|
|
def enterMainMenu(self):
|
|
self.checkObscurred()
|
|
if self.localAvatar.canChat() or self.cr.wantMagicWords:
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 1
|
|
self.acceptOnce('enterNormalChat', self.fsm.request, ['normalChat'])
|
|
|
|
def checkObscurred(self):
|
|
if not self.__scObscured:
|
|
self.scButton.show()
|
|
if not self.__normalObscured:
|
|
self.normalButton.show()
|
|
|
|
def exitMainMenu(self):
|
|
self.scButton.hide()
|
|
self.normalButton.hide()
|
|
self.ignore('enterNormalChat')
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
|
|
def whisperTo(self, avatarName, avatarId, playerId = None):
|
|
self.fsm.request('whisper', [avatarName, avatarId, playerId])
|
|
|
|
def noWhisper(self):
|
|
self.fsm.request('mainMenu')
|
|
|
|
def handleWhiteListSelect(self):
|
|
self.fsm.request('whiteListOpenChat')
|
|
|
|
def enterWhiteListOpenChat(self):
|
|
self.checkObscurred()
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
base.localAvatar.chatMgr.chatInputWhiteList.activateByData()
|
|
|
|
def exitWhiteListOpenChat(self):
|
|
pass
|
|
|
|
def enterWhiteListAvatarChat(self, receiverId):
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
base.localAvatar.chatMgr.chatInputWhiteList.activateByData(receiverId, 0)
|
|
|
|
def exitWhiteListAvatarChat(self):
|
|
pass
|
|
|
|
def enterWhiteListPlayerChat(self, receiverId):
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
base.localAvatar.chatMgr.chatInputWhiteList.activateByData(receiverId, 1)
|
|
|
|
def exitWhiteListPlayerChat(self):
|
|
pass
|
|
|
|
def enterWhisper(self, avatarName, avatarId, playerId = None):
|
|
self.whisperScButton['extraArgs'] = [avatarName, avatarId, playerId]
|
|
self.whisperButton['extraArgs'] = [avatarName, avatarId, playerId]
|
|
playerName = None
|
|
chatToToon = 1
|
|
online = 0
|
|
if avatarId in self.cr.doId2do:
|
|
online = 1
|
|
elif self.cr.isFriend(avatarId):
|
|
online = self.cr.isFriendOnline(avatarId)
|
|
hasManager = hasattr(base.cr, 'playerFriendsManager')
|
|
if hasManager:
|
|
if base.cr.playerFriendsManager.askAvatarOnline(avatarId):
|
|
online = 1
|
|
avatarUnderstandable = 0
|
|
playerUnderstandable = 0
|
|
av = None
|
|
if avatarId:
|
|
av = self.cr.identifyAvatar(avatarId)
|
|
if av != None:
|
|
avatarUnderstandable = av.isUnderstandable()
|
|
if playerId:
|
|
if playerId in base.cr.playerFriendsManager.playerId2Info:
|
|
playerInfo = base.cr.playerFriendsManager.playerId2Info.get(playerId)
|
|
playerName = playerInfo.playerName
|
|
online = 1
|
|
playerUnderstandable = playerInfo.understandableYesNo
|
|
if playerUnderstandable or not avatarId:
|
|
chatToToon = 0
|
|
if chatToToon:
|
|
chatName = avatarName
|
|
else:
|
|
chatName = playerName
|
|
normalButtonObscured, scButtonObscured = self.isObscured()
|
|
if (avatarUnderstandable or playerUnderstandable) and online and not normalButtonObscured:
|
|
self.whisperButton['state'] = 'normal'
|
|
self.enablewhisperButton()
|
|
else:
|
|
self.whisperButton['state'] = 'inactive'
|
|
self.disablewhisperButton()
|
|
if online:
|
|
self.whisperScButton['state'] = 'normal'
|
|
self.whisperButton['state'] = 'normal'
|
|
self.changeFrameText(OTPLocalizer.ChatManagerWhisperToName % chatName)
|
|
else:
|
|
self.whisperScButton['state'] = 'inactive'
|
|
self.whisperButton['state'] = 'inactive'
|
|
self.changeFrameText(OTPLocalizer.ChatManagerWhisperOffline % chatName)
|
|
self.whisperFrame.show()
|
|
self.refreshWhisperFrame()
|
|
if avatarUnderstandable or playerUnderstandable:
|
|
if playerId and not chatToToon:
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 1
|
|
self.acceptOnce('enterNormalChat', self.fsm.request, ['whisperChatPlayer', [avatarName, playerId]])
|
|
elif online and chatToToon:
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 1
|
|
self.acceptOnce('enterNormalChat', self.fsm.request, ['whisperChat', [avatarName, avatarId]])
|
|
return
|
|
|
|
def disablewhisperButton(self):
|
|
pass
|
|
|
|
def enablewhisperButton(self):
|
|
pass
|
|
|
|
def refreshWhisperFrame(self):
|
|
pass
|
|
|
|
def changeFrameText(self, newText):
|
|
self.whisperFrame['text'] = newText
|
|
|
|
def exitWhisper(self):
|
|
self.whisperFrame.hide()
|
|
self.ignore('enterNormalChat')
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
|
|
def enterWhisperSpeedChat(self, avatarId):
|
|
self.whisperFrame.show()
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
self.chatInputSpeedChat.show(avatarId)
|
|
|
|
def exitWhisperSpeedChat(self):
|
|
self.whisperFrame.hide()
|
|
self.chatInputSpeedChat.hide()
|
|
|
|
def enterWhisperSpeedChatPlayer(self, playerId):
|
|
self.whisperFrame.show()
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
self.chatInputSpeedChat.show(playerId, 1)
|
|
|
|
def exitWhisperSpeedChatPlayer(self):
|
|
self.whisperFrame.hide()
|
|
self.chatInputSpeedChat.hide()
|
|
|
|
def enterWhisperChat(self, avatarName, avatarId):
|
|
result = self.chatInputNormal.activateByData(avatarId)
|
|
return result
|
|
|
|
def exitWhisperChat(self):
|
|
self.chatInputNormal.deactivate()
|
|
|
|
def enterWhisperChatPlayer(self, avatarName, playerId):
|
|
playerInfo = base.cr.playerFriendsManager.getFriendInfo(playerId)
|
|
if playerInfo:
|
|
avatarName = playerInfo.playerName
|
|
result = self.chatInputNormal.activateByData(playerId, 1)
|
|
return result
|
|
|
|
def exitWhisperChatPlayer(self):
|
|
self.chatInputNormal.deactivate()
|
|
|
|
def enterSpeedChat(self):
|
|
messenger.send('enterSpeedChat')
|
|
if not self.__scObscured:
|
|
self.scButton.show()
|
|
if not self.__normalObscured:
|
|
self.normalButton.show()
|
|
if self.wantBackgroundFocus:
|
|
self.chatInputNormal.chatEntry['backgroundFocus'] = 0
|
|
self.chatInputSpeedChat.show()
|
|
|
|
def exitSpeedChat(self):
|
|
self.scButton.hide()
|
|
self.normalButton.hide()
|
|
self.chatInputSpeedChat.hide()
|
|
|
|
def enterNormalChat(self):
|
|
result = self.chatInputNormal.activateByData()
|
|
return result
|
|
|
|
def exitNormalChat(self):
|
|
self.chatInputNormal.deactivate()
|
|
|
|
def enterNoSecretChatAtAll(self):
|
|
self.notify.error('called enterNoSecretChatAtAll() on parent class')
|
|
|
|
def exitNoSecretChatAtAll(self):
|
|
self.notify.error('called exitNoSecretChatAtAll() on parent class')
|
|
|
|
def enterNoSecretChatAtAllAndNoWhitelist(self):
|
|
self.notify.error('called enterNoSecretChatAtAllAndNoWhitelist() on parent class')
|
|
|
|
def exitNoSecretChatAtAllAndNoWhitelist(self):
|
|
self.notify.error('called exitNoSecretChatAtAllAndNoWhitelist() on parent class')
|
|
|
|
def enterOtherDialog(self):
|
|
pass
|
|
|
|
def exitOtherDialog(self):
|
|
pass |