mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-01 01:10:21 +08:00
fix: keep WebUI assets in sync with core version (#8901)
* fix: keep WebUI assets in sync with core version * fix: import dashboard version before bundled fallback * fix: remove stale WebUI dist robustly
This commit is contained in:
@@ -16,8 +16,11 @@ venv*/
|
|||||||
ENV/
|
ENV/
|
||||||
.conda/
|
.conda/
|
||||||
dashboard/
|
dashboard/
|
||||||
|
!astrbot/dashboard/
|
||||||
|
!astrbot/dashboard/dist/
|
||||||
|
!astrbot/dashboard/dist/**
|
||||||
data/
|
data/
|
||||||
tests/
|
tests/
|
||||||
.ruff_cache/
|
.ruff_cache/
|
||||||
.astrbot
|
.astrbot
|
||||||
astrbot.lock
|
astrbot.lock
|
||||||
|
|||||||
20
.github/workflows/docker-image.yml
vendored
20
.github/workflows/docker-image.yml
vendored
@@ -46,14 +46,21 @@ jobs:
|
|||||||
|
|
||||||
- name: Build Dashboard
|
- name: Build Dashboard
|
||||||
run: |
|
run: |
|
||||||
|
dashboard_version=$(python3 - <<'PY'
|
||||||
|
import tomllib
|
||||||
|
with open("pyproject.toml", "rb") as f:
|
||||||
|
print("v" + tomllib.load(f)["project"]["version"])
|
||||||
|
PY
|
||||||
|
)
|
||||||
cd dashboard
|
cd dashboard
|
||||||
npm install
|
npm install
|
||||||
npm run build
|
npm run build
|
||||||
mkdir -p dist/assets
|
mkdir -p dist/assets
|
||||||
echo $(git rev-parse HEAD) > dist/assets/version
|
echo "$dashboard_version" > dist/assets/version
|
||||||
cd ..
|
cd ..
|
||||||
mkdir -p data
|
mkdir -p astrbot/dashboard
|
||||||
cp -r dashboard/dist data/
|
rm -rf astrbot/dashboard/dist
|
||||||
|
cp -r dashboard/dist astrbot/dashboard/dist
|
||||||
|
|
||||||
- name: Determine test image tags
|
- name: Determine test image tags
|
||||||
id: test-meta
|
id: test-meta
|
||||||
@@ -157,10 +164,11 @@ jobs:
|
|||||||
npm install
|
npm install
|
||||||
npm run build
|
npm run build
|
||||||
mkdir -p dist/assets
|
mkdir -p dist/assets
|
||||||
echo $(git rev-parse HEAD) > dist/assets/version
|
echo "${{ steps.release-meta.outputs.version }}" > dist/assets/version
|
||||||
cd ..
|
cd ..
|
||||||
mkdir -p data
|
mkdir -p astrbot/dashboard
|
||||||
cp -r dashboard/dist data/
|
rm -rf astrbot/dashboard/dist
|
||||||
|
cp -r dashboard/dist astrbot/dashboard/dist
|
||||||
|
|
||||||
- name: Set QEMU
|
- name: Set QEMU
|
||||||
uses: docker/setup-qemu-action@v4.1.0
|
uses: docker/setup-qemu-action@v4.1.0
|
||||||
|
|||||||
@@ -183,8 +183,22 @@ async def download_file(
|
|||||||
path: str,
|
path: str,
|
||||||
show_progress: bool = False,
|
show_progress: bool = False,
|
||||||
progress_callback=None,
|
progress_callback=None,
|
||||||
|
allow_insecure_ssl_fallback: bool = True,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""从指定 url 下载文件到指定路径 path"""
|
"""Download a remote file to a local path.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
url: Remote URL to download.
|
||||||
|
path: Local destination path.
|
||||||
|
show_progress: Whether to print progress to stdout.
|
||||||
|
progress_callback: Optional callback for progress payloads.
|
||||||
|
allow_insecure_ssl_fallback: Whether certificate failures may retry with
|
||||||
|
TLS certificate verification disabled.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None.
|
||||||
|
"""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
ssl_context = ssl.create_default_context(
|
ssl_context = ssl.create_default_context(
|
||||||
cafile=certifi.where(),
|
cafile=certifi.where(),
|
||||||
@@ -259,6 +273,8 @@ async def download_file(
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
except (aiohttp.ClientConnectorSSLError, aiohttp.ClientConnectorCertificateError):
|
except (aiohttp.ClientConnectorSSLError, aiohttp.ClientConnectorCertificateError):
|
||||||
|
if not allow_insecure_ssl_fallback:
|
||||||
|
raise
|
||||||
# 关闭SSL验证(仅在证书验证失败时作为fallback)
|
# 关闭SSL验证(仅在证书验证失败时作为fallback)
|
||||||
logger.warning(
|
logger.warning(
|
||||||
f"SSL certificate verification failed for {_safe_url_for_log(url)}. "
|
f"SSL certificate verification failed for {_safe_url_for_log(url)}. "
|
||||||
@@ -355,10 +371,22 @@ def get_local_ip_addresses():
|
|||||||
return network_ips
|
return network_ips
|
||||||
|
|
||||||
|
|
||||||
def _read_dashboard_dist_version(dist_dir: str | Path) -> str | None:
|
def get_dashboard_dist_version(dist_dir: str | Path) -> str | None:
|
||||||
|
"""Read the WebUI version from a dashboard dist directory.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
dist_dir: Dashboard dist directory path.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
The version string from assets/version, or None when unavailable.
|
||||||
|
"""
|
||||||
|
|
||||||
version_file = Path(dist_dir) / "assets" / "version"
|
version_file = Path(dist_dir) / "assets" / "version"
|
||||||
if version_file.exists():
|
try:
|
||||||
return version_file.read_text(encoding="utf-8").strip()
|
if version_file.exists():
|
||||||
|
return version_file.read_text(encoding="utf-8").strip()
|
||||||
|
except (OSError, UnicodeDecodeError) as exc:
|
||||||
|
logger.warning("Failed to read WebUI version from %s: %s", version_file, exc)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
@@ -380,42 +408,106 @@ def _normalize_dashboard_version(version: str) -> str:
|
|||||||
return version
|
return version
|
||||||
|
|
||||||
|
|
||||||
def should_use_bundled_dashboard_dist(
|
def is_dashboard_version_compatible(
|
||||||
user_dist: str | Path, current_version: str
|
dashboard_version: str | None, current_version: str
|
||||||
) -> bool:
|
) -> bool:
|
||||||
user_version = _read_dashboard_dist_version(user_dist)
|
"""Check whether a WebUI version matches the current core version.
|
||||||
bundled_dist = get_bundled_dashboard_dist_path()
|
|
||||||
if user_version is None or not bundled_dist.exists():
|
Args:
|
||||||
|
dashboard_version: Version read from the WebUI assets/version file.
|
||||||
|
current_version: Current AstrBot core version.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
True when both versions are valid SemVer values and compare equal.
|
||||||
|
"""
|
||||||
|
|
||||||
|
if dashboard_version is None:
|
||||||
return False
|
return False
|
||||||
try:
|
try:
|
||||||
return (
|
return (
|
||||||
VersionComparator.compare_version(
|
VersionComparator.compare_version(
|
||||||
|
_normalize_dashboard_version(dashboard_version),
|
||||||
_normalize_dashboard_version(current_version),
|
_normalize_dashboard_version(current_version),
|
||||||
_normalize_dashboard_version(user_version),
|
|
||||||
)
|
)
|
||||||
> 0
|
== 0
|
||||||
)
|
)
|
||||||
except (TypeError, ValueError):
|
except (TypeError, ValueError):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def is_dashboard_dist_compatible(dist_dir: str | Path, current_version: str) -> bool:
|
||||||
|
"""Check whether a WebUI dist is complete and matches the core version.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
dist_dir: Dashboard dist directory path.
|
||||||
|
current_version: Current AstrBot core version.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
True when the dist has an index file and a compatible assets/version.
|
||||||
|
"""
|
||||||
|
|
||||||
|
dist_path = Path(dist_dir)
|
||||||
|
return (dist_path / "index.html").is_file() and is_dashboard_version_compatible(
|
||||||
|
get_dashboard_dist_version(dist_path),
|
||||||
|
current_version,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def should_use_bundled_dashboard_dist(
|
||||||
|
user_dist: str | Path, current_version: str
|
||||||
|
) -> bool:
|
||||||
|
"""Decide whether bundled WebUI should replace a user data dist.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_dist: Runtime dashboard dist directory under data/.
|
||||||
|
current_version: Current AstrBot core version.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
True when user_dist exists but is missing or mismatched against the
|
||||||
|
current core version, and bundled WebUI matches the current core version.
|
||||||
|
"""
|
||||||
|
|
||||||
|
user_dist = Path(user_dist)
|
||||||
|
user_version = get_dashboard_dist_version(user_dist)
|
||||||
|
bundled_dist = get_bundled_dashboard_dist_path()
|
||||||
|
if not user_dist.exists() or not is_dashboard_dist_compatible(
|
||||||
|
bundled_dist,
|
||||||
|
current_version,
|
||||||
|
):
|
||||||
|
return False
|
||||||
|
if user_version is None or not (user_dist / "index.html").is_file():
|
||||||
|
return True
|
||||||
|
try:
|
||||||
|
return not is_dashboard_version_compatible(user_version, current_version)
|
||||||
|
except (TypeError, ValueError):
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
async def get_dashboard_version():
|
async def get_dashboard_version():
|
||||||
|
"""Return the effective WebUI version for the current runtime.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
The matching data/dist version, matching bundled version, or the raw
|
||||||
|
data/dist version when no compatible bundled WebUI is available.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from astrbot.core.config.default import VERSION
|
||||||
|
|
||||||
# First check user data directory (manually updated / downloaded dashboard).
|
# First check user data directory (manually updated / downloaded dashboard).
|
||||||
dist_dir = os.path.join(get_astrbot_data_path(), "dist")
|
dist_dir = os.path.join(get_astrbot_data_path(), "dist")
|
||||||
if os.path.exists(dist_dir):
|
if os.path.exists(dist_dir):
|
||||||
from astrbot.core.config.default import VERSION
|
user_version = get_dashboard_dist_version(dist_dir)
|
||||||
|
if is_dashboard_dist_compatible(dist_dir, VERSION):
|
||||||
|
return user_version
|
||||||
|
|
||||||
if should_use_bundled_dashboard_dist(dist_dir, VERSION):
|
bundled = get_bundled_dashboard_dist_path()
|
||||||
bundled_version = _read_dashboard_dist_version(
|
if is_dashboard_dist_compatible(bundled, VERSION):
|
||||||
get_bundled_dashboard_dist_path()
|
return get_dashboard_dist_version(bundled)
|
||||||
)
|
return user_version
|
||||||
if bundled_version is not None:
|
|
||||||
return bundled_version
|
|
||||||
return _read_dashboard_dist_version(dist_dir)
|
|
||||||
|
|
||||||
bundled = get_bundled_dashboard_dist_path()
|
bundled = get_bundled_dashboard_dist_path()
|
||||||
if bundled.exists():
|
if is_dashboard_dist_compatible(bundled, VERSION):
|
||||||
return _read_dashboard_dist_version(bundled)
|
return get_dashboard_dist_version(bundled)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
@@ -427,6 +519,7 @@ async def download_dashboard(
|
|||||||
proxy: str | None = None,
|
proxy: str | None = None,
|
||||||
progress_callback=None,
|
progress_callback=None,
|
||||||
extract: bool = True,
|
extract: bool = True,
|
||||||
|
allow_insecure_ssl_fallback: bool = True,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Download dashboard assets and optionally extract them.
|
"""Download dashboard assets and optionally extract them.
|
||||||
|
|
||||||
@@ -438,6 +531,8 @@ async def download_dashboard(
|
|||||||
proxy: Optional download proxy prefix.
|
proxy: Optional download proxy prefix.
|
||||||
progress_callback: Optional callback for download progress payloads.
|
progress_callback: Optional callback for download progress payloads.
|
||||||
extract: Whether to extract the archive after download.
|
extract: Whether to extract the archive after download.
|
||||||
|
allow_insecure_ssl_fallback: Whether certificate failures may retry with
|
||||||
|
TLS certificate verification disabled.
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
None.
|
None.
|
||||||
@@ -460,6 +555,7 @@ async def download_dashboard(
|
|||||||
str(zip_path),
|
str(zip_path),
|
||||||
show_progress=True,
|
show_progress=True,
|
||||||
progress_callback=progress_callback,
|
progress_callback=progress_callback,
|
||||||
|
allow_insecure_ssl_fallback=allow_insecure_ssl_fallback,
|
||||||
)
|
)
|
||||||
if not zipfile.is_zipfile(zip_path):
|
if not zipfile.is_zipfile(zip_path):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
@@ -491,6 +587,7 @@ async def download_dashboard(
|
|||||||
str(zip_path),
|
str(zip_path),
|
||||||
show_progress=True,
|
show_progress=True,
|
||||||
progress_callback=progress_callback,
|
progress_callback=progress_callback,
|
||||||
|
allow_insecure_ssl_fallback=allow_insecure_ssl_fallback,
|
||||||
)
|
)
|
||||||
if not zipfile.is_zipfile(zip_path):
|
if not zipfile.is_zipfile(zip_path):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
@@ -506,6 +603,7 @@ async def download_dashboard(
|
|||||||
str(zip_path),
|
str(zip_path),
|
||||||
show_progress=True,
|
show_progress=True,
|
||||||
progress_callback=progress_callback,
|
progress_callback=progress_callback,
|
||||||
|
allow_insecure_ssl_fallback=allow_insecure_ssl_fallback,
|
||||||
)
|
)
|
||||||
if not zipfile.is_zipfile(zip_path):
|
if not zipfile.is_zipfile(zip_path):
|
||||||
raise RuntimeError("Downloaded dashboard package is not a valid ZIP file")
|
raise RuntimeError("Downloaded dashboard package is not a valid ZIP file")
|
||||||
|
|||||||
@@ -22,7 +22,9 @@ from astrbot.core.db import BaseDatabase
|
|||||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||||
from astrbot.core.utils.io import (
|
from astrbot.core.utils.io import (
|
||||||
get_bundled_dashboard_dist_path,
|
get_bundled_dashboard_dist_path,
|
||||||
|
get_dashboard_dist_version,
|
||||||
get_local_ip_addresses,
|
get_local_ip_addresses,
|
||||||
|
is_dashboard_dist_compatible,
|
||||||
should_use_bundled_dashboard_dist,
|
should_use_bundled_dashboard_dist,
|
||||||
)
|
)
|
||||||
from astrbot.dashboard.asgi_runtime import (
|
from astrbot.dashboard.asgi_runtime import (
|
||||||
@@ -182,21 +184,32 @@ class AstrBotDashboard:
|
|||||||
|
|
||||||
# Path priority:
|
# Path priority:
|
||||||
# 1. Explicit webui_dir argument
|
# 1. Explicit webui_dir argument
|
||||||
# 2. data/dist/ (user-installed / manually updated dashboard)
|
# 2. data/dist/ when it matches the core version
|
||||||
# 3. astrbot/dashboard/dist/ (bundled with the wheel)
|
# 3. astrbot/dashboard/dist/ when it matches the core version
|
||||||
if webui_dir and os.path.exists(webui_dir):
|
if webui_dir and os.path.exists(webui_dir):
|
||||||
self.data_path = os.path.abspath(webui_dir)
|
self.data_path = os.path.abspath(webui_dir)
|
||||||
else:
|
else:
|
||||||
user_dist = os.path.join(get_astrbot_data_path(), "dist")
|
user_dist = os.path.join(get_astrbot_data_path(), "dist")
|
||||||
bundled_dist = get_bundled_dashboard_dist_path()
|
bundled_dist = get_bundled_dashboard_dist_path()
|
||||||
if os.path.exists(user_dist) and not should_use_bundled_dashboard_dist(
|
user_version = get_dashboard_dist_version(user_dist)
|
||||||
|
if os.path.exists(user_dist) and is_dashboard_dist_compatible(
|
||||||
user_dist,
|
user_dist,
|
||||||
VERSION,
|
VERSION,
|
||||||
):
|
):
|
||||||
self.data_path = os.path.abspath(user_dist)
|
self.data_path = os.path.abspath(user_dist)
|
||||||
elif bundled_dist.exists():
|
elif should_use_bundled_dashboard_dist(
|
||||||
|
user_dist,
|
||||||
|
VERSION,
|
||||||
|
) or is_dashboard_dist_compatible(bundled_dist, VERSION):
|
||||||
self.data_path = str(bundled_dist)
|
self.data_path = str(bundled_dist)
|
||||||
logger.info("Using bundled dashboard dist: %s", self.data_path)
|
logger.info("Using bundled dashboard dist: %s", self.data_path)
|
||||||
|
elif os.path.exists(user_dist):
|
||||||
|
logger.warning(
|
||||||
|
"Ignoring data/dist because WebUI version mismatches core: %s, expected v%s.",
|
||||||
|
user_version,
|
||||||
|
VERSION,
|
||||||
|
)
|
||||||
|
self.data_path = None
|
||||||
else:
|
else:
|
||||||
# Fall back to expected user path (will fail gracefully later)
|
# Fall back to expected user path (will fail gracefully later)
|
||||||
self.data_path = os.path.abspath(user_dist)
|
self.data_path = os.path.abspath(user_dist)
|
||||||
@@ -545,7 +558,7 @@ class AstrBotDashboard:
|
|||||||
|
|
||||||
raise Exception(f"端口 {port} 已被占用")
|
raise Exception(f"端口 {port} 已被占用")
|
||||||
|
|
||||||
if (Path(self.data_path) / "index.html").is_file():
|
if self.data_path and (Path(self.data_path) / "index.html").is_file():
|
||||||
webui_status = "WebUI is ready"
|
webui_status = "WebUI is ready"
|
||||||
else:
|
else:
|
||||||
webui_status = (
|
webui_status = (
|
||||||
|
|||||||
91
main.py
91
main.py
@@ -2,6 +2,7 @@ import argparse
|
|||||||
import asyncio
|
import asyncio
|
||||||
import mimetypes
|
import mimetypes
|
||||||
import os
|
import os
|
||||||
|
import shutil
|
||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
@@ -46,7 +47,10 @@ from astrbot.core.utils.astrbot_path import ( # noqa: E402
|
|||||||
from astrbot.core.utils.io import ( # noqa: E402
|
from astrbot.core.utils.io import ( # noqa: E402
|
||||||
download_dashboard,
|
download_dashboard,
|
||||||
get_bundled_dashboard_dist_path,
|
get_bundled_dashboard_dist_path,
|
||||||
get_dashboard_version,
|
get_dashboard_dist_version,
|
||||||
|
is_dashboard_dist_compatible,
|
||||||
|
is_dashboard_version_compatible,
|
||||||
|
remove_dir,
|
||||||
should_use_bundled_dashboard_dist,
|
should_use_bundled_dashboard_dist,
|
||||||
)
|
)
|
||||||
from astrbot.core.utils.runtime_env import is_packaged_desktop_runtime # noqa: E402
|
from astrbot.core.utils.runtime_env import is_packaged_desktop_runtime # noqa: E402
|
||||||
@@ -91,7 +95,15 @@ def check_env() -> None:
|
|||||||
|
|
||||||
|
|
||||||
async def check_dashboard_files(webui_dir: str | None = None):
|
async def check_dashboard_files(webui_dir: str | None = None):
|
||||||
"""下载管理面板文件"""
|
"""Resolve and repair dashboard static files for startup.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
webui_dir: Optional explicit WebUI directory path from CLI.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
The directory path to serve, or None when no usable WebUI can be prepared.
|
||||||
|
"""
|
||||||
|
|
||||||
# 指定webui目录
|
# 指定webui目录
|
||||||
if webui_dir:
|
if webui_dir:
|
||||||
if os.path.exists(webui_dir):
|
if os.path.exists(webui_dir):
|
||||||
@@ -99,40 +111,81 @@ async def check_dashboard_files(webui_dir: str | None = None):
|
|||||||
return webui_dir
|
return webui_dir
|
||||||
logger.warning("WebUI directory not found: %s. Using default.", webui_dir)
|
logger.warning("WebUI directory not found: %s. Using default.", webui_dir)
|
||||||
|
|
||||||
data_dist_path = os.path.join(get_astrbot_data_path(), "dist")
|
data_dist_path = Path(get_astrbot_data_path()) / "dist"
|
||||||
if os.path.exists(data_dist_path):
|
bundled_dist = get_bundled_dashboard_dist_path()
|
||||||
v = await get_dashboard_version()
|
if data_dist_path.exists():
|
||||||
|
v = get_dashboard_dist_version(data_dist_path)
|
||||||
|
if is_dashboard_dist_compatible(data_dist_path, VERSION):
|
||||||
|
logger.info("WebUI is up to date.")
|
||||||
|
return str(data_dist_path)
|
||||||
|
|
||||||
if should_use_bundled_dashboard_dist(data_dist_path, VERSION):
|
if should_use_bundled_dashboard_dist(data_dist_path, VERSION):
|
||||||
bundled_dist = get_bundled_dashboard_dist_path()
|
|
||||||
logger.info(
|
logger.info(
|
||||||
"Using bundled WebUI because data/dist is older than core version v%s.",
|
"Replacing data/dist with bundled WebUI because its version does not match core version v%s.",
|
||||||
VERSION,
|
VERSION,
|
||||||
)
|
)
|
||||||
return str(bundled_dist)
|
try:
|
||||||
if v is not None:
|
remove_dir(str(data_dist_path))
|
||||||
# 存在文件
|
shutil.copytree(bundled_dist, data_dist_path)
|
||||||
if v == f"v{VERSION}":
|
return str(data_dist_path)
|
||||||
logger.info("WebUI is up to date.")
|
except Exception as e:
|
||||||
else:
|
|
||||||
logger.warning(
|
logger.warning(
|
||||||
"WebUI version mismatch: %s, expected v%s.",
|
"Failed to replace data/dist with bundled WebUI: %s. Using bundled WebUI directly.",
|
||||||
v,
|
e,
|
||||||
VERSION,
|
|
||||||
)
|
)
|
||||||
return data_dist_path
|
return str(bundled_dist)
|
||||||
|
|
||||||
|
if is_dashboard_version_compatible(v, VERSION):
|
||||||
|
logger.warning(
|
||||||
|
"WebUI files are incomplete for v%s. Re-downloading WebUI.",
|
||||||
|
VERSION,
|
||||||
|
)
|
||||||
|
elif v is not None:
|
||||||
|
logger.warning(
|
||||||
|
"WebUI version mismatch: %s, expected v%s. Re-downloading WebUI.",
|
||||||
|
v,
|
||||||
|
VERSION,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
logger.warning(
|
||||||
|
"WebUI version file is missing. Re-downloading WebUI v%s.",
|
||||||
|
VERSION,
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
await download_dashboard(
|
||||||
|
version=f"v{VERSION}",
|
||||||
|
latest=False,
|
||||||
|
allow_insecure_ssl_fallback=False,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logger.critical(f"下载管理面板文件失败: {e}。")
|
||||||
|
return None
|
||||||
|
logger.info("管理面板下载完成。")
|
||||||
|
return str(data_dist_path)
|
||||||
|
|
||||||
|
if is_dashboard_dist_compatible(bundled_dist, VERSION):
|
||||||
|
logger.info(
|
||||||
|
"Using bundled WebUI v%s.", get_dashboard_dist_version(bundled_dist)
|
||||||
|
)
|
||||||
|
return str(bundled_dist)
|
||||||
|
|
||||||
logger.info(
|
logger.info(
|
||||||
"Downloading WebUI. If it fails, download dist.zip from https://github.com/AstrBotDevs/AstrBot/releases/latest and extract dist to data/.",
|
"Downloading WebUI. If it fails, download dist.zip from https://github.com/AstrBotDevs/AstrBot/releases/latest and extract dist to data/.",
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
await download_dashboard(version=f"v{VERSION}", latest=False)
|
await download_dashboard(
|
||||||
|
version=f"v{VERSION}",
|
||||||
|
latest=False,
|
||||||
|
allow_insecure_ssl_fallback=False,
|
||||||
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.critical(f"下载管理面板文件失败: {e}。")
|
logger.critical(f"下载管理面板文件失败: {e}。")
|
||||||
return None
|
return None
|
||||||
|
|
||||||
logger.info("管理面板下载完成。")
|
logger.info("管理面板下载完成。")
|
||||||
return data_dist_path
|
return str(data_dist_path)
|
||||||
|
|
||||||
|
|
||||||
async def main_async(webui_dir_arg: str | None) -> None:
|
async def main_async(webui_dir_arg: str | None) -> None:
|
||||||
|
|||||||
@@ -273,6 +273,7 @@ def test_dashboard_uses_bundled_dist_when_data_dist_is_stale(
|
|||||||
bundled_dist = tmp_path / "bundled-dist"
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
user_dist.mkdir(parents=True)
|
user_dist.mkdir(parents=True)
|
||||||
bundled_dist.mkdir()
|
bundled_dist.mkdir()
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
"astrbot.dashboard.server.get_astrbot_data_path",
|
"astrbot.dashboard.server.get_astrbot_data_path",
|
||||||
@@ -293,6 +294,32 @@ def test_dashboard_uses_bundled_dist_when_data_dist_is_stale(
|
|||||||
assert server.data_path == str(bundled_dist)
|
assert server.data_path == str(bundled_dist)
|
||||||
|
|
||||||
|
|
||||||
|
def test_dashboard_ignores_mismatched_data_dist_without_bundled(
|
||||||
|
core_lifecycle_td: AstrBotCoreLifecycle,
|
||||||
|
monkeypatch,
|
||||||
|
tmp_path,
|
||||||
|
):
|
||||||
|
data_dir = tmp_path / "data"
|
||||||
|
user_dist = data_dir / "dist"
|
||||||
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
|
(user_dist / "assets").mkdir(parents=True)
|
||||||
|
(user_dist / "assets" / "version").write_text("v0.0.1", encoding="utf-8")
|
||||||
|
|
||||||
|
monkeypatch.setattr(
|
||||||
|
"astrbot.dashboard.server.get_astrbot_data_path",
|
||||||
|
lambda: str(data_dir),
|
||||||
|
)
|
||||||
|
monkeypatch.setattr(
|
||||||
|
"astrbot.dashboard.server.get_bundled_dashboard_dist_path",
|
||||||
|
lambda: bundled_dist,
|
||||||
|
)
|
||||||
|
|
||||||
|
shutdown_event = asyncio.Event()
|
||||||
|
server = AstrBotDashboard(core_lifecycle_td, core_lifecycle_td.db, shutdown_event)
|
||||||
|
|
||||||
|
assert server.data_path is None
|
||||||
|
|
||||||
|
|
||||||
async def _set_dashboard_password_change_required(
|
async def _set_dashboard_password_change_required(
|
||||||
core_lifecycle_td: AstrBotCoreLifecycle,
|
core_lifecycle_td: AstrBotCoreLifecycle,
|
||||||
required: bool,
|
required: bool,
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from unittest import mock
|
|||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from astrbot.core.utils.io import should_use_bundled_dashboard_dist
|
from astrbot.core.utils.io import get_dashboard_version, should_use_bundled_dashboard_dist
|
||||||
from main import (
|
from main import (
|
||||||
DASHBOARD_RESET_PASSWORD_ENV,
|
DASHBOARD_RESET_PASSWORD_ENV,
|
||||||
_apply_startup_env_flags,
|
_apply_startup_env_flags,
|
||||||
@@ -173,49 +173,108 @@ def test_version_info_comparisons():
|
|||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_check_dashboard_files_not_exists(monkeypatch):
|
async def test_check_dashboard_files_not_exists(tmp_path):
|
||||||
"""Tests dashboard download when files do not exist."""
|
"""Tests dashboard download when files do not exist."""
|
||||||
monkeypatch.setattr(os.path, "exists", lambda x: False)
|
data_dir = tmp_path / "data"
|
||||||
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
|
|
||||||
with mock.patch("main.download_dashboard") as mock_download:
|
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
||||||
await check_dashboard_files()
|
with mock.patch(
|
||||||
|
"main.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=bundled_dist,
|
||||||
|
):
|
||||||
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
|
result = await check_dashboard_files()
|
||||||
|
|
||||||
|
from main import VERSION
|
||||||
|
|
||||||
|
assert result == str(data_dir / "dist")
|
||||||
mock_download.assert_called_once()
|
mock_download.assert_called_once()
|
||||||
|
mock_download.assert_called_once_with(
|
||||||
|
version=f"v{VERSION}",
|
||||||
|
latest=False,
|
||||||
|
allow_insecure_ssl_fallback=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_check_dashboard_files_exists_and_version_match(monkeypatch):
|
async def test_check_dashboard_files_exists_and_version_match(tmp_path):
|
||||||
"""Tests that dashboard is not downloaded when it exists and version matches."""
|
"""Tests that dashboard is not downloaded when it exists and version matches."""
|
||||||
# Mock os.path.exists to return True
|
from main import VERSION
|
||||||
monkeypatch.setattr(os.path, "exists", lambda x: True)
|
|
||||||
|
|
||||||
# Mock get_dashboard_version to return the current version
|
data_dir = tmp_path / "data"
|
||||||
with mock.patch("main.get_dashboard_version") as mock_get_version:
|
data_dist = data_dir / "dist"
|
||||||
# We need to import VERSION from main's context
|
(data_dist / "assets").mkdir(parents=True)
|
||||||
from main import VERSION
|
(data_dist / "assets" / "version").write_text(f"v{VERSION}", encoding="utf-8")
|
||||||
|
(data_dist / "index.html").write_text("user", encoding="utf-8")
|
||||||
mock_get_version.return_value = f"v{VERSION}"
|
|
||||||
|
|
||||||
|
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
||||||
with mock.patch("main.download_dashboard") as mock_download:
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
await check_dashboard_files()
|
result = await check_dashboard_files()
|
||||||
# Assert that download_dashboard was NOT called
|
assert result == str(data_dist)
|
||||||
mock_download.assert_not_called()
|
mock_download.assert_not_called()
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_check_dashboard_files_exists_but_version_mismatch(monkeypatch):
|
async def test_check_dashboard_files_exists_but_version_mismatch_downloads(tmp_path):
|
||||||
"""Tests that a warning is logged when dashboard version mismatches."""
|
"""Tests that a mismatched dashboard is downloaded on startup."""
|
||||||
monkeypatch.setattr(os.path, "exists", lambda x: True)
|
from main import VERSION
|
||||||
|
|
||||||
with mock.patch(
|
data_dir = tmp_path / "data"
|
||||||
"main.get_dashboard_version", mock.AsyncMock(return_value="v0.0.1")
|
data_dist = data_dir / "dist"
|
||||||
):
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
with mock.patch("main.logger.warning") as mock_logger_warning:
|
(data_dist / "assets").mkdir(parents=True)
|
||||||
await check_dashboard_files()
|
(data_dist / "assets" / "version").write_text("v0.0.1", encoding="utf-8")
|
||||||
|
|
||||||
|
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
||||||
|
with mock.patch(
|
||||||
|
"main.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=bundled_dist,
|
||||||
|
):
|
||||||
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
|
with mock.patch("main.logger.warning") as mock_logger_warning:
|
||||||
|
result = await check_dashboard_files()
|
||||||
|
|
||||||
|
assert result == str(data_dist)
|
||||||
|
mock_download.assert_called_once_with(
|
||||||
|
version=f"v{VERSION}",
|
||||||
|
latest=False,
|
||||||
|
allow_insecure_ssl_fallback=False,
|
||||||
|
)
|
||||||
mock_logger_warning.assert_called_once()
|
mock_logger_warning.assert_called_once()
|
||||||
call_args, _ = mock_logger_warning.call_args
|
call_args, _ = mock_logger_warning.call_args
|
||||||
assert "WebUI version mismatch" in call_args[0]
|
assert "WebUI version mismatch" in call_args[0]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_check_dashboard_files_downloads_when_matching_dist_is_incomplete(
|
||||||
|
tmp_path,
|
||||||
|
):
|
||||||
|
"""Tests that a version match alone is not enough to serve WebUI."""
|
||||||
|
from main import VERSION
|
||||||
|
|
||||||
|
data_dir = tmp_path / "data"
|
||||||
|
data_dist = data_dir / "dist"
|
||||||
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
|
(data_dist / "assets").mkdir(parents=True)
|
||||||
|
(data_dist / "assets" / "version").write_text(f"v{VERSION}", encoding="utf-8")
|
||||||
|
|
||||||
|
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
||||||
|
with mock.patch(
|
||||||
|
"main.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=bundled_dist,
|
||||||
|
):
|
||||||
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
|
result = await check_dashboard_files()
|
||||||
|
|
||||||
|
assert result == str(data_dist)
|
||||||
|
mock_download.assert_called_once_with(
|
||||||
|
version=f"v{VERSION}",
|
||||||
|
latest=False,
|
||||||
|
allow_insecure_ssl_fallback=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_should_use_bundled_dashboard_dist_when_data_dist_is_stale(tmp_path):
|
def test_should_use_bundled_dashboard_dist_when_data_dist_is_stale(tmp_path):
|
||||||
user_dist = tmp_path / "user-dist"
|
user_dist = tmp_path / "user-dist"
|
||||||
bundled_dist = tmp_path / "bundled-dist"
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
@@ -223,6 +282,7 @@ def test_should_use_bundled_dashboard_dist_when_data_dist_is_stale(tmp_path):
|
|||||||
(bundled_dist / "assets").mkdir(parents=True)
|
(bundled_dist / "assets").mkdir(parents=True)
|
||||||
(user_dist / "assets" / "version").write_text("v4.24.2", encoding="utf-8")
|
(user_dist / "assets" / "version").write_text("v4.24.2", encoding="utf-8")
|
||||||
(bundled_dist / "assets" / "version").write_text("v4.24.4", encoding="utf-8")
|
(bundled_dist / "assets" / "version").write_text("v4.24.4", encoding="utf-8")
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
with mock.patch(
|
with mock.patch(
|
||||||
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
||||||
@@ -231,46 +291,94 @@ def test_should_use_bundled_dashboard_dist_when_data_dist_is_stale(tmp_path):
|
|||||||
assert should_use_bundled_dashboard_dist(user_dist, "v4.24.4") is True
|
assert should_use_bundled_dashboard_dist(user_dist, "v4.24.4") is True
|
||||||
|
|
||||||
|
|
||||||
def test_should_keep_data_dist_when_version_file_is_malformed(tmp_path):
|
def test_should_use_bundled_dashboard_dist_when_version_file_is_malformed(tmp_path):
|
||||||
user_dist = tmp_path / "user-dist"
|
user_dist = tmp_path / "user-dist"
|
||||||
bundled_dist = tmp_path / "bundled-dist"
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
(user_dist / "assets").mkdir(parents=True)
|
(user_dist / "assets").mkdir(parents=True)
|
||||||
(bundled_dist / "assets").mkdir(parents=True)
|
(bundled_dist / "assets").mkdir(parents=True)
|
||||||
(user_dist / "assets" / "version").write_text("not-a-version", encoding="utf-8")
|
(user_dist / "assets" / "version").write_text("not-a-version", encoding="utf-8")
|
||||||
|
(bundled_dist / "assets" / "version").write_text("v4.24.4", encoding="utf-8")
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
with mock.patch(
|
with mock.patch(
|
||||||
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
||||||
return_value=bundled_dist,
|
return_value=bundled_dist,
|
||||||
):
|
):
|
||||||
assert should_use_bundled_dashboard_dist(user_dist, "4.24.4") is False
|
assert should_use_bundled_dashboard_dist(user_dist, "4.24.4") is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_should_use_bundled_dashboard_dist_when_data_version_file_is_missing(tmp_path):
|
||||||
|
user_dist = tmp_path / "user-dist"
|
||||||
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
|
(user_dist / "assets").mkdir(parents=True)
|
||||||
|
(bundled_dist / "assets").mkdir(parents=True)
|
||||||
|
(bundled_dist / "assets" / "version").write_text("v4.24.4", encoding="utf-8")
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
|
with mock.patch(
|
||||||
|
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=bundled_dist,
|
||||||
|
):
|
||||||
|
assert should_use_bundled_dashboard_dist(user_dist, "4.24.4") is True
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_check_dashboard_files_uses_bundled_dist_when_data_dist_is_stale(
|
async def test_get_dashboard_version_uses_bundled_dist_when_data_dist_is_missing(
|
||||||
tmp_path,
|
tmp_path,
|
||||||
):
|
):
|
||||||
"""Tests that a stale data/dist does not override bundled dashboard assets."""
|
"""Tests bundled WebUI version lookup when data/dist is absent."""
|
||||||
|
from main import VERSION
|
||||||
|
|
||||||
|
data_dir = tmp_path / "data"
|
||||||
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
|
(bundled_dist / "assets").mkdir(parents=True)
|
||||||
|
(bundled_dist / "assets" / "version").write_text(f"v{VERSION}", encoding="utf-8")
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
|
with mock.patch(
|
||||||
|
"astrbot.core.utils.io.get_astrbot_data_path",
|
||||||
|
return_value=str(data_dir),
|
||||||
|
):
|
||||||
|
with mock.patch(
|
||||||
|
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=bundled_dist,
|
||||||
|
):
|
||||||
|
assert await get_dashboard_version() == f"v{VERSION}"
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_check_dashboard_files_replaces_stale_data_dist_with_bundled_dist(
|
||||||
|
tmp_path,
|
||||||
|
):
|
||||||
|
"""Tests that a stale data/dist is repaired from bundled dashboard assets."""
|
||||||
|
from main import VERSION
|
||||||
|
|
||||||
data_dir = tmp_path / "data"
|
data_dir = tmp_path / "data"
|
||||||
data_dist = data_dir / "dist"
|
data_dist = data_dir / "dist"
|
||||||
bundled_dist = tmp_path / "bundled-dist"
|
bundled_dist = tmp_path / "bundled-dist"
|
||||||
data_dist.mkdir(parents=True)
|
(data_dist / "assets").mkdir(parents=True)
|
||||||
bundled_dist.mkdir()
|
(bundled_dist / "assets").mkdir(parents=True)
|
||||||
|
(data_dist / "assets" / "version").write_text("v0.0.1", encoding="utf-8")
|
||||||
|
(data_dist / "old.txt").write_text("old", encoding="utf-8")
|
||||||
|
(bundled_dist / "assets" / "version").write_text(f"v{VERSION}", encoding="utf-8")
|
||||||
|
(bundled_dist / "index.html").write_text("bundled", encoding="utf-8")
|
||||||
|
|
||||||
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
with mock.patch("main.get_astrbot_data_path", return_value=str(data_dir)):
|
||||||
with mock.patch(
|
with mock.patch(
|
||||||
"main.get_dashboard_version", mock.AsyncMock(return_value="v0.0.1")
|
"main.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=Path(bundled_dist),
|
||||||
):
|
):
|
||||||
with mock.patch(
|
with mock.patch(
|
||||||
"main.should_use_bundled_dashboard_dist", return_value=True
|
"astrbot.core.utils.io.get_bundled_dashboard_dist_path",
|
||||||
|
return_value=Path(bundled_dist),
|
||||||
):
|
):
|
||||||
with mock.patch(
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
"main.get_bundled_dashboard_dist_path",
|
result = await check_dashboard_files()
|
||||||
return_value=Path(bundled_dist),
|
|
||||||
):
|
|
||||||
with mock.patch("main.download_dashboard") as mock_download:
|
|
||||||
result = await check_dashboard_files()
|
|
||||||
|
|
||||||
assert result == str(bundled_dist)
|
assert result == str(data_dist)
|
||||||
|
assert (data_dist / "assets" / "version").read_text(encoding="utf-8") == f"v{VERSION}"
|
||||||
|
assert (data_dist / "index.html").read_text(encoding="utf-8") == "bundled"
|
||||||
|
assert not (data_dist / "old.txt").exists()
|
||||||
mock_download.assert_not_called()
|
mock_download.assert_not_called()
|
||||||
|
|
||||||
|
|
||||||
@@ -281,7 +389,7 @@ async def test_check_dashboard_files_with_webui_dir_arg(monkeypatch):
|
|||||||
monkeypatch.setattr(os.path, "exists", lambda path: path == valid_dir)
|
monkeypatch.setattr(os.path, "exists", lambda path: path == valid_dir)
|
||||||
|
|
||||||
with mock.patch("main.download_dashboard") as mock_download:
|
with mock.patch("main.download_dashboard") as mock_download:
|
||||||
with mock.patch("main.get_dashboard_version") as mock_get_version:
|
with mock.patch("main.get_dashboard_dist_version") as mock_get_version:
|
||||||
result = await check_dashboard_files(webui_dir=valid_dir)
|
result = await check_dashboard_files(webui_dir=valid_dir)
|
||||||
assert result == valid_dir
|
assert result == valid_dir
|
||||||
mock_download.assert_not_called()
|
mock_download.assert_not_called()
|
||||||
|
|||||||
@@ -440,6 +440,7 @@ async def test_download_dashboard_falls_back_when_hosted_package_is_not_zip(
|
|||||||
path: str,
|
path: str,
|
||||||
show_progress: bool = False, # noqa: ARG001
|
show_progress: bool = False, # noqa: ARG001
|
||||||
progress_callback=None, # noqa: ARG001
|
progress_callback=None, # noqa: ARG001
|
||||||
|
allow_insecure_ssl_fallback: bool = True, # noqa: ARG001
|
||||||
) -> None:
|
) -> None:
|
||||||
calls.append(url)
|
calls.append(url)
|
||||||
parsed = urlparse(url)
|
parsed = urlparse(url)
|
||||||
|
|||||||
Reference in New Issue
Block a user