✨ 自动更新

2025-12-15 06:12:53 +08:00 · 2024-08-19 23:07:43 +08:00 · 2024-08-19 23:07:43 +08:00 · 1c58a8f021
commit 1c58a8f021
parent 09f586be25
4 changed files with 282 additions and 1 deletions
--- a/zhenxun/plugins/auto_update/init.py
+++ b/zhenxun/plugins/auto_update/init.py
@ -0,0 +1,67 @@
+from nonebot.adapters import Bot
+from nonebot.permission import SUPERUSER
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import Alconna, Args, Match, on_alconna
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.utils import PluginExtraData, RegisterConfig
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType
+from zhenxun.utils.message import MessageUtils
+
+from ._data_source import UpdateManage
+
+__plugin_meta__ = PluginMetadata(
+    name="自动更新",
+    description="就算是真寻也会成长的",
+    usage="""
+    usage：
+        检查更新真寻最新版本，包括了自动更新
+        指令：
+            检查更新真寻
+    """.strip(),
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        plugin_type=PluginType.SUPERUSER,
+        configs=[
+            RegisterConfig(
+                key="UPDATE_REMIND",
+                value=True,
+                help="是否检测更新版本",
+                default_value=True,
+            ),
+            RegisterConfig(
+                key="UPDATE_REMIND",
+                value=True,
+                help="是否检测更新版本",
+                default_value=True,
+            ),
+        ],
+    ).dict(),
+)
+
+_matcher = on_alconna(
+    Alconna("检查更新", Args["ver_type?", ["main", "dev", "release"]]),
+    priority=1,
+    block=True,
+    permission=SUPERUSER,
+)
+
+
+@_matcher.handle()
+async def _(bot: Bot, session: EventSession, ver_type: Match[str]):
+    if not session.id1:
+        await MessageUtils.build_message("用户id为空...").finish()
+    if not ver_type.available:
+        result = await UpdateManage.check_version()
+        logger.info("查看当前版本...", "检查更新", session=session)
+        await MessageUtils.build_message(result).finish()
+    try:
+        result = await UpdateManage.update(bot, session.id1, ver_type.result)
+    except Exception as e:
+        logger.error("版本更新失败...", "检查更新", session=session, e=e)
+        await MessageUtils.build_message(f"更新版本失败...e: {e}").finish()
+    if result:
+        await MessageUtils.build_message(result).finish()
+    await MessageUtils.build_message("更新版本失败...").finish()
--- a/zhenxun/plugins/auto_update/_data_source.py
+++ b/zhenxun/plugins/auto_update/_data_source.py
@ -0,0 +1,189 @@
+import os
+import shutil
+import tarfile
+from pathlib import Path
+
+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 .config import (
+    BACKUP_PATH,
+    BASE_PATH,
+    DEV_URL,
+    DOWNLOAD_FILE,
+    MAIN_URL,
+    PYPROJECT_FILE,
+    PYPROJECT_LOCK_FILE,
+    RELEASE_URL,
+    REPLACE_FOLDERS,
+    TMP_PATH,
+    VERSION_FILE,
+)
+
+
+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 f"检测到当前版本更新\n当前版本：{cur_version}\n最新版本：{data.get('name')}\n创建日期：{data.get('created_at')}\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(f"开始下载真寻最新版文件....", "检查更新")
+        cur_version = cls.__get_version()
+        new_version = "main"
+        url = MAIN_URL
+        if version_type == "dev":
+            url = DEV_URL
+            new_version = "dev"
+        if version_type == "release":
+            data = await cls.__get_latest_data()
+            if not data:
+                return "获取更新版本失败..."
+            url = data.get("tarball_url")
+            new_version = data.get("name")
+            url = (await AsyncHttpx.get(url)).headers.get("Location")  # type: ignore
+        if not url:
+            return "获取版本下载链接失败..."
+        logger.debug(
+            f"更新版本：{cur_version} -> {new_version} | 下载链接：{url}", "检查更新"
+        )
+        await PlatformUtils.send_superuser(
+            bot,
+            f"检测真寻已更新，版本更新：{cur_version} -> {new_version}\n开始更新...",
+            user_id,
+        )
+        if await AsyncHttpx.download_file(url, DOWNLOAD_FILE):
+            logger.debug("下载真寻最新版文件完成...", "检查更新")
+            if version_type != "release":
+                new_version = None
+            await cls.__file_handle(new_version)
+            return f"版本更新完成\n版本: {cur_version} -> {new_version}\n请重新启动真寻以完成更新!"
+        else:
+            logger.debug("下载真寻最新版文件失败...", "检查更新")
+        return None
+
+    @run_sync
+    @classmethod
+    def __file_handle(cls, latest_version: str | None):
+        """文件移动操作
+
+        参数:
+            latest_version: 版本号
+        """
+        TMP_PATH.mkdir(exist_ok=True, parents=True)
+        BACKUP_PATH.mkdir(exist_ok=True, parents=True)
+        if BACKUP_PATH.exists():
+            shutil.rmtree(BACKUP_PATH)
+        tf = None
+        logger.debug("开始解压文件压缩包...", "检查更新")
+        tf = tarfile.open(DOWNLOAD_FILE)
+        tf.extractall(TMP_PATH)
+        logger.debug("解压文件压缩包完成...", "检查更新")
+        download_file_path = TMP_PATH / os.listdir(TMP_PATH)[0]
+        _pyproject = download_file_path / "pyproject.toml"
+        _lock_file = download_file_path / "poetry.lock"
+        extract_path = TMP_PATH / os.listdir(TMP_PATH)[0] / "zhenxun"
+        target_path = BASE_PATH
+        if PYPROJECT_FILE.exists():
+            logger.debug(f"备份文件: {PYPROJECT_FILE}", "检查更新")
+            shutil.move(PYPROJECT_FILE, BACKUP_PATH / "pyproject.toml")
+        if PYPROJECT_LOCK_FILE.exists():
+            logger.debug(f"备份文件: {PYPROJECT_FILE}", "检查更新")
+            shutil.move(PYPROJECT_LOCK_FILE, BACKUP_PATH / "poetry.lock")
+        if _pyproject.exists():
+            logger.debug("移动文件: pyproject.toml", "检查更新")
+            shutil.move(_pyproject, Path() / "pyproject.toml")
+        if _lock_file.exists():
+            logger.debug("移动文件: pyproject.toml", "检查更新")
+            shutil.move(_lock_file, Path() / "poetry.lock")
+        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.rmtree(_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 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 tf:
+            tf.close()
+        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}")
+        os.system(f"poetry run pip install -r {(Path() / 'pyproject.toml').absolute()}")
+
+    @classmethod
+    def __get_version(cls) -> str:
+        """获取当前版本
+
+        返回:
+            str: 当前版本号
+        """
+        _version = "v0.0.0"
+        if VERSION_FILE.exists():
+            text = VERSION_FILE.open("w", 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(f"检查更新真寻获取版本失败", e=e)
+        return {}
--- a/zhenxun/plugins/auto_update/config.py
+++ b/zhenxun/plugins/auto_update/config.py
@ -0,0 +1,23 @@
+from pathlib import Path
+
+from zhenxun.configs.path_config import TEMP_PATH
+
+DEV_URL = "https://ghproxy.cc/https://github.com/HibiKier/zhenxun_bot/archive/refs/heads/dev.zip"
+MAIN_URL = "https://ghproxy.cc/https://github.com/HibiKier/zhenxun_bot/archive/refs/heads/main.zip"
+RELEASE_URL = "https://api.github.com/repos/HibiKier/zhenxun_bot/releases/latest"
+
+
+VERSION_FILE = Path() / "__version__"
+
+PYPROJECT_FILE = Path() / "pyproject.toml"
+PYPROJECT_LOCK_FILE = Path() / "poetry.lock"
+
+BASE_PATH = Path() / "zhenxun"
+
+TMP_PATH = TEMP_PATH / "auto_update"
+
+BACKUP_PATH = Path() / "backup"
+
+DOWNLOAD_FILE = TMP_PATH / "download_latest_file.tar.gz"
+
+REPLACE_FOLDERS = ["builtin_plugins", "plugins", "services", "utils", "models"]
--- a/zhenxun/utils/platform.py
+++ b/zhenxun/utils/platform.py
@ -61,7 +61,7 @@ class PlatformUtils:
    async def send_superuser(
        cls,
        bot: Bot,
-        message: UniMessage,
+        message: UniMessage | str,
        superuser_id: str | None = None,
    ) -> Receipt | None:
        """发送消息给超级用户
@ -83,6 +83,8 @@ class PlatformUtils:
            if not platform_superusers:
                raise NotFindSuperuser()
            superuser_id = random.choice(platform_superusers)
+        if isinstance(message, str):
+            message = MessageUtils.build_message(message)
        return await cls.send_message(bot, superuser_id, None, message)

    @classmethod