From b808d08eb3bfb221d16b788d5ff7fae23e764a99 Mon Sep 17 00:00:00 2001
From: Jack Robison <jackrobison@lbry.io>
Date: Mon, 2 Apr 2018 16:49:48 -0400
Subject: [PATCH] support callbacks to component setups

---
 lbrynet/daemon/ComponentManager.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/lbrynet/daemon/ComponentManager.py b/lbrynet/daemon/ComponentManager.py
index 8b645c014..0cb7057dc 100644
--- a/lbrynet/daemon/ComponentManager.py
+++ b/lbrynet/daemon/ComponentManager.py
@@ -50,15 +50,27 @@ class ComponentManager(object):
 
     @classmethod
     @defer.inlineCallbacks
-    def setup(cls):
+    def setup(cls, **callbacks):
         """
         Start Components in sequence sorted by requirements
 
         :return: (defer.Deferred)
         """
+        for component_name, cb in callbacks.iteritems():
+            if not callable(cb):
+                raise ValueError("%s is not callable" % cb)
+            cls.get_component(component_name)
+
+        def _setup(component):
+            if component.component_name in callbacks:
+                d = component._setup()
+                d.addCallback(callbacks[component.component_name])
+                return d
+            return component.setup()
+
         stages = cls.sort_components()
         for stage in stages:
-            yield defer.DeferredList([component._setup() for component in stage])
+            yield defer.DeferredList([_setup(component) for component in stage])
 
     @classmethod
     @defer.inlineCallbacks
@@ -75,13 +87,15 @@ class ComponentManager(object):
     @classmethod
     def all_components_running(cls, *component_names):
         """
+        Check if components are running
+
         :return: (bool) True if all specified components are running
         """
-        c = {component.component_name: component for component in cls.components}
+        components = {component.component_name: component for component in cls.components}
         for component in component_names:
-            if component not in c:
+            if component not in components:
                 raise NameError("%s is not a known Component" % component)
-            if not c[component].running:
+            if not components[component].running:
                 return False
         return True