[core] Add API for platform core

kibicara/platformapi.py

@@ -0,0 +1,96 @@
+# Copyright (C) 2020 by Thomas Lindner <tom@dl6tom.de>
+# Copyright (C) 2020 by Cathy Hu <cathy.hu@fau.de>
+#
+# SPDX-License-Identifier: 0BSD
+
+from asyncio import create_task, Queue
+from kibicara.model import BadWord, Trigger
+from logging import getLogger
+from re import match
+
+
+logger = getLogger(__name__)
+
+
+class Message:
+    def __init__(self, text, **kwargs):
+        self.text = text
+        self.__dict__.update(kwargs)
+
+
+class Censor:
+    instances = {}
+
+    def __init__(self, hood):
+        self.hood = hood
+        self.inbox = Queue()
+        self.task = None
+        self.hood_censors = self.instances.setdefault(hood.id, [])
+        self.hood_censors.append(self)
+
+    def start(self):
+        if self.task is None:
+            self.task = create_task(self.__run())
+
+    def stop(self):
+        if self.task is not None:
+            self.task.cancel()
+            self.task = None
+
+    async def __run(self):
+        await self.hood.load()
+        self.task.set_name('%s %s' % (self.__class__.__name__, self.hood.name))
+        await self.run()
+
+    # override this in derived class
+    async def run(self):
+        pass
+
+    async def publish(self, message):
+        if not await self.__is_appropriate(message):
+            return
+        for censor in self.hood_censors:
+            await censor.inbox.put(message)
+
+    async def receive(self):
+        return await self.inbox.get()
+
+    async def __is_appropriate(self, message):
+        for badword in await BadWord.objects.filter(hood=self.hood).all():
+            if match(badword.pattern, message.text):
+                return False
+        for trigger in await Trigger.objects.filter(hood=self.hood).all():
+            if match(trigger.pattern, message.text):
+                return True
+        return False
+
+
+class Spawner:
+    instances = []
+
+    def __init__(self, ORMClass, BotClass):
+        self.ORMClass = ORMClass
+        self.BotClass = BotClass
+        self.bots = {}
+        self.instances.append(self)
+
+    @classmethod
+    async def init_all(cls):
+        for spawner in cls.instances:
+            await spawner._init()
+
+    async def _init(self):
+        for item in await self.ORMClass.objects.all():
+            self.start(item)
+
+    def start(self, item):
+        bot = self.bots.setdefault(item.pk, self.BotClass(item))
+        bot.start()
+
+    def stop(self, item):
+        bot = self.bots.pop(item.pk, None)
+        if bot is not None:
+            bot.stop()
+
+    def get(self, item):
+        return self.bots.get(item.pk)