fxdyz

core/event_push.py

@@ -15,6 +15,7 @@ from astrbot.api.star import Context
 
 from .sqlite import AsyncSQLiteDB
 from .server_binding import ServerBindingService
+from .session_control import SessionControlService
 
 
 DEFAULT_WSS_URL = "wss://socket.nicemoe.cn"
@@ -294,11 +295,13 @@ class EventPushService:
         config: AstrBotConfig,
         sqlite: AsyncSQLiteDB,
         server_binding: ServerBindingService,
+        session_control: SessionControlService,
     ):
         self.context = context
         self.config = config
         self.sql = sqlite
         self.server_binding = server_binding
+        self.session_control = session_control
         self.url = str(config.get("jx3api_wss", "") or DEFAULT_WSS_URL).strip()
         self.token = str(config.get("jx3api_wss_token", "") or "").strip()
         self._runner: Optional[asyncio.Task] = None
@@ -520,6 +523,7 @@ class EventPushService:
         return [
             (str(row["session_id"]), str(row.get("server") or "").strip())
             for row in rows
+            if self.session_control.is_allowed(row["session_id"])
         ]
 
     async def list_subscription_statuses(self) -> list[dict[str, Any]]: