mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 01:40:15 +08:00
83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
import os
|
||
import uuid
|
||
|
||
from astrbot.core import astrbot_config, logger
|
||
from astrbot.core.config.default import WEBHOOK_SUPPORTED_PLATFORMS
|
||
|
||
|
||
def _get_callback_api_base() -> str:
|
||
try:
|
||
return astrbot_config.get("callback_api_base", "").rstrip("/")
|
||
except Exception as e:
|
||
logger.error(f"获取 callback_api_base 失败: {e!s}")
|
||
return ""
|
||
|
||
|
||
def _get_dashboard_port() -> int:
|
||
try:
|
||
return astrbot_config.get("dashboard", {}).get("port", 6185)
|
||
except Exception as e:
|
||
logger.error(f"获取 dashboard 端口失败: {e!s}")
|
||
return 6185
|
||
|
||
|
||
def _is_dashboard_ssl_enabled() -> bool:
|
||
env_ssl = os.environ.get("ASTRBOT_DASHBOARD_SSL_ENABLE") or os.environ.get(
|
||
"DASHBOARD_SSL_ENABLE"
|
||
)
|
||
if env_ssl is not None:
|
||
return env_ssl.strip().lower() in {"1", "true", "yes", "on"}
|
||
|
||
try:
|
||
return bool(astrbot_config.get("dashboard", {}).get("ssl", {}).get("enable"))
|
||
except Exception as e:
|
||
logger.error(f"获取 dashboard SSL 配置失败: {e!s}")
|
||
return False
|
||
|
||
|
||
def log_webhook_info(platform_name: str, webhook_uuid: str) -> None:
|
||
"""打印美观的 webhook 信息日志
|
||
|
||
Args:
|
||
platform_name: 平台名称
|
||
webhook_uuid: webhook 的 UUID
|
||
"""
|
||
|
||
callback_base = _get_callback_api_base()
|
||
|
||
if not callback_base:
|
||
callback_base = "http(s)://<your-astrbot-domain>"
|
||
|
||
if not callback_base.startswith("http"):
|
||
callback_base = f"http(s)://{callback_base}"
|
||
|
||
callback_base = callback_base.rstrip("/")
|
||
webhook_url = f"{callback_base}/api/platform/webhook/{webhook_uuid}"
|
||
scheme = "https" if _is_dashboard_ssl_enabled() else "http"
|
||
|
||
display_log = (
|
||
"\n====================\n"
|
||
f"🔗 机器人平台 {platform_name} 已启用统一 Webhook 模式\n"
|
||
f"📍 Webhook 回调地址: \n"
|
||
f" ➜ {scheme}://<your-ip>:{_get_dashboard_port()}/api/platform/webhook/{webhook_uuid}\n"
|
||
f" ➜ {webhook_url}\n"
|
||
"====================\n"
|
||
)
|
||
logger.info(display_log)
|
||
|
||
|
||
def ensure_platform_webhook_config(platform_cfg: dict) -> bool:
|
||
"""为支持统一 webhook 的平台自动生成 webhook_uuid
|
||
|
||
Args:
|
||
platform_cfg (dict): 平台配置字典
|
||
|
||
Returns:
|
||
bool: 如果生成了 webhook_uuid 则返回 True，否则返回 False
|
||
"""
|
||
pt = platform_cfg.get("type", "")
|
||
if pt in WEBHOOK_SUPPORTED_PLATFORMS and not platform_cfg.get("webhook_uuid"):
|
||
platform_cfg["webhook_uuid"] = uuid.uuid4().hex[:16]
|
||
return True
|
||
return False
|