mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 01:40:15 +08:00
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
import asyncio
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import click
|
|
from filelock import FileLock, Timeout
|
|
|
|
from ..utils import check_dashboard, get_astrbot_root
|
|
|
|
DASHBOARD_INITIAL_PASSWORD_ENV = "ASTRBOT_DASHBOARD_INITIAL_PASSWORD"
|
|
|
|
|
|
def _initialize_config_from_env(astrbot_root: Path) -> None:
|
|
if DASHBOARD_INITIAL_PASSWORD_ENV not in os.environ:
|
|
return
|
|
|
|
from astrbot.core.config.astrbot_config import AstrBotConfig
|
|
|
|
AstrBotConfig(config_path=str(astrbot_root / "data" / "cmd_config.json"))
|
|
click.echo("Initialized data/cmd_config.json with dashboard initial password.")
|
|
|
|
|
|
async def initialize_astrbot(astrbot_root: Path) -> None:
|
|
"""Execute AstrBot initialization logic.
|
|
|
|
Args:
|
|
astrbot_root: Runtime root directory to initialize.
|
|
"""
|
|
dot_astrbot = astrbot_root / ".astrbot"
|
|
|
|
if not dot_astrbot.exists():
|
|
dot_astrbot.touch()
|
|
click.echo(f"Created {dot_astrbot}")
|
|
|
|
paths = {
|
|
"data": astrbot_root / "data",
|
|
"config": astrbot_root / "data" / "config",
|
|
"plugins": astrbot_root / "data" / "plugins",
|
|
"temp": astrbot_root / "data" / "temp",
|
|
}
|
|
|
|
for name, path in paths.items():
|
|
path_exists = path.exists()
|
|
path.mkdir(parents=True, exist_ok=True)
|
|
click.echo(f"{'Directory exists' if path_exists else 'Created'}: {path}")
|
|
|
|
_initialize_config_from_env(astrbot_root)
|
|
|
|
await check_dashboard(astrbot_root / "data")
|
|
|
|
|
|
@click.command()
|
|
def init() -> None:
|
|
"""Initialize AstrBot"""
|
|
click.echo("Initializing AstrBot...")
|
|
if os.environ.get("ASTRBOT_ROOT"):
|
|
astrbot_root = get_astrbot_root()
|
|
click.echo(f"Using ASTRBOT_ROOT: {astrbot_root}")
|
|
else:
|
|
user_root = (Path.home() / ".astrbot").resolve()
|
|
current_root = Path.cwd().resolve()
|
|
click.echo("Choose AstrBot runtime directory:")
|
|
click.echo(f"1. {user_root} (recommended)")
|
|
click.echo(f"2. Current directory: {current_root}")
|
|
choice = click.prompt(
|
|
"Select",
|
|
type=click.Choice(["1", "2"]),
|
|
default="1",
|
|
show_choices=False,
|
|
)
|
|
astrbot_root = user_root if choice == "1" else current_root
|
|
|
|
astrbot_root.mkdir(parents=True, exist_ok=True)
|
|
os.environ["ASTRBOT_ROOT"] = str(astrbot_root)
|
|
lock_file = astrbot_root / "astrbot.lock"
|
|
lock = FileLock(lock_file, timeout=5)
|
|
|
|
try:
|
|
with lock.acquire():
|
|
asyncio.run(initialize_astrbot(astrbot_root))
|
|
click.echo("Done! You can now run 'astrbot run' to start AstrBot")
|
|
except Timeout:
|
|
raise click.ClickException(
|
|
"Cannot acquire lock file. Please check if another instance is running"
|
|
)
|
|
except click.Abort:
|
|
raise
|
|
|
|
except Exception as e:
|
|
raise click.ClickException(f"Initialization failed: {e!s}")
|