mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
477 lines
15 KiB
Python
477 lines
15 KiB
Python
# 插件管理模块
|
||
import asyncio
|
||
import logging
|
||
import importlib
|
||
import os
|
||
import pkgutil
|
||
import sys
|
||
import shutil
|
||
import traceback
|
||
import re
|
||
|
||
import pkg.utils.updater as updater
|
||
import pkg.utils.context as context
|
||
import pkg.plugin.switch as switch
|
||
import pkg.plugin.settings as settings
|
||
import pkg.qqbot.adapter as msadapter
|
||
import pkg.utils.network as network
|
||
|
||
from mirai import Mirai
|
||
import requests
|
||
|
||
from CallingGPT.session.session import Session
|
||
|
||
__plugins__ = {}
|
||
"""插件列表
|
||
|
||
示例:
|
||
{
|
||
"example": {
|
||
"path": "plugins/example/main.py",
|
||
"enabled: True,
|
||
"name": "example",
|
||
"description": "example",
|
||
"version": "0.0.1",
|
||
"author": "RockChinQ",
|
||
"class": <class 'plugins.example.ExamplePlugin'>,
|
||
"hooks": {
|
||
"person_message": [
|
||
<function ExamplePlugin.person_message at 0x0000020E1D1B8D38>
|
||
]
|
||
},
|
||
"instance": None
|
||
}
|
||
}
|
||
"""
|
||
|
||
__plugins_order__ = []
|
||
"""插件顺序"""
|
||
|
||
__enable_content_functions__ = True
|
||
"""是否启用内容函数"""
|
||
|
||
__callable_functions__ = []
|
||
"""供GPT调用的函数结构"""
|
||
|
||
__function_inst_map__: dict[str, callable] = {}
|
||
"""函数名:实例 映射"""
|
||
|
||
|
||
def generate_plugin_order():
|
||
"""根据__plugin__生成插件初始顺序，无视是否启用"""
|
||
global __plugins_order__
|
||
__plugins_order__ = []
|
||
for plugin_name in __plugins__:
|
||
__plugins_order__.append(plugin_name)
|
||
|
||
|
||
def iter_plugins():
|
||
"""按照顺序迭代插件"""
|
||
for plugin_name in __plugins_order__:
|
||
yield __plugins__[plugin_name]
|
||
|
||
|
||
def iter_plugins_name():
|
||
"""迭代插件名"""
|
||
for plugin_name in __plugins_order__:
|
||
yield plugin_name
|
||
|
||
|
||
__current_module_path__ = ""
|
||
|
||
|
||
def walk_plugin_path(module, prefix='', path_prefix=''):
|
||
global __current_module_path__
|
||
"""遍历插件路径"""
|
||
for item in pkgutil.iter_modules(module.__path__):
|
||
if item.ispkg:
|
||
logging.debug("扫描插件包: plugins/{}".format(path_prefix + item.name))
|
||
walk_plugin_path(__import__(module.__name__ + '.' + item.name, fromlist=['']),
|
||
prefix + item.name + '.', path_prefix + item.name + '/')
|
||
else:
|
||
try:
|
||
logging.debug("扫描插件模块: plugins/{}".format(path_prefix + item.name + '.py'))
|
||
__current_module_path__ = "plugins/"+path_prefix + item.name + '.py'
|
||
|
||
importlib.import_module(module.__name__ + '.' + item.name)
|
||
logging.debug('加载模块: plugins/{} 成功'.format(path_prefix + item.name + '.py'))
|
||
except:
|
||
logging.error('加载模块: plugins/{} 失败: {}'.format(path_prefix + item.name + '.py', sys.exc_info()))
|
||
traceback.print_exc()
|
||
|
||
|
||
def load_plugins():
|
||
"""加载插件"""
|
||
logging.debug("加载插件")
|
||
PluginHost()
|
||
walk_plugin_path(__import__('plugins'))
|
||
|
||
logging.debug(__plugins__)
|
||
|
||
# 加载开关数据
|
||
switch.load_switch()
|
||
|
||
# 生成初始顺序
|
||
generate_plugin_order()
|
||
# 加载插件顺序
|
||
settings.load_settings()
|
||
|
||
# 输出已注册的内容函数列表
|
||
logging.debug("registered content functions: {}".format(__callable_functions__))
|
||
logging.debug("function instance map: {}".format(__function_inst_map__))
|
||
|
||
|
||
def initialize_plugins():
|
||
"""初始化插件"""
|
||
logging.info("初始化插件")
|
||
import pkg.plugin.models as models
|
||
|
||
successfully_initialized_plugins = []
|
||
|
||
for plugin in iter_plugins():
|
||
# if not plugin['enabled']:
|
||
# continue
|
||
try:
|
||
models.__current_registering_plugin__ = plugin['name']
|
||
plugin['instance'] = plugin["class"](plugin_host=context.get_plugin_host())
|
||
# logging.info("插件 {} 已初始化".format(plugin['name']))
|
||
successfully_initialized_plugins.append(plugin['name'])
|
||
except:
|
||
logging.error("插件{}初始化时发生错误: {}".format(plugin['name'], sys.exc_info()))
|
||
|
||
logging.info("以下插件已初始化: {}".format(", ".join(successfully_initialized_plugins)))
|
||
|
||
|
||
def unload_plugins():
|
||
"""卸载插件"""
|
||
# 不再显式卸载插件，因为当程序结束时，插件的析构函数会被系统执行
|
||
# for plugin in __plugins__.values():
|
||
# if plugin['enabled'] and plugin['instance'] is not None:
|
||
# if not hasattr(plugin['instance'], '__del__'):
|
||
# logging.warning("插件{}没有定义析构函数".format(plugin['name']))
|
||
# else:
|
||
# try:
|
||
# plugin['instance'].__del__()
|
||
# logging.info("卸载插件: {}".format(plugin['name']))
|
||
# plugin['instance'] = None
|
||
# except:
|
||
# logging.error("插件{}卸载时发生错误: {}".format(plugin['name'], sys.exc_info()))
|
||
|
||
|
||
def get_github_plugin_repo_label(repo_url: str) -> list[str]:
|
||
"""获取username, repo"""
|
||
|
||
# 提取 username/repo , 正则表达式
|
||
repo = re.findall(r'(?:https?://github\.com/|git@github\.com:)([^/]+/[^/]+?)(?:\.git|/|$)', repo_url)
|
||
|
||
if len(repo) > 0: # github
|
||
return repo[0].split("/")
|
||
else:
|
||
return None
|
||
|
||
|
||
def download_plugin_source_coder(repo_url: str, target_path: str):
|
||
"""下载插件源码"""
|
||
# 检查源类型
|
||
|
||
# 提取 username/repo , 正则表达式
|
||
repo = get_github_plugin_repo_label(repo_url)
|
||
|
||
if repo is not None: # github
|
||
logging.info("从 GitHub 下载插件源码...")
|
||
|
||
zipball_url = f"https://api.github.com/repos/{'/'.join(repo)}/zipball/HEAD"
|
||
|
||
zip_resp = requests.get(
|
||
url=zipball_url,
|
||
proxies=network.wrapper_proxies(),
|
||
stream=True
|
||
)
|
||
|
||
if zip_resp.status_code != 200:
|
||
raise Exception("下载源码失败: {}".format(zip_resp.text))
|
||
|
||
if os.path.exists("temp/"+target_path):
|
||
shutil.rmtree("temp/"+target_path)
|
||
|
||
if os.path.exists(target_path):
|
||
shutil.rmtree(target_path)
|
||
|
||
os.makedirs("temp/"+target_path)
|
||
|
||
with open("temp/"+target_path+"/source.zip", "wb") as f:
|
||
for chunk in zip_resp.iter_content(chunk_size=1024):
|
||
if chunk:
|
||
f.write(chunk)
|
||
|
||
logging.info("下载完成, 解压...")
|
||
import zipfile
|
||
with zipfile.ZipFile("temp/"+target_path+"/source.zip", 'r') as zip_ref:
|
||
zip_ref.extractall("temp/"+target_path)
|
||
os.remove("temp/"+target_path+"/source.zip")
|
||
|
||
# 目标是 username-repo-hash , 用正则表达式提取完整的文件夹名，复制到 plugins/repo
|
||
import glob
|
||
|
||
# 获取解压后的文件夹名
|
||
unzip_dir = glob.glob("temp/"+target_path+"/*")[0]
|
||
|
||
# 复制到 plugins/repo
|
||
shutil.copytree(unzip_dir, target_path+"/")
|
||
|
||
# 删除解压后的文件夹
|
||
shutil.rmtree(unzip_dir)
|
||
|
||
logging.info("解压完成")
|
||
else:
|
||
raise Exception("暂不支持的源类型，请使用 GitHub 仓库发行插件。")
|
||
|
||
|
||
def install_plugin(repo_url: str):
|
||
"""安装插件，从git储存库获取并解决依赖"""
|
||
|
||
repo_label = None
|
||
|
||
repo_label = get_github_plugin_repo_label(repo_url)
|
||
|
||
if repo_label is None:
|
||
raise Exception("暂不支持的源类型，请使用 GitHub 仓库发行插件。")
|
||
|
||
download_plugin_source_coder(repo_url, "plugins/"+repo_label[1])
|
||
|
||
# 检查此目录是否包含requirements.txt
|
||
if os.path.exists("plugins/"+repo_label[1]+"/requirements.txt"):
|
||
logging.info("检测到requirements.txt，正在安装依赖")
|
||
import pkg.utils.pkgmgr
|
||
pkg.utils.pkgmgr.install_requirements("plugins/"+repo_label[1]+"/requirements.txt")
|
||
|
||
import pkg.utils.log as log
|
||
log.reset_logging()
|
||
|
||
|
||
def uninstall_plugin(plugin_name: str) -> str:
|
||
"""卸载插件"""
|
||
if plugin_name not in __plugins__:
|
||
raise Exception("插件不存在")
|
||
|
||
# 获取文件夹路径
|
||
plugin_path = __plugins__[plugin_name]['path'].replace("\\", "/")
|
||
|
||
# 剪切路径为plugins/插件名
|
||
plugin_path = plugin_path.split("plugins/")[1].split("/")[0]
|
||
|
||
# 删除文件夹
|
||
shutil.rmtree("plugins/"+plugin_path)
|
||
return "plugins/"+plugin_path
|
||
|
||
|
||
def update_plugin(plugin_name: str):
|
||
"""更新插件"""
|
||
# 检查是否有远程地址记录
|
||
target_plugin_dir = "plugins/" + __plugins__[plugin_name]['path'].replace("\\", "/").split("plugins/")[1].split("/")[0]
|
||
|
||
remote_url = updater.get_remote_url(target_plugin_dir)
|
||
if remote_url == "https://github.com/RockChinQ/QChatGPT" or remote_url == "https://gitee.com/RockChin/QChatGPT" \
|
||
or remote_url == "" or remote_url is None or remote_url == "http://github.com/RockChinQ/QChatGPT" or remote_url == "http://gitee.com/RockChin/QChatGPT":
|
||
raise Exception("插件没有远程地址记录，无法更新")
|
||
|
||
# 把远程clone到temp/plugins/update/插件名
|
||
logging.info("克隆插件储存库: {}".format(remote_url))
|
||
|
||
from dulwich import porcelain
|
||
clone_target_dir = "temp/plugins/update/"+target_plugin_dir.split("/")[-1]+"/"
|
||
|
||
if os.path.exists(clone_target_dir):
|
||
shutil.rmtree(clone_target_dir)
|
||
|
||
if not os.path.exists(clone_target_dir):
|
||
os.makedirs(clone_target_dir)
|
||
repo = porcelain.clone(remote_url, clone_target_dir, checkout=True)
|
||
|
||
# 检查此目录是否包含requirements.txt
|
||
if os.path.exists(clone_target_dir+"requirements.txt"):
|
||
logging.info("检测到requirements.txt，正在安装依赖")
|
||
import pkg.utils.pkgmgr
|
||
pkg.utils.pkgmgr.install_requirements(clone_target_dir+"requirements.txt")
|
||
|
||
import pkg.utils.log as log
|
||
log.reset_logging()
|
||
|
||
# 将temp/plugins/update/插件名 覆盖到 plugins/插件名
|
||
shutil.rmtree(target_plugin_dir)
|
||
|
||
shutil.copytree(clone_target_dir, target_plugin_dir)
|
||
|
||
class EventContext:
|
||
"""事件上下文"""
|
||
eid = 0
|
||
"""事件编号"""
|
||
|
||
name = ""
|
||
|
||
__prevent_default__ = False
|
||
"""是否阻止默认行为"""
|
||
|
||
__prevent_postorder__ = False
|
||
"""是否阻止后续插件的执行"""
|
||
|
||
__return_value__ = {}
|
||
""" 返回值
|
||
示例:
|
||
{
|
||
"example": [
|
||
'value1',
|
||
'value2',
|
||
3,
|
||
4,
|
||
{
|
||
'key1': 'value1',
|
||
},
|
||
['value1', 'value2']
|
||
]
|
||
}
|
||
"""
|
||
|
||
def add_return(self, key: str, ret):
|
||
"""添加返回值"""
|
||
if key not in self.__return_value__:
|
||
self.__return_value__[key] = []
|
||
self.__return_value__[key].append(ret)
|
||
|
||
def get_return(self, key: str) -> list:
|
||
"""获取key的所有返回值"""
|
||
if key in self.__return_value__:
|
||
return self.__return_value__[key]
|
||
return None
|
||
|
||
def get_return_value(self, key: str):
|
||
"""获取key的首个返回值"""
|
||
if key in self.__return_value__:
|
||
return self.__return_value__[key][0]
|
||
return None
|
||
|
||
def prevent_default(self):
|
||
"""阻止默认行为"""
|
||
self.__prevent_default__ = True
|
||
|
||
def prevent_postorder(self):
|
||
"""阻止后续插件执行"""
|
||
self.__prevent_postorder__ = True
|
||
|
||
def is_prevented_default(self):
|
||
"""是否阻止默认行为"""
|
||
return self.__prevent_default__
|
||
|
||
def is_prevented_postorder(self):
|
||
"""是否阻止后序插件执行"""
|
||
return self.__prevent_postorder__
|
||
|
||
def __init__(self, name: str):
|
||
self.name = name
|
||
self.eid = EventContext.eid
|
||
self.__prevent_default__ = False
|
||
self.__prevent_postorder__ = False
|
||
self.__return_value__ = {}
|
||
EventContext.eid += 1
|
||
|
||
|
||
def emit(event_name: str, **kwargs) -> EventContext:
|
||
"""触发事件"""
|
||
import pkg.utils.context as context
|
||
if context.get_plugin_host() is None:
|
||
return None
|
||
return context.get_plugin_host().emit(event_name, **kwargs)
|
||
|
||
|
||
class PluginHost:
|
||
"""插件宿主"""
|
||
|
||
def __init__(self):
|
||
"""初始化插件宿主"""
|
||
context.set_plugin_host(self)
|
||
self.calling_gpt_session = Session([])
|
||
|
||
def get_runtime_context(self) -> context:
|
||
"""获取运行时上下文（pkg.utils.context模块的对象）
|
||
|
||
此上下文用于和主程序其他模块交互（数据库、QQ机器人、OpenAI接口等）
|
||
详见pkg.utils.context模块
|
||
其中的context变量保存了其他重要模块的类对象，可以使用这些对象进行交互
|
||
"""
|
||
return context
|
||
|
||
def get_bot(self) -> Mirai:
|
||
"""获取机器人对象"""
|
||
return context.get_qqbot_manager().bot
|
||
|
||
def get_bot_adapter(self) -> msadapter.MessageSourceAdapter:
|
||
"""获取消息源适配器"""
|
||
return context.get_qqbot_manager().adapter
|
||
|
||
def send_person_message(self, person, message):
|
||
"""发送私聊消息"""
|
||
self.get_bot_adapter().send_message("person", person, message)
|
||
|
||
def send_group_message(self, group, message):
|
||
"""发送群消息"""
|
||
self.get_bot_adapter().send_message("group", group, message)
|
||
|
||
def notify_admin(self, message):
|
||
"""通知管理员"""
|
||
context.get_qqbot_manager().notify_admin(message)
|
||
|
||
def emit(self, event_name: str, **kwargs) -> EventContext:
|
||
"""触发事件"""
|
||
import json
|
||
|
||
event_context = EventContext(event_name)
|
||
logging.debug("触发事件: {} ({})".format(event_name, event_context.eid))
|
||
for plugin in iter_plugins():
|
||
|
||
if not plugin['enabled']:
|
||
continue
|
||
|
||
# if plugin['instance'] is None:
|
||
# # 从关闭状态切到开启状态之后，重新加载插件
|
||
# try:
|
||
# plugin['instance'] = plugin["class"](plugin_host=self)
|
||
# logging.info("插件 {} 已初始化".format(plugin['name']))
|
||
# except:
|
||
# logging.error("插件 {} 初始化时发生错误: {}".format(plugin['name'], sys.exc_info()))
|
||
# continue
|
||
|
||
if 'hooks' not in plugin or event_name not in plugin['hooks']:
|
||
continue
|
||
|
||
hooks = []
|
||
if event_name in plugin["hooks"]:
|
||
hooks = plugin["hooks"][event_name]
|
||
for hook in hooks:
|
||
try:
|
||
already_prevented_default = event_context.is_prevented_default()
|
||
|
||
kwargs['host'] = context.get_plugin_host()
|
||
kwargs['event'] = event_context
|
||
|
||
hook(plugin['instance'], **kwargs)
|
||
|
||
if event_context.is_prevented_default() and not already_prevented_default:
|
||
logging.debug("插件 {} 已要求阻止事件 {} 的默认行为".format(plugin['name'], event_name))
|
||
|
||
except Exception as e:
|
||
logging.error("插件{}响应事件{}时发生错误".format(plugin['name'], event_name))
|
||
logging.error(traceback.format_exc())
|
||
|
||
# print("done:{}".format(plugin['name']))
|
||
if event_context.is_prevented_postorder():
|
||
logging.debug("插件 {} 阻止了后序插件的执行".format(plugin['name']))
|
||
break
|
||
|
||
logging.debug("事件 {} ({}) 处理完毕，返回值: {}".format(event_name, event_context.eid,
|
||
event_context.__return_value__))
|
||
|
||
return event_context
|
||
|
||
if __name__ == "__main__":
|
||
pass
|