mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
86 lines
3.8 KiB
Python
86 lines
3.8 KiB
Python
import threading, traceback, uuid
|
||
from .. import Route, Response, logger
|
||
from quart import Quart, request
|
||
from type.types import Context
|
||
from model.plugin.manager import PluginManager
|
||
from util.updator.astrbot_updator import AstrBotUpdator
|
||
|
||
class PluginRoute(Route):
|
||
def __init__(self, context: Context, app: Quart, astrbot_updator: AstrBotUpdator, plugin_manager: PluginManager) -> None:
|
||
super().__init__(context, app)
|
||
self.routes = {
|
||
'/plugin/get': ('GET', self.get_plugins),
|
||
'/plugin/install': ('POST', self.install_plugin),
|
||
'/plugin/install-upload': ('POST', self.install_plugin_upload),
|
||
'/plugin/update': ('POST', self.update_plugin),
|
||
'/plugin/uninstall': ('POST', self.uninstall_plugin),
|
||
}
|
||
self.astrbot_updator = astrbot_updator
|
||
self.plugin_manager = plugin_manager
|
||
self.register_routes()
|
||
|
||
async def get_plugins(self):
|
||
_plugin_resp = []
|
||
for plugin in self.context.cached_plugins:
|
||
_p = plugin.metadata
|
||
_t = {
|
||
"name": _p.plugin_name,
|
||
"repo": '' if _p.repo is None else _p.repo,
|
||
"author": _p.author,
|
||
"desc": _p.desc,
|
||
"version": _p.version
|
||
}
|
||
_plugin_resp.append(_t)
|
||
return Response().ok(_plugin_resp).__dict__
|
||
|
||
async def install_plugin(self):
|
||
post_data = await request.json
|
||
repo_url = post_data["url"]
|
||
try:
|
||
logger.info(f"正在安装插件 {repo_url}")
|
||
await self.plugin_manager.install_plugin(repo_url)
|
||
threading.Thread(target=self.astrbot_updator._reboot, args=(2, self.context)).start()
|
||
logger.info(f"安装插件 {repo_url} 成功, 2秒后重启")
|
||
return Response().ok(None, "安装成功，程序将在 2 秒内重启。").__dict__
|
||
except Exception as e:
|
||
logger.error(traceback.format_exc())
|
||
return Response().error(str(e)).__dict__
|
||
|
||
async def install_plugin_upload(self):
|
||
try:
|
||
file = request.files['file']
|
||
print(file.filename)
|
||
logger.info(f"正在安装用户上传的插件 {file.filename}")
|
||
file_path = f"data/temp/{uuid.uuid4()}.zip"
|
||
file.save(file_path)
|
||
self.plugin_manager.install_plugin_from_file(file_path)
|
||
logger.info(f"安装插件 {file.filename} 成功")
|
||
return Response().ok(None, "安装成功！！").__dict__
|
||
except Exception as e:
|
||
logger.error(traceback.format_exc())
|
||
return Response().error(str(e)).__dict__
|
||
|
||
async def uninstall_plugin(self):
|
||
post_data = await request.json
|
||
plugin_name = post_data["name"]
|
||
try:
|
||
logger.info(f"正在卸载插件 {plugin_name}")
|
||
self.plugin_manager.uninstall_plugin(plugin_name)
|
||
logger.info(f"卸载插件 {plugin_name} 成功")
|
||
return Response().ok(None, "卸载成功").__dict__
|
||
except Exception as e:
|
||
logger.error(traceback.format_exc())
|
||
return Response().error(str(e)).__dict__
|
||
|
||
async def update_plugin(self):
|
||
post_data = await request.json
|
||
plugin_name = post_data["name"]
|
||
try:
|
||
logger.info(f"正在更新插件 {plugin_name}")
|
||
await self.plugin_manager.update_plugin(plugin_name)
|
||
threading.Thread(target=self.astrbot_updator._reboot, args=(2, self.context)).start()
|
||
logger.info(f"更新插件 {plugin_name} 成功，2秒后重启")
|
||
return Response().ok(None, "更新成功，程序将在 2 秒内重启。").__dict__
|
||
except Exception as e:
|
||
logger.error(f"/api/extensions/update: {traceback.format_exc()}")
|
||
return Response().error(str(e)).__dict__ |