Soulter
88 lines
3.6 KiB
Python
88 lines
3.6 KiB
Python
import asyncio
|
|
|
|
from util.io import port_checker
|
|
from type.register import RegisteredPlatform
|
|
from type.types import Context
|
|
from SparkleLogging.utils.core import LogManager
|
|
from logging import Logger
|
|
from astrbot.message.handler import MessageHandler
|
|
|
|
logger: Logger = LogManager.GetLogger(log_name='astrbot')
|
|
|
|
|
|
class PlatformManager():
|
|
def __init__(self, context: Context, message_handler: MessageHandler) -> None:
|
|
self.context = context
|
|
self.config = context.base_config
|
|
self.msg_handler = message_handler
|
|
|
|
def load_platforms(self):
|
|
tasks = []
|
|
|
|
if 'gocqbot' in self.config and self.config['gocqbot']['enable']:
|
|
logger.info("启用 QQ(nakuru 适配器)")
|
|
tasks.append(asyncio.create_task(self.gocq_bot(), name="nakuru-adapter"))
|
|
|
|
if 'aiocqhttp' in self.config and self.config['aiocqhttp']['enable']:
|
|
logger.info("启用 QQ(aiocqhttp 适配器)")
|
|
tasks.append(asyncio.create_task(self.aiocq_bot(), name="aiocqhttp-adapter"))
|
|
|
|
# QQ频道
|
|
if 'qqbot' in self.config and self.config['qqbot']['enable'] and self.config['qqbot']['appid'] != None:
|
|
logger.info("启用 QQ(官方 API) 机器人消息平台")
|
|
tasks.append(asyncio.create_task(self.qqchan_bot(), name="qqofficial-adapter"))
|
|
|
|
return tasks
|
|
|
|
async def gocq_bot(self):
|
|
'''
|
|
运行 QQ(nakuru 适配器)
|
|
'''
|
|
from model.platform.qq_nakuru import QQGOCQ
|
|
noticed = False
|
|
host = self.config.get("gocq_host", "127.0.0.1")
|
|
port = self.config.get("gocq_websocket_port", 6700)
|
|
http_port = self.config.get("gocq_http_port", 5700)
|
|
logger.info(
|
|
f"正在检查连接...host: {host}, ws port: {port}, http port: {http_port}")
|
|
while True:
|
|
if not port_checker(port=port, host=host) or not port_checker(port=http_port, host=host):
|
|
if not noticed:
|
|
noticed = True
|
|
logger.warning(
|
|
f"连接到{host}:{port}（或{http_port}）失败。程序会每隔 5s 自动重试。")
|
|
await asyncio.sleep(5)
|
|
else:
|
|
logger.info("nakuru 适配器已连接。")
|
|
break
|
|
try:
|
|
qq_gocq = QQGOCQ(self.context, self.msg_handler)
|
|
self.context.platforms.append(RegisteredPlatform(
|
|
platform_name="gocq", platform_instance=qq_gocq, origin="internal"))
|
|
await qq_gocq.run()
|
|
except BaseException as e:
|
|
logger.error("启动 nakuru 适配器时出现错误: " + str(e))
|
|
|
|
def aiocq_bot(self):
|
|
'''
|
|
运行 QQ(aiocqhttp 适配器)
|
|
'''
|
|
from model.platform.qq_aiocqhttp import AIOCQHTTP
|
|
qq_aiocqhttp = AIOCQHTTP(self.context, self.msg_handler)
|
|
self.context.platforms.append(RegisteredPlatform(
|
|
platform_name="aiocqhttp", platform_instance=qq_aiocqhttp, origin="internal"))
|
|
return qq_aiocqhttp.run_aiocqhttp()
|
|
|
|
def qqchan_bot(self):
|
|
'''
|
|
运行 QQ 官方机器人适配器
|
|
'''
|
|
try:
|
|
from model.platform.qq_official import QQOfficial
|
|
qqchannel_bot = QQOfficial(self.context, self.msg_handler)
|
|
self.context.platforms.append(RegisteredPlatform(
|
|
platform_name="qqchan", platform_instance=qqchannel_bot, origin="internal"))
|
|
return qqchannel_bot.run()
|
|
except BaseException as e:
|
|
logger.error("启动 QQ官方机器人适配器时出现错误: " + str(e))
|