Merge pull request #894 from thsrite/main

app/chain/__init__.py

@@ -65,6 +65,15 @@ class ChainBase(metaclass=ABCMeta):
             del cache
             gc.collect()
 
+    @staticmethod
+    def remove_cache(filename: str) -> None:
+        """
+        删除本地缓存
+        """
+        cache_path = settings.TEMP_PATH / filename
+        if cache_path.exists():
+            Path(cache_path).unlink()
+
     def run_module(self, method: str, *args, **kwargs) -> Any:
         """
         运行包含该方法的所有模块，然后返回结果

app/chain/system.py

@@ -1,7 +1,13 @@
+import json
+import os
+import re
 from typing import Union
 
 from app.chain import ChainBase
+from app.core.config import settings
+from app.log import logger
 from app.schemas import Notification, MessageChannel
+from app.utils.http import RequestUtils
 from app.utils.system import SystemUtils
 
 
@@ -10,6 +16,9 @@ class SystemChain(ChainBase):
     系统级处理链
     """
 
+    _restart_file = "__system_restart__"
+    _update_file = "__system_update__"
+
     def remote_clear_cache(self, channel: MessageChannel, userid: Union[int, str]):
         """
         清理系统缓存
@@ -22,6 +31,131 @@ class SystemChain(ChainBase):
         """
         重启系统
         """
-        self.post_message(Notification(channel=channel,
-                                       title=f"系统正在重启，请耐心等候！", userid=userid))
+        if channel and userid:
+            self.post_message(Notification(channel=channel,
+                                           title="系统正在重启，请耐心等候！", userid=userid))
+            # 保存重启信息
+            self.save_cache({
+                "channel": channel.value,
+                "userid": userid
+            }, self._restart_file)
         SystemUtils.restart()
+
+    def update(self, channel: MessageChannel = None, userid: Union[int, str] = None):
+        """
+        重启系统
+        """
+        if SystemUtils.is_windows():
+            logger.error("windows暂不支持")
+            return
+        if channel and userid:
+            self.post_message(Notification(channel=channel,
+                                           title="系统正在更新，请耐心等候！", userid=userid))
+            # 保存重启信息
+            self.save_cache({
+                "channel": channel.value,
+                "userid": userid
+            }, self._update_file)
+
+        # 重启系统
+        os.system("bash /usr/local/bin/mp_update")
+        if channel and userid:
+            self.post_message(Notification(channel=channel,
+                                           title="暂无新版本！", userid=userid))
+            self.remove_cache(self._update_file)
+
+    def version(self, channel: MessageChannel, userid: Union[int, str]):
+        """
+        查看当前版本、远程版本
+        """
+        release_version = self.__get_release_version()
+        local_version = self.get_local_version()
+        if release_version == local_version:
+            title = f"当前版本：{local_version}，已是最新版本"
+        else:
+            title = f"当前版本：{local_version}，远程版本：{release_version}"
+
+        self.post_message(Notification(channel=channel,
+                                       title=title, userid=userid))
+
+    def restart_finish(self):
+        """
+        如通过交互命令重启，
+        重启完发送msg
+        """
+        cache_file, action, channel, userid = None, None, None, None
+        # 重启消息
+        restart_channel = self.load_cache(self._restart_file)
+        if restart_channel:
+            cache_file = self._restart_file
+            action = "重启"
+            # 发送重启完成msg
+            if not isinstance(restart_channel, dict):
+                restart_channel = json.loads(restart_channel)
+            channel = next(
+                (channel for channel in MessageChannel.__members__.values() if
+                 channel.value == restart_channel.get('channel')), None)
+            userid = restart_channel.get('userid')
+
+        # 更新消息
+        update_channel = self.load_cache(self._update_file)
+        if update_channel:
+            cache_file = self._update_file
+            action = "更新"
+            # 发送重启完成msg
+            if not isinstance(update_channel, dict):
+                update_channel = json.loads(update_channel)
+            channel = next(
+                (channel for channel in MessageChannel.__members__.values() if
+                 channel.value == update_channel.get('channel')), None)
+            userid = update_channel.get('userid')
+
+        # 发送消息
+        if channel and userid:
+            # 版本号
+            release_version = self.__get_release_version()
+            local_version = self.get_local_version()
+            if release_version == local_version:
+                title = f"当前版本：{local_version}"
+            else:
+                title = f"当前版本：{local_version}，远程版本：{release_version}"
+            self.post_message(Notification(channel=channel,
+                                           title=f"系统已{action}完成！{title}",
+                                           userid=userid))
+            self.remove_cache(cache_file)
+
+    @staticmethod
+    def __get_release_version():
+        """
+        获取最新版本
+        """
+        version_res = RequestUtils(proxies=settings.PROXY).get_res(
+            "https://api.github.com/repos/jxxghp/MoviePilot/releases/latest")
+        if version_res:
+            ver_json = version_res.json()
+            version = f"{ver_json['tag_name']}"
+            return version
+        else:
+            return None
+
+    @staticmethod
+    def get_local_version():
+        """
+        查看当前版本
+        """
+        version_file = settings.ROOT_PATH / "version.py"
+        if version_file.exists():
+            try:
+                with open(version_file, 'rb') as f:
+                    version = f.read()
+                pattern = r"v(\d+\.\d+\.\d+)"
+                match = re.search(pattern, str(version))
+
+                if match:
+                    version = match.group(1)
+                    return f"v{version}"
+                else:
+                    logger.warn("未找到版本号")
+                    return None
+            except Exception as err:
+                logger.error(f"加载版本文件 {version_file} 出错：{err}")