From 185b1d8a215b84579ec8a2e89507e8cffc3c9755 Mon Sep 17 00:00:00 2001
From: snowy <snowykami@outlook.com>
Date: Fri, 12 Apr 2024 01:07:53 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=87=AA=E5=8A=A8=E6=9B=B4=E6=96=B0?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docs/deployment/config.md                   |  9 +--
 liteyuki/liteyuki_main/api.py               | 67 +++++++++++++++++++++
 liteyuki/liteyuki_main/core.py              | 47 ++++++---------
 liteyuki/resources/templates/css/status.css |  0
 liteyuki/resources/templates/status.html    |  2 +-
 requirements.txt                            |  1 +
 6 files changed, 92 insertions(+), 34 deletions(-)
 create mode 100644 liteyuki/liteyuki_main/api.py
 create mode 100644 liteyuki/resources/templates/css/status.css

diff --git a/docs/deployment/config.md b/docs/deployment/config.md
index 6ab3ceb..53c04a3 100644
--- a/docs/deployment/config.md
+++ b/docs/deployment/config.md
@@ -21,12 +21,13 @@ nickname: [ "liteyuki" ]  # 机器人昵称
 superusers: [ "1919810" ]  # 超级用户
 
 # 未列出的配置项（如要自定义请手动修改）
-onebot_access_token: "" # 访问令牌，对公网开放时建议设置
-default_language: "zh-CN" # 默认语言，支持i18n部分语言和自行扩展的语言代码
+onebot_access_token: "" # 访问令牌，对公开放时建议设置
+default_language: "zh-CN" # 默认语言
 log_level: "INFO" # 日志等级
 log_icon: true # 是否显示日志等级图标（某些控制台字体不可用）
-auto_report: true # 是否自动上报问题给轻雪服务器，仅包含硬件信息和运行软件版本
-fake_device_info: # 统计卡片显示的虚假设备信息，用于保护隐私
+auto_report: true # 是否自动上报问题给轻雪服务器
+auto_update: true # 是否自动更新轻雪，每天4点检查更新
+fake_device_info: # 用于保护设备隐私
   cpu:
     brand: AMD
     cores: 16 # 物理核心数
diff --git a/liteyuki/liteyuki_main/api.py b/liteyuki/liteyuki_main/api.py
new file mode 100644
index 0000000..f208e56
--- /dev/null
+++ b/liteyuki/liteyuki_main/api.py
@@ -0,0 +1,67 @@
+import nonebot
+from git import Repo
+
+remote_urls = [
+        "https://github.com/snowykami/LiteyukiBot.git",
+        "https://gitee.com/snowykami/LiteyukiBot.git"
+]
+
+
+def detect_update() -> bool:
+    # 对每个远程仓库进行检查，只要有一个仓库有更新，就返回True
+    for remote_url in remote_urls:
+        repo = Repo(".")
+        repo.remotes.origin.set_url(remote_url)
+        repo.remotes.origin.fetch()
+        if repo.head.commit != repo.commit('origin/main'):
+            return True
+
+
+def update_liteyuki() -> tuple[bool, str]:
+    """更新轻雪
+    :return: 是否更新成功，更新变动"""
+    # origins = ["origin", "origin2"]
+    # repo = Repo(".")
+    #
+    # # Get the current HEAD commit
+    # current_head_commit = repo.head.commit
+    #
+    # # Fetch the latest information from the cloud
+    # repo.remotes.origin.fetch()
+    #
+    # # Get the latest HEAD commit
+    # new_head_commit = repo.commit('origin/main')
+    #
+    # # If the new HEAD commit is different from the current HEAD commit, there is a new commit
+    # diffs = current_head_commit.diff(new_head_commit)
+    # logs = ""
+    # for diff in diffs.iter_change_type('M'):
+    #     logs += f"\n{diff.a_path}"
+    #
+    # for origin in origins:
+    #     try:
+    #         repo.remotes[origin].pull()
+    #         break
+    #     except Exception as e:
+    #         nonebot.logger.error(f"Pull from {origin} failed: {e}")
+    #         continue
+    # else:
+    #     return False, 0
+    #
+    # return True, len(logs)
+    new_commit_detected = detect_update()
+    if new_commit_detected:
+        repo = Repo(".")
+        logs = ""
+        # 对每个远程仓库进行更新
+        for remote_url in remote_urls:
+            repo.remotes.origin.set_url(remote_url)
+            repo.remotes.origin.pull()
+            diffs = repo.head.commit.diff("origin/main")
+
+            for diff in diffs.iter_change_type('M'):
+                logs += f"\n{diff.a_path}"
+        return True, logs
+    else:
+        return False, "Nothing Changed"
+
diff --git a/liteyuki/liteyuki_main/core.py b/liteyuki/liteyuki_main/core.py
index 32ad783..c675831 100644
--- a/liteyuki/liteyuki_main/core.py
+++ b/liteyuki/liteyuki_main/core.py
@@ -1,25 +1,25 @@
-import asyncio
 import base64
 import time
 from typing import Any
 
 import nonebot
 import pip
-from git import Repo
 from nonebot import Bot, get_driver, require
 from nonebot.exception import MockApiException
 from nonebot.internal.matcher import Matcher
 from nonebot.permission import SUPERUSER
 
-from liteyuki.utils.config import load_from_yaml
+from liteyuki.utils.config import get_config, load_from_yaml
 from liteyuki.utils.data_manager import StoredConfig, TempConfig, common_db
 from liteyuki.utils.language import get_user_lang
 from liteyuki.utils.ly_typing import T_Bot, T_MessageEvent
 from liteyuki.utils.message import Markdown as md
 from liteyuki.utils.reloader import Reloader
+from .api import update_liteyuki
 
-require("nonebot_plugin_alconna"), require("nonebot_plugin_htmlrender")
+require("nonebot_plugin_alconna"), require("nonebot_plugin_apscheduler")
 from nonebot_plugin_alconna import on_alconna, Alconna, Args, Subcommand, Arparma
+from nonebot_plugin_apscheduler import scheduler
 
 driver = get_driver()
 
@@ -46,30 +46,7 @@ async def _(bot: T_Bot, matcher: Matcher):
 async def _(bot: T_Bot, event: T_MessageEvent):
     # 使用git pull更新
     ulang = get_user_lang(str(event.user_id))
-    origins = ["origin", "origin2"]
-    repo = Repo(".")
-
-    # Get the current HEAD commit
-    current_head_commit = repo.head.commit
-
-    # Fetch the latest information from the cloud
-    repo.remotes.origin.fetch()
-
-    # Get the latest HEAD commit
-    new_head_commit = repo.commit('origin/main')
-
-    # If the new HEAD commit is different from the current HEAD commit, there is a new commit
-    diffs = current_head_commit.diff(new_head_commit)
-    logs = ""
-    for diff in diffs.iter_change_type('M'):
-        logs += f"\n{diff.a_path}"
-
-    for origin in origins:
-        try:
-            repo.remotes[origin].pull()
-            break
-        except Exception as e:
-            nonebot.logger.error(f"Pull from {origin} failed: {e}")
+    success, logs = update_liteyuki()
     reply = "Liteyuki updated!\n"
     reply += f"```\n{logs}\n```\n"
     btn_restart = md.btn_cmd(ulang.get("liteyuki.restart_now"), "reload-liteyuki")
@@ -173,7 +150,7 @@ async def _(event: T_MessageEvent, matcher: Matcher):
     aliases={"轻雪文档"},
 ).handle()
 async def _(matcher: Matcher):
-    matcher.finish("https://bot.liteyuki.icu/usage")
+    await matcher.finish("https://bot.liteyuki.icu/usage")
 
 
 # system hook
@@ -240,3 +217,15 @@ async def _(bot: T_Bot):
             group_id=reload_session_id,
             message="Liteyuki reloaded in %.2f s" % delta_time
         )
+
+
+# 每天4点更新
+@scheduler.scheduled_job("cron", hour=4)
+async def every_day_update():
+    if get_config("auto_update", True):
+        result, logs = update_liteyuki()
+        if result:
+            nonebot.logger.info(f"Liteyuki updated: {logs}")
+            Reloader.reload(1)
+        else:
+            nonebot.logger.info(logs)
\ No newline at end of file
diff --git a/liteyuki/resources/templates/css/status.css b/liteyuki/resources/templates/css/status.css
new file mode 100644
index 0000000..e69de29
diff --git a/liteyuki/resources/templates/status.html b/liteyuki/resources/templates/status.html
index 0e88f15..b819ce3 100644
--- a/liteyuki/resources/templates/status.html
+++ b/liteyuki/resources/templates/status.html
@@ -7,7 +7,7 @@
 </head>
 <body>
 <div class="data-storage" id="data">{{ data | tojson }}</div>
-<div class="info-box" id="device-info"></div>
+<div class="info-box" id="system-info"></div>
 <div class="info-box" id="disk-info"></div>
 <div class="info-box" id="motto-info"></div>
 </body>
diff --git a/requirements.txt b/requirements.txt
index 00f6d81..41657cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,6 +8,7 @@ nonebot2[fastapi,httpx,websockets]==2.2.1
 nonebot-plugin-htmlrender==0.3.1
 nonebot-adapter-onebot==2.4.3
 nonebot-plugin-alconna==0.41.0
+nonebot_plugin_apscheduler==0.4.0
 playwright==1.17.2
 psutil==5.9.8
 py-cpuinfo==9.0.0