Load plugins directly

Author: berticus2016

plugin_loader.py

@@ -1,36 +1,34 @@
-import os
-import sys
-import importlib
-from pathlib import Path
+from plugins.health_plugin import Plugin as HealthPlugin
+from plugins.map_plugin import Plugin as MapPlugin
+from plugins.mesh_relay_plugin import Plugin as MeshRelayPlugin
+from plugins.ping_plugin import Plugin as PingPlugin
+from plugins.telemetry_plugin import Plugin as TelemetryPlugin
+from plugins.weather_plugin import Plugin as WeatherPlugin
+
 from log_utils import get_logger
 
 logger = get_logger(name="Plugins")
 
-plugins = []
+active_plugins = []
 
 
 def load_plugins():
     global plugins
-    if plugins:
-        return plugins
-
-    plugins = []
-    plugin_dirs = [Path("plugins")]
-
-    for plugin_folder in plugin_dirs:
-        sys.path.insert(0, str(plugin_folder.resolve()))
-
-        for plugin_file in plugin_folder.glob("*.py"):
-            plugin_name = plugin_file.stem
-            if plugin_name == "__init__":
-                continue
-            plugin_module = importlib.import_module(plugin_name)
-            if hasattr(plugin_module, "Plugin"):
-                plugin = plugin_module.Plugin()
-                if plugin.config["active"]:
-                    logger.info(
-                        f"Loaded {os.path.basename(plugin_folder)}/{plugin_name}"
-                    )
-                    plugins.append(plugin)
-
-    return plugins
+    if active_plugins:
+        return active_plugins
+
+    plugins = [
+        HealthPlugin(),
+        MapPlugin(),
+        MeshRelayPlugin(),
+        PingPlugin(),
+        TelemetryPlugin(),
+        WeatherPlugin(),
+    ]
+
+    for plugin in plugins:
+        if plugin.config["active"]:
+            logger.info(f"Loaded {plugin.plugin_name}")
+            active_plugins.append(plugin)
+
+    return active_plugins

plugins/base_plugin.py

@@ -7,7 +7,6 @@
     get_plugin_data_for_node,
     delete_plugin_data,
 )
-from matrix_utils import bot_command
 
 
 class BasePlugin(ABC):
@@ -44,6 +43,8 @@ def get_data(self):
         return get_plugin_data(self.plugin_name)
 
     def matches(self, payload):
+        from matrix_utils import bot_command
+
         if type(payload) == str:
             return bot_command(self.plugin_name, payload)
         return False

plugins/health_plugin.py

@@ -1,15 +1,14 @@
 import re
 import statistics
-from base_plugin import BasePlugin
-
-from matrix_utils import connect_matrix
-from meshtastic_utils import connect_meshtastic
+from plugins.base_plugin import BasePlugin
 
 
 class Plugin(BasePlugin):
     plugin_name = "health"
 
     def generate_response(self):
+        from meshtastic_utils import connect_meshtastic
+
         meshtastic_client = connect_meshtastic()
         battery_levels = []
         air_util_tx = []
@@ -44,6 +43,8 @@ async def handle_meshtastic_message(
         return False
 
     async def handle_room_message(self, room, event, full_message):
+        from matrix_utils import connect_matrix
+
         full_message = full_message.strip()
         if not self.matches(full_message):
             return False

plugins/map_plugin.py

@@ -5,11 +5,7 @@
 import re
 from PIL import Image
 from nio import AsyncClient, UploadResponse
-from base_plugin import BasePlugin
-
-
-from matrix_utils import connect_matrix
-from meshtastic_utils import connect_meshtastic
+from plugins.base_plugin import BasePlugin
 
 
 def anonymize_location(lat, lon, radius=1000):
@@ -93,6 +89,9 @@ async def handle_room_message(self, room, event, full_message):
         if not self.matches(full_message):
             return False
 
+        from matrix_utils import connect_matrix
+        from meshtastic_utils import connect_meshtastic
+
         matrix_client = await connect_matrix()
         meshtastic_client = connect_meshtastic()
 

plugins/mesh_relay_plugin.py

@@ -8,8 +8,6 @@
 
 from plugins.base_plugin import BasePlugin
 from config import relay_config
-from matrix_utils import connect_matrix
-from meshtastic_utils import connect_meshtastic
 
 matrix_rooms: List[dict] = relay_config["matrix_rooms"]
 
@@ -57,6 +55,8 @@ def process(self, packet):
     async def handle_meshtastic_message(
         self, packet, formatted_message, longname, meshnet_name
     ):
+        from matrix_utils import connect_matrix
+
         packet = self.process(packet)
         matrix_client = await connect_matrix()
 
@@ -121,6 +121,8 @@ async def handle_room_message(self, room, event, full_message):
             self.logger.error(f"Error processing embedded packet: {e}")
             return
 
+        from meshtastic_utils import connect_meshtastic
+
         meshtastic_client = connect_meshtastic()
         meshPacket = mesh_pb2.MeshPacket()
         meshPacket.channel = channel

plugins/ping.py

@@ -2,8 +2,6 @@
 import requests
 
 from plugins.base_plugin import BasePlugin
-from matrix_utils import connect_matrix
-from meshtastic_utils import connect_meshtastic
 
 
 class Plugin(BasePlugin):
@@ -80,6 +78,8 @@ async def handle_meshtastic_message(
             if f"!{self.plugin_name}" not in message:
                 return False
 
+            from meshtastic_utils import connect_meshtastic
+
             meshtastic_client = connect_meshtastic()
             if packet["fromId"] in meshtastic_client.nodes:
                 weather_notice = "Cannot determine location"