149 lines
4.0 KiB
Python
149 lines
4.0 KiB
Python
"""
|
|
更新检查工具
|
|
在插件加载时检查是否有新版本
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
from typing import Optional
|
|
|
|
|
|
def get_current_version() -> Optional[str]:
|
|
"""
|
|
获取当前版本号
|
|
|
|
Returns:
|
|
版本号字符串，如果读取失败返回 None
|
|
"""
|
|
version_file = os.path.join(os.path.dirname(os.path.dirname(__file__)), "version.txt")
|
|
try:
|
|
with open(version_file, 'r', encoding='utf-8') as f:
|
|
return f.read().strip()
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
def check_for_updates() -> bool:
|
|
"""
|
|
检查是否有更新
|
|
|
|
Returns:
|
|
True 如果有更新，False 如果已是最新或检查失败
|
|
"""
|
|
try:
|
|
# 获取当前目录
|
|
plugin_dir = os.path.dirname(os.path.dirname(__file__))
|
|
|
|
# 检查是否是 Git 仓库
|
|
git_dir = os.path.join(plugin_dir, '.git')
|
|
if not os.path.exists(git_dir):
|
|
return False
|
|
|
|
# 执行 git fetch（禁止弹出认证弹框，失败时静默处理）
|
|
env = os.environ.copy()
|
|
env['GIT_TERMINAL_PROMPT'] = '0'
|
|
subprocess.run(
|
|
['git', 'fetch', 'origin'],
|
|
cwd=plugin_dir,
|
|
capture_output=True,
|
|
timeout=10,
|
|
env=env
|
|
)
|
|
|
|
# 检查本地和远程版本
|
|
local = subprocess.run(
|
|
['git', 'rev-parse', '@'],
|
|
cwd=plugin_dir,
|
|
capture_output=True,
|
|
text=True
|
|
).stdout.strip()
|
|
|
|
remote = subprocess.run(
|
|
['git', 'rev-parse', '@{u}'],
|
|
cwd=plugin_dir,
|
|
capture_output=True,
|
|
text=True
|
|
).stdout.strip()
|
|
|
|
return local != remote
|
|
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def get_update_changelog() -> list:
|
|
"""从远程 CHANGELOG.md 最新版本块中提取更新内容（最多5条）"""
|
|
try:
|
|
plugin_dir = os.path.dirname(os.path.dirname(__file__))
|
|
result = subprocess.run(
|
|
['git', 'show', 'origin/main:CHANGELOG.md'],
|
|
cwd=plugin_dir,
|
|
capture_output=True,
|
|
text=True,
|
|
encoding='utf-8'
|
|
)
|
|
lines = result.stdout.splitlines()
|
|
|
|
in_block = False
|
|
items = []
|
|
for line in lines:
|
|
if line.startswith('## [') and not line.startswith('## [Unreleased]'):
|
|
if in_block:
|
|
break
|
|
in_block = True
|
|
continue
|
|
if in_block:
|
|
stripped = line.strip()
|
|
if stripped and not stripped.startswith('#') and not stripped.startswith('---'):
|
|
text = stripped.lstrip('- ').replace('**', '').strip()
|
|
if text and len(text) > 3:
|
|
items.append(text)
|
|
if len(items) >= 5:
|
|
break
|
|
|
|
return items
|
|
except Exception:
|
|
return []
|
|
|
|
|
|
def notify_new_version():
|
|
"""检测到新版本时，推送蓝色更新通知弹框"""
|
|
changelog = get_update_changelog()
|
|
|
|
try:
|
|
import threading
|
|
from server import PromptServer
|
|
|
|
def _send():
|
|
try:
|
|
PromptServer.instance.send_sync(
|
|
"o1key.new_version",
|
|
{"changelog": changelog}
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
threading.Timer(3.0, _send).start()
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def notify_update_available():
|
|
"""通知用户有更新可用（前端弹窗）"""
|
|
try:
|
|
import threading
|
|
from server import PromptServer
|
|
|
|
def _send():
|
|
try:
|
|
PromptServer.instance.send_sync(
|
|
"o1key.update_available",
|
|
{"message": "欢迎使用o1key工作流，祝您马年，马上有福，马上有钱，马到成功！！！"}
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
threading.Timer(3.0, _send).start()
|
|
except Exception:
|
|
pass
|