Files
ai-agent/internal/middleware/chat_middleware.go
T
t 18c9354095 refactor: 将客服后端重构为宿主可嵌入模块
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力,补充业务读写工具与客户快捷操作契约。

- 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现,将身份权限与业务主体交由宿主管理。

- 使用 libSQL 重构向量存储,并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。

- 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议,补齐模块初始化、业务动作和公共载荷等回归测试。
2026-08-28 22:23:13 +08:00

46 lines
1.2 KiB
Go

package middleware
import (
"net/url"
"strings"
"code.tczkiot.com/wlw/ai-agent/internal/pkg/httpx"
"code.tczkiot.com/wlw/ai-agent/internal/services"
"github.com/gin-gonic/gin"
)
func ExternalUserMiddleware(ctx *gin.Context) {
channel := services.ChannelService.GetEnabledChannel(ctx)
if channel == nil {
httpx.AbortJSON(ctx, 200, httpx.JsonErrorMsg(ctx, "error.e0210"))
return
}
external, err := services.SubjectService.ResolveExternal(
ctx.Request.Context(),
requestExternalValue(ctx, "X-External-Id", "external_id"),
decodeExternalName(requestExternalValue(ctx, "X-External-Name", "external_name")),
)
if err != nil {
httpx.AbortJSON(ctx, 200, httpx.JsonErrorMsg(ctx, "error.auth.expired"))
return
}
httpx.SetExternalUser(ctx, external)
ctx.Next()
}
func requestExternalValue(ctx *gin.Context, header, query string) string {
if value := strings.TrimSpace(ctx.GetHeader(header)); value != "" {
return value
}
return strings.TrimSpace(ctx.Query(query))
}
func decodeExternalName(value string) string {
decoded, err := url.QueryUnescape(strings.TrimSpace(value))
if err != nil {
return strings.TrimSpace(value)
}
return strings.TrimSpace(decoded)
}