103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
import os
|
|
import sys
|
|
from pathlib import Path
|
|
import click
|
|
from astrbot.core.config.default import VERSION
|
|
|
|
|
|
def get_astrbot_root(path: str | None) -> Path:
|
|
"""获取astrbot根目录"""
|
|
match path:
|
|
case None:
|
|
match ASTRBOT_ROOT := os.getenv("ASTRBOT_ROOT"):
|
|
case None:
|
|
astrbot_root = Path.cwd() / "data"
|
|
case _:
|
|
astrbot_root = Path(ASTRBOT_ROOT).resolve()
|
|
case str():
|
|
astrbot_root = Path(path).resolve()
|
|
|
|
dot_astrbot = astrbot_root / ".astrbot"
|
|
if not dot_astrbot.exists():
|
|
if click.confirm(
|
|
f"运行前必须先执行初始化！请检查当前目录是否正确，回车以继续: {astrbot_root}",
|
|
default=True,
|
|
abort=True,
|
|
):
|
|
dot_astrbot.touch()
|
|
astrbot_root.mkdir(parents=True, exist_ok=True)
|
|
click.echo(f"Created {dot_astrbot}")
|
|
|
|
return astrbot_root
|
|
|
|
|
|
def check_astrbot_root(astrbot_root: Path) -> None:
|
|
"""验证"""
|
|
dot_astrbot = astrbot_root / ".astrbot"
|
|
if not astrbot_root.exists():
|
|
click.echo(f"AstrBot root directory does not exist: {astrbot_root}")
|
|
click.echo("Please run 'astrbot init' to create the directory.")
|
|
sys.exit(1)
|
|
else:
|
|
click.echo(f"AstrBot root directory exists: {astrbot_root}")
|
|
if not dot_astrbot.exists():
|
|
click.echo(
|
|
"如果你确认这是 Astrbot root directory, 你需要在当前目录下创建一个 .astrbot 文件标记该目录为 AstrBot 的数据目录。"
|
|
)
|
|
if click.confirm(
|
|
f"请检查当前目录是否正确，确认正确请回车: {astrbot_root}",
|
|
default=True,
|
|
abort=True,
|
|
):
|
|
dot_astrbot.touch()
|
|
click.echo(f"Created {dot_astrbot}")
|
|
else:
|
|
click.echo(f"Welcome back! AstrBot root directory: {astrbot_root}")
|
|
|
|
|
|
async def check_dashboard(astrbot_root: Path) -> None:
|
|
"""检查是否安装了dashboard"""
|
|
try:
|
|
from ..core.utils.io import get_dashboard_version, download_dashboard
|
|
except ImportError:
|
|
from astrbot.core.utils.io import get_dashboard_version, download_dashboard
|
|
|
|
try:
|
|
dashboard_version = await get_dashboard_version()
|
|
match dashboard_version:
|
|
case None:
|
|
click.echo("未安装管理面板")
|
|
if click.confirm(
|
|
"是否安装管理面板？",
|
|
default=True,
|
|
abort=True,
|
|
):
|
|
click.echo("正在安装管理面板...")
|
|
await download_dashboard(
|
|
path="data/dashboard.zip", extract_path=str(astrbot_root)
|
|
)
|
|
click.echo("管理面板安装完成")
|
|
|
|
case str():
|
|
if dashboard_version == f"v{VERSION}":
|
|
click.echo("无需更新")
|
|
else:
|
|
try:
|
|
version = dashboard_version.split("v")[1]
|
|
click.echo(f"管理面板版本: {version}")
|
|
await download_dashboard(
|
|
path="data/dashboard.zip", extract_path=str(astrbot_root)
|
|
)
|
|
except Exception as e:
|
|
click.echo(f"下载管理面板失败: {e}")
|
|
return
|
|
except FileNotFoundError:
|
|
click.echo("初始化管理面板目录...")
|
|
try:
|
|
await download_dashboard(
|
|
path=str(astrbot_root / "dashboard.zip"), extract_path=str(astrbot_root)
|
|
)
|
|
click.echo("管理面板初始化完成")
|
|
except Exception as e:
|
|
click.echo(f"下载管理面板失败: {e}")
|
|
return |