\m/

stdborg.py

@@ -0,0 +1,9 @@
+import logging
+
+from uniborg import Uniborg
+
+logging.basicConfig(level=logging.INFO)
+
+borg = Uniborg("stdborg", plugin_path="stdplugins")
+
+borg.run()

stdplugins/testplugin.py

@@ -0,0 +1,5 @@
+from telethon import events
+
+@borg.on(events.NewMessage)
+async def asdf(e):
+    print(e.raw_text)

test.py

@@ -1,13 +0,0 @@
-from uniborg import Uniborg
-from telethon import events
-
-borg = Uniborg("uniborg")
-
-print(borg.uid)
-print(dir(borg))
-
-@borg.on(events.NewMessage)
-async def asdf(e):
-    print(e.raw_text)
-
-borg.run()

uniborg/hacks.py

@@ -0,0 +1,7 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+class ReverseList(list):
+    def __iter__(self):
+        return reversed(self)

uniborg/telethon.py

@@ -3,25 +3,62 @@
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 import asyncio
+import importlib.util
 import logging
+from pathlib import Path
 
 from telethon import TelegramClient
 import telethon.utils
 
+from . import hacks
+
 class Uniborg(TelegramClient):
-    def __init__(self, session, **kwargs):
+    def __init__(self, session, *, plugin_path="plugins", **kwargs):
-        self.logger = logging.getLogger(session)
+        # TODO: handle non-string session
+        self._name = session
+        self._logger = logging.getLogger(session)
+        self._plugins = {}
+        self._plugin_path = plugin_path
 
         super().__init__(session,
                 17349, "344583e45741c457fe1862106095a5eb", # yarr
                 **kwargs)
 
-        asyncio.get_event_loop().run_until_complete(self.async_init())
+        # This is a hack, please avert your eyes
+        # We want this in order for the most recently added handler to take
+        # precedence
+        self._event_builders = hacks.ReverseList()
 
-    async def async_init(self):
+        self._loop.run_until_complete(self._async_init())
+
+        core_plugin = Path(__file__).parent / "_core.py"
+        self.load_plugin_from_file(core_plugin)
+
+        for p in Path().glob(f"{self._plugin_path}/*.py"):
+            self.load_plugin_from_file(p)
+
+    async def _async_init(self):
         await self.start()
 
         self.uid = telethon.utils.get_peer_id(await self.get_me())
 
     def run(self):
-        asyncio.get_event_loop().run_forever()
+        self._loop.run_forever()
+
+    def load_plugin(self, shortname):
+        self.load_plugin_from_file(f"{self._plugin_path}/{shortname}.py")
+
+    def load_plugin_from_file(self, path):
+        path = Path(path)
+        shortname = path.stem
+        name = f"_UniborgPlugins.{self._name}.{shortname}"
+
+        spec = importlib.util.spec_from_file_location(name, path)
+        mod = importlib.util.module_from_spec(spec)
+
+        mod.borg = self
+        mod.logger = logging.getLogger(shortname)
+
+        spec.loader.exec_module(mod)
+        self._plugins[shortname] = mod
+        self._logger.info(f"Successfully loaded plugin {shortname}")