Handle emitting capabilities in handshake

Remove superflous message handler

Author: matburt

receptor/protocol.py

@@ -97,7 +97,6 @@ async def wait_greeting(self):
                     self.transport.close()
             await asyncio.sleep(.1)
         logger.debug("handshake complete, starting normal handle loop")
-        self.loop.create_task(self.connection.message_handler(self.incoming_buffer)) # Duplicated (see handle_handshake)?
 
     def handle_handshake(self, data):
         self.greeted = True
@@ -110,6 +109,7 @@ def send_handshake(self):
             "cmd": "HI",
             "id": self.receptor.node_id,
             "expire_time": time.time() + 10,
+            "capabilities": self.receptor.work_manager.get_capabilities(),
         }).encode("utf-8")
         self.transport.write(msg + DELIM)
 

receptor/work.py

@@ -18,6 +18,9 @@ def load_receptor_worker(self, name):
             raise exceptions.UnknownDirective(f"Error loading directive handlers for {name}")
         return entry_points[0].load()
 
+    def get_capabilities(self):
+        return [x.name for x in pkg_resources.iter_entry_points('receptor.worker')]
+
     async def handle(self, inner_env):
         logger.info(f'Handling work for {inner_env.message_id} as {inner_env.directive}')
         namespace, action = inner_env.directive.split(':', 1)