LIghtJUNction
04aee2890a
- Add local file caching for dashboard downloads with version validation - Implement fallback to 'latest' version if specific version download fails - Add robust error handling in CLI check_dashboard to prevent crashes - Remove dashboard caching from smoke tests (backend-only mode)
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
from pathlib import Path
|
||
|
||
import click
|
||
|
||
# Static assets bundled inside the installed wheel (built by hatch_build.py).
|
||
_BUNDLED_DIST = Path(__file__).parent.parent.parent / "dashboard" / "dist"
|
||
|
||
|
||
def check_astrbot_root(path: str | Path) -> bool:
|
||
"""Check if the path is an AstrBot root directory"""
|
||
if not isinstance(path, Path):
|
||
path = Path(path)
|
||
if not path.exists() or not path.is_dir():
|
||
return False
|
||
if not (path / ".astrbot").exists():
|
||
return False
|
||
return True
|
||
|
||
|
||
def get_astrbot_root() -> Path:
|
||
"""Get the AstrBot root directory path"""
|
||
return Path.cwd()
|
||
|
||
|
||
async def check_dashboard(astrbot_root: Path) -> None:
|
||
"""Check if the dashboard is installed"""
|
||
from astrbot.core.config.default import VERSION
|
||
from astrbot.core.utils.io import download_dashboard, get_dashboard_version
|
||
|
||
from .version_comparator import VersionComparator
|
||
|
||
# If the wheel ships bundled dashboard assets, no network download is needed.
|
||
if _BUNDLED_DIST.exists():
|
||
click.echo("Dashboard is bundled with the package – skipping download.")
|
||
return
|
||
|
||
try:
|
||
dashboard_version = await get_dashboard_version()
|
||
match dashboard_version:
|
||
case None:
|
||
click.echo("Dashboard is not installed")
|
||
if click.confirm(
|
||
"Install dashboard?",
|
||
default=True,
|
||
abort=True,
|
||
):
|
||
click.echo("Installing dashboard...")
|
||
try:
|
||
await download_dashboard(
|
||
path="data/dashboard.zip",
|
||
extract_path=str(astrbot_root / "data"),
|
||
version=f"v{VERSION}",
|
||
latest=False,
|
||
)
|
||
click.echo("Dashboard installed successfully")
|
||
except Exception as e:
|
||
click.echo(f"Failed to install dashboard: {e}")
|
||
|
||
case str():
|
||
if VersionComparator.compare_version(VERSION, dashboard_version) <= 0:
|
||
click.echo("Dashboard is already up to date")
|
||
return
|
||
try:
|
||
version = dashboard_version.split("v")[1]
|
||
click.echo(f"Dashboard version: {version}")
|
||
await download_dashboard(
|
||
path="data/dashboard.zip",
|
||
extract_path=str(astrbot_root / "data"),
|
||
version=f"v{VERSION}",
|
||
latest=False,
|
||
)
|
||
except Exception as e:
|
||
click.echo(f"Failed to download dashboard: {e}")
|
||
return
|
||
except FileNotFoundError:
|
||
click.echo("Initializing dashboard directory...")
|
||
try:
|
||
await download_dashboard(
|
||
path=str(astrbot_root / "data" / "dashboard.zip"),
|
||
extract_path=str(astrbot_root / "data"),
|
||
version=f"v{VERSION}",
|
||
latest=False,
|
||
)
|
||
click.echo("Dashboard initialized successfully")
|
||
except Exception as e:
|
||
click.echo(f"Failed to download dashboard: {e}")
|
||
return
|