JiuContinent
fbd0aaa9e4
refactor(api): 使用常量替代硬编码字符串 - 在health_check接口中使用ApiResponseKey.STATUS和ApiStatus.OK常量 - 替换硬编码的状态返回值为枚举常量 refactor(core): 配置模块错误信息统一使用常量 - 从constants模块导入ConfigErrorDetail并替换CORS_ORIGINS和LEGACY_ALLOWED_QUERIES的验证错误信息 - 配置类中的默认值使用constants中定义的常量 feat(constants): 添加API响应、安全错误和配置错误常量类 - 新增ApiResponseKey用于API状态键名 - 新增ApiStatus用于API状态值 - 新增SecurityErrorDetail用于安全认证错误详情 - 新增ConfigErrorDetail用于配置验证错误详情 - 添加DEFAULT_MODEL_PROVIDER和DEFAULT_OPENCLAW_ACTION_JSON常量 refactor(security): 安全认证模块使用错误常量 - 将硬编码的安全错误信息替换为SecurityErrorDetail常量 - 包括API密钥、审批密钥和审计密钥的相关错误信息 refactor(ai-agent): AI代理适配器改进错误处理 - 将HTTP状态码替换为FastAPI状态常量 - 添加OpenClaw工具和操作的错误常量 - 修复健康检查和工具调用中的状态码比较逻辑 - 添加AIToolAuditKey用于工具审计键名 feat(ai-agent): 扩展AI代理常量定义 - 新增AIToolAuditKey用于工具审计字段 - 添加OpenClaw相关的错误常量如OPENCLAW_CHAT_PROVIDER_REQUIRED等 - 添加UNSUPPORTED_AI_SKILL_TEMPLATE模板字符串 refactor(approvals): 审批模块常量化重构 - 新增ApprovalPayloadKey用于审批载荷字段 - 添加approval_action函数和APPROVAL_ACTION_SEPARATOR分隔符 - 使用常量替换字面量值 feat(audit): 审计模块新增飞书事件动作类型 - 添加FEISHU_WEBHOOK_EVENT和FEISHU_LONG_CONNECTION_EVENT审计动作 refactor(business): 业务模块全面常量化 - 新增BusinessDomain枚举包含所有业务域 - 添加BusinessResponseKey、BusinessPayloadKey等常量类 - 重构DOMAIN_MODELS为frozenset以提高性能 - 添加normalize_domain等辅助函数用于域标准化 - 使用常量替换路由和业务服务中的硬编码字符串 - 添加业务错误常量和字段验证模板 refactor(feishu): 飞书客户端错误处理优化 - 将HTTP状态码替换为FastAPI标准状态常量 - 改进错误处理的一致性 refactor(approvals): 审批服务使用新常量结构 - 使用ApprovalPayloadKey常量重构载荷字段 - 使用approval_action函数统一动作命名格式 - 优化高风险域判断逻辑 ```
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
import json
|
|
import logging
|
|
from typing import Any
|
|
from urllib.parse import urlsplit
|
|
|
|
from app.core.config import get_settings
|
|
from app.core.database import SessionLocal
|
|
from app.modules.feishu.constants import FEISHU_DEFAULT_OPEN_API_DOMAIN, FeishuEventSource
|
|
from app.modules.feishu.events import FeishuEventService
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _sdk_domain(base_url: str) -> str:
|
|
parsed = urlsplit(base_url)
|
|
if not parsed.scheme or not parsed.netloc:
|
|
return FEISHU_DEFAULT_OPEN_API_DOMAIN
|
|
return f"{parsed.scheme}://{parsed.netloc}"
|
|
|
|
|
|
def _sdk_event_to_payload(event: Any) -> dict[str, Any]:
|
|
try:
|
|
from lark_oapi.core.json import JSON
|
|
except ImportError as exc:
|
|
raise RuntimeError("lark-oapi is required for Feishu long connection") from exc
|
|
|
|
data = JSON.marshal(event)
|
|
if not data:
|
|
return {}
|
|
return json.loads(data)
|
|
|
|
|
|
def _handle_message_event(event: Any) -> None:
|
|
payload = _sdk_event_to_payload(event)
|
|
db = SessionLocal()
|
|
try:
|
|
result = FeishuEventService(db).handle_event(
|
|
payload,
|
|
source=FeishuEventSource.LONG_CONNECTION,
|
|
auto_reply=True,
|
|
)
|
|
logger.info("Handled Feishu long connection event: %s", result)
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def run_long_connection() -> None:
|
|
"""Start the Feishu long connection client and block forever."""
|
|
|
|
settings = get_settings()
|
|
if not settings.feishu_app_id or not settings.feishu_app_secret:
|
|
raise RuntimeError("FEISHU_APP_ID and FEISHU_APP_SECRET are required")
|
|
|
|
try:
|
|
import lark_oapi as lark
|
|
except ImportError as exc:
|
|
raise RuntimeError("Install lark-oapi before starting Feishu long connection") from exc
|
|
|
|
event_handler = (
|
|
lark.EventDispatcherHandler.builder(
|
|
settings.feishu_encrypt_key or "",
|
|
settings.feishu_verification_token or "",
|
|
)
|
|
.register_p2_im_message_receive_v1(_handle_message_event)
|
|
.build()
|
|
)
|
|
client = lark.ws.Client(
|
|
app_id=settings.feishu_app_id,
|
|
app_secret=settings.feishu_app_secret,
|
|
event_handler=event_handler,
|
|
log_level=lark.LogLevel.WARNING,
|
|
domain=_sdk_domain(settings.feishu_base_url),
|
|
)
|
|
logger.info("Starting Feishu long connection client")
|
|
client.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.INFO)
|
|
run_long_connection()
|