fix: skip unavailable WebUI startup

astrbot/core/initial_loader.py

@@ -22,6 +22,7 @@ def __init__(self, db: BaseDatabase, log_broker: LogBroker) -> None:
         self.logger = logger
         self.log_broker = log_broker
         self.webui_dir: str | None = None
+        self.webui_available = True
 
     async def start(self) -> None:
         core_lifecycle = AstrBotCoreLifecycle(self.log_broker, self.db)
@@ -35,6 +36,14 @@ async def start(self) -> None:
 
         core_task = core_lifecycle.start()
 
+        if not self.webui_available:
+            try:
+                await core_task
+            except asyncio.CancelledError:
+                logger.info("🌈 正在关闭 AstrBot...")
+                await core_lifecycle.stop()
+            return
+
         webui_dir = self.webui_dir
 
         self.dashboard_server = AstrBotDashboard(

main.py

@@ -130,6 +130,7 @@ async def main_async(webui_dir_arg: str | None) -> None:
 
     core_lifecycle = InitialLoader(db, log_broker)
     core_lifecycle.webui_dir = webui_dir
+    core_lifecycle.webui_available = webui_dir is not None
     await core_lifecycle.start()
 
 

tests/test_main.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 import sys
 from pathlib import Path
@@ -257,3 +258,68 @@ async def test_check_dashboard_files_with_webui_dir_arg(monkeypatch):
             assert result == valid_dir
             mock_download.assert_not_called()
             mock_get_version.assert_not_called()
+
+
+@pytest.mark.asyncio
+async def test_main_async_disables_webui_when_dashboard_check_fails():
+    with mock.patch(
+        "main.check_dashboard_files",
+        mock.AsyncMock(return_value=None),
+    ):
+        with mock.patch("main.log_broker", create=True):
+            with mock.patch("main.InitialLoader") as mock_loader:
+                mock_loader.return_value.start = mock.AsyncMock()
+
+                from main import main_async
+
+                await main_async(None)
+
+    assert mock_loader.return_value.webui_dir is None
+    assert mock_loader.return_value.webui_available is False
+    mock_loader.return_value.start.assert_awaited_once()
+
+
+@pytest.mark.asyncio
+async def test_initial_loader_skips_unavailable_webui():
+    from astrbot.core.initial_loader import InitialLoader
+
+    core_lifecycle = mock.MagicMock()
+    core_lifecycle.initialize = mock.AsyncMock()
+    core_lifecycle.start = mock.AsyncMock()
+
+    with mock.patch(
+        "astrbot.core.initial_loader.AstrBotCoreLifecycle",
+        return_value=core_lifecycle,
+    ):
+        with mock.patch(
+            "astrbot.core.initial_loader.AstrBotDashboard",
+        ) as mock_dashboard:
+            loader = InitialLoader(mock.MagicMock(), mock.MagicMock())
+            loader.webui_available = False
+
+            await loader.start()
+
+    core_lifecycle.initialize.assert_awaited_once()
+    core_lifecycle.start.assert_awaited_once()
+    mock_dashboard.assert_not_called()
+
+
+@pytest.mark.asyncio
+async def test_initial_loader_stops_core_when_unavailable_webui_is_cancelled():
+    from astrbot.core.initial_loader import InitialLoader
+
+    core_lifecycle = mock.MagicMock()
+    core_lifecycle.initialize = mock.AsyncMock()
+    core_lifecycle.start = mock.AsyncMock(side_effect=asyncio.CancelledError)
+    core_lifecycle.stop = mock.AsyncMock()
+
+    with mock.patch(
+        "astrbot.core.initial_loader.AstrBotCoreLifecycle",
+        return_value=core_lifecycle,
+    ):
+        loader = InitialLoader(mock.MagicMock(), mock.MagicMock())
+        loader.webui_available = False
+
+        await loader.start()
+
+    core_lifecycle.stop.assert_awaited_once()