package vip.mate.channel.webchat; import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.ObjectMapper; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.tags.Tag; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Value; import org.springframework.core.io.FileSystemResource; import org.springframework.core.io.Resource; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import javax.crypto.Mac; import javax.crypto.spec.SecretKeySpec; import java.net.URLEncoder; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Path; import java.security.GeneralSecurityException; import java.security.MessageDigest; import java.util.ArrayList; import java.util.Base64; import vip.mate.channel.web.Utf8SseEmitter; import vip.mate.agent.AgentService; import vip.mate.channel.model.ChannelEntity; import vip.mate.channel.service.ChannelService; import vip.mate.channel.web.ChatStreamTracker; import vip.mate.common.result.R; import vip.mate.memory.event.ConversationCompletionPublisher; import vip.mate.workspace.conversation.ConversationService; import vip.mate.workspace.conversation.model.ConversationEntity; import vip.mate.workspace.conversation.model.MessageContentPart; import vip.mate.workspace.conversation.model.MessageEntity; import java.io.IOException; import java.time.LocalDateTime; import java.util.List; import java.util.Map; import java.util.UUID; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.regex.Pattern; import java.util.stream.Collectors; /** * WebChat 嵌入式对话接口 *

* 独立于 ChatController,使用 API Key 认证(不依赖 JWT)。 * 供外部网站通过 JS SDK 嵌入 MateClaw 对话能力。 *

* 认证方式:请求头 X-MC-Key 携带 API Key * * @author MateClaw Team */ @Tag(name = "WebChat 嵌入式对话") @Slf4j @RestController @RequestMapping("/api/v1/channels/webchat") @RequiredArgsConstructor public class WebChatController { private final ChannelService channelService; private final AgentService agentService; private final ConversationService conversationService; private final ChatStreamTracker streamTracker; private final ObjectMapper objectMapper; private final ConversationCompletionPublisher completionPublisher; private final vip.mate.memory.identity.MemoryOwnerResolver memoryOwnerResolver; private final WebChatFileService fileService; private final WebChatTokenRevocationService tokenRevocationService; private final vip.mate.audit.service.AuditEventService auditService; /** Visitor-token TTL in seconds (7 days). Mirrors GeneratedFileCache's TTL. */ static final long VISITOR_TOKEN_TTL_SECONDS = 7 * 24 * 3600L; /** * Server-only secret used to sign per-visitor tokens. Reuses the JWT secret so no extra * config/migration is needed; it is never sent to the client (unlike the public channel API key). */ @Value("${mateclaw.jwt.secret:MateClaw-JWT-Secret-Key-2024-Please-Change-In-Production}") private String visitorTokenSecret; private final ExecutorService sseExecutor = Executors.newCachedThreadPool(); /** * WebChat SSE 流式对话 */ @Operation(summary = "WebChat SSE 流式对话") @PostMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE) public SseEmitter chatStream( @RequestHeader("X-MC-Key") String apiKey, @RequestBody WebChatRequest request) { // RFC-058 PR-1: Utf8SseEmitter 显式 charset=UTF-8,防止中文 SSE 乱码 SseEmitter emitter = new Utf8SseEmitter(10 * 60 * 1000L); // 验证 API Key 并获取关联的 Channel 配置 ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { sendErrorAndComplete(emitter, "Invalid API Key"); return emitter; } // Resolve the target agent: an explicit request agentId overrides the channel's // bound agent, but must belong to the channel's workspace (anti privilege-escalation: // a shared channel Key must not be able to drive arbitrary agents in other workspaces). Long agentId = channel.getAgentId(); if (request.getAgentId() != null) { var requested = agentService.getAgent(request.getAgentId()); if (requested == null) { sendErrorAndComplete(emitter, "Requested agent not found"); return emitter; } if (channel.getWorkspaceId() != null && requested.getWorkspaceId() != null && !channel.getWorkspaceId().equals(requested.getWorkspaceId())) { sendErrorAndComplete(emitter, "Requested agent does not belong to this channel's workspace"); return emitter; } agentId = request.getAgentId(); } if (agentId == null) { sendErrorAndComplete(emitter, "No agent configured for this WebChat channel"); return emitter; } final Long resolvedAgentId = agentId; // Optional sessionId lets one visitor hold multiple isolated threads. It is only ever // composed into the server-derived conversationId (kept under the key+visitor namespace), // never accepted as a raw conversationId — so a caller can't reach another tenant's history. final String visitorId; final String effectiveSessionId; try { visitorId = normalizeVisitorId(request.getVisitorId()); effectiveSessionId = normalizeSessionId(request.getSessionId()); } catch (IllegalArgumentException ex) { sendErrorAndComplete(emitter, ex.getMessage()); return emitter; } String conversationId = deriveConversationId(apiKey, visitorId, effectiveSessionId); // Server-issued, unforgeable proof that this caller owns this visitorId. Returned in the // meta event below; the session-management endpoints require it back (see verifyVisitorToken). final String visitorToken = computeVisitorToken(visitorTokenSecret, channel.getId(), visitorId); String message = request.getMessage() != null ? request.getMessage() : ""; if (message.isBlank()) { sendErrorAndComplete(emitter, "Message is required"); return emitter; } log.info("[WebChat] Stream: agentId={}, conversationId={}, visitor={}", agentId, conversationId, visitorId); // 注册 emitter 回调 emitter.onCompletion(() -> log.debug("[WebChat] SSE completed: {}", conversationId)); emitter.onTimeout(() -> { log.debug("[WebChat] SSE timeout: {}", conversationId); streamTracker.complete(conversationId); }); emitter.onError(e -> { log.debug("[WebChat] SSE error: {} - {}", conversationId, e.getMessage()); streamTracker.complete(conversationId); }); sseExecutor.execute(() -> { try { // 创建或获取会话(workspace 从 agent 获取) var webAgent = agentService.getAgent(resolvedAgentId); Long webWsId = webAgent != null ? webAgent.getWorkspaceId() : 1L; var conv = conversationService.getOrCreateWebchatConversation( conversationId, resolvedAgentId, webchatUsername(visitorId), webWsId, effectiveSessionId); // 保存用户消息(含访客本轮引用的附件)。附件元数据一律服务端按 fileId 回查, // 不信客户端传入;path 用于 Agent 侧工具读取,对外消息视图会被剥离。 List userParts = buildUserParts(conversationId, message, request.getAttachmentIds()); conversationService.saveMessage(conversationId, "user", message, userParts); // 初始化 SSE 流跟踪 streamTracker.register(conversationId); streamTracker.attach(conversationId, emitter); // Echo the effective session so the caller can persist it (especially when // sessionId was omitted) and address the same thread on subsequent calls. The // visitorToken must be stored by the caller and sent back on list/messages/delete. streamTracker.broadcast(conversationId, "meta", "{\"sessionId\":" + escapeJson(effectiveSessionId) + ",\"conversationId\":" + escapeJson(conversationId) + ",\"visitorToken\":" + escapeJson(visitorToken) + "}"); // Accumulate the assistant reply so it can be persisted on stream completion. // Pattern mirrors ChatController: always accumulate, only broadcast when the // delta is not a persistence-only echo of content already streamed by inner nodes. StringBuilder assistantReply = new StringBuilder(); // Token usage + model attribution: capture _usage_final event emitted at stream end final int[] usage = {0, 0}; // [promptTokens, completionTokens] final String[] modelInfo = {null, null}; // [runtimeModel, runtimeProvider] // Attribute memory to this external visitor so each end-user // behind the shared webchat account is isolated. The same origin // resolves the owner key for both the read (recall) and write // (publish) paths below. vip.mate.agent.context.ChatOrigin webchatOrigin = vip.mate.agent.context.ChatOrigin.web(conversationId, visitorId, webWsId, null) .withSender(null, "api", null); String webchatOwnerKey = memoryOwnerResolver.resolve(webchatOrigin); reactor.core.Disposable disposable = agentService.chatStructuredStream(resolvedAgentId, message, conversationId, visitorId, null, webchatOrigin) .doOnNext(delta -> { if (delta.isEvent() && "_usage_final".equals(delta.eventType())) { Map data = delta.eventData(); usage[0] = ((Number) data.getOrDefault("promptTokens", 0)).intValue(); usage[1] = ((Number) data.getOrDefault("completionTokens", 0)).intValue(); Object model = data.get("runtimeModelName"); Object provider = data.get("runtimeProviderId"); if (model != null) modelInfo[0] = model.toString(); if (provider != null) modelInfo[1] = provider.toString(); } // Forward a curated subset of agent lifecycle events to the // visitor SSE stream. The full event vocabulary (iteration_*, // perf_summary, _routing_decision, feedback_event, ...) is // internal — exposing it to 3rd-party websites would leak // graph internals and complicate the SDK contract. The four // types below are the ones that drive visible UX: typing // indicator (phase), tool execution badges (tool_start/end), // plan-execute checklist (plan). See docs/zh/webchat.md. if (delta.isEvent()) { forwardVisitorEvent(conversationId, delta.eventType(), delta.eventData()); } if (delta.content() != null && !delta.content().isEmpty()) { assistantReply.append(delta.content()); if (!delta.persistenceOnly()) { streamTracker.broadcast(conversationId, "content_delta", "{\"text\":" + escapeJson(delta.content()) + "}"); } } if (delta.thinking() != null && !delta.thinking().isEmpty() && !delta.persistenceOnly()) { streamTracker.broadcast(conversationId, "thinking_delta", "{\"text\":" + escapeJson(delta.thinking()) + "}"); } }) .doOnComplete(() -> { String reply = assistantReply.toString(); try { if (!reply.isBlank()) { conversationService.saveMessage( conversationId, "assistant", reply, List.of(), "completed", usage[0], usage[1], modelInfo[0], modelInfo[1]); } completionPublisher.publish( resolvedAgentId, conversationId, message, reply, "webchat", webchatOwnerKey); } catch (Exception persistErr) { log.warn("[WebChat] Failed to persist assistant reply / publish event: {}", persistErr.getMessage()); } streamTracker.broadcast(conversationId, "done", "{\"status\":\"completed\"}"); streamTracker.complete(conversationId); }) .doOnError(e -> { log.error("[WebChat] Stream error: {}", e.getMessage()); streamTracker.broadcast(conversationId, "error", "{\"message\":" + escapeJson(e.getMessage()) + "}"); streamTracker.complete(conversationId); }) .subscribe(); // Bind the subscription's Disposable so requestStop() (invoked by // POST /sessions/stop) can actually dispose the Flux and interrupt // the LLM stream. Without this, stopRequested is set but the underlying // HTTP call keeps running — token burn + side-effect tools still fire. // Mirrors ChatController#chatStream line 495. streamTracker.setDisposable(conversationId, disposable); } catch (Exception e) { log.error("[WebChat] Error: {}", e.getMessage(), e); try { emitter.send(SseEmitter.event().name("error") .data(Map.of("message", e.getMessage()))); emitter.complete(); } catch (IOException ex) { emitter.completeWithError(ex); } } }); return emitter; } /** * 获取 WebChat 配置(前端 SDK 初始化用) */ @Operation(summary = "获取 WebChat 配置") @GetMapping("/config") public R> getConfig(@RequestHeader("X-MC-Key") String apiKey) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } JsonNode config = parseConfig(channel.getConfigJson()); return R.ok(Map.of( "channelName", channel.getName(), "agentId", channel.getAgentId() != null ? channel.getAgentId() : 0, "title", textOrDefault(config, "title", channel.getName()), "placeholder", textOrDefault(config, "placeholder", "Type a message..."), "primaryColor", textOrDefault(config, "primary_color", "#409eff"), "welcomeMessage", textOrDefault(config, "welcome_message", "") )); } /** Cap on how many empty (message_count = 0) threads one visitor may hold on a * channel at once. Guards against pathologic clients churning placeholder * sessions without ever sending a message. */ private static final int MAX_EMPTY_SESSIONS_PER_VISITOR = 5; /** * 显式创建一条访客会话线程(空会话)。 *

* 与 {@code POST /stream} 的隐式 getOrCreate 互补:本端点先建一条 message_count=0 * 的占位线程,调用方拿到 {@code sessionId/conversationId/visitorToken} 之后,再决定 * 何时通过 {@code /stream} 发首条消息。鉴权为访客的首次接触:仅校验 * {@code X-MC-Key},不要求 {@code X-MC-Visitor-Token},后端会签发并回传 token, * 调用方在后续 GET/PUT/DELETE 上必须回带。 *

* 行为: *

*/ @Operation(summary = "显式创建访客会话线程(空会话)") @PostMapping("/sessions") public R> createSession( @RequestHeader("X-MC-Key") String apiKey, @RequestBody(required = false) WebChatCreateSessionRequest request) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } // Resolve agent: explicit request.agentId overrides channel's bound agent, // but must belong to channel's workspace (mirrors /stream). final Long agentId; if (request != null && request.getAgentId() != null) { var requested = agentService.getAgent(request.getAgentId()); if (requested == null) { return R.fail(400, "Requested agent not found"); } if (channel.getWorkspaceId() != null && requested.getWorkspaceId() != null && !channel.getWorkspaceId().equals(requested.getWorkspaceId())) { return R.fail(400, "Requested agent does not belong to this channel's workspace"); } agentId = request.getAgentId(); } else { agentId = channel.getAgentId(); if (agentId == null) { return R.fail(400, "No agent configured for this WebChat channel"); } } final String visitorId; final String sessionId; try { visitorId = normalizeVisitorId(request != null ? request.getVisitorId() : null); sessionId = normalizeSessionId(request != null ? request.getSessionId() : null); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String title = (request != null && request.getTitle() != null) ? request.getTitle().trim() : null; if (title != null && (title.isEmpty() || title.length() > 100)) { return R.fail(400, "title 不合法(1-100 字)"); } String conversationId = deriveConversationId(apiKey, visitorId, sessionId); String owner = webchatUsername(visitorId); // Idempotency: existing thread is returned as-is. Title and every other // field are left untouched — a re-create call must not clobber a previously // set title. Existing rows are exempt from the empty-session quota. ConversationEntity existing = conversationService.findByConversationId(conversationId); if (existing != null && owner.equals(existing.getUsername())) { audit(channel, visitorId, "webchat.create-session", conversationId, "{\"sessionId\":\"" + sessionId + "\",\"idempotent\":true}"); return R.ok(buildCreateSessionResponse(existing, sessionId, channel.getId(), visitorId)); } // Quota: count empty threads this visitor already holds on this channel. // loadVisitorSessions already scopes to (channel prefix ∩ visitor owner). long emptyCount = loadVisitorSessions(apiKey, visitorId).stream() .filter(s -> s.getMessageCount() == null || s.getMessageCount() == 0) .count(); if (emptyCount >= MAX_EMPTY_SESSIONS_PER_VISITOR) { return R.fail(409, "未活跃会话数已达上限(" + MAX_EMPTY_SESSIONS_PER_VISITOR + "),请先发送消息或删除旧会话"); } ConversationEntity conv = conversationService.getOrCreateWebchatConversation( conversationId, agentId, owner, channel.getWorkspaceId(), sessionId, title); audit(channel, visitorId, "webchat.create-session", conversationId, "{\"sessionId\":\"" + sessionId + "\",\"idempotent\":false}"); return R.ok(buildCreateSessionResponse(conv, sessionId, channel.getId(), visitorId)); } private Map buildCreateSessionResponse(ConversationEntity conv, String sessionId, Long channelId, String visitorId) { String visitorToken = computeVisitorToken(visitorTokenSecret, channelId, visitorId); // LinkedHashMap (not Map.of) because Map.of rejects null and we want a // stable key order for the response payload. Map m = new java.util.LinkedHashMap<>(); m.put("sessionId", sessionId != null ? sessionId : ""); m.put("conversationId", conv.getConversationId()); m.put("visitorToken", visitorToken); m.put("title", conv.getTitle() != null ? conv.getTitle() : ""); m.put("createTime", conv.getCreateTime()); return m; } /** * Audit a visitor-side write. Actor is {@code "webchat::"} * so audit searches can filter by channel / visitor. detailJson should be * a JSON object capturing whatever the operator would need to reconstruct * the call (sessionId, before/after state, etc). */ private void audit(ChannelEntity channel, String visitorId, String action, String conversationId, String detailJson) { String actor = "webchat:" + channel.getId() + ":" + visitorId; auditService.recordAs(actor, channel.getWorkspaceId(), action, "CONVERSATION", conversationId, null, detailJson); } /** * 列出某访客的会话线程 *

* 仅返回属于本 Key + visitorId 的会话(按 conversationId 前缀过滤), * 不暴露裸 conversationId,调用方按 sessionId 寻址。 */ @Operation(summary = "列出访客会话线程") @GetMapping("/sessions") public R> listSessions( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(defaultValue = "false") boolean includeArchived) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } return R.ok(loadVisitorSessions(apiKey, visitorId, includeArchived)); } /** * 分页 + 关键词搜索某访客的会话线程。 *

访客的会话集是按 visitor 命名空间限定的(数量有界),故在内存里做关键词过滤与分页。 * keyword 不区分大小写、匹配标题子串。 *

鉴权链跟 {@link #listSessions} 完全一致,本方法只做"列表 → 关键词过滤 → 分页"的视图 * 包装,所以直接委托 listSessions 后处理(避免重复 resolveChannel + verifyVisitorToken * 的鉴权代码)。 */ @Operation(summary = "分页查询访客会话线程") @GetMapping("/sessions/page") public R> pageSessions( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(defaultValue = "1") int page, @RequestParam(defaultValue = "20") int size, @RequestParam(required = false) String keyword, @RequestParam(defaultValue = "false") boolean includeArchived) { @SuppressWarnings("unchecked") R> base = (R>) (R) listSessions(apiKey, visitorToken, visitorId, includeArchived); if (base.getCode() != 200) { return R.fail(base.getCode(), base.getMsg()); } if (page < 1) page = 1; if (size < 1 || size > 200) size = 20; List all = base.getData(); if (keyword != null && !keyword.isBlank()) { String kw = keyword.trim().toLowerCase(java.util.Locale.ROOT); all = all.stream() .filter(s -> s.getTitle() != null && s.getTitle().toLowerCase(java.util.Locale.ROOT).contains(kw)) .collect(Collectors.toList()); } long total = all.size(); int from = Math.min((page - 1) * size, all.size()); int to = Math.min(from + size, all.size()); List pageItems = all.subList(from, to); return R.ok(Map.of( "items", pageItems, "total", total, "page", page, "size", size )); } /** * 重命名某会话线程。标题非空、长度 ≤ 100。 */ @Operation(summary = "重命名会话线程") @PutMapping("/sessions/title") public R renameSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestBody Map body) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } String title = body != null && body.get("title") != null ? body.get("title").trim() : ""; if (title.isEmpty() || title.length() > 100) { return R.fail(400, "标题不合法(1-100 字)"); } conversationService.renameConversation(conversationId, title); audit(channel, visitorId, "webchat.rename-session", conversationId, "{\"sessionId\":\"" + sid + "\",\"title\":\"" + title + "\"}"); return R.ok(); } /** * 置顶 / 取消置顶某会话线程。Pinned 线程在访客的 /sessions 列表里排在最前 * (沿用 {@link ConversationService#listWebchatConversations} 的 pinned DESC 排序)。 */ @Operation(summary = "置顶 / 取消置顶会话线程") @PutMapping("/sessions/pinned") public R pinSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestBody Map body) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } Object v = body != null ? body.get("pinned") : null; if (!(v instanceof Boolean)) { return R.fail(400, "body must contain {pinned: true|false}"); } conversationService.setPinned(conversationId, (Boolean) v); audit(channel, visitorId, "webchat.pin-session", conversationId, "{\"sessionId\":\"" + sid + "\",\"pinned\":" + v + "}"); return R.ok(); } /** * 归档 / 取消归档某会话线程。归档后线程仍在 DB(历史保留、按 sessionId 寻址、文件可下载), * 但默认从 /sessions 列表隐藏;调用方需传 {@code includeArchived=true} 才能看到。 */ @Operation(summary = "归档 / 取消归档会话线程") @PutMapping("/sessions/archive") public R archiveSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestBody Map body) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } Object v = body != null ? body.get("archived") : null; if (!(v instanceof Boolean)) { return R.fail(400, "body must contain {archived: true|false}"); } conversationService.setArchived(conversationId, (Boolean) v); audit(channel, visitorId, "webchat.archive-session", conversationId, "{\"sessionId\":\"" + sid + "\",\"archived\":" + v + "}"); return R.ok(); } /** * Load this visitor's session threads (own namespace only), mapped to the * compact view. Sorted as {@code listConversations} returns them (pinned * desc, last-active desc). Shared by the list and paginated endpoints. *

* Enumeration is keyed by the visitor's username plus the channel prefix * ({@code webchat::}) rather than the full conversationId prefix, so it * still catches threads whose conversationId hashed (long visitorId + * sessionId). The sessionId is read from the persisted {@code webchatSessionId} * column (set on creation) and only falls back to parsing the conversationId * for legacy rows created before that column existed. */ private List loadVisitorSessions(String apiKey, String visitorId) { return loadVisitorSessions(apiKey, visitorId, false); } /** * Overload that lets the caller opt into archived threads. By default * (used by /sessions listing and the empty-session quota check) archived * rows are filtered out — they still exist on disk and are addressable * by sessionId, but don't pollute the active listing and don't count * against the "≤ 5 empty threads" quota (the visitor already declared * they're done with them). */ private List loadVisitorSessions(String apiKey, String visitorId, boolean includeArchived) { String base = deriveConversationId(apiKey, visitorId, null); String channelPrefix = "webchat:" + apiKey.substring(0, Math.min(8, apiKey.length())) + ":"; String owner = webchatUsername(visitorId); // Query is scoped to this visitor's own rows only (no system rows), so // listing a visitor's threads doesn't load every IM/cron conversation. // The channel prefix is matched in-memory with a literal startsWith so a // '_' / '%' in the api key's first 8 chars can't act as a LIKE wildcard. return conversationService.listWebchatConversations(owner).stream() .filter(c -> c.getConversationId() != null && c.getConversationId().startsWith(channelPrefix)) .filter(c -> includeArchived || c.getArchived() == null || c.getArchived() == 0) .map(c -> { String sid = recoverSessionId(c, base); return new WebChatSessionView(sid, c.getTitle(), c.getLastActiveTime(), c.getMessageCount(), c.getPinned() != null ? c.getPinned() : 0, c.getArchived() != null ? c.getArchived() : 0, c.getStreamStatus() != null ? c.getStreamStatus() : "idle"); }) .collect(Collectors.toList()); } /** * Recover a thread's sessionId. Prefers the persisted column; for legacy * rows (column null) falls back to parsing the non-hashed conversationId. * Returns null for the default (no-session) thread and for legacy hashed rows * whose sessionId can no longer be reconstructed. */ private String recoverSessionId(vip.mate.workspace.conversation.model.ConversationEntity c, String base) { if (c.getWebchatSessionId() != null) { return c.getWebchatSessionId(); } String cid = c.getConversationId(); if (cid.equals(base)) { return null; } String prefix = base + ":"; if (cid.startsWith(prefix)) { return cid.substring(prefix.length()); } return null; } /** * 获取某会话线程的消息列表(支持分页)。 *

不传 limit 时返回全部消息(向后兼容);传 limit 返回最新 limit 条 + hasMore; * 传 beforeId + limit 时返回该 ID 之前的 limit 条(上拉加载更早消息)。 */ @Operation(summary = "获取会话消息(支持分页)") @GetMapping("/sessions/messages") public R sessionMessages( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestParam(required = false) Long beforeId, @RequestParam(required = false) Integer limit) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } // Backward-compatible: no limit → full external (path-stripped) list. if (limit == null || limit <= 0) { return R.ok(conversationService.listMessageViewsExternal(conversationId)); } // Paginated: mirror ConversationController#listMessages but with the // external view so visitors never see server-side file paths. List messages; boolean hasMore; if (beforeId != null) { messages = conversationService.listMessagesBefore(conversationId, beforeId, limit + 1); hasMore = messages.size() > limit; if (hasMore) { messages = messages.subList(messages.size() - limit, messages.size()); } } else { long total = conversationService.countMessages(conversationId); messages = conversationService.listRecentMessages(conversationId, limit); hasMore = total > limit; } return R.ok(Map.of( "messages", conversationService.toExternalMessageViews(messages), "hasMore", hasMore )); } /** * 删除某会话线程 */ @Operation(summary = "删除会话线程") @DeleteMapping("/sessions") public R deleteSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } conversationService.deleteConversation(conversationId); audit(channel, visitorId, "webchat.delete-session", conversationId, "{\"sessionId\":\"" + sid + "\"}"); return R.ok(); } /** * 停止访客某线程正在进行中的 SSE 流。 *

* 鉴权同其他会话管理端点(API Key + visitorToken + 会话归属)。内部调 * {@link ChatStreamTracker#requestStop(String)}——靠 chatStream 注册时绑定的 * Disposable 实际中断 Flux;返回 {@code stopped=false} 表示当前没有活跃流 * (幂等,不报错)。 *

* 不做 approval sweep:webchat 渠道目前不暴露 approval UI,且无 MateClaw * username 可传给 {@code denyAllByConversation}。若未来 webchat 接入审批流, * 再单独评估是否补这层。 */ @Operation(summary = "停止访客会话线程的进行中流") @PostMapping("/sessions/stop") public R> stopSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, visitorId, sid); // ownsConversation is the existence + ownership guard: an unknown sessionId // maps to a conversationId that either doesn't exist or belongs to someone // else — both return 404 so the caller can't probe the namespace. if (!ownsConversation(conversationId, visitorId)) { return R.fail(404, "Session not found"); } boolean stopped = streamTracker.requestStop(conversationId); log.info("[WebChat] Stop requested: conversationId={}, visitor={}, stopped={}", conversationId, visitorId, stopped); audit(channel, visitorId, "webchat.stop-session", conversationId, "{\"sessionId\":\"" + sid + "\",\"stopped\":" + stopped + "}"); return R.ok(Map.of("stopped", stopped)); } /** * 重新生成最后一条助手回复。 *

* 语义:找到会话最后一条 {@code role=user} 消息 → stop 当前流(如有)→ 删除最后一条 * {@code role=assistant} 消息 → 用 last user message 重新启动 agent turn。 * 实际启动复用 {@link #chatStream},它会重新 saveMessage user(新消息 id,内容相同)。 * 这样不重复 100 行 SSE 代码,代价是用户消息多一条(语义上等同"重发")。 *

* 没有任何 user 消息时返回 400(无内容可重新生成)。 */ @Operation(summary = "重新生成最后一条助手回复") @PostMapping(value = "/sessions/regenerate", produces = MediaType.TEXT_EVENT_STREAM_VALUE) public SseEmitter regenerateSession( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId) { SseEmitter emitter = new Utf8SseEmitter(10 * 60 * 1000L); ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { sendErrorAndComplete(emitter, "Invalid API Key"); return emitter; } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { sendErrorAndComplete(emitter, "Invalid or missing visitor token"); return emitter; } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { sendErrorAndComplete(emitter, ex.getMessage()); return emitter; } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { sendErrorAndComplete(emitter, "Session not found"); return emitter; } // Stop any in-flight stream first so its doOnComplete doesn't race the // delete/save below. Single-node webchat means requestStop hits the // right disposable; multi-node is a separate epic. streamTracker.requestStop(conversationId); MessageEntity lastAssistant = conversationService.findLastMessageByRole(conversationId, "assistant"); if (lastAssistant != null) { conversationService.deleteMessageById(lastAssistant.getId()); } MessageEntity lastUser = conversationService.findLastMessageByRole(conversationId, "user"); if (lastUser == null) { sendErrorAndComplete(emitter, "No user message to regenerate from"); return emitter; } log.info("[WebChat] Regenerate: conversationId={}, visitor={}, seedMessageId={}", conversationId, visitorId, lastUser.getId()); audit(channel, visitorId, "webchat.regenerate-session", conversationId, "{\"sessionId\":\"" + sid + "\",\"seedMessageId\":" + lastUser.getId() + "}"); // Reuse chatStream: it'll resolve the agent again (cheap), re-derive // conversationId, saveMessage user (new id, same content), and start // the agent turn. visitorId echoes through to keep the visitor-scoped // memory owner consistent. WebChatRequest req = new WebChatRequest(); req.setMessage(lastUser.getContent()); req.setVisitorId(visitorId); req.setSessionId(sid); return chatStream(apiKey, req); } /** * 上传文件(入站)。访客先上传拿到 fileId,再在 /stream 的 attachmentIds 中引用。 *

鉴权同会话接口:API Key + visitor token;conversationId 服务端派生。 */ @Operation(summary = "WebChat 上传文件") @PostMapping(value = "/upload", consumes = MediaType.MULTIPART_FORM_DATA_VALUE) public R> uploadFile( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestPart("file") MultipartFile file) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return R.fail(401, "Invalid API Key"); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return R.fail(401, "Invalid or missing visitor token"); } // Upload requires an established visitor identity (the token is bound to it); // unlike /stream we never mint a fresh visitorId here. String vid; String sid; try { if (visitorId == null || visitorId.trim().isEmpty()) { return R.fail(400, "visitorId is required"); } vid = normalizeVisitorId(visitorId); sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return R.fail(400, ex.getMessage()); } String conversationId = deriveConversationId(apiKey, vid, sid); try { WebChatFileService.StagedFile stored = fileService.store(conversationId, file); audit(channel, vid, "webchat.upload-file", conversationId, "{\"sessionId\":\"" + sid + "\",\"fileId\":\"" + stored.storedName() + "\",\"size\":" + stored.size() + "}"); return R.ok(Map.of( "fileId", stored.storedName(), "fileName", stored.originalName(), "contentType", stored.contentType() != null ? stored.contentType() : "application/octet-stream", "size", stored.size() )); } catch (WebChatFileService.UploadRejectedException ex) { return R.fail(400, ex.getMessage()); } catch (IOException ex) { log.error("[WebChat] Upload failed conv={}: {}", conversationId, ex.getMessage()); return R.fail(500, "Upload failed"); } } /** * 下载文件(出站)。serves both visitor-uploaded files and agent-produced files * written under the conversation dir. 鉴权同上,路径在服务端派生目录内防穿越。 */ @Operation(summary = "WebChat 下载文件") @GetMapping("/files") public ResponseEntity downloadFile( @RequestHeader("X-MC-Key") String apiKey, @RequestHeader(value = "X-MC-Visitor-Token", required = false) String visitorToken, @RequestParam String visitorId, @RequestParam(required = false) String sessionId, @RequestParam String storedName) { ChannelEntity channel = resolveChannel(apiKey); if (channel == null) { return ResponseEntity.status(401).build(); } if (!verifyVisitorToken(visitorTokenSecret, channel.getId(), visitorId, visitorToken)) { return ResponseEntity.status(401).build(); } String sid; try { sid = normalizeSessionId(sessionId); } catch (IllegalArgumentException ex) { return ResponseEntity.badRequest().build(); } String conversationId = deriveConversationId(apiKey, visitorId, sid); if (!ownsConversation(conversationId, visitorId)) { return ResponseEntity.status(404).build(); } Path file = fileService.resolve(conversationId, storedName).orElse(null); if (file == null) { return ResponseEntity.notFound().build(); } String contentType; try { contentType = Files.probeContentType(file); } catch (IOException e) { contentType = null; } MediaType mediaType = MediaType.APPLICATION_OCTET_STREAM; if (contentType != null) { try { mediaType = MediaType.parseMediaType(contentType); } catch (Exception ignored) { // fall back to octet-stream } } // Only inline images; everything else downloads as an attachment. nosniff // stops the browser from re-interpreting the bytes as active content. boolean inlineImage = contentType != null && contentType.startsWith("image/"); String encodedName = URLEncoder.encode(file.getFileName().toString(), StandardCharsets.UTF_8) .replace("+", "%20"); return ResponseEntity.ok() .contentType(mediaType) .header("X-Content-Type-Options", "nosniff") .header(HttpHeaders.CONTENT_DISPOSITION, (inlineImage ? "inline" : "attachment") + "; filename*=UTF-8''" + encodedName) .body(new FileSystemResource(file)); } /** * Build the user message's content parts: a text part for the message plus a * file/media part for each referenced attachment. Attachment metadata is * resolved server-side from the staging registry (the client only sends opaque * ids); an id that is unknown, expired, or belongs to another conversation is * silently dropped. */ private List buildUserParts(String conversationId, String message, List attachmentIds) { List parts = new ArrayList<>(); if (message != null && !message.isBlank()) { MessageContentPart text = new MessageContentPart(); text.setType("text"); text.setText(message); parts.add(text); } if (attachmentIds != null) { for (String fileId : attachmentIds) { fileService.consume(conversationId, fileId).ifPresent(sf -> { MessageContentPart p = new MessageContentPart(); p.setType(WebChatFileService.partTypeFor(sf.contentType())); p.setFileName(sf.originalName()); p.setContentType(sf.contentType()); p.setStoredName(sf.storedName()); p.setFileSize(sf.size()); // Relative download ref (caller adds auth headers + visitorId/sessionId). p.setFileUrl("/api/v1/channels/webchat/files?storedName=" + URLEncoder.encode(sf.storedName(), StandardCharsets.UTF_8)); // Server path lets the agent's file tools read the upload; stripped from // the external message view (listMessageViewsExternal). fileService.resolve(conversationId, sf.storedName()) .ifPresent(path -> p.setPath(path.toString())); parts.add(p); }); } } return parts; } // ==================== 内部方法 ==================== private static final Pattern SESSION_ID_PATTERN = Pattern.compile("[A-Za-z0-9_-]{1,64}"); /** * 归一化调用方传入的 sessionId:空白 → null;非空必须满足白名单字符集,否则抛出。 */ private String normalizeSessionId(String raw) { if (raw == null) { return null; } String s = raw.trim(); if (s.isEmpty()) { return null; } if (!SESSION_ID_PATTERN.matcher(s).matches()) { throw new IllegalArgumentException( "Invalid sessionId (allowed: letters, digits, '-', '_', length 1-64)"); } return s; } private static final Pattern VISITOR_ID_PATTERN = Pattern.compile("[A-Za-z0-9_.:\\-]{1,128}"); /** * 归一化调用方传入的 visitorId:空白 → 新 UUID;非空必须满足白名单字符集,否则抛出。 * 限制字符集既防注入/控制字符,也为派生的 conversationId / username 提供可预期的边界。 */ private String normalizeVisitorId(String raw) { if (raw == null || raw.trim().isEmpty()) { return UUID.randomUUID().toString(); } String s = raw.trim(); if (!VISITOR_ID_PATTERN.matcher(s).matches()) { throw new IllegalArgumentException( "Invalid visitorId (allowed: letters, digits, '-', '_', '.', ':', length 1-128)"); } return s; } /** * 由服务端拼装 conversationId,始终钳在 key + visitor 命名空间内。 * 绝不接受调用方传入的裸 conversationId。 *

conversation_id 列为 VARCHAR(64);当 visitorId + sessionId 过长导致超出列宽时, * 把可变部分折叠为稳定哈希,保证 id 唯一且有界(否则 INSERT 会在 /stream 处 500)。 */ static String deriveConversationId(String apiKey, String visitorId, String sessionId) { String key8 = apiKey.substring(0, Math.min(8, apiKey.length())); String full = "webchat:" + key8 + ":" + visitorId + (sessionId != null ? ":" + sessionId : ""); if (full.length() <= 64) { return full; } return "webchat:" + key8 + ":#" + sha256Hex(visitorId + "" + (sessionId == null ? "" : sessionId)).substring(0, 40); } /** * 由 visitorId 派生 username(mate_conversation.username,VARCHAR(64))。 * 同样在超长时折叠为哈希,避免 username 溢出列宽。 */ static String webchatUsername(String visitorId) { String u = "webchat:" + visitorId; return u.length() <= 64 ? u : "webchat:#" + sha256Hex(visitorId).substring(0, 40); } private static String sha256Hex(String s) { try { byte[] d = java.security.MessageDigest.getInstance("SHA-256") .digest(s.getBytes(java.nio.charset.StandardCharsets.UTF_8)); StringBuilder sb = new StringBuilder(d.length * 2); for (byte b : d) { sb.append(Character.forDigit((b >> 4) & 0xF, 16)).append(Character.forDigit(b & 0xF, 16)); } return sb.toString(); } catch (java.security.NoSuchAlgorithmException e) { throw new IllegalStateException("SHA-256 unavailable", e); } } /** * 存在性守卫:会话存在且属于本 visitor 命名空间时返回 true,否则 404。 *

注意:这不是鉴权边界——conversationId 由调用方自报的 visitorId 派生, * 等式两边同源,单凭它无法防越权。真正的鉴权由 {@link #verifyVisitorToken} 完成。 */ private boolean ownsConversation(String conversationId, String visitorId) { ConversationEntity conv = conversationService.findByConversationId(conversationId); return conv != null && webchatUsername(visitorId).equals(conv.getUsername()); } /** * 用服务端密钥对 (channelId, visitorId) 做 HMAC-SHA256,签发不可伪造的 visitor token。 * 载荷含 channelId,使 token 不能跨渠道复用。 *

Token 默认 7 天后过期({@link #VISITOR_TOKEN_TTL_SECONDS});过期时间作为后缀 * 明文附加在 HMAC 之后({@code .}),既参与签名也方便解析。 * 过期后访客可通过 {@code /stream} 重新签发({@code /stream} 不校验 token,只签发)。 */ static String computeVisitorToken(String secret, Long channelId, String visitorId) { return computeVisitorToken(secret, channelId, visitorId, java.time.Instant.now().getEpochSecond() + VISITOR_TOKEN_TTL_SECONDS); } /** Test/override hook: explicit expiration epoch second. */ static String computeVisitorToken(String secret, Long channelId, String visitorId, long expiresAtEpochSecond) { try { Mac mac = Mac.getInstance("HmacSHA256"); mac.init(new SecretKeySpec(secret.getBytes(StandardCharsets.UTF_8), "HmacSHA256")); String payload = channelId + ":" + visitorId + ":" + expiresAtEpochSecond; byte[] sig = mac.doFinal(payload.getBytes(StandardCharsets.UTF_8)); return Base64.getUrlEncoder().withoutPadding().encodeToString(sig) + "." + expiresAtEpochSecond; } catch (GeneralSecurityException e) { throw new IllegalStateException("HMAC-SHA256 unavailable", e); } } /** * 校验调用方回传的 token 的签名 + 过期。不查撤销表(撤销是实例层职责, * 见 {@link #verifyVisitorToken})。Static 是为了让单测可以直接验证 HMAC 语义, * 不需要起 Spring context。 */ static boolean verifyVisitorTokenSignature(String secret, Long channelId, String visitorId, String presented) { if (presented == null || presented.isEmpty() || visitorId == null || channelId == null) { return false; } int dot = presented.lastIndexOf('.'); if (dot <= 0 || dot == presented.length() - 1) { return false; } long exp; try { exp = Long.parseLong(presented.substring(dot + 1)); } catch (NumberFormatException e) { return false; } if (java.time.Instant.now().getEpochSecond() >= exp) { return false; } // Constant-time comparison of the full token (sig + ".exp"). HMAC covers // both channelId:visitorId and exp, so any tampering with exp invalidates sig. byte[] expected = computeVisitorToken(secret, channelId, visitorId, exp).getBytes(StandardCharsets.UTF_8); byte[] actual = presented.getBytes(StandardCharsets.UTF_8); return MessageDigest.isEqual(expected, actual); } /** * 完整校验:签名 + 过期 + 撤销。任一不通过返回 false。实例方法,接入 * {@link WebChatTokenRevocationService}。{@code /stream} 第一次接触不调用本方法 * (只签发 token,不校验),所以被撤销的 visitor 仍能发起新会话——撤销只让旧的 * 管理 token 失效,符合 issue #351 的设计。 */ boolean verifyVisitorToken(String secret, Long channelId, String visitorId, String presented) { if (!verifyVisitorTokenSignature(secret, channelId, visitorId, presented)) { return false; } if (tokenRevocationService != null && tokenRevocationService.isRevoked(channelId, visitorId)) { return false; } return true; } /** * 通过 API Key 查找 WebChat 渠道 */ private ChannelEntity resolveChannel(String apiKey) { if (apiKey == null || apiKey.isBlank()) { return null; } List channels = channelService.listChannelsByType("webchat"); for (ChannelEntity channel : channels) { if (!Boolean.TRUE.equals(channel.getEnabled())) continue; JsonNode config = parseConfig(channel.getConfigJson()); String configuredApiKey = textOrDefault(config, "api_key", null); if (configuredApiKey != null && apiKey.equals(configuredApiKey)) { return channel; } } return null; } private JsonNode parseConfig(String configJson) { if (configJson == null || configJson.isBlank()) { return objectMapper.createObjectNode(); } try { return objectMapper.readTree(configJson); } catch (Exception e) { log.warn("[WebChat] Failed to parse configJson: {}", e.getMessage()); return objectMapper.createObjectNode(); } } private String textOrDefault(JsonNode node, String fieldName, String defaultValue) { if (node != null) { JsonNode value = node.get(fieldName); if (value != null && !value.isNull()) { String text = value.asText(); if (!text.isBlank()) { return text; } } } return defaultValue; } private void sendErrorAndComplete(SseEmitter emitter, String message) { try { emitter.send(SseEmitter.event().name("error").data(Map.of("message", message))); emitter.complete(); } catch (IOException e) { emitter.completeWithError(e); } } /** * Forward a curated subset of agent lifecycle events to the visitor SSE * stream as visitor-friendly {@code phase} / {@code tool_start} / * {@code tool_end} / {@code plan} events. Internal event types * ({@code _usage_final}, {@code _routing_decision}, {@code iteration_*}, * {@code perf_summary}, {@code feedback_event}, {@code finish_reason}, * {@code plan_step_*}) are dropped — they leak graph internals and have * no visitor-facing value. * *

Tool arguments are deliberately not forwarded. The agent may * invoke tools with PII / sensitive arguments (file paths, user queries, * credentials); relaying those to a 3rd-party website frontend is a data * leak. The frontend gets only the tool name and renders a localized * label via its own lookup table. * *

Payloads are serialized via the injected {@link ObjectMapper} so * nested maps/lists are encoded correctly (the hand-rolled {@link #escapeJson} * helper is string-only). * *

Backward compat: visitors / SDKs that don't know these event types * silently ignore them per the SSE spec. */ private void forwardVisitorEvent(String conversationId, String eventType, Map data) { if (eventType == null || data == null) return; Map payload; String sseName; switch (eventType) { case "phase": // Graph phase transition (planning / thinking / generating / // summarizing / ...). Lets the SDK show a typing indicator // before the first content_delta lands. sseName = "phase"; payload = Map.of( "phase", String.valueOf(data.getOrDefault("phase", "")), "timestamp", System.currentTimeMillis()); break; case "tool_call_started": // Tool invocation started. Args intentionally omitted — see javadoc. sseName = "tool_start"; payload = Map.of( "tool", String.valueOf(data.getOrDefault("toolName", data.getOrDefault("tool", "")))); break; case "tool_call_completed": // Tool invocation finished. Result content intentionally omitted. sseName = "tool_end"; payload = new java.util.LinkedHashMap<>(); payload.put("tool", String.valueOf(data.getOrDefault("toolName", data.getOrDefault("tool", "")))); Object success = data.get("success"); payload.put("success", success != null ? success : Boolean.TRUE); break; case "plan_created": // Plan-Execute agents expose their step list. The SDK can render // a checklist; subsequent plan_step_* events are dropped (too // granular for a visitor view). sseName = "plan"; payload = Map.of("steps", data.getOrDefault("steps", List.of())); break; default: // Curated allow-list: anything else is internal — silently drop. return; } try { String json = objectMapper.writeValueAsString(payload); streamTracker.broadcast(conversationId, sseName, json); } catch (Exception e) { log.debug("[WebChat] Failed to serialize visitor event {} for {}: {}", eventType, conversationId, e.getMessage()); } } private String escapeJson(String value) { if (value == null) return "null"; return "\"" + value .replace("\\", "\\\\") .replace("\"", "\\\"") .replace("\n", "\\n") .replace("\r", "\\r") .replace("\t", "\\t") + "\""; } // ==================== 请求体 ==================== @lombok.Data public static class WebChatRequest { private String message; private String visitorId; /** Optional: route this call to a specific agent instead of the channel's bound agent. * Must belong to the channel's workspace. *

Only applied when the (visitorId + sessionId) conversation is first created. Once that * conversation exists, its agent is fixed: a different agentId on later requests is silently * ignored. To talk to another agent, use a new sessionId (or a new visitorId). */ private Long agentId; /** Optional: open a distinct conversation thread for the same visitor. * Composed into the server-derived conversationId; never used as a raw conversationId. */ private String sessionId; /** Optional: ids returned by POST /upload, referencing files this visitor uploaded * for this conversation. Metadata is resolved server-side; unknown / foreign / expired * ids are dropped. */ private List attachmentIds; } /** Compact view of one of a visitor's conversation threads. */ @lombok.Data @lombok.AllArgsConstructor public static class WebChatSessionView { /** null for the visitor's default (no-session) thread. */ private String sessionId; private String title; private LocalDateTime lastActiveTime; private Integer messageCount; /** 1 if the visitor pinned this thread, 0 otherwise. */ private Integer pinned; /** 1 if the visitor archived this thread, 0 otherwise. */ private Integer archived; /** {@code running} if a stream is in progress on this thread, else {@code idle}. */ private String streamStatus; } /** Body for {@code POST /sessions} — explicitly create an empty thread. */ @lombok.Data public static class WebChatCreateSessionRequest { /** Optional; server mints a UUID when absent (same convention as /stream). */ private String visitorId; /** Optional; server generates one when absent. Whitelisted charset, ≤ 64 chars. */ private String sessionId; /** Optional; 1–100 chars when non-blank, otherwise left null so the first * /stream message still derives the title (mirrors PUT /sessions/title rules). */ private String title; /** Optional; override the channel's bound agent. Must belong to the channel's * workspace. Only applied on first creation — once the thread exists, a * different agentId is ignored. */ private Long agentId; } }