added grpc api to core-daemon, added get sessions and get session rpc calls

daemon/core/grpc/client.py

@@ -16,6 +16,11 @@ class CoreApiClient(object):
     def get_sessions(self):
         return self.stub.GetSessions(core_pb2.SessionsRequest())
 
+    def get_session(self, _id):
+        request = core_pb2.SessionRequest()
+        request.id = _id
+        return self.stub.GetSession(request)
+
     @contextmanager
     def connect(self):
         channel = grpc.insecure_channel(self.address)
@@ -32,6 +37,11 @@ def main():
         response = client.get_sessions()
         print("core client received: %s" % response)
 
+        if len(response.sessions) > 0:
+            session_data = response.sessions[0]
+            session = client.get_session(session_data.id)
+            print(session)
+
 
 if __name__ == "__main__":
     logging.basicConfig()

daemon/core/grpc/server.py

@@ -1,3 +1,5 @@
+from core.enumerations import NodeTypes
+
 from concurrent import futures
 import time
 import logging
@@ -6,23 +8,81 @@ import grpc
 
 import core_pb2
 import core_pb2_grpc
+from core.misc import nodeutils
 
 _ONE_DAY_IN_SECONDS = 60 * 60 * 24
 
 
 class CoreApiServer(core_pb2_grpc.CoreApiServicer):
+    def __init__(self, coreemu):
+        super(CoreApiServer, self).__init__()
+        self.coreemu = coreemu
 
     def GetSessions(self, request, context):
         response = core_pb2.SessionsResponse()
-        session = response.sessions.add()
-        session.id = 1
+        for session_id in self.coreemu.sessions:
+            session = self.coreemu.sessions[session_id]
+            session_data = response.sessions.add()
+            session_data.id = session_id
+            session_data.state = session.state
+            session_data.nodes = session.get_node_count()
+        return response
+
+    def GetSession(self, request, context):
+        session = self.coreemu.sessions.get(request.id)
+        if not request:
+            pass
+
+        response = core_pb2.SessionResponse()
+        for node_id in session.objects:
+            node = session.objects[node_id]
+
+            if not isinstance(node.objid, int):
+                continue
+
+            node_data = response.nodes.add()
+            node_data.id = node.objid
+            node_data.name = node.name
+            node_data.type = nodeutils.get_node_type(node.__class__).value
+            model = getattr(node, "type", None)
+            if model:
+                node_data.model = model
+
+            x = node.position.x
+            if x is not None:
+                node_data.position.x = x
+            y = node.position.y
+            if y is not None:
+                node_data.position.y = y
+            z = node.position.z
+            if z is not None:
+                node_data.position.z = z
+
+            services = getattr(node, "services", [])
+            if services is None:
+                services = []
+            services = [x.name for x in services]
+            node_data.services.extend(services)
+
+            emane_model = None
+            if nodeutils.is_node(node, NodeTypes.EMANE):
+                emane_model = node.model.name
+            if emane_model:
+                node_data.emane = emane_model
+
+            links_data = node.all_link_data(0)
+            for link_data in links_data:
+                pass
+                # link = core_utils.convert_link(session, link_data)
+                # links.append(link)
         return response
 
 
-def main():
+def listen(coreemu, address="[::]:50051"):
+    logging.info("starting grpc api: %s", address)
     server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
-    core_pb2_grpc.add_CoreApiServicer_to_server(CoreApiServer(), server)
-    server.add_insecure_port("[::]:50051")
+    core_pb2_grpc.add_CoreApiServicer_to_server(CoreApiServer(coreemu), server)
+    server.add_insecure_port(address)
     server.start()
 
     try:
@@ -30,8 +90,3 @@ def main():
             time.sleep(_ONE_DAY_IN_SECONDS)
     except KeyboardInterrupt:
         server.stop(0)
-
-
-if __name__ == "__main__":
-    logging.basicConfig()
-    main()

daemon/core/misc/nodeutils.py

@@ -63,6 +63,19 @@ def get_node_class(node_type):
     return _NODE_MAP[node_type]
 
 
+def get_node_type(node_class):
+    """
+    Retrieve the node type given a node class.
+
+    :param class node_class: node class to get type for
+    :return: node type
+    :rtype: core.enumerations.NodeTypes
+    """
+    global _NODE_MAP
+    node_type_map = {v: k for k, v in _NODE_MAP.iteritems()}
+    return node_type_map.get(node_class)
+
+
 def is_node(obj, node_types):
     """
     Validates if an object is one of the provided node types.