feat: cli supports ASTRBOT_DASHBOARD_INITIAL_PASSWORD env

Author: Soulter

astrbot/cli/commands/cmd_init.py

@@ -1,11 +1,24 @@
 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"""
@@ -31,6 +44,8 @@ async def initialize_astrbot(astrbot_root: Path) -> None:
         path.mkdir(parents=True, exist_ok=True)
         click.echo(f"{'Created' if not path.exists() else 'Directory exists'}: {path}")
 
+    _initialize_config_from_env(astrbot_root)
+
     await check_dashboard(astrbot_root / "data")
 
 

tests/test_cli_init.py

@@ -0,0 +1,54 @@
+import json
+
+import pytest
+
+from astrbot.cli.commands import cmd_init
+from astrbot.core.utils.auth_password import verify_dashboard_password
+
+
+@pytest.mark.asyncio
+async def test_init_without_initial_password_env_does_not_create_config(
+    monkeypatch,
+    tmp_path,
+):
+    async def fake_check_dashboard(_data_path):
+        return None
+
+    monkeypatch.delenv(cmd_init.DASHBOARD_INITIAL_PASSWORD_ENV, raising=False)
+    monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
+    (tmp_path / ".astrbot").touch()
+
+    await cmd_init.initialize_astrbot(tmp_path)
+
+    assert not (tmp_path / "data" / "cmd_config.json").exists()
+
+
+@pytest.mark.asyncio
+async def test_init_uses_initial_password_env_to_create_config(
+    monkeypatch,
+    tmp_path,
+):
+    async def fake_check_dashboard(_data_path):
+        return None
+
+    initial_password = "AstrBotInitialPassword123"
+    monkeypatch.setenv(cmd_init.DASHBOARD_INITIAL_PASSWORD_ENV, initial_password)
+    monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
+    (tmp_path / ".astrbot").touch()
+
+    await cmd_init.initialize_astrbot(tmp_path)
+
+    config_path = tmp_path / "data" / "cmd_config.json"
+    config = json.loads(config_path.read_text(encoding="utf-8-sig"))
+    dashboard_config = config["dashboard"]
+
+    assert verify_dashboard_password(
+        dashboard_config["pbkdf2_password"],
+        initial_password,
+    )
+    assert verify_dashboard_password(
+        dashboard_config["password"],
+        initial_password,
+    )
+    assert dashboard_config["password_change_required"] is True
+    assert dashboard_config["password_storage_upgraded"] is True