mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
264 lines
9.1 KiB
Python
264 lines
9.1 KiB
Python
import os
|
|
import shutil
|
|
import tarfile
|
|
import zipfile
|
|
import subprocess
|
|
|
|
from nonebot.adapters import Bot
|
|
from nonebot.utils import run_sync
|
|
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.http_utils import AsyncHttpx
|
|
from zhenxun.utils.platform import PlatformUtils
|
|
from zhenxun.utils.github_utils.models import RepoInfo
|
|
from zhenxun.utils.github_utils import parse_github_url
|
|
|
|
from .config import (
|
|
TMP_PATH,
|
|
BASE_PATH,
|
|
BACKUP_PATH,
|
|
RELEASE_URL,
|
|
REQ_TXT_FILE,
|
|
VERSION_FILE,
|
|
PYPROJECT_FILE,
|
|
REPLACE_FOLDERS,
|
|
BASE_PATH_STRING,
|
|
DOWNLOAD_GZ_FILE,
|
|
DOWNLOAD_ZIP_FILE,
|
|
DEFAULT_GITHUB_URL,
|
|
PYPROJECT_LOCK_FILE,
|
|
REQ_TXT_FILE_STRING,
|
|
PYPROJECT_FILE_STRING,
|
|
PYPROJECT_LOCK_FILE_STRING,
|
|
)
|
|
|
|
|
|
def install_requirement():
|
|
requirement_path = (REQ_TXT_FILE).absolute()
|
|
|
|
if not requirement_path.exists():
|
|
logger.debug(
|
|
f"没有找到zhenxun的requirement.txt,目标路径为{requirement_path}", "插件管理"
|
|
)
|
|
return
|
|
try:
|
|
result = subprocess.run(
|
|
["pip", "install", "-r", str(requirement_path)],
|
|
check=True,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
logger.debug(f"成功安装真寻依赖，日志:\n{result.stdout}", "插件管理")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"安装真寻依赖失败，错误:\n{e.stderr}", "插件管理", e=e)
|
|
|
|
|
|
@run_sync
|
|
def _file_handle(latest_version: str | None):
|
|
"""文件移动操作
|
|
|
|
参数:
|
|
latest_version: 版本号
|
|
"""
|
|
BACKUP_PATH.mkdir(exist_ok=True, parents=True)
|
|
logger.debug("开始解压文件压缩包...", "检查更新")
|
|
download_file = DOWNLOAD_GZ_FILE
|
|
if DOWNLOAD_GZ_FILE.exists():
|
|
tf = tarfile.open(DOWNLOAD_GZ_FILE)
|
|
else:
|
|
download_file = DOWNLOAD_ZIP_FILE
|
|
tf = zipfile.ZipFile(DOWNLOAD_ZIP_FILE)
|
|
tf.extractall(TMP_PATH)
|
|
logger.debug("解压文件压缩包完成...", "检查更新")
|
|
download_file_path = TMP_PATH / next(
|
|
x for x in os.listdir(TMP_PATH) if (TMP_PATH / x).is_dir()
|
|
)
|
|
_pyproject = download_file_path / PYPROJECT_FILE_STRING
|
|
_lock_file = download_file_path / PYPROJECT_LOCK_FILE_STRING
|
|
_req_file = download_file_path / REQ_TXT_FILE_STRING
|
|
extract_path = download_file_path / BASE_PATH_STRING
|
|
target_path = BASE_PATH
|
|
if PYPROJECT_FILE.exists():
|
|
logger.debug(f"移除备份文件: {PYPROJECT_FILE}", "检查更新")
|
|
shutil.move(PYPROJECT_FILE, BACKUP_PATH / PYPROJECT_FILE_STRING)
|
|
if PYPROJECT_LOCK_FILE.exists():
|
|
logger.debug(f"移除备份文件: {PYPROJECT_LOCK_FILE}", "检查更新")
|
|
shutil.move(PYPROJECT_LOCK_FILE, BACKUP_PATH / PYPROJECT_LOCK_FILE_STRING)
|
|
if REQ_TXT_FILE.exists():
|
|
logger.debug(f"移除备份文件: {REQ_TXT_FILE}", "检查更新")
|
|
shutil.move(REQ_TXT_FILE, BACKUP_PATH / REQ_TXT_FILE_STRING)
|
|
if _pyproject.exists():
|
|
logger.debug("移动文件: pyproject.toml", "检查更新")
|
|
shutil.move(_pyproject, PYPROJECT_FILE)
|
|
if _lock_file.exists():
|
|
logger.debug("移动文件: poetry.lock", "检查更新")
|
|
shutil.move(_lock_file, PYPROJECT_LOCK_FILE)
|
|
if _req_file.exists():
|
|
logger.debug("移动文件: requirements.txt", "检查更新")
|
|
shutil.move(_req_file, REQ_TXT_FILE)
|
|
for folder in REPLACE_FOLDERS:
|
|
"""移动指定文件夹"""
|
|
_dir = BASE_PATH / folder
|
|
_backup_dir = BACKUP_PATH / folder
|
|
if _backup_dir.exists():
|
|
logger.debug(f"删除备份文件夹 {_backup_dir}", "检查更新")
|
|
shutil.rmtree(_backup_dir)
|
|
if _dir.exists():
|
|
logger.debug(f"移动旧文件夹 {_dir}", "检查更新")
|
|
shutil.move(_dir, _backup_dir)
|
|
else:
|
|
logger.warning(f"文件夹 {_dir} 不存在，跳过删除", "检查更新")
|
|
for folder in REPLACE_FOLDERS:
|
|
src_folder_path = extract_path / folder
|
|
dest_folder_path = target_path / folder
|
|
if src_folder_path.exists():
|
|
logger.debug(
|
|
f"移动文件夹: {src_folder_path} -> {dest_folder_path}", "检查更新"
|
|
)
|
|
shutil.move(src_folder_path, dest_folder_path)
|
|
else:
|
|
logger.debug(f"源文件夹不存在: {src_folder_path}", "检查更新")
|
|
if tf:
|
|
tf.close()
|
|
if download_file.exists():
|
|
logger.debug(f"删除下载文件: {download_file}", "检查更新")
|
|
download_file.unlink()
|
|
if extract_path.exists():
|
|
logger.debug(f"删除解压文件夹: {extract_path}", "检查更新")
|
|
shutil.rmtree(extract_path)
|
|
if TMP_PATH.exists():
|
|
shutil.rmtree(TMP_PATH)
|
|
if latest_version:
|
|
with open(VERSION_FILE, "w", encoding="utf8") as f:
|
|
f.write(f"__version__: {latest_version}")
|
|
install_requirement()
|
|
|
|
|
|
class UpdateManage:
|
|
@classmethod
|
|
async def check_version(cls) -> str:
|
|
"""检查更新版本
|
|
|
|
返回:
|
|
str: 更新信息
|
|
"""
|
|
cur_version = cls.__get_version()
|
|
data = await cls.__get_latest_data()
|
|
if not data:
|
|
return "检查更新获取版本失败..."
|
|
return (
|
|
"检测到当前版本更新\n"
|
|
f"当前版本：{cur_version}\n"
|
|
f"最新版本：{data.get('name')}\n"
|
|
f"创建日期：{data.get('created_at')}\n"
|
|
f"更新内容：\n{data.get('body')}"
|
|
)
|
|
|
|
@classmethod
|
|
async def update(cls, bot: Bot, user_id: str, version_type: str) -> str | None:
|
|
"""更新操作
|
|
|
|
参数:
|
|
bot: Bot
|
|
user_id: 用户id
|
|
version_type: 更新版本类型
|
|
|
|
返回:
|
|
str | None: 返回消息
|
|
"""
|
|
logger.info("开始下载真寻最新版文件....", "检查更新")
|
|
cur_version = cls.__get_version()
|
|
url = None
|
|
new_version = None
|
|
repo_info = parse_github_url(DEFAULT_GITHUB_URL)
|
|
if version_type in {"dev", "main"}:
|
|
repo_info.branch = version_type
|
|
new_version = await cls.__get_version_from_repo(repo_info)
|
|
if new_version:
|
|
new_version = new_version.split(":")[-1].strip()
|
|
url = await repo_info.get_archive_download_url()
|
|
elif version_type == "release":
|
|
data = await cls.__get_latest_data()
|
|
if not data:
|
|
return "获取更新版本失败..."
|
|
new_version = data.get("name", "")
|
|
url = await repo_info.get_release_source_download_url_tgz(new_version)
|
|
if not url:
|
|
return "获取版本下载链接失败..."
|
|
if TMP_PATH.exists():
|
|
logger.debug(f"删除临时文件夹 {TMP_PATH}", "检查更新")
|
|
shutil.rmtree(TMP_PATH)
|
|
logger.debug(
|
|
f"开始更新版本：{cur_version} -> {new_version} | 下载链接：{url}",
|
|
"检查更新",
|
|
)
|
|
await PlatformUtils.send_superuser(
|
|
bot,
|
|
f"检测真寻已更新，版本更新：{cur_version} -> {new_version}\n开始更新...",
|
|
user_id,
|
|
)
|
|
download_file = (
|
|
DOWNLOAD_GZ_FILE if version_type == "release" else DOWNLOAD_ZIP_FILE
|
|
)
|
|
if await AsyncHttpx.download_file(url, download_file):
|
|
logger.debug("下载真寻最新版文件完成...", "检查更新")
|
|
await _file_handle(new_version)
|
|
return (
|
|
f"版本更新完成\n"
|
|
f"版本: {cur_version} -> {new_version}\n"
|
|
"请重新启动真寻以完成更新!"
|
|
)
|
|
else:
|
|
logger.debug("下载真寻最新版文件失败...", "检查更新")
|
|
return None
|
|
|
|
@classmethod
|
|
def __get_version(cls) -> str:
|
|
"""获取当前版本
|
|
|
|
返回:
|
|
str: 当前版本号
|
|
"""
|
|
_version = "v0.0.0"
|
|
if VERSION_FILE.exists():
|
|
if text := VERSION_FILE.open(encoding="utf8").readline():
|
|
_version = text.split(":")[-1].strip()
|
|
return _version
|
|
|
|
@classmethod
|
|
async def __get_latest_data(cls) -> dict:
|
|
"""获取最新版本信息
|
|
|
|
返回:
|
|
dict: 最新版本数据
|
|
"""
|
|
for _ in range(3):
|
|
try:
|
|
res = await AsyncHttpx.get(RELEASE_URL)
|
|
if res.status_code == 200:
|
|
return res.json()
|
|
except TimeoutError:
|
|
pass
|
|
except Exception as e:
|
|
logger.error("检查更新真寻获取版本失败", e=e)
|
|
return {}
|
|
|
|
@classmethod
|
|
async def __get_version_from_repo(cls, repo_info: RepoInfo) -> str:
|
|
"""从指定分支获取版本号
|
|
|
|
参数:
|
|
branch: 分支名称
|
|
|
|
返回:
|
|
str: 版本号
|
|
"""
|
|
version_url = await repo_info.get_raw_download_url(path="__version__")
|
|
try:
|
|
res = await AsyncHttpx.get(version_url)
|
|
if res.status_code == 200:
|
|
return res.text.strip()
|
|
except Exception as e:
|
|
logger.error(f"获取 {repo_info.branch} 分支版本失败", e=e)
|
|
return "未知版本"
|