forked from capedayo/minya_hotomoe
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
|
from pyfiglet import Figlet
|
||
|
|
f = Figlet(font='small')
|
||
|
|
|
||
|
|
import os
|
||
|
|
import importlib.util
|
||
|
|
import re
|
||
|
|
import inspect
|
||
|
|
from dotenv import load_dotenv
|
||
|
|
|
||
|
|
from module_interface import ModuleInterface, ModuleManager
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from aiohttp import ClientWebSocketResponse
|
||
|
|
from mipac.models.notification import NotificationNote
|
||
|
|
from mipa.ext.commands.bot import Bot
|
||
|
|
|
||
|
|
def load_modules():
|
||
|
|
module_dir = "modules"
|
||
|
|
modules = []
|
||
|
|
for filename in os.listdir(module_dir):
|
||
|
|
if filename.endswith(".py"):
|
||
|
|
module_name = os.path.splitext(filename)[0]
|
||
|
|
module_path = os.path.join(module_dir, filename)
|
||
|
|
|
||
|
|
spec = importlib.util.spec_from_file_location(module_name, module_path)
|
||
|
|
module = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(module)
|
||
|
|
|
||
|
|
for attribute_name in dir(module):
|
||
|
|
attribute = getattr(module, attribute_name)
|
||
|
|
if inspect.isclass(attribute) and issubclass(attribute, ModuleInterface) and attribute != ModuleInterface:
|
||
|
|
modules.append(attribute())
|
||
|
|
return modules
|
||
|
|
|
||
|
|
class MiNya(Bot):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
self.__modules = []
|
||
|
|
self.__manager = ModuleManager(self)
|
||
|
|
|
||
|
|
async def _connect_channel(self):
|
||
|
|
await self.router.connect_channel(['main'])
|
||
|
|
async def on_reconnect(self, ws: ClientWebSocketResponse):
|
||
|
|
await self._connect_channel()
|
||
|
|
|
||
|
|
async def on_ready(self, ws: ClientWebSocketResponse):
|
||
|
|
await self._connect_channel()
|
||
|
|
print("[Main] Bot System Ready.")
|
||
|
|
print("")
|
||
|
|
print("[Main] Loading Modules...")
|
||
|
|
print("")
|
||
|
|
self.__modules = load_modules()
|
||
|
|
print("")
|
||
|
|
print(f"[Main] Loaded Modules : {len(self.__modules)}")
|
||
|
|
print("")
|
||
|
|
|
||
|
|
print("[Main] Registering Modules...")
|
||
|
|
|
||
|
|
for module in self.__modules:
|
||
|
|
self.__manager.register_module(module)
|
||
|
|
print("")
|
||
|
|
|
||
|
|
async def on_mention(self, notice: NotificationNote):
|
||
|
|
print("[Main] Got Mentioned!")
|
||
|
|
for _, module in self.__manager.modules.items():
|
||
|
|
if module.regex_pattern and re.match(module.regex_pattern, notice.note.content):
|
||
|
|
print("[Main] Regex Matching Module found : ", module.name)
|
||
|
|
if await module.execute_module(notice.note):
|
||
|
|
break
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print(f.renderText('MiNya V1'))
|
||
|
|
print('"MIsskey Networked Yet Ambitious"')
|
||
|
|
print('')
|
||
|
|
print("[Main] Minya V1 Bootstap...")
|
||
|
|
print("[Main] Loading settings...")
|
||
|
|
load_dotenv()
|
||
|
|
instance = os.environ.get('BOT_INSTANCE')
|
||
|
|
key = os.environ.get('BOT_KEY')
|
||
|
|
|
||
|
|
if not (instance and key):
|
||
|
|
print("[MAIN] ERR! please check .env or os environment!")
|
||
|
|
return
|
||
|
|
bot = MiNya()
|
||
|
|
asyncio.run(bot.start(f'wss://{instance}/streaming', key))
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|