mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
109 lines
3.6 KiB
Python
109 lines
3.6 KiB
Python
import os
|
||
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 whether a path is an AstrBot root directory.
|
||
|
||
Args:
|
||
path: Directory path to inspect.
|
||
|
||
Returns:
|
||
Whether the directory contains the AstrBot root marker.
|
||
"""
|
||
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.
|
||
|
||
Returns:
|
||
The explicit root, current local root, default user root, or current
|
||
directory when no initialized root exists.
|
||
"""
|
||
if root := os.environ.get("ASTRBOT_ROOT"):
|
||
return Path(root).expanduser().resolve()
|
||
|
||
current_root = Path.cwd().resolve()
|
||
if check_astrbot_root(current_root):
|
||
return current_root
|
||
|
||
user_root = (Path.home() / ".astrbot").resolve()
|
||
if check_astrbot_root(user_root):
|
||
return user_root
|
||
|
||
return current_root
|
||
|
||
|
||
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...")
|
||
await download_dashboard(
|
||
path="data/dashboard.zip",
|
||
extract_path=str(astrbot_root),
|
||
version=f"v{VERSION}",
|
||
latest=False,
|
||
)
|
||
click.echo("Dashboard installed successfully")
|
||
|
||
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),
|
||
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 / "dashboard.zip"),
|
||
extract_path=str(astrbot_root),
|
||
version=f"v{VERSION}",
|
||
latest=False,
|
||
)
|
||
click.echo("Dashboard initialized successfully")
|
||
except Exception as e:
|
||
click.echo(f"Failed to download dashboard: {e}")
|
||
return
|