feat: Implement bot orchestration, command handling, and runtime coordination for background services.

This commit is contained in:
2569718930@qq.com
2026-03-13 05:53:41 +08:00
parent d5bbc43b52
commit f7b649bb0a
7 changed files with 206 additions and 64 deletions
+46 -32
View File
@@ -1,61 +1,75 @@
from __future__ import annotations
import os
from typing import Any
from loguru import logger
from src.bot.io_layer import BotIOLayer
from src.bot.services.entitlement_service import BotEntitlementService
class CommandGuard:
"""Unified command gate: entitlement + points charge."""
"""Unified command gate: group membership + points charge."""
def __init__(self, io_layer: BotIOLayer, entitlement_service: BotEntitlementService):
_ALLOWED_MEMBER_STATUSES = {"creator", "administrator", "member", "restricted"}
def __init__(self, io_layer: BotIOLayer, group_chat_id: str | None = None):
self.io_layer = io_layer
self.entitlement_service = entitlement_service
self.group_chat_id = str(group_chat_id or os.getenv("TELEGRAM_CHAT_ID") or "").strip()
self.group_invite_url = str(os.getenv("POLYWEATHER_BOT_GROUP_INVITE_URL") or "").strip()
def ensure_entitled(self, message: Any, command_label: str) -> bool:
def _reply_group_required(self, message: Any) -> None:
lines = ["🔒 该指令仅限群成员使用,请先加入官方群。"]
if self.group_invite_url:
lines.append(f"👉 入群链接: {self.group_invite_url}")
self.io_layer.bot.reply_to(message, "\n".join(lines))
def ensure_group_member(self, message: Any, command_label: str) -> bool:
user = getattr(message, "from_user", None)
user_id = getattr(user, "id", None)
if user_id is None:
return False
decision = self.entitlement_service.check(int(user_id), command_label)
if decision.allowed:
if not self.group_chat_id:
logger.error(
"group member blocked command={} user_id={} reason=missing_TELEGRAM_CHAT_ID",
command_label,
user_id,
)
self.io_layer.bot.reply_to(
message,
"⚠️ 机器人未配置群组准入(TELEGRAM_CHAT_ID),请联系管理员。",
)
return False
try:
member = self.io_layer.bot.get_chat_member(self.group_chat_id, int(user_id))
status = str(getattr(member, "status", "") or "").strip().lower()
except Exception as exc:
logger.info(
"group member blocked command={} user_id={} chat_id={} reason=lookup_failed error={}",
command_label,
user_id,
self.group_chat_id,
exc,
)
self._reply_group_required(message)
return False
if status in self._ALLOWED_MEMBER_STATUSES:
return True
if decision.reason == "bind_required":
denial_text = (
"🔒 当前指令需要订阅权限。\n"
"请先绑定账号后再试:\n"
"<code>/bind &lt;supabase_user_id&gt; [email]</code>"
)
elif decision.reason in {"supabase_subscription_required", "premium_required"}:
denial_text = (
"🔒 当前指令需要高级权限。\n"
"请先开通订阅后再使用。"
)
else:
denial_text = (
"🔒 当前指令需要高级权限。\n"
"请先开通订阅后再使用。"
)
self.io_layer.bot.reply_to(
message,
denial_text,
parse_mode="HTML",
)
logger.info(
"bot entitlement blocked command={} user_id={} reason={}",
"group member blocked command={} user_id={} chat_id={} status={}",
command_label,
user_id,
decision.reason,
self.group_chat_id,
status or "unknown",
)
self._reply_group_required(message)
return False
def ensure_access_and_points(self, message: Any, cost: int, command_label: str) -> bool:
if not self.ensure_entitled(message, command_label):
if not self.ensure_group_member(message, command_label):
return False
return self.io_layer.ensure_query_points(message, cost, command_label)