mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 20:17:13 +08:00
232 lines
10 KiB
Python
232 lines
10 KiB
Python
import json
|
|
import platform
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Callable
|
|
|
|
from app.runtime import config as _runtime_config
|
|
from app.runtime.log import logger
|
|
from app.adapters.network.http import RequestUtils
|
|
from app.foundation.version import compare_version
|
|
from app.adapters.system.host import SystemUtils
|
|
from app.runtime.settings import get_runtime_setting
|
|
|
|
|
|
# 保留模块级旧 Settings 入口，旧插件和测试可能仍会对其做运行时覆盖；实现读取统一走 runtime 端口。
|
|
settings = _runtime_config.settings
|
|
|
|
|
|
ResourceVersionProvider = Callable[[], tuple[str, str]]
|
|
|
|
|
|
def _unavailable_resource_versions() -> tuple[str, str]:
|
|
"""站点能力尚未装配时返回可安全比较的空版本。"""
|
|
return "0", "0"
|
|
|
|
|
|
_resource_version_provider: ResourceVersionProvider = _unavailable_resource_versions
|
|
|
|
|
|
def configure_resource_version_provider(provider: ResourceVersionProvider) -> None:
|
|
"""由组合根注入已加载的认证与索引版本，保持适配器不依赖应用层。"""
|
|
global _resource_version_provider
|
|
_resource_version_provider = provider
|
|
|
|
|
|
class ResourceHelper:
|
|
"""
|
|
检测和更新资源包
|
|
"""
|
|
|
|
_base_dir: Path = get_runtime_setting("ROOT_PATH")
|
|
_resource_target = Path("app/application/site")
|
|
_version_flag = get_runtime_setting("RESOURCE_VERSION_FLAG")
|
|
_repo = (
|
|
f"{get_runtime_setting('GITHUB_PROXY')}https://raw.githubusercontent.com/"
|
|
f"jxxghp/MoviePilot-Resources/main/package.{_version_flag}.json"
|
|
)
|
|
_files_api = (
|
|
"https://api.github.com/repos/jxxghp/"
|
|
f"MoviePilot-Resources/contents/resources.{_version_flag}"
|
|
)
|
|
|
|
@property
|
|
def proxies(self):
|
|
"""返回访问 GitHub 资源时应使用的代理配置。"""
|
|
return (
|
|
None
|
|
if get_runtime_setting("GITHUB_PROXY")
|
|
else get_runtime_setting("PROXY")
|
|
)
|
|
|
|
@staticmethod
|
|
def _get_python_version_tag() -> str:
|
|
"""返回资源文件名使用的 CPython ABI 标签。"""
|
|
version = sys.version_info
|
|
return f"cp{version.major}{version.minor}"
|
|
|
|
@staticmethod
|
|
def _get_machine_tag() -> str:
|
|
"""将系统架构名称归一为资源文件使用的标签。"""
|
|
machine = platform.machine().lower()
|
|
if machine in {"arm64", "aarch64"}:
|
|
return "aarch64"
|
|
elif machine in {"x86_64", "amd64"}:
|
|
return "x86_64"
|
|
return machine
|
|
|
|
@classmethod
|
|
def _get_needed_files(cls) -> list[str]:
|
|
"""返回 V3 资源在当前平台需要下载的文件名。"""
|
|
python_version = ResourceHelper._get_python_version_tag()
|
|
python_ver = python_version.replace("cp", "")
|
|
system = platform.system().lower()
|
|
machine = ResourceHelper._get_machine_tag()
|
|
files = [f"user.sites.{cls._version_flag}.bin"]
|
|
if system == "linux":
|
|
files.append(f"sites.cpython-{python_ver}-{machine}-linux-gnu.so")
|
|
elif system == "darwin":
|
|
files.append(f"sites.cpython-{python_ver}-darwin.so")
|
|
elif system == "windows":
|
|
files.append(f"sites.cp{python_ver}-win_amd64.pyd")
|
|
return files
|
|
|
|
def _load_resource_info(self):
|
|
"""读取 V3 资源清单。"""
|
|
response = RequestUtils(
|
|
proxies=self.proxies,
|
|
headers=get_runtime_setting("GITHUB_HEADERS"),
|
|
timeout=10,
|
|
).get_res(self._repo)
|
|
return response if response and response.status_code == 200 else None
|
|
|
|
def check(
|
|
self,
|
|
*,
|
|
auth_version: str | None = None,
|
|
indexer_version: str | None = None,
|
|
) -> bool:
|
|
"""
|
|
检测并安装当前平台的资源更新。
|
|
|
|
:param auth_version: 当前已加载的站点认证资源版本；省略时使用组合根注入值
|
|
:param indexer_version: 当前已加载的站点索引资源版本；省略时使用组合根注入值
|
|
:return: 是否成功安装了需要由上层处理重启的新资源
|
|
"""
|
|
if not get_runtime_setting("AUTO_UPDATE_RESOURCE"):
|
|
return False
|
|
if SystemUtils.is_frozen():
|
|
return False
|
|
if auth_version is None or indexer_version is None:
|
|
configured_auth_version, configured_indexer_version = (
|
|
_resource_version_provider()
|
|
)
|
|
auth_version = auth_version or configured_auth_version
|
|
indexer_version = indexer_version or configured_indexer_version
|
|
logger.info("开始检测资源包版本...")
|
|
res = self._load_resource_info()
|
|
if res:
|
|
try:
|
|
resource_info = json.loads(res.text)
|
|
online_version = resource_info.get("version")
|
|
if online_version:
|
|
logger.info(f"最新资源包版本：v{online_version}")
|
|
# 需要更新的资源包
|
|
need_updates = {}
|
|
# 资源明细
|
|
resources: dict = resource_info.get("resources") or {}
|
|
for rname, resource in resources.items():
|
|
rtype = resource.get("type")
|
|
platform = resource.get("platform")
|
|
declared_target = Path(str(resource.get("target") or ""))
|
|
version = resource.get("version")
|
|
# 判断平台
|
|
if platform and platform != SystemUtils.platform():
|
|
continue
|
|
# 判断版本号
|
|
if rtype == "auth":
|
|
# 站点认证资源
|
|
local_version = auth_version
|
|
elif rtype == "sites":
|
|
# 站点索引资源
|
|
local_version = indexer_version
|
|
else:
|
|
continue
|
|
if compare_version(version, ">", local_version):
|
|
logger.info(f"{rname} 资源包有更新，最新版本：v{version}")
|
|
else:
|
|
continue
|
|
# 需要安装
|
|
if declared_target != self._resource_target:
|
|
logger.warning(
|
|
"忽略资源 %s 的非 canonical 目标目录：%s",
|
|
rname,
|
|
declared_target,
|
|
)
|
|
continue
|
|
need_updates[rname] = self._resource_target
|
|
if need_updates:
|
|
# 下载文件信息列表
|
|
r = RequestUtils(
|
|
proxies=get_runtime_setting("PROXY"),
|
|
headers=get_runtime_setting("GITHUB_HEADERS"),
|
|
timeout=30,
|
|
).get_res(self._files_api)
|
|
if r and not r.ok:
|
|
logger.error(
|
|
f"连接仓库失败：{r.status_code} - {r.reason}"
|
|
)
|
|
return False
|
|
elif not r:
|
|
logger.error("连接仓库失败")
|
|
return False
|
|
files_info = r.json()
|
|
# 下载资源文件
|
|
needed_files = self._get_needed_files()
|
|
logger.info(f"需要下载的资源文件：{needed_files}")
|
|
success = True
|
|
for item in files_info:
|
|
file_name = item.get("name")
|
|
if file_name not in needed_files:
|
|
continue
|
|
save_path = need_updates.get(file_name)
|
|
if not save_path:
|
|
continue
|
|
if item.get("download_url"):
|
|
logger.info(f"开始更新资源文件：{file_name} ...")
|
|
download_url = (
|
|
f"{get_runtime_setting('GITHUB_PROXY')}{item.get('download_url')}"
|
|
)
|
|
res = RequestUtils(
|
|
proxies=self.proxies,
|
|
headers=get_runtime_setting("GITHUB_HEADERS"),
|
|
timeout=180,
|
|
).get_res(download_url)
|
|
if not res:
|
|
logger.error(f"文件 {file_name} 下载失败！")
|
|
success = False
|
|
break
|
|
elif res.status_code != 200:
|
|
logger.error(
|
|
f"下载文件 {file_name} 失败：{res.status_code} - {res.reason}"
|
|
)
|
|
success = False
|
|
break
|
|
file_path = self._base_dir / save_path / file_name
|
|
if not file_path.parent.exists():
|
|
file_path.parent.mkdir(parents=True, exist_ok=True)
|
|
file_path.write_bytes(res.content)
|
|
if success:
|
|
logger.info("资源包更新完成，等待启动层处理后续重启")
|
|
return True
|
|
else:
|
|
logger.warning("资源包更新失败，跳过升级！")
|
|
else:
|
|
logger.info("所有资源已最新，无需更新")
|
|
except json.JSONDecodeError:
|
|
logger.error("资源包仓库数据解析失败！")
|
|
return False
|
|
else:
|
|
logger.warning("无法连接资源包仓库！")
|
|
return False
|