31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.distributed.DistributedObjectGlobalAI import DistributedObjectGlobalAI
|
|
from direct.distributed.PyDatagram import *
|
|
|
|
|
|
class ToontownFriendsManagerAI(DistributedObjectGlobalAI):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('ToontownFriendsManagerAI')
|
|
|
|
def sendFriendOnline(self, avId, friendId, commonChatFlags, whitelistChatFlags):
|
|
datagram = PyDatagram()
|
|
datagram.addUint32(friendId) # doId
|
|
datagram.addUint8(commonChatFlags) # commonChatFlags
|
|
datagram.addUint8(whitelistChatFlags) # whitelistChatFlags
|
|
self.sendUpdateToAvatarId(avId, 'friendOnline', [datagram.getMessage()])
|
|
|
|
def sendMakeFriends(self, avatarAId, avatarBId, flags, context):
|
|
self.sendUpdate('makeFriends', [avatarAId, avatarBId, flags, context])
|
|
|
|
def makeFriendsResponse(self, avatarAId, avatarBId, result, context):
|
|
if result == 1:
|
|
avatarA = self.air.doId2do.get(avatarAId)
|
|
avatarB = self.air.doId2do.get(avatarBId)
|
|
if avatarA and avatarB:
|
|
self.sendFriendOnline(avatarAId, avatarBId, 0, 1)
|
|
self.sendFriendOnline(avatarBId, avatarAId, 0, 1)
|
|
|
|
messenger.send("makeFriendsReply", [result, context])
|
|
|
|
def sendRequestSecret(self, requesterId):
|
|
self.sendUpdate('requestSecret', [requesterId])
|