重构webui主页与好友/群组api

plugins/web_ui/api/tabs/main/data_source.py

@@ -0,0 +1,36 @@
+import time
+from typing import Optional
+
+import nonebot
+from nonebot import Driver
+from nonebot.adapters.onebot.v12 import Bot
+
+driver: Driver = nonebot.get_driver()
+
+
+class BotLive:
+    def __init__(self):
+        self._data = {}
+
+    def add(self, bot_id: str):
+        self._data[bot_id] = time.time()
+
+    def get(self, bot_id: str) -> Optional[int]:
+        return self._data.get(bot_id)
+
+    def remove(self, bot_id: str):
+        if bot_id in self._data:
+            del self._data[bot_id]
+
+
+bot_live = BotLive()
+
+
+@driver.on_bot_connect
+async def _(bot: Bot):
+    bot_live.add(bot.self_id)
+
+
+@driver.on_bot_disconnect
+async def _(bot: Bot):
+    bot_live.remove(bot.self_id)