From d6ef91477905912b7ea81da70f75e2d4a10afdaf Mon Sep 17 00:00:00 2001
From: John Cote <jwcotejr@gmail.com>
Date: Thu, 8 Jul 2021 22:17:07 -0400
Subject: [PATCH] friends: Generate FriendManagerAI

---
 otp/friends/FriendManagerAI.py      | 8 ++++++--
 toontown/ai/ToontownAIRepository.py | 4 ++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/otp/friends/FriendManagerAI.py b/otp/friends/FriendManagerAI.py
index 616cfbc..6db8ef8 100644
--- a/otp/friends/FriendManagerAI.py
+++ b/otp/friends/FriendManagerAI.py
@@ -1,5 +1,9 @@
 from direct.directnotify import DirectNotifyGlobal
-from direct.distributed.DistributedObjectAI import DistributedObjectAI
+from direct.distributed.DistributedObjectGlobalAI import DistributedObjectGlobalAI
 
-class FriendManagerAI(DistributedObjectAI):
+
+class FriendManagerAI(DistributedObjectGlobalAI):
     notify = DirectNotifyGlobal.directNotify.newCategory('FriendManagerAI')
+
+    def friendQuery(self, inviteeId):
+        self.notify.info('Hello world: %s' % inviteeId)
diff --git a/toontown/ai/ToontownAIRepository.py b/toontown/ai/ToontownAIRepository.py
index 122949b..ae74e88 100644
--- a/toontown/ai/ToontownAIRepository.py
+++ b/toontown/ai/ToontownAIRepository.py
@@ -86,6 +86,7 @@ class ToontownAIRepository(ToontownInternalRepository):
         self.trophyMgr = None
         self.safeZoneManager = None
         self.magicWordManager = None
+        self.friendManager = None
         self.zoneTable = {}
         self.dnaStoreMap = {}
         self.dnaDataMap = {}
@@ -208,6 +209,9 @@ class ToontownAIRepository(ToontownInternalRepository):
         self.magicWordManager = ToontownMagicWordManagerAI(self)
         self.magicWordManager.generateWithRequired(OTP_ZONE_ID_MANAGEMENT)
 
+        # Generate our friend manager...
+        self.friendManager = self.generateGlobalObject(OTP_DO_ID_FRIEND_MANAGER, 'FriendManager')
+
     def generateHood(self, hoodConstructor, zoneId):
         # Bossbot HQ doesn't use DNA, so we skip over that.
         if zoneId != ToontownGlobals.BossbotHQ: