main initializes services from plugins

Author: GabrielSalla

src/main.py

@@ -11,6 +11,7 @@
 import databases as databases
 import internal_database as internal_database
 import message_queue as message_queue
+import plugins as plugins
 import registry as registry
 import utils.app as app
 import utils.log as log
@@ -25,6 +26,21 @@ async def protected_task(task):
         _logger.warning(f"Exception with task '{task}'")
 
 
+async def init_plugins_services(controller_enabled: bool, executor_enabled: bool):
+    """Initialize the plugins services"""
+    for plugin_name, plugin in plugins.loaded_plugins.items():
+        _logger.info(f"Loading plugin {plugin_name}")
+
+        plugin_services = getattr(plugin, "services", None)
+        if plugin_services is None:
+            continue
+
+        for service_name in plugin_services.__all__:
+            service = getattr(plugin_services, service_name)
+            if hasattr(service, "init"):
+                await service.init(controller_enabled, executor_enabled)
+
+
 async def init(controller_enabled: bool, executor_enabled: bool):
     """Initialize the application dependencies. Some of the components will behave differently if
     they start with or without the controller."""
@@ -38,13 +54,31 @@ async def init(controller_enabled: bool, executor_enabled: bool):
     await message_queue.init()
     await http_server.init(controller_enabled)
 
+    await init_plugins_services(controller_enabled, executor_enabled)
+
+
+async def stop_plugins_services():
+    """Stop the plugins services"""
+    for plugin_name, plugin in plugins.loaded_plugins.items():
+        _logger.info(f"Loading plugin {plugin_name}")
+
+        plugin_services = getattr(plugin, "services", None)
+        if plugin_services is None:
+            continue
+
+        for service_name in plugin_services.__all__:
+            service = getattr(plugin_services, service_name)
+            if hasattr(service, "stop"):
+                await protected_task(service.stop())
+
 
 async def finish():
     """Finish the application, making sure any exception won't impact other closing tasks"""
     await protected_task(http_server.wait_stop())
     await protected_task(monitors_loader.wait_stop())
     await protected_task(databases.close())
     await protected_task(internal_database.close())
+    await protected_task(stop_plugins_services())
 
 
 async def main():