Files
ai-gateway-go/internal/portal/http.go
T
LLMGuardX Dev 87c2b04174 0.11.3: 旗舰版第四轮完善(统一审批中心/工具治理/平台环境变量/数字员工入口/个人渠道/报表多维/租户配额)
- 统一审批中心:模型/资源/渠道/工具四类申请聚合审批,通过自动开通
  (marketplace 安装/渠道授权),outbox 双向站内信;门户可发起/撤回。
- 工具治理:rate_limit_rpm(固定窗口原子 upsert,多实例共享)+ approval_required
  (首次调用自动发起审批,批准前一律拒绝)。
- 平台环境变量:平台级注入 skill/MCP 运行时,个人可覆盖;系统管理员可写。
- 数字员工会话入口:门户列表/对话/调用记录,复用用户运行时凭据。
- 个人渠道:webhook 入站令牌 SHA-256 摘要 + constant-time 校验,绑定已批准
  模型,用量归属用户 Key。
- 报表多维:工具调用/审批授权/安全事件三组统计端点与页面。
- 租户配额:部门 Key/月 Token 上限,运行时凭据开通强制校验,概览展示用量。
- 迁移 000042-000045;修复渠道空 API Key NOT NULL 违约与 inet 扫描;
  25 包测试通过,前后端构建通过,端到端验证完成。
2026-08-13 13:41:22 +08:00

928 lines
28 KiB
Go

package portal
import (
"encoding/json"
"errors"
"net/http"
"strconv"
"strings"
"time"
"aigateway.local/core/internal/identity"
"aigateway.local/core/internal/platform/apiresponse"
)
type HTTPHandler struct {
service *Service
identity *identity.Service
mux *http.ServeMux
}
func NewHTTPHandler(service *Service, identityService *identity.Service) *HTTPHandler {
h := &HTTPHandler{service: service, identity: identityService, mux: http.NewServeMux()}
h.mux.HandleFunc("POST /api/v1/portal/password", h.changePassword)
h.mux.HandleFunc("GET /api/v1/portal/login-logs", h.loginLogs)
h.mux.HandleFunc("GET /api/v1/portal/applications", h.applications)
h.mux.HandleFunc("GET /api/v1/portal/catalog", h.catalog)
h.mux.HandleFunc("GET /api/v1/portal/knowledge", h.knowledge)
h.mux.HandleFunc("GET /api/v1/portal/tools", h.tools)
h.mux.HandleFunc("GET /api/v1/portal/prompts", h.prompts)
h.mux.HandleFunc("GET /api/v1/portal/prompts/{id}", h.prompt)
h.mux.HandleFunc("POST /api/v1/portal/prompts/{id}/favorite", h.favorite(true))
h.mux.HandleFunc("DELETE /api/v1/portal/prompts/{id}/favorite", h.favorite(false))
h.mux.HandleFunc("GET /api/v1/portal/model-requests/available", h.models)
h.mux.HandleFunc("GET /api/v1/portal/model-requests", h.modelRequests)
h.mux.HandleFunc("POST /api/v1/portal/model-requests", h.createModelRequest)
h.mux.HandleFunc("GET /api/v1/portal/resource-requests", h.resourceRequests)
h.mux.HandleFunc("POST /api/v1/portal/resource-requests", h.createResourceRequest)
h.mux.HandleFunc("DELETE /api/v1/portal/resource-requests/{id}", h.cancelResourceRequest)
h.mux.HandleFunc("GET /api/v1/portal/logs", h.logs)
h.mux.HandleFunc("GET /api/v1/portal/logs/{id}", h.logDetail)
h.mux.HandleFunc("GET /api/v1/portal/stats", h.stats)
h.mux.HandleFunc("GET /api/v1/portal/cost", h.cost)
h.mux.HandleFunc("GET /api/v1/portal/docs-info", h.docsInfo)
h.mux.HandleFunc("POST /api/v1/portal/apps/{code}/chat", h.chat)
h.mux.HandleFunc("GET /api/v1/portal/apps/{code}/conversations", h.listConversations)
h.mux.HandleFunc("POST /api/v1/portal/apps/{code}/conversations", h.createConversation)
h.mux.HandleFunc("PATCH /api/v1/portal/apps/{code}/conversations/{id}", h.renameConversation)
h.mux.HandleFunc("DELETE /api/v1/portal/apps/{code}/conversations/{id}", h.deleteConversation)
h.mux.HandleFunc("GET /api/v1/portal/apps/{code}/conversations/{id}", h.getConversation)
h.mux.HandleFunc("POST /api/v1/portal/apps/{code}/conversations/{id}/messages", h.appendConversationMessage)
// 通用聊天:选择已批准模型直接对话(复用用户运行时凭据)。
h.mux.HandleFunc("GET /api/v1/portal/chat/models", h.chatModels)
h.mux.HandleFunc("POST /api/v1/portal/chat/completions", h.chatOnce)
h.mux.HandleFunc("GET /api/v1/portal/chat/sessions", h.listChatSessions)
h.mux.HandleFunc("POST /api/v1/portal/chat/sessions", h.createChatSession)
h.mux.HandleFunc("PATCH /api/v1/portal/chat/sessions/{id}", h.renameChatSession)
h.mux.HandleFunc("DELETE /api/v1/portal/chat/sessions/{id}", h.deleteChatSession)
h.mux.HandleFunc("GET /api/v1/portal/chat/sessions/{id}", h.getChatSession)
h.mux.HandleFunc("POST /api/v1/portal/chat/sessions/{id}/messages", h.appendChatMessage)
// 个人渠道:webhook 入站(公开,令牌鉴权) + 个人管理。
h.mux.HandleFunc("POST /v1/personal-channels/{code}/inbound", h.personalChannelInbound)
h.mux.HandleFunc("GET /api/v1/portal/personal-channels", h.personalChannels)
h.mux.HandleFunc("POST /api/v1/portal/personal-channels", h.createPersonalChannel)
h.mux.HandleFunc("POST /api/v1/portal/personal-channels/{id}/token", h.regeneratePersonalToken)
h.mux.HandleFunc("DELETE /api/v1/portal/personal-channels/{id}", h.deletePersonalChannel)
// 数字员工:会话入口 + 调用记录。
h.mux.HandleFunc("GET /api/v1/portal/digital-employees", h.digitalEmployees)
h.mux.HandleFunc("POST /api/v1/portal/digital-employees/{code}/chat", h.runDigitalEmployee)
h.mux.HandleFunc("GET /api/v1/portal/digital-employees/runs", h.myEmployeeRuns)
h.mux.HandleFunc("GET /api/v1/portal/marketplace", h.marketplace)
h.mux.HandleFunc("GET /api/v1/portal/marketplace/categories", h.marketplaceCategories)
h.mux.HandleFunc("GET /api/v1/portal/marketplace/installed", h.marketplaceInstalled)
h.mux.HandleFunc("GET /api/v1/portal/marketplace/{type}/{code}", h.marketplaceDetail)
h.mux.HandleFunc("POST /api/v1/portal/marketplace/{type}/{code}/install", h.marketplaceInstall)
h.mux.HandleFunc("DELETE /api/v1/portal/marketplace/{type}/{code}/install", h.marketplaceUninstall)
return h
}
func (h *HTTPHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { h.mux.ServeHTTP(w, r) }
func (h *HTTPHandler) account(w http.ResponseWriter, r *http.Request) (identity.Account, bool) {
account, err := h.identity.Authenticate(r.Context(), identity.KindPortal, r.Header.Get("Authorization"))
if err != nil {
apiresponse.Error(w, http.StatusUnauthorized, "登录状态无效或已过期")
return identity.Account{}, false
}
return account, true
}
func decode(w http.ResponseWriter, r *http.Request, target any) bool {
decoder := json.NewDecoder(http.MaxBytesReader(w, r.Body, 1<<20))
decoder.DisallowUnknownFields()
if err := decoder.Decode(target); err != nil {
apiresponse.Error(w, http.StatusBadRequest, "请求格式无效")
return false
}
return true
}
func portalError(w http.ResponseWriter, err error) {
if errors.Is(err, ErrNotFound) {
apiresponse.Error(w, http.StatusNotFound, "资源不存在或无权访问")
return
}
apiresponse.Error(w, http.StatusBadRequest, err.Error())
}
func (h *HTTPHandler) loginLogs(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
logs, err := h.identity.ListLoginLogs(r.Context(), identity.KindPortal, a.Login, 50)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, logs)
}
func (h *HTTPHandler) changePassword(w http.ResponseWriter, r *http.Request) {
account, ok := h.account(w, r)
if !ok {
return
}
var input struct {
OldPassword string `json:"old_password"`
NewPassword string `json:"new_password"`
}
if !decode(w, r, &input) {
return
}
if err := h.identity.ChangePassword(r.Context(), account, input.OldPassword, input.NewPassword); err != nil {
if errors.Is(err, identity.ErrInvalidCredentials) {
apiresponse.Error(w, http.StatusUnauthorized, "当前口令错误")
return
}
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"changed": true})
}
func (h *HTTPHandler) applications(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.Applications(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) knowledge(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.Knowledge(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) tools(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.Tools(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) prompts(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.Prompts(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
query := strings.ToLower(strings.TrimSpace(r.URL.Query().Get("q")))
if query != "" {
filtered := items[:0]
for _, item := range items {
if strings.Contains(strings.ToLower(item.Name+" "+item.Description+" "+strings.Join(item.Tags, " ")), query) {
filtered = append(filtered, item)
}
}
items = filtered
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) prompt(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
item, err := h.service.Prompt(r.Context(), a, r.PathValue("id"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) favorite(value bool) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.SetFavorite(r.Context(), a, r.PathValue("id"), value); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"favorite": value})
}
}
func (h *HTTPHandler) catalog(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
apps, err := h.service.Applications(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
knowledge, err := h.service.Knowledge(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
tools, err := h.service.Tools(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
prompts, err := h.service.Prompts(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"applications": apps, "knowledge": knowledge, "tools": tools, "prompts": prompts})
}
// --- 资源市场 ---
func (h *HTTPHandler) marketplace(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
q := r.URL.Query()
items, err := h.service.MarketplaceCatalog(r.Context(), a, q.Get("type"), q.Get("category_id"), q.Get("tag"), q.Get("q"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) marketplaceCategories(w http.ResponseWriter, r *http.Request) {
if _, ok := h.account(w, r); !ok {
return
}
items, err := h.service.MarketplaceCategories(r.Context())
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) marketplaceInstalled(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.MarketplaceInstalled(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) marketplaceDetail(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
item, raw, err := h.service.MarketplaceDetail(r.Context(), a, r.PathValue("type"), r.PathValue("code"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"item": item, "detail": raw})
}
func (h *HTTPHandler) marketplaceInstall(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
level := strings.TrimSpace(r.URL.Query().Get("permission_level"))
created, err := h.service.MarketplaceInstall(r.Context(), a, r.PathValue("type"), r.PathValue("code"), level)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"installed": true, "created": created})
}
func (h *HTTPHandler) marketplaceUninstall(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.MarketplaceUninstall(r.Context(), a, r.PathValue("type"), r.PathValue("code")); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"installed": false})
}
func (h *HTTPHandler) models(w http.ResponseWriter, r *http.Request) {
if _, ok := h.account(w, r); !ok {
return
}
items, err := h.service.Models(r.Context())
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) modelRequests(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.ModelRequests(r.Context(), a.ID, "")
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) createModelRequest(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input ModelRequest
if !decode(w, r, &input) {
return
}
item, err := h.service.CreateModelRequest(r.Context(), a, input)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
// --- 资源/渠道权限申请 ---
func (h *HTTPHandler) resourceRequests(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.ResourceRequests(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) createResourceRequest(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
ResourceType string `json:"resource_type"`
ResourceCode string `json:"resource_code"`
Reason string `json:"reason"`
}
if !decode(w, r, &input) {
return
}
item, err := h.service.CreateResourceRequest(r.Context(), a, input.ResourceType, input.ResourceCode, input.Reason)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) cancelResourceRequest(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.CancelResourceRequest(r.Context(), a, r.PathValue("id")); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"cancelled": true})
}
func limitParam(r *http.Request) int {
value, _ := strconv.Atoi(r.URL.Query().Get("limit"))
if value < 1 {
value = 50
}
if value > 200 {
value = 200
}
return value
}
func daysParam(r *http.Request) int {
value, _ := strconv.Atoi(r.URL.Query().Get("days"))
if value != 1 && value != 7 && value != 30 {
value = 7
}
return value
}
func (h *HTTPHandler) logs(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
days, limit := daysParam(r), limitParam(r)
rows, err := h.service.pool.Query(r.Context(), `SELECT e.id::text,e.request_id,e.provider_code,e.model,e.protocol,e.status_code,e.prompt_tokens,e.completion_tokens,e.cost_microunits,e.latency_ms,e.labels,e.recorded_at FROM gateway.audit_events e JOIN gateway.api_keys k ON k.id=e.api_key_id WHERE k.portal_user_id=$1 AND e.recorded_at >= clock_timestamp()-make_interval(days=>$2) ORDER BY e.recorded_at DESC LIMIT $3`, a.ID, days, limit)
if err != nil {
portalError(w, err)
return
}
defer rows.Close()
items := []map[string]any{}
for rows.Next() {
var id, requestID, protocol string
var provider, model *string
var status, latency *int
var pt, ct, cost *int64
var labels json.RawMessage
var at time.Time
if err = rows.Scan(&id, &requestID, &provider, &model, &protocol, &status, &pt, &ct, &cost, &latency, &labels, &at); err != nil {
portalError(w, err)
return
}
items = append(items, map[string]any{"id": id, "request_id": requestID, "provider_code": provider, "model": model, "protocol": protocol, "status_code": status, "prompt_tokens": pt, "completion_tokens": ct, "cost_microunits": cost, "latency_ms": latency, "labels": labels, "recorded_at": at})
}
apiresponse.OK(w, map[string]any{"items": items})
}
func (h *HTTPHandler) logDetail(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var id, requestID, protocol string
var provider, model, requestPreview, responsePreview *string
var status, latency *int
var pt, ct, cost *int64
var labels json.RawMessage
var at time.Time
err := h.service.pool.QueryRow(r.Context(), `SELECT e.id::text,e.request_id,e.provider_code,e.model,e.protocol,e.status_code,e.prompt_tokens,e.completion_tokens,e.cost_microunits,e.latency_ms,e.labels,e.request_preview,e.response_preview,e.recorded_at FROM gateway.audit_events e JOIN gateway.api_keys k ON k.id=e.api_key_id WHERE k.portal_user_id=$1 AND e.id=$2 ORDER BY e.recorded_at DESC LIMIT 1`, a.ID, r.PathValue("id")).Scan(&id, &requestID, &provider, &model, &protocol, &status, &pt, &ct, &cost, &latency, &labels, &requestPreview, &responsePreview, &at)
if err != nil {
if errors.Is(err, ErrNotFound) {
portalError(w, err)
} else {
apiresponse.Error(w, http.StatusNotFound, "日志不存在或无权访问")
}
return
}
apiresponse.OK(w, map[string]any{"id": id, "request_id": requestID, "provider_code": provider, "model": model, "protocol": protocol, "status_code": status, "prompt_tokens": pt, "completion_tokens": ct, "cost_microunits": cost, "latency_ms": latency, "labels": labels, "request_preview": requestPreview, "response_preview": responsePreview, "recorded_at": at})
}
func (h *HTTPHandler) stats(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
days := daysParam(r)
var requests, failed, pt, ct, cost int64
var latency *float64
err := h.service.pool.QueryRow(r.Context(), `SELECT count(*),count(*) FILTER (WHERE coalesce(e.status_code,500)>=400),coalesce(sum(e.prompt_tokens),0),coalesce(sum(e.completion_tokens),0),coalesce(sum(e.cost_microunits),0),avg(e.latency_ms) FROM gateway.audit_events e JOIN gateway.api_keys k ON k.id=e.api_key_id WHERE k.portal_user_id=$1 AND e.recorded_at>=clock_timestamp()-make_interval(days=>$2)`, a.ID, days).Scan(&requests, &failed, &pt, &ct, &cost, &latency)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"days": days, "requests": requests, "failed_requests": failed, "prompt_tokens": pt, "completion_tokens": ct, "total_tokens": pt + ct, "cost_microunits": cost, "avg_latency_ms": latency})
}
func (h *HTTPHandler) cost(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var cost int64
err := h.service.pool.QueryRow(r.Context(), `SELECT coalesce(sum(e.cost_microunits),0) FROM gateway.audit_events e JOIN gateway.api_keys k ON k.id=e.api_key_id WHERE k.portal_user_id=$1 AND e.recorded_at>=date_trunc('month',clock_timestamp())`, a.ID).Scan(&cost)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"cost_microunits_this_month": cost, "period": time.Now().UTC().Format("2006-01")})
}
func (h *HTTPHandler) docsInfo(w http.ResponseWriter, r *http.Request) {
if _, ok := h.account(w, r); !ok {
return
}
apiresponse.OK(w, map[string]any{"base_url": "/v1", "protocols": []string{"OpenAI chat/completions", "OpenAI responses", "Anthropic messages", "embeddings"}, "authentication": "Authorization: Bearer <API_KEY>", "note": "密钥只在创建时显示一次,请由管理员审批后通过安全渠道交付。"})
}
type chatInput struct {
Message string `json:"message"`
Messages []map[string]any `json:"messages"`
Variables map[string]any `json:"variables"`
}
func messageFromInput(input chatInput) (string, bool) {
if strings.TrimSpace(input.Message) != "" {
return input.Message, true
}
if len(input.Messages) != 1 {
return "", false
}
role, _ := input.Messages[0]["role"].(string)
content, _ := input.Messages[0]["content"].(string)
return content, role == "user" && strings.TrimSpace(content) != ""
}
func writeApplicationResponse(w http.ResponseWriter, response map[string]any) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
_ = json.NewEncoder(w).Encode(response)
}
func (h *HTTPHandler) chat(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input chatInput
if !decode(w, r, &input) {
return
}
message, ok := messageFromInput(input)
if !ok {
apiresponse.Error(w, 400, "兼容入口只接受一条 user 文本消息")
return
}
response, err := h.service.Chat(r.Context(), a, r.PathValue("code"), message, input.Variables)
if err != nil {
portalError(w, err)
return
}
writeApplicationResponse(w, response)
}
func (h *HTTPHandler) listConversations(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.ListConversations(r.Context(), a, r.PathValue("code"), 100)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) renameConversation(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
Title string `json:"title"`
}
decoder := json.NewDecoder(http.MaxBytesReader(w, r.Body, 1<<20))
decoder.DisallowUnknownFields()
if decoder.Decode(&input) != nil {
apiresponse.Error(w, http.StatusBadRequest, "请求格式无效")
return
}
item, err := h.service.RenameConversation(r.Context(), a, r.PathValue("code"), r.PathValue("id"), input.Title)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) deleteConversation(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.DeleteConversation(r.Context(), a, r.PathValue("code"), r.PathValue("id")); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"deleted": true})
}
func (h *HTTPHandler) createConversation(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
item, err := h.service.CreateConversation(r.Context(), a, r.PathValue("code"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) getConversation(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
item, err := h.service.Conversation(r.Context(), a, r.PathValue("code"), r.PathValue("id"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) appendConversationMessage(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input chatInput
if !decode(w, r, &input) {
return
}
message, ok := messageFromInput(input)
if !ok {
apiresponse.Error(w, 400, "消息不能为空")
return
}
response, err := h.service.AppendConversationMessage(r.Context(), a, r.PathValue("code"), r.PathValue("id"), message, input.Variables)
if err != nil {
portalError(w, err)
return
}
writeApplicationResponse(w, response)
}
// --- 通用聊天 ---
type chatCompletionsInput struct {
ProviderCode string `json:"provider_code"`
Model string `json:"model"`
Message string `json:"message"`
}
func (h *HTTPHandler) chatModels(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.ChatModels(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) chatOnce(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input chatCompletionsInput
if !decode(w, r, &input) {
return
}
response, err := h.service.ChatOnce(r.Context(), a, input.ProviderCode, input.Model, input.Message)
if err != nil {
portalError(w, err)
return
}
writeApplicationResponse(w, response)
}
func (h *HTTPHandler) listChatSessions(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
items, err := h.service.ListChatSessions(r.Context(), a, limit)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) createChatSession(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input chatCompletionsInput
if !decode(w, r, &input) {
return
}
item, err := h.service.CreateChatSession(r.Context(), a, input.ProviderCode, input.Model)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) renameChatSession(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
Title string `json:"title"`
}
if !decode(w, r, &input) {
return
}
item, err := h.service.RenameChatSession(r.Context(), a, r.PathValue("id"), input.Title)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) deleteChatSession(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.DeleteChatSession(r.Context(), a, r.PathValue("id")); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"deleted": true})
}
func (h *HTTPHandler) getChatSession(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
item, err := h.service.ChatSession(r.Context(), a, r.PathValue("id"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, item)
}
func (h *HTTPHandler) appendChatMessage(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
Message string `json:"message"`
}
if !decode(w, r, &input) {
return
}
response, err := h.service.AppendChatMessage(r.Context(), a, r.PathValue("id"), input.Message)
if err != nil {
portalError(w, err)
return
}
writeApplicationResponse(w, response)
}
// --- 个人渠道 ---
func (h *HTTPHandler) personalChannels(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.PersonalChannels(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) createPersonalChannel(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
Code string `json:"code"`
Name string `json:"name"`
ProviderCode string `json:"provider_code"`
Model string `json:"model"`
}
if !decode(w, r, &input) {
return
}
item, token, err := h.service.CreatePersonalChannel(r.Context(), a, input.Code, input.Name, input.ProviderCode, input.Model)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]any{"channel": item, "inbound_token": token, "inbound_url": "/v1/personal-channels/" + item.Code + "/inbound"})
}
func (h *HTTPHandler) regeneratePersonalToken(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
token, err := h.service.RegenerateToken(r.Context(), a, r.PathValue("id"))
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]string{"inbound_token": token})
}
func (h *HTTPHandler) deletePersonalChannel(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
if err := h.service.DeletePersonalChannel(r.Context(), a, r.PathValue("id")); err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, map[string]bool{"deleted": true})
}
// personalChannelInbound 个人渠道入站(公开端点,令牌鉴权,同步返回文本)。
func (h *HTTPHandler) personalChannelInbound(w http.ResponseWriter, r *http.Request) {
token := strings.TrimSpace(r.Header.Get("X-Inbound-Token"))
if token == "" {
token = strings.TrimSpace(r.URL.Query().Get("token"))
}
var input struct {
Message string `json:"message"`
Content string `json:"content"`
}
if !decode(w, r, &input) {
return
}
message := strings.TrimSpace(input.Message)
if message == "" {
message = strings.TrimSpace(input.Content)
}
reply, err := h.service.HandlePersonalInbound(r.Context(), r.PathValue("code"), token, message)
if err != nil {
apiresponse.Error(w, http.StatusBadRequest, err.Error())
return
}
apiresponse.OK(w, map[string]string{"reply": reply})
}
// --- 数字员工 ---
func (h *HTTPHandler) digitalEmployees(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
items, err := h.service.DigitalEmployees(r.Context(), a)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}
func (h *HTTPHandler) runDigitalEmployee(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
var input struct {
Message string `json:"message"`
}
if !decode(w, r, &input) {
return
}
response, err := h.service.RunDigitalEmployee(r.Context(), a, r.PathValue("code"), input.Message)
if err != nil {
portalError(w, err)
return
}
writeApplicationResponse(w, response)
}
func (h *HTTPHandler) myEmployeeRuns(w http.ResponseWriter, r *http.Request) {
a, ok := h.account(w, r)
if !ok {
return
}
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
items, err := h.service.MyEmployeeRuns(r.Context(), a, limit)
if err != nil {
portalError(w, err)
return
}
apiresponse.OK(w, items)
}