515 lines
19 KiB
Python
515 lines
19 KiB
Python
import string
|
|
from direct.showbase import DirectObject
|
|
from direct.fsm import ClassicFSM
|
|
from direct.fsm import State
|
|
from otp.login import SecretFriendsInfoPanel
|
|
from otp.login import PrivacyPolicyPanel
|
|
from otp.otpbase import OTPLocalizer
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
from otp.login import LeaveToPayDialog
|
|
from direct.gui.DirectGui import *
|
|
from panda3d.core 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 string.find(message, 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')
|
|
execChat = ConfigVariableBool('exec-chat', 0).value
|
|
|
|
def __init__(self, cr, localAvatar):
|
|
self.cr = cr
|
|
self.localAvatar = localAvatar
|
|
self.wantBackgroundFocus = 1
|
|
self.__scObscured = 0
|
|
self.__normalObscured = 0
|
|
self.openChatWarning = None
|
|
self.unpaidChatWarning = None
|
|
self.teaser = None
|
|
self.paidNoParentPassword = None
|
|
self.noSecretChatAtAll = None
|
|
self.noSecretChatAtAllAndNoWhitelist = None
|
|
self.noSecretChatWarning = None
|
|
self.activateChatGui = None
|
|
self.chatMoreInfo = None
|
|
self.chatPrivacyPolicy = None
|
|
self.secretChatActivated = None
|
|
self.problemActivatingChat = None
|
|
self.leaveToPayDialog = 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('openChatWarning', self.enterOpenChatWarning, self.exitOpenChatWarning),
|
|
State.State('leaveToPayDialog', self.enterLeaveToPayDialog, self.exitLeaveToPayDialog),
|
|
State.State('unpaidChatWarning', self.enterUnpaidChatWarning, self.exitUnpaidChatWarning),
|
|
State.State('noSecretChatAtAll', self.enterNoSecretChatAtAll, self.exitNoSecretChatAtAll),
|
|
State.State('noSecretChatAtAllAndNoWhitelist', self.enterNoSecretChatAtAllAndNoWhitelist, self.exitNoSecretChatAtAllAndNoWhitelist),
|
|
State.State('noSecretChatWarning', self.enterNoSecretChatWarning, self.exitNoSecretChatWarning),
|
|
State.State('noFriendsWarning', self.enterNoFriendsWarning, self.exitNoFriendsWarning),
|
|
State.State('otherDialog', self.enterOtherDialog, self.exitOtherDialog),
|
|
State.State('activateChat', self.enterActivateChat, self.exitActivateChat),
|
|
State.State('chatMoreInfo', self.enterChatMoreInfo, self.exitChatMoreInfo),
|
|
State.State('chatPrivacyPolicy', self.enterChatPrivacyPolicy, self.exitChatPrivacyPolicy),
|
|
State.State('secretChatActivated', self.enterSecretChatActivated, self.exitSecretChatActivated),
|
|
State.State('problemActivatingChat', self.enterProblemActivatingChat, self.exitProblemActivatingChat),
|
|
State.State('whiteListOpenChat', self.enterWhiteListOpenChat, self.exitWhiteListOpenChat),
|
|
State.State('whiteListAvatarChat', self.enterWhiteListAvatarChat, self.exitWhiteListAvatarChat),
|
|
State.State('whiteListPlayerChat', self.enterWhiteListPlayerChat, self.exitWhiteListPlayerChat),
|
|
State.State('trueFriendTeaserPanel', self.enterTrueFriendTeaserPanel, self.exitTrueFriendTeaserPanel)], '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.openChatWarning:
|
|
self.openChatWarning.destroy()
|
|
self.openChatWarning = None
|
|
if self.unpaidChatWarning:
|
|
self.payButton = None
|
|
self.unpaidChatWarning.destroy()
|
|
self.unpaidChatWarning = None
|
|
if self.teaser:
|
|
self.teaser.cleanup()
|
|
self.teaser.unload()
|
|
self.teaser = None
|
|
if self.noSecretChatAtAll:
|
|
self.noSecretChatAtAll.destroy()
|
|
self.noSecretChatAtAll = None
|
|
if self.noSecretChatAtAllAndNoWhitelist:
|
|
self.noSecretChatAtAllAndNoWhitelist.destroy()
|
|
self.noSecretChatAtAllAndNoWhitelist = None
|
|
if self.noSecretChatWarning:
|
|
self.noSecretChatWarning.destroy()
|
|
self.noSecretChatWarning = None
|
|
if self.activateChatGui:
|
|
self.activateChatGui.destroy()
|
|
self.activateChatGui = None
|
|
if self.chatMoreInfo:
|
|
self.chatMoreInfo.destroy()
|
|
self.chatMoreInfo = None
|
|
if self.chatPrivacyPolicy:
|
|
self.chatPrivacyPolicy.destroy()
|
|
self.chatPrivacyPolicy = None
|
|
if self.secretChatActivated:
|
|
self.secretChatActivated.destroy()
|
|
self.secretChatActivated = None
|
|
if self.problemActivatingChat:
|
|
self.problemActivatingChat.destroy()
|
|
self.problemActivatingChat = 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.changeFrameText(OTPLocalizer.ChatManagerWhisperToName % chatName)
|
|
else:
|
|
self.whisperScButton['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]])
|
|
if ConfigVariableBool('force-typed-whisper-enabled', 0).value:
|
|
self.whisperButton['state'] = 'normal'
|
|
self.enablewhisperButton()
|
|
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 enterOpenChatWarning(self):
|
|
self.notify.error('called enterOpenChatWarning() on parent class')
|
|
|
|
def exitOpenChatWarning(self):
|
|
self.notify.error('called exitOpenChatWarning() on parent class')
|
|
|
|
def enterLeaveToPayDialog(self):
|
|
if self.leaveToPayDialog == None:
|
|
self.leaveToPayDialog = LeaveToPayDialog.LeaveToPayDialog(self.paidNoParentPassword)
|
|
self.leaveToPayDialog.setCancel(self.__handleLeaveToPayCancel)
|
|
self.leaveToPayDialog.show()
|
|
return
|
|
|
|
def exitLeaveToPayDialog(self):
|
|
if self.leaveToPayDialog:
|
|
self.leaveToPayDialog.destroy()
|
|
self.leaveToPayDialog = None
|
|
return
|
|
|
|
def enterUnpaidChatWarning(self):
|
|
self.notify.error('called enterUnpaidChatWarning() on parent class')
|
|
|
|
def exitUnpaidChatWarning(self):
|
|
self.notify.error('called exitUnpaidChatWarning() on parent class')
|
|
|
|
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 enterNoSecretChatWarning(self):
|
|
self.notify.error('called enterNoSecretChatWarning() on parent class')
|
|
|
|
def exitNoSecretChatWarning(self):
|
|
self.notify.error('called exitNoSecretChatWarning() on parent class')
|
|
|
|
def enterNoFriendsWarning(self):
|
|
self.notify.error('called enterNoFriendsWarning() on parent class')
|
|
|
|
def exitNoFriendsWarning(self):
|
|
self.notify.error('called exitNoFriendsWarning() on parent class')
|
|
|
|
def enterActivateChat(self):
|
|
self.notify.error('called enterActivateChat() on parent class')
|
|
|
|
def exitActivateChat(self):
|
|
self.notify.error('called exitActivateChat() on parent class')
|
|
|
|
def enterOtherDialog(self):
|
|
pass
|
|
|
|
def exitOtherDialog(self):
|
|
pass
|
|
|
|
def enterChatMoreInfo(self):
|
|
if self.chatMoreInfo == None:
|
|
self.chatMoreInfo = SecretFriendsInfoPanel.SecretFriendsInfoPanel('secretFriendsInfoDone')
|
|
self.chatMoreInfo.show()
|
|
self.accept('secretFriendsInfoDone', self.__secretFriendsInfoDone)
|
|
return
|
|
|
|
def exitChatMoreInfo(self):
|
|
self.chatMoreInfo.hide()
|
|
self.ignore('secretFriendsInfoDone')
|
|
|
|
def enterChatPrivacyPolicy(self):
|
|
if self.chatPrivacyPolicy == None:
|
|
self.chatPrivacyPolicy = PrivacyPolicyPanel.PrivacyPolicyPanel('privacyPolicyDone')
|
|
self.chatPrivacyPolicy.show()
|
|
self.accept('privacyPolicyDone', self.__privacyPolicyDone)
|
|
return
|
|
|
|
def exitChatPrivacyPolicy(self):
|
|
cleanupDialog('privacyPolicyDialog')
|
|
self.chatPrivacyPolicy = None
|
|
self.ignore('privacyPolicyDone')
|
|
return
|
|
|
|
def enterSecretChatActivated(self):
|
|
self.notify.error('called enterSecretChatActivated() on parent class')
|
|
|
|
def exitSecretChatActivated(self):
|
|
self.notify.error('called exitSecretChatActivated() on parent class')
|
|
|
|
def enterProblemActivatingChat(self):
|
|
self.notify.error('called enterProblemActivatingChat() on parent class')
|
|
|
|
def exitProblemActivatingChat(self):
|
|
self.notify.error('called exitProblemActivatingChat() on parent class')
|
|
|
|
def enterTrueFriendTeaserPanel(self):
|
|
self.notify.error('called enterTrueFriendTeaserPanel () on parent class')
|
|
|
|
def exitTrueFriendTeaserPanel(self):
|
|
self.notify.error('called exitTrueFriendTeaserPanel () on parent class')
|
|
|
|
def __handleLeaveToPayCancel(self):
|
|
self.fsm.request('mainMenu')
|
|
|
|
def __secretFriendsInfoDone(self):
|
|
self.fsm.request('activateChat')
|
|
|
|
def __privacyPolicyDone(self):
|
|
self.fsm.request('activateChat')
|