From 3fe7dccae14d3dc387e41d1e22a2ec8050890846 Mon Sep 17 00:00:00 2001
From: Sam Edwards <CFSworks@gmail.com>
Date: Tue, 6 May 2014 10:18:39 -0600
Subject: [PATCH] distributed: Add function to query objects' activated status
 from AstronInternalRepository.

---
 .../distributed/AstronInternalRepository.py   | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/direct/src/distributed/AstronInternalRepository.py b/direct/src/distributed/AstronInternalRepository.py
index 311ec3e4d5..fcaf56e1ad 100644
--- a/direct/src/distributed/AstronInternalRepository.py
+++ b/direct/src/distributed/AstronInternalRepository.py
@@ -46,6 +46,7 @@ class AstronInternalRepository(ConnectionRepository):
         self.__contextCounter = 0
 
         self.dbInterface = AstronDatabaseInterface(self)
+        self.__callbacks = {}
 
         self.ourChannel = self.allocateChannel()
 
@@ -157,6 +158,8 @@ class AstronInternalRepository(ConnectionRepository):
                          DBSERVER_OBJECT_SET_FIELD_IF_EQUALS_RESP,
                          DBSERVER_OBJECT_SET_FIELDS_IF_EQUALS_RESP):
             self.dbInterface.handleDatagram(msgType, di)
+        elif msgType == DBSS_OBJECT_GET_ACTIVATED_RESP:
+            self.handleGetActivatedResp(di)
         else:
             self.notify.warning('Received message with unknown MsgType=%d' % msgType)
 
@@ -215,6 +218,30 @@ class AstronInternalRepository(ConnectionRepository):
         do.delete()
         do.sendDeleteEvent()
 
+    def handleGetActivatedResp(self, di):
+        ctx = di.getUint32()
+        doId = di.getUint32()
+        activated = di.getUint8()
+
+        if ctx not in self.__callbacks:
+            self.notify.warning('Received unexpected DBSS_OBJECT_GET_ACTIVATED_RESP (ctx: %d)' %ctx)
+            return
+
+        try:
+            self.__callbacks[ctx](doId, activated)
+        finally:
+            del self.__callbacks[ctx]
+
+    def getActivated(self, doId, callback):
+        ctx = self.getContext()
+        self.__callbacks[ctx] = callback
+
+        dg = PyDatagram()
+        dg.addServerHeader(doId, self.ourChannel, DBSS_OBJECT_GET_ACTIVATED)
+        dg.addUint32(ctx)
+        dg.addUint32(doId)
+        self.send(dg)
+
     def sendUpdate(self, do, fieldName, args):
         """
         Send a field update for the given object.