mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-11-01 01:07:54 +00:00
283 lines
9.8 KiB
Python
283 lines
9.8 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.fsm import FSM
|
||
|
|
from direct.gui.DirectGui import *
|
||
|
|
from direct.task import Task
|
||
|
|
from pandac.PandaModules import *
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from otp.chat.ChatInputTyped import ChatInputTyped
|
||
|
|
from otp.otpbase import OTPGlobals
|
||
|
|
from otp.otpbase import OTPLocalizer
|
||
|
|
from toontown.chat.ChatGlobals import *
|
||
|
|
|
||
|
|
|
||
|
|
class ChatInputWhiteListFrame(FSM.FSM, DirectFrame):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('ChatInputWhiteList')
|
||
|
|
|
||
|
|
def __init__(self, entryOptions, parent = None, **kw):
|
||
|
|
FSM.FSM.__init__(self, 'ChatInputWhiteListFrame')
|
||
|
|
self.okayToSubmit = True
|
||
|
|
self.receiverId = None
|
||
|
|
DirectFrame.__init__(self, parent=aspect2dp, pos=(0, 0, 0.3), relief=None, image=DGG.getDefaultDialogGeom(), image_scale=(1.6, 1, 1.4), image_pos=(0, 0, -0.05), image_color=OTPGlobals.GlobalDialogColor, borderWidth=(0.01, 0.01))
|
||
|
|
optiondefs = {'parent': self,
|
||
|
|
'relief': DGG.SUNKEN,
|
||
|
|
'scale': 0.05,
|
||
|
|
'frameSize': (-0.2,
|
||
|
|
25.3,
|
||
|
|
-0.5,
|
||
|
|
1.2),
|
||
|
|
'borderWidth': (0.1, 0.1),
|
||
|
|
'frameColor': (0.9, 0.9, 0.85, 0.8),
|
||
|
|
'pos': (-0.2, 0, 0.11),
|
||
|
|
'entryFont': OTPGlobals.getInterfaceFont(),
|
||
|
|
'width': 8.6,
|
||
|
|
'numLines': 3,
|
||
|
|
'cursorKeys': 1,
|
||
|
|
'backgroundFocus': 0,
|
||
|
|
'suppressKeys': 1,
|
||
|
|
'suppressMouse': 1,
|
||
|
|
'command': self.sendChat,
|
||
|
|
'failedCommand': self.sendFailed,
|
||
|
|
'focus': 0,
|
||
|
|
'text': '',
|
||
|
|
'sortOrder': DGG.FOREGROUND_SORT_INDEX}
|
||
|
|
entryOptions['parent'] = self
|
||
|
|
self.chatEntry = DirectEntry(**entryOptions)
|
||
|
|
self.whisperId = None
|
||
|
|
self.chatEntry.bind(DGG.OVERFLOW, self.chatOverflow)
|
||
|
|
wantHistory = 0
|
||
|
|
if __dev__:
|
||
|
|
wantHistory = 1
|
||
|
|
self.wantHistory = base.config.GetBool('want-chat-history', wantHistory)
|
||
|
|
self.history = ['']
|
||
|
|
self.historySize = base.config.GetInt('chat-history-size', 10)
|
||
|
|
self.historyIndex = 0
|
||
|
|
self.promoteWhiteList = 0
|
||
|
|
self.checkBeforeSend = base.config.GetBool('white-list-check-before-send', 0)
|
||
|
|
self.whiteList = None
|
||
|
|
self.active = 0
|
||
|
|
self.autoOff = 0
|
||
|
|
self.sendBy = 'Mode'
|
||
|
|
self.prefilter = 1
|
||
|
|
from direct.gui import DirectGuiGlobals
|
||
|
|
self.chatEntry.bind(DirectGuiGlobals.TYPE, self.applyFilter)
|
||
|
|
self.chatEntry.bind(DirectGuiGlobals.ERASE, self.applyFilter)
|
||
|
|
tpMgr = TextPropertiesManager.getGlobalPtr()
|
||
|
|
Red = tpMgr.getProperties('red')
|
||
|
|
Red.setTextColor(1.0, 0.0, 0.0, 1)
|
||
|
|
tpMgr.setProperties('WLRed', Red)
|
||
|
|
del tpMgr
|
||
|
|
self.origFrameColor = self.chatEntry['frameColor']
|
||
|
|
return
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
from direct.gui import DirectGuiGlobals
|
||
|
|
self.chatEntry.unbind(DGG.OVERFLOW)
|
||
|
|
self.chatEntry.unbind(DirectGuiGlobals.TYPE)
|
||
|
|
self.chatEntry.unbind(DirectGuiGlobals.ERASE)
|
||
|
|
self.chatEntry.ignoreAll()
|
||
|
|
DirectFrame.destroy(self)
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
self.ignore('arrow_up-up')
|
||
|
|
self.ignore('arrow_down-up')
|
||
|
|
|
||
|
|
def requestMode(self, mode, *args):
|
||
|
|
return self.request(mode, *args)
|
||
|
|
|
||
|
|
def defaultFilter(self, request, *args):
|
||
|
|
if request == 'AllChat':
|
||
|
|
if not base.talkAssistant.checkAnyTypedChat():
|
||
|
|
messenger.send('Chat-Failed open typed chat test')
|
||
|
|
self.notify.warning('Chat-Failed open typed chat test')
|
||
|
|
return None
|
||
|
|
elif request == 'PlayerWhisper':
|
||
|
|
if not base.talkAssistant.checkWhisperTypedChatPlayer(self.whisperId):
|
||
|
|
messenger.send('Chat-Failed player typed chat test')
|
||
|
|
self.notify.warning('Chat-Failed player typed chat test')
|
||
|
|
return None
|
||
|
|
elif request == 'AvatarWhisper':
|
||
|
|
if not base.talkAssistant.checkWhisperTypedChatAvatar(self.whisperId):
|
||
|
|
messenger.send('Chat-Failed avatar typed chat test')
|
||
|
|
self.notify.warning('Chat-Failed avatar typed chat test')
|
||
|
|
return None
|
||
|
|
return FSM.FSM.defaultFilter(self, request, *args)
|
||
|
|
|
||
|
|
def enterOff(self):
|
||
|
|
self.deactivate()
|
||
|
|
localAvatar.chatMgr.fsm.request('mainMenu')
|
||
|
|
|
||
|
|
def exitOff(self):
|
||
|
|
self.activate()
|
||
|
|
|
||
|
|
def enterAllChat(self):
|
||
|
|
self.chatEntry['focus'] = 1
|
||
|
|
self.show()
|
||
|
|
|
||
|
|
def exitAllChat(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def enterPlayerWhisper(self):
|
||
|
|
self.tempText = self.chatEntry.get()
|
||
|
|
self.activate()
|
||
|
|
|
||
|
|
def exitPlayerWhisper(self):
|
||
|
|
self.chatEntry.set(self.tempText)
|
||
|
|
self.whisperId = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def enterAvatarWhisper(self):
|
||
|
|
self.tempText = self.chatEntry.get()
|
||
|
|
self.activate()
|
||
|
|
|
||
|
|
def exitAvatarWhisper(self):
|
||
|
|
self.chatEntry.set(self.tempText)
|
||
|
|
self.whisperId = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def activateByData(self, receiverId = None, toPlayer = 0):
|
||
|
|
self.receiverId = receiverId
|
||
|
|
self.toPlayer = toPlayer
|
||
|
|
result = None
|
||
|
|
if not self.receiverId:
|
||
|
|
result = self.requestMode('AllChat')
|
||
|
|
elif self.receiverId and not self.toPlayer:
|
||
|
|
self.whisperId = receiverId
|
||
|
|
result = self.requestMode('AvatarWhisper')
|
||
|
|
elif self.receiverId and self.toPlayer:
|
||
|
|
self.whisperId = receiverId
|
||
|
|
result = self.requestMode('PlayerWhisper')
|
||
|
|
return result
|
||
|
|
|
||
|
|
def activate(self):
|
||
|
|
self.chatEntry['focus'] = 1
|
||
|
|
self.show()
|
||
|
|
self.active = 1
|
||
|
|
self.chatEntry.guiItem.setAcceptEnabled(False)
|
||
|
|
|
||
|
|
def deactivate(self):
|
||
|
|
self.chatEntry.set('')
|
||
|
|
self.chatEntry['focus'] = 0
|
||
|
|
self.hide()
|
||
|
|
self.active = 0
|
||
|
|
|
||
|
|
def isActive(self):
|
||
|
|
return self.active
|
||
|
|
|
||
|
|
def sendChat(self, text, overflow = False):
|
||
|
|
if not (len(text) > 0 and text[0] in ['~', '>']):
|
||
|
|
if self.prefilter:
|
||
|
|
words = text.split(' ')
|
||
|
|
newwords = []
|
||
|
|
for word in words:
|
||
|
|
if word == '' or self.whiteList.isWord(word) or self.promoteWhiteList:
|
||
|
|
newwords.append(word)
|
||
|
|
else:
|
||
|
|
newwords.append(base.whiteList.defaultWord)
|
||
|
|
|
||
|
|
text = ' '.join(newwords)
|
||
|
|
else:
|
||
|
|
text = self.chatEntry.get(plain=True)
|
||
|
|
|
||
|
|
if text:
|
||
|
|
self.chatEntry.set('')
|
||
|
|
self.sendChatBySwitch(text)
|
||
|
|
if self.wantHistory:
|
||
|
|
self.addToHistory(text)
|
||
|
|
|
||
|
|
if not overflow:
|
||
|
|
self.hide()
|
||
|
|
if self.autoOff:
|
||
|
|
self.requestMode('Off')
|
||
|
|
|
||
|
|
localAvatar.chatMgr.messageSent()
|
||
|
|
|
||
|
|
def sendChatBySwitch(self, text):
|
||
|
|
if len(text) > 0 and text[0] == '~':
|
||
|
|
base.talkAssistant.sendOpenTalk(text)
|
||
|
|
elif self.sendBy == 'Mode':
|
||
|
|
self.sendChatByMode(text)
|
||
|
|
elif self.sendBy == 'Data':
|
||
|
|
self.sendChatByData(text)
|
||
|
|
else:
|
||
|
|
self.sendChatByMode(text)
|
||
|
|
|
||
|
|
def sendChatByData(self, text):
|
||
|
|
if not self.receiverId:
|
||
|
|
base.talkAssistant.sendOpenTalk(text)
|
||
|
|
elif self.receiverId and not self.toPlayer:
|
||
|
|
base.talkAssistant.sendWhisperTalk(text, self.receiverId)
|
||
|
|
|
||
|
|
def sendChatByMode(self, text):
|
||
|
|
state = self.getCurrentOrNextState()
|
||
|
|
messenger.send('sentRegularChat')
|
||
|
|
if state == 'PlayerWhisper':
|
||
|
|
base.talkAssistant.sendPlayerWhisperWLChat(text, self.whisperId)
|
||
|
|
elif state == 'AvatarWhisper':
|
||
|
|
base.talkAssistant.sendAvatarWhisperWLChat(text, self.whisperId)
|
||
|
|
else:
|
||
|
|
base.talkAssistant.sendOpenTalk(text)
|
||
|
|
|
||
|
|
def sendFailed(self, text):
|
||
|
|
if not self.checkBeforeSend:
|
||
|
|
self.sendChat(text)
|
||
|
|
return
|
||
|
|
self.chatEntry['frameColor'] = (0.9, 0.0, 0.0, 0.8)
|
||
|
|
|
||
|
|
def resetFrameColor(task = None):
|
||
|
|
self.chatEntry['frameColor'] = self.origFrameColor
|
||
|
|
return Task.done
|
||
|
|
|
||
|
|
taskMgr.doMethodLater(0.1, resetFrameColor, 'resetFrameColor')
|
||
|
|
self.applyFilter(keyArgs=None, strict=True)
|
||
|
|
self.okayToSubmit = True
|
||
|
|
self.chatEntry.guiItem.setAcceptEnabled(True)
|
||
|
|
return
|
||
|
|
|
||
|
|
def chatOverflow(self, overflowText):
|
||
|
|
self.notify.debug('chatOverflow')
|
||
|
|
self.sendChat(self.chatEntry.get(plain=True), overflow=True)
|
||
|
|
|
||
|
|
def addToHistory(self, text):
|
||
|
|
self.history = [text] + self.history[:self.historySize - 1]
|
||
|
|
self.historyIndex = 0
|
||
|
|
|
||
|
|
def getPrevHistory(self):
|
||
|
|
self.chatEntry.set(self.history[self.historyIndex])
|
||
|
|
self.historyIndex += 1
|
||
|
|
self.historyIndex %= len(self.history)
|
||
|
|
|
||
|
|
def getNextHistory(self):
|
||
|
|
self.chatEntry.set(self.history[self.historyIndex])
|
||
|
|
self.historyIndex -= 1
|
||
|
|
self.historyIndex %= len(self.history)
|
||
|
|
|
||
|
|
def applyFilter(self, keyArgs, strict = False):
|
||
|
|
text = self.chatEntry.get(plain=True)
|
||
|
|
if text.startswith('~'):
|
||
|
|
self.okayToSubmit = True
|
||
|
|
else:
|
||
|
|
words = text.split(' ')
|
||
|
|
newwords = []
|
||
|
|
self.notify.debug('%s' % words)
|
||
|
|
self.okayToSubmit = True
|
||
|
|
for word in words:
|
||
|
|
if word == '' or self.whiteList.isWord(word):
|
||
|
|
newwords.append(word)
|
||
|
|
else:
|
||
|
|
if self.checkBeforeSend:
|
||
|
|
self.okayToSubmit = False
|
||
|
|
else:
|
||
|
|
self.okayToSubmit = True
|
||
|
|
newwords.append('\x01WLEnter\x01' + word + '\x02')
|
||
|
|
|
||
|
|
if not strict:
|
||
|
|
lastword = words[-1]
|
||
|
|
if lastword == '' or self.whiteList.isPrefix(lastword):
|
||
|
|
newwords[-1] = lastword
|
||
|
|
else:
|
||
|
|
newwords[-1] = '\x01WLEnter\x01' + lastword + '\x02'
|
||
|
|
newtext = ' '.join(newwords)
|
||
|
|
self.chatEntry.set(newtext)
|
||
|
|
self.chatEntry.guiItem.setAcceptEnabled(self.okayToSubmit)
|