mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-22 08:15:53 +00:00
style: 完成所有文件的lint
This commit is contained in:
claude-code-best
@@ -1,4 +1,4 @@
|
||||
import type { SetStateAction } from "react";
|
||||
import type { SetStateAction } from 'react'
|
||||
import {
|
||||
apiFetchSession,
|
||||
apiFetchSessionHistory,
|
||||
@@ -7,9 +7,9 @@ import {
|
||||
apiSendControl,
|
||||
apiInterrupt,
|
||||
getUuid,
|
||||
} from "../api/client";
|
||||
import { generateMessageUuid } from "./utils";
|
||||
import type { SessionEvent, EventPayload } from "../types";
|
||||
} from '../api/client'
|
||||
import { generateMessageUuid } from './utils'
|
||||
import type { SessionEvent, EventPayload } from '../types'
|
||||
import type {
|
||||
ThreadEntry,
|
||||
ToolCallData,
|
||||
@@ -19,352 +19,405 @@ import type {
|
||||
ToolCallEntry,
|
||||
UserMessageImage,
|
||||
PendingPermission,
|
||||
} from "./types";
|
||||
} from './types'
|
||||
|
||||
// SSE Event Bus — 复用自 rcs-transport.ts，仅保留连接管理
|
||||
type SSEEventHandler = (event: SessionEvent) => void;
|
||||
type SSEEventHandler = (event: SessionEvent) => void
|
||||
|
||||
class SSEBus {
|
||||
private listeners: Set<SSEEventHandler> = new Set();
|
||||
private eventSource: EventSource | null = null;
|
||||
private listeners: Set<SSEEventHandler> = new Set()
|
||||
private eventSource: EventSource | null = null
|
||||
|
||||
onEvent(handler: SSEEventHandler): () => void {
|
||||
this.listeners.add(handler);
|
||||
return () => this.listeners.delete(handler);
|
||||
this.listeners.add(handler)
|
||||
return () => this.listeners.delete(handler)
|
||||
}
|
||||
|
||||
connect(sessionId: string): void {
|
||||
this.disconnect();
|
||||
const uuid = getUuid();
|
||||
const url = `/web/sessions/${sessionId}/events?uuid=${encodeURIComponent(uuid)}`;
|
||||
const es = new EventSource(url);
|
||||
this.eventSource = es;
|
||||
this.disconnect()
|
||||
const uuid = getUuid()
|
||||
const url = `/web/sessions/${sessionId}/events?uuid=${encodeURIComponent(uuid)}`
|
||||
const es = new EventSource(url)
|
||||
this.eventSource = es
|
||||
|
||||
es.addEventListener("message", (e: MessageEvent) => {
|
||||
es.addEventListener('message', (e: MessageEvent) => {
|
||||
try {
|
||||
const data = JSON.parse(e.data) as SessionEvent;
|
||||
const data = JSON.parse(e.data) as SessionEvent
|
||||
for (const handler of this.listeners) {
|
||||
handler(data);
|
||||
handler(data)
|
||||
}
|
||||
} catch {
|
||||
// ignore parse errors
|
||||
}
|
||||
});
|
||||
})
|
||||
}
|
||||
|
||||
disconnect(): void {
|
||||
if (this.eventSource) {
|
||||
this.eventSource.close();
|
||||
this.eventSource = null;
|
||||
this.eventSource.close()
|
||||
this.eventSource = null
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 全局 SSE bus 实例
|
||||
export const sseBus = new SSEBus();
|
||||
export const sseBus = new SSEBus()
|
||||
|
||||
// =============================================================================
|
||||
// RCS Chat Adapter — 将 SSE 事件转为 ThreadEntry
|
||||
// =============================================================================
|
||||
|
||||
function mapToolStatus(status: string): ToolCallStatus {
|
||||
if (status === "completed") return "complete";
|
||||
if (status === "failed") return "error";
|
||||
return "running";
|
||||
if (status === 'completed') return 'complete'
|
||||
if (status === 'failed') return 'error'
|
||||
return 'running'
|
||||
}
|
||||
|
||||
function extractEventText(payload: EventPayload): string {
|
||||
if (typeof payload.content === "string") return payload.content;
|
||||
if (payload.message && typeof payload.message === "object") {
|
||||
const msg = payload.message as Record<string, unknown>;
|
||||
if (typeof msg.content === "string") return msg.content;
|
||||
if (typeof payload.content === 'string') return payload.content
|
||||
if (payload.message && typeof payload.message === 'object') {
|
||||
const msg = payload.message as Record<string, unknown>
|
||||
if (typeof msg.content === 'string') return msg.content
|
||||
if (Array.isArray(msg.content)) {
|
||||
return (msg.content as Array<Record<string, unknown>>)
|
||||
.filter((b) => b.type === "text" && typeof b.text === "string")
|
||||
.map((b) => b.text as string)
|
||||
.join("");
|
||||
.filter(b => b.type === 'text' && typeof b.text === 'string')
|
||||
.map(b => b.text as string)
|
||||
.join('')
|
||||
}
|
||||
}
|
||||
return "";
|
||||
return ''
|
||||
}
|
||||
|
||||
function findToolCallIndex(entries: ThreadEntry[], toolCallId: string): number {
|
||||
for (let i = entries.length - 1; i >= 0; i--) {
|
||||
const entry = entries[i];
|
||||
if (entry && entry.type === "tool_call" && entry.toolCall.id === toolCallId) {
|
||||
return i;
|
||||
const entry = entries[i]
|
||||
if (
|
||||
entry &&
|
||||
entry.type === 'tool_call' &&
|
||||
entry.toolCall.id === toolCallId
|
||||
) {
|
||||
return i
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
return -1
|
||||
}
|
||||
|
||||
export class RCSChatAdapter {
|
||||
private sessionId: string;
|
||||
private setEntries: React.Dispatch<SetStateAction<ThreadEntry[]>>;
|
||||
private unsub: (() => void) | null = null;
|
||||
private onStatusChange?: (status: string) => void;
|
||||
private onError?: (error: string) => void;
|
||||
private onPermissionRequest?: (permission: PendingPermission) => void;
|
||||
private sessionId: string
|
||||
private setEntries: React.Dispatch<SetStateAction<ThreadEntry[]>>
|
||||
private unsub: (() => void) | null = null
|
||||
private onStatusChange?: (status: string) => void
|
||||
private onError?: (error: string) => void
|
||||
private onPermissionRequest?: (permission: PendingPermission) => void
|
||||
|
||||
constructor(
|
||||
sessionId: string,
|
||||
setEntries: React.Dispatch<SetStateAction<ThreadEntry[]>>,
|
||||
options?: {
|
||||
onStatusChange?: (status: string) => void;
|
||||
onError?: (error: string) => void;
|
||||
onPermissionRequest?: (permission: PendingPermission) => void;
|
||||
onStatusChange?: (status: string) => void
|
||||
onError?: (error: string) => void
|
||||
onPermissionRequest?: (permission: PendingPermission) => void
|
||||
},
|
||||
) {
|
||||
this.sessionId = sessionId;
|
||||
this.setEntries = setEntries;
|
||||
this.onStatusChange = options?.onStatusChange;
|
||||
this.onError = options?.onError;
|
||||
this.onPermissionRequest = options?.onPermissionRequest;
|
||||
this.sessionId = sessionId
|
||||
this.setEntries = setEntries
|
||||
this.onStatusChange = options?.onStatusChange
|
||||
this.onError = options?.onError
|
||||
this.onPermissionRequest = options?.onPermissionRequest
|
||||
}
|
||||
|
||||
/** 初始化：绑定会话、加载历史、连接 SSE */
|
||||
async init(): Promise<void> {
|
||||
try {
|
||||
await apiBind(this.sessionId);
|
||||
await apiBind(this.sessionId)
|
||||
} catch {
|
||||
// may already be bound
|
||||
}
|
||||
|
||||
await this.loadHistory();
|
||||
this.connectSSE();
|
||||
await this.loadHistory()
|
||||
this.connectSSE()
|
||||
}
|
||||
|
||||
/** 加载历史事件并转为 ThreadEntry */
|
||||
async loadHistory(): Promise<void> {
|
||||
const { events } = await apiFetchSessionHistory(this.sessionId);
|
||||
if (!events || events.length === 0) return;
|
||||
const { events } = await apiFetchSessionHistory(this.sessionId)
|
||||
if (!events || events.length === 0) return
|
||||
|
||||
const historyEntries: ThreadEntry[] = [];
|
||||
let currentAssistant: AssistantMessageEntry | null = null;
|
||||
const historyEntries: ThreadEntry[] = []
|
||||
let currentAssistant: AssistantMessageEntry | null = null
|
||||
|
||||
const flushAssistant = () => {
|
||||
if (currentAssistant) {
|
||||
historyEntries.push(currentAssistant);
|
||||
currentAssistant = null;
|
||||
historyEntries.push(currentAssistant)
|
||||
currentAssistant = null
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
for (const event of events) {
|
||||
const payload = event.payload || ({} as EventPayload);
|
||||
const payload = event.payload || ({} as EventPayload)
|
||||
|
||||
if (event.type === "user") {
|
||||
if (event.direction === "outbound") continue; // skip echoed user messages
|
||||
flushAssistant();
|
||||
const text = extractEventText(payload);
|
||||
if (event.type === 'user') {
|
||||
if (event.direction === 'outbound') continue // skip echoed user messages
|
||||
flushAssistant()
|
||||
const text = extractEventText(payload)
|
||||
if (text) {
|
||||
historyEntries.push({
|
||||
type: "user_message",
|
||||
type: 'user_message',
|
||||
id: event.id || `hist-user-${historyEntries.length}`,
|
||||
content: text,
|
||||
});
|
||||
})
|
||||
}
|
||||
} else if (event.type === "assistant") {
|
||||
flushAssistant();
|
||||
const text = extractEventText(payload);
|
||||
const toolParts: ThreadEntry[] = [];
|
||||
} else if (event.type === 'assistant') {
|
||||
flushAssistant()
|
||||
const text = extractEventText(payload)
|
||||
const toolParts: ThreadEntry[] = []
|
||||
|
||||
const msg = payload.message as Record<string, unknown> | undefined;
|
||||
if (msg && typeof msg === "object" && Array.isArray(msg.content)) {
|
||||
const msg = payload.message as Record<string, unknown> | undefined
|
||||
if (msg && typeof msg === 'object' && Array.isArray(msg.content)) {
|
||||
for (const block of msg.content as Array<Record<string, unknown>>) {
|
||||
if (block.type === "tool_use") {
|
||||
if (block.type === 'tool_use') {
|
||||
toolParts.push({
|
||||
type: "tool_call",
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
id: (block.id as string) || `hist-tool-${historyEntries.length}`,
|
||||
title: (block.name as string) || "tool",
|
||||
status: "complete",
|
||||
id:
|
||||
(block.id as string) ||
|
||||
`hist-tool-${historyEntries.length}`,
|
||||
title: (block.name as string) || 'tool',
|
||||
status: 'complete',
|
||||
rawInput: (block.input as Record<string, unknown>) || {},
|
||||
},
|
||||
});
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (text || toolParts.length > 0) {
|
||||
currentAssistant = {
|
||||
type: "assistant_message",
|
||||
type: 'assistant_message',
|
||||
id: event.id || `hist-asst-${historyEntries.length}`,
|
||||
chunks: text ? [{ type: "message", text }] : [],
|
||||
};
|
||||
historyEntries.push(currentAssistant);
|
||||
chunks: text ? [{ type: 'message', text }] : [],
|
||||
}
|
||||
historyEntries.push(currentAssistant)
|
||||
// Push tool calls after assistant message
|
||||
for (const tp of toolParts) {
|
||||
historyEntries.push(tp);
|
||||
historyEntries.push(tp)
|
||||
}
|
||||
currentAssistant = null; // Tool calls are separate entries
|
||||
currentAssistant = null // Tool calls are separate entries
|
||||
}
|
||||
} else if (event.type === "tool_use") {
|
||||
const p = payload as Record<string, unknown>;
|
||||
} else if (event.type === 'tool_use') {
|
||||
const p = payload as Record<string, unknown>
|
||||
const tc: ToolCallEntry = {
|
||||
type: "tool_call",
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
id: (p.tool_call_id as string) || `hist-tool-${historyEntries.length}`,
|
||||
title: (p.tool_name as string) || "tool",
|
||||
status: "complete",
|
||||
id:
|
||||
(p.tool_call_id as string) ||
|
||||
`hist-tool-${historyEntries.length}`,
|
||||
title: (p.tool_name as string) || 'tool',
|
||||
status: 'complete',
|
||||
rawInput: (p.tool_input as Record<string, unknown>) || {},
|
||||
},
|
||||
};
|
||||
historyEntries.push(tc);
|
||||
} else if (event.type === "tool_result") {
|
||||
const p = payload as Record<string, unknown>;
|
||||
}
|
||||
historyEntries.push(tc)
|
||||
} else if (event.type === 'tool_result') {
|
||||
const p = payload as Record<string, unknown>
|
||||
// Find last tool call and update with output
|
||||
const idx = findToolCallIndex(historyEntries, (p.tool_call_id as string) || "");
|
||||
const idx = findToolCallIndex(
|
||||
historyEntries,
|
||||
(p.tool_call_id as string) || '',
|
||||
)
|
||||
if (idx >= 0) {
|
||||
const entry = historyEntries[idx] as ToolCallEntry;
|
||||
const entry = historyEntries[idx] as ToolCallEntry
|
||||
historyEntries[idx] = {
|
||||
type: "tool_call",
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
...entry.toolCall,
|
||||
rawOutput: { output: p.content || p.output || "" },
|
||||
rawOutput: { output: p.content || p.output || '' },
|
||||
},
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
flushAssistant();
|
||||
this.setEntries(historyEntries);
|
||||
flushAssistant()
|
||||
this.setEntries(historyEntries)
|
||||
}
|
||||
|
||||
/** 连接 SSE 事件流 */
|
||||
connectSSE(): void {
|
||||
sseBus.connect(this.sessionId);
|
||||
this.unsub = sseBus.onEvent((event) => this.handleEvent(event));
|
||||
sseBus.connect(this.sessionId)
|
||||
this.unsub = sseBus.onEvent(event => this.handleEvent(event))
|
||||
}
|
||||
|
||||
/** 断开 SSE */
|
||||
disconnect(): void {
|
||||
if (this.unsub) {
|
||||
this.unsub();
|
||||
this.unsub = null;
|
||||
this.unsub()
|
||||
this.unsub = null
|
||||
}
|
||||
sseBus.disconnect();
|
||||
sseBus.disconnect()
|
||||
}
|
||||
|
||||
/** 处理 SSE 事件 */
|
||||
handleEvent(event: SessionEvent): void {
|
||||
const type = event.type;
|
||||
const payload = event.payload || ({} as EventPayload);
|
||||
const type = event.type
|
||||
const payload = event.payload || ({} as EventPayload)
|
||||
|
||||
// Skip bridge init noise
|
||||
const serialized = JSON.stringify(event);
|
||||
if (/Remote Control connecting/i.test(serialized)) return;
|
||||
const serialized = JSON.stringify(event)
|
||||
if (/Remote Control connecting/i.test(serialized)) return
|
||||
|
||||
switch (type) {
|
||||
// ---- 助手消息 ----
|
||||
case "assistant": {
|
||||
const content = typeof payload.content === "string" ? payload.content : "";
|
||||
this.setEntries((prev) => {
|
||||
const lastEntry = prev[prev.length - 1];
|
||||
case 'assistant': {
|
||||
const content =
|
||||
typeof payload.content === 'string' ? payload.content : ''
|
||||
this.setEntries(prev => {
|
||||
const lastEntry = prev[prev.length - 1]
|
||||
|
||||
// If last entry is AssistantMessage, append to it
|
||||
if (lastEntry?.type === "assistant_message") {
|
||||
const lastChunk = lastEntry.chunks[lastEntry.chunks.length - 1];
|
||||
if (lastChunk?.type === "message") {
|
||||
if (lastEntry?.type === 'assistant_message') {
|
||||
const lastChunk = lastEntry.chunks[lastEntry.chunks.length - 1]
|
||||
if (lastChunk?.type === 'message') {
|
||||
return [
|
||||
...prev.slice(0, -1),
|
||||
{ ...lastEntry, chunks: [...lastEntry.chunks.slice(0, -1), { type: "message", text: lastChunk.text + content }] },
|
||||
];
|
||||
{
|
||||
...lastEntry,
|
||||
chunks: [
|
||||
...lastEntry.chunks.slice(0, -1),
|
||||
{ type: 'message', text: lastChunk.text + content },
|
||||
],
|
||||
},
|
||||
]
|
||||
}
|
||||
return [
|
||||
...prev.slice(0, -1),
|
||||
{ ...lastEntry, chunks: [...lastEntry.chunks, { type: "message", text: content }] },
|
||||
];
|
||||
{
|
||||
...lastEntry,
|
||||
chunks: [
|
||||
...lastEntry.chunks,
|
||||
{ type: 'message', text: content },
|
||||
],
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
// Create new AssistantMessage
|
||||
if (content && content.trim()) {
|
||||
const newEntry: AssistantMessageEntry = {
|
||||
type: "assistant_message",
|
||||
type: 'assistant_message',
|
||||
id: `assistant-${Date.now()}`,
|
||||
chunks: [{ type: "message", text: content }],
|
||||
};
|
||||
return [...prev, newEntry];
|
||||
chunks: [{ type: 'message', text: content }],
|
||||
}
|
||||
return [...prev, newEntry]
|
||||
}
|
||||
return prev;
|
||||
});
|
||||
return prev
|
||||
})
|
||||
|
||||
// Check for embedded tool_use blocks
|
||||
const msg = payload.message as Record<string, unknown> | undefined;
|
||||
if (msg && typeof msg === "object" && Array.isArray(msg.content)) {
|
||||
const toolBlocks = (msg.content as Array<Record<string, unknown>>).filter((b) => b.type === "tool_use");
|
||||
const msg = payload.message as Record<string, unknown> | undefined
|
||||
if (msg && typeof msg === 'object' && Array.isArray(msg.content)) {
|
||||
const toolBlocks = (
|
||||
msg.content as Array<Record<string, unknown>>
|
||||
).filter(b => b.type === 'tool_use')
|
||||
for (const block of toolBlocks) {
|
||||
const toolCallId = (block.id as string) || `call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`;
|
||||
const toolCallId =
|
||||
(block.id as string) ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`
|
||||
const toolData: ToolCallData = {
|
||||
id: toolCallId,
|
||||
title: (block.name as string) || "tool",
|
||||
status: "running",
|
||||
title: (block.name as string) || 'tool',
|
||||
status: 'running',
|
||||
rawInput: (block.input as Record<string, unknown>) || {},
|
||||
};
|
||||
this.setEntries((prev) => [...prev, { type: "tool_call", toolCall: toolData }]);
|
||||
}
|
||||
this.setEntries(prev => [
|
||||
...prev,
|
||||
{ type: 'tool_call', toolCall: toolData },
|
||||
])
|
||||
}
|
||||
}
|
||||
break;
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 工具调用 ----
|
||||
case "tool_use": {
|
||||
const p = payload as Record<string, unknown>;
|
||||
const toolCallId = (p.tool_call_id as string) || `call-${Date.now()}`;
|
||||
case 'tool_use': {
|
||||
const p = payload as Record<string, unknown>
|
||||
const toolCallId = (p.tool_call_id as string) || `call-${Date.now()}`
|
||||
const toolData: ToolCallData = {
|
||||
id: toolCallId,
|
||||
title: (p.tool_name as string) || "tool",
|
||||
status: "running",
|
||||
title: (p.tool_name as string) || 'tool',
|
||||
status: 'running',
|
||||
rawInput: (p.tool_input as Record<string, unknown>) || {},
|
||||
};
|
||||
this.setEntries((prev) => [...prev, { type: "tool_call", toolCall: toolData }]);
|
||||
break;
|
||||
}
|
||||
this.setEntries(prev => [
|
||||
...prev,
|
||||
{ type: 'tool_call', toolCall: toolData },
|
||||
])
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 工具结果 ----
|
||||
case "tool_result": {
|
||||
const p = payload as Record<string, unknown>;
|
||||
const callId = (p.tool_call_id as string) || "";
|
||||
this.setEntries((prev) => {
|
||||
const idx = findToolCallIndex(prev, callId);
|
||||
if (idx < 0) return prev;
|
||||
const entry = prev[idx] as ToolCallEntry;
|
||||
case 'tool_result': {
|
||||
const p = payload as Record<string, unknown>
|
||||
const callId = (p.tool_call_id as string) || ''
|
||||
this.setEntries(prev => {
|
||||
const idx = findToolCallIndex(prev, callId)
|
||||
if (idx < 0) return prev
|
||||
const entry = prev[idx] as ToolCallEntry
|
||||
return prev.map((e, i) =>
|
||||
i === idx
|
||||
? { type: "tool_call", toolCall: { ...entry.toolCall, status: "complete" as ToolCallStatus, rawOutput: { output: p.content || p.output || "" } } }
|
||||
? {
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
...entry.toolCall,
|
||||
status: 'complete' as ToolCallStatus,
|
||||
rawOutput: { output: p.content || p.output || '' },
|
||||
},
|
||||
}
|
||||
: e,
|
||||
);
|
||||
});
|
||||
break;
|
||||
)
|
||||
})
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 权限请求 ----
|
||||
case "control_request":
|
||||
case "permission_request": {
|
||||
const req = payload.request as Record<string, unknown> | undefined;
|
||||
if (req && req.subtype === "can_use_tool") {
|
||||
const requestId = payload.request_id || "";
|
||||
const toolName = (req.tool_name as string) || "unknown";
|
||||
const toolInput = (req.input || req.tool_input || {}) as Record<string, unknown>;
|
||||
const description = (req.description as string) || "";
|
||||
case 'control_request':
|
||||
case 'permission_request': {
|
||||
const req = payload.request as Record<string, unknown> | undefined
|
||||
if (req && req.subtype === 'can_use_tool') {
|
||||
const requestId = payload.request_id || ''
|
||||
const toolName = (req.tool_name as string) || 'unknown'
|
||||
const toolInput = (req.input || req.tool_input || {}) as Record<
|
||||
string,
|
||||
unknown
|
||||
>
|
||||
const description = (req.description as string) || ''
|
||||
|
||||
// Update tool call status
|
||||
this.setEntries((prev) => {
|
||||
this.setEntries(prev => {
|
||||
// Find matching tool call
|
||||
const idx = [...prev].reverse().findIndex((e) => e.type === "tool_call");
|
||||
const idx = [...prev]
|
||||
.reverse()
|
||||
.findIndex(e => e.type === 'tool_call')
|
||||
if (idx >= 0) {
|
||||
const realIdx = prev.length - 1 - idx;
|
||||
const entry = prev[realIdx] as ToolCallEntry;
|
||||
if (entry.toolCall.status === "running") {
|
||||
const realIdx = prev.length - 1 - idx
|
||||
const entry = prev[realIdx] as ToolCallEntry
|
||||
if (entry.toolCall.status === 'running') {
|
||||
return prev.map((e, i) =>
|
||||
i === realIdx
|
||||
? { type: "tool_call", toolCall: { ...entry.toolCall, status: "waiting_for_confirmation" as ToolCallStatus, permissionRequest: { requestId, options: [] } } }
|
||||
? {
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
...entry.toolCall,
|
||||
status: 'waiting_for_confirmation' as ToolCallStatus,
|
||||
permissionRequest: { requestId, options: [] },
|
||||
},
|
||||
}
|
||||
: e,
|
||||
);
|
||||
)
|
||||
}
|
||||
}
|
||||
return prev;
|
||||
});
|
||||
return prev
|
||||
})
|
||||
|
||||
// Notify parent
|
||||
this.onPermissionRequest?.({
|
||||
@@ -372,102 +425,117 @@ export class RCSChatAdapter {
|
||||
toolName,
|
||||
toolInput,
|
||||
description,
|
||||
});
|
||||
})
|
||||
}
|
||||
break;
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 会话状态 ----
|
||||
case "session_status": {
|
||||
if (typeof payload.status === "string") {
|
||||
this.onStatusChange?.(payload.status);
|
||||
case 'session_status': {
|
||||
if (typeof payload.status === 'string') {
|
||||
this.onStatusChange?.(payload.status)
|
||||
}
|
||||
break;
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 错误 ----
|
||||
case "error": {
|
||||
const errorMsg = String(payload.message || payload.content || "Unknown error");
|
||||
this.onError?.(errorMsg);
|
||||
break;
|
||||
case 'error': {
|
||||
const errorMsg = String(
|
||||
payload.message || payload.content || 'Unknown error',
|
||||
)
|
||||
this.onError?.(errorMsg)
|
||||
break
|
||||
}
|
||||
|
||||
// ---- 忽略的事件类型 ----
|
||||
case "partial_assistant":
|
||||
case "result":
|
||||
case "result_success":
|
||||
case "control_response":
|
||||
case "permission_response":
|
||||
case "system":
|
||||
case "task_state":
|
||||
case "automation_state":
|
||||
case "status":
|
||||
break;
|
||||
case 'partial_assistant':
|
||||
case 'result':
|
||||
case 'result_success':
|
||||
case 'control_response':
|
||||
case 'permission_response':
|
||||
case 'system':
|
||||
case 'task_state':
|
||||
case 'automation_state':
|
||||
case 'status':
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
/** 发送用户消息 */
|
||||
async sendMessage(text: string, images?: UserMessageImage[]): Promise<void> {
|
||||
if (!text.trim() && (!images || images.length === 0)) return;
|
||||
if (!text.trim() && (!images || images.length === 0)) return
|
||||
|
||||
// Add user message to entries
|
||||
const userEntry: UserMessageEntry = {
|
||||
type: "user_message",
|
||||
type: 'user_message',
|
||||
id: `user-${Date.now()}`,
|
||||
content: text,
|
||||
images: images && images.length > 0 ? images : undefined,
|
||||
};
|
||||
this.setEntries((prev) => [...prev, userEntry]);
|
||||
}
|
||||
this.setEntries(prev => [...prev, userEntry])
|
||||
|
||||
// Send to backend
|
||||
await apiSendEvent(this.sessionId, {
|
||||
type: "user",
|
||||
type: 'user',
|
||||
uuid: generateMessageUuid(),
|
||||
content: text,
|
||||
message: { content: text },
|
||||
});
|
||||
})
|
||||
}
|
||||
|
||||
/** 响应权限请求 */
|
||||
async respondPermission(requestId: string, approved: boolean, extra?: Record<string, unknown>): Promise<void> {
|
||||
async respondPermission(
|
||||
requestId: string,
|
||||
approved: boolean,
|
||||
extra?: Record<string, unknown>,
|
||||
): Promise<void> {
|
||||
await apiSendControl(this.sessionId, {
|
||||
type: "permission_response",
|
||||
type: 'permission_response',
|
||||
approved,
|
||||
request_id: requestId,
|
||||
...extra,
|
||||
});
|
||||
})
|
||||
|
||||
// Update tool call status
|
||||
this.setEntries((prev) =>
|
||||
prev.map((entry) => {
|
||||
if (entry.type !== "tool_call") return entry;
|
||||
if (entry.toolCall.permissionRequest?.requestId !== requestId) return entry;
|
||||
this.setEntries(prev =>
|
||||
prev.map(entry => {
|
||||
if (entry.type !== 'tool_call') return entry
|
||||
if (entry.toolCall.permissionRequest?.requestId !== requestId)
|
||||
return entry
|
||||
return {
|
||||
type: "tool_call",
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
...entry.toolCall,
|
||||
status: approved ? "running" : ("rejected" as ToolCallStatus),
|
||||
status: approved ? 'running' : ('rejected' as ToolCallStatus),
|
||||
permissionRequest: undefined,
|
||||
},
|
||||
};
|
||||
}
|
||||
}),
|
||||
);
|
||||
)
|
||||
}
|
||||
|
||||
/** 中断当前操作 */
|
||||
async interrupt(): Promise<void> {
|
||||
// Mark running tools as canceled
|
||||
this.setEntries((prev) =>
|
||||
prev.map((entry) => {
|
||||
if (entry.type !== "tool_call") return entry;
|
||||
if (entry.toolCall.status !== "running" && entry.toolCall.status !== "waiting_for_confirmation") return entry;
|
||||
this.setEntries(prev =>
|
||||
prev.map(entry => {
|
||||
if (entry.type !== 'tool_call') return entry
|
||||
if (
|
||||
entry.toolCall.status !== 'running' &&
|
||||
entry.toolCall.status !== 'waiting_for_confirmation'
|
||||
)
|
||||
return entry
|
||||
return {
|
||||
type: "tool_call",
|
||||
toolCall: { ...entry.toolCall, status: "canceled" as ToolCallStatus, permissionRequest: undefined },
|
||||
};
|
||||
type: 'tool_call',
|
||||
toolCall: {
|
||||
...entry.toolCall,
|
||||
status: 'canceled' as ToolCallStatus,
|
||||
permissionRequest: undefined,
|
||||
},
|
||||
}
|
||||
}),
|
||||
);
|
||||
)
|
||||
|
||||
await apiInterrupt(this.sessionId);
|
||||
await apiInterrupt(this.sessionId)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,336 +1,363 @@
|
||||
import type { ChatTransport, UIMessage, UIMessageChunk } from "ai";
|
||||
import { getUuid } from "../api/client";
|
||||
import { generateMessageUuid } from "./utils";
|
||||
import type { SessionEvent, EventPayload } from "../types";
|
||||
import type { ChatTransport, UIMessage, UIMessageChunk } from 'ai'
|
||||
import { getUuid } from '../api/client'
|
||||
import { generateMessageUuid } from './utils'
|
||||
import type { SessionEvent, EventPayload } from '../types'
|
||||
|
||||
// ============================================================
|
||||
// SSE Event Bus — shared between SSE listener and transport
|
||||
// ============================================================
|
||||
|
||||
type SSEEventHandler = (event: SessionEvent) => void;
|
||||
type SSEEventHandler = (event: SessionEvent) => void
|
||||
|
||||
class SSEEventBus {
|
||||
private listeners: Set<SSEEventHandler> = new Set();
|
||||
private eventSource: EventSource | null = null;
|
||||
private _lastSeqNum = 0;
|
||||
private listeners: Set<SSEEventHandler> = new Set()
|
||||
private eventSource: EventSource | null = null
|
||||
private _lastSeqNum = 0
|
||||
|
||||
get lastSeqNum() {
|
||||
return this._lastSeqNum;
|
||||
return this._lastSeqNum
|
||||
}
|
||||
|
||||
/** Register a listener for SSE events */
|
||||
onEvent(handler: SSEEventHandler): () => void {
|
||||
this.listeners.add(handler);
|
||||
return () => this.listeners.delete(handler);
|
||||
this.listeners.add(handler)
|
||||
return () => this.listeners.delete(handler)
|
||||
}
|
||||
|
||||
/** Connect to the SSE stream for a session */
|
||||
connect(sessionId: string): void {
|
||||
this.disconnect();
|
||||
const uuid = getUuid();
|
||||
const url = `/web/sessions/${sessionId}/events?uuid=${encodeURIComponent(uuid)}`;
|
||||
const es = new EventSource(url);
|
||||
this.eventSource = es;
|
||||
this.disconnect()
|
||||
const uuid = getUuid()
|
||||
const url = `/web/sessions/${sessionId}/events?uuid=${encodeURIComponent(uuid)}`
|
||||
const es = new EventSource(url)
|
||||
this.eventSource = es
|
||||
|
||||
es.addEventListener("message", (e: MessageEvent) => {
|
||||
es.addEventListener('message', (e: MessageEvent) => {
|
||||
try {
|
||||
const data = JSON.parse(e.data) as SessionEvent;
|
||||
if (data.seqNum !== undefined && data.seqNum <= this._lastSeqNum) return;
|
||||
if (data.seqNum !== undefined) this._lastSeqNum = data.seqNum;
|
||||
const data = JSON.parse(e.data) as SessionEvent
|
||||
if (data.seqNum !== undefined && data.seqNum <= this._lastSeqNum) return
|
||||
if (data.seqNum !== undefined) this._lastSeqNum = data.seqNum
|
||||
for (const handler of this.listeners) {
|
||||
handler(data);
|
||||
handler(data)
|
||||
}
|
||||
} catch {
|
||||
// ignore parse errors
|
||||
}
|
||||
});
|
||||
})
|
||||
}
|
||||
|
||||
/** Disconnect the SSE stream */
|
||||
disconnect(): void {
|
||||
if (this.eventSource) {
|
||||
this.eventSource.close();
|
||||
this.eventSource = null;
|
||||
this.eventSource.close()
|
||||
this.eventSource = null
|
||||
}
|
||||
this._lastSeqNum = 0;
|
||||
this._lastSeqNum = 0
|
||||
}
|
||||
}
|
||||
|
||||
// Singleton event bus
|
||||
export const sseBus = new SSEEventBus();
|
||||
export const sseBus = new SSEEventBus()
|
||||
|
||||
// ============================================================
|
||||
// RCS ChatTransport — bridges RCS SSE to AI SDK UIMessageChunk
|
||||
// ============================================================
|
||||
|
||||
interface RCSTransportOptions {
|
||||
sessionId: string;
|
||||
onPermissionRequest?: (event: SessionEvent) => void;
|
||||
onSessionStatus?: (status: string) => void;
|
||||
onError?: (error: string) => void;
|
||||
sessionId: string
|
||||
onPermissionRequest?: (event: SessionEvent) => void
|
||||
onSessionStatus?: (status: string) => void
|
||||
onError?: (error: string) => void
|
||||
}
|
||||
|
||||
export class RCSTransport implements ChatTransport<UIMessage> {
|
||||
private sessionId: string;
|
||||
private onPermissionRequest?: (event: SessionEvent) => void;
|
||||
private onSessionStatus?: (status: string) => void;
|
||||
private onError?: (error: string) => void;
|
||||
private unsub: (() => void) | null = null;
|
||||
private sessionId: string
|
||||
private onPermissionRequest?: (event: SessionEvent) => void
|
||||
private onSessionStatus?: (status: string) => void
|
||||
private onError?: (error: string) => void
|
||||
private unsub: (() => void) | null = null
|
||||
|
||||
constructor(options: RCSTransportOptions) {
|
||||
this.sessionId = options.sessionId;
|
||||
this.onPermissionRequest = options.onPermissionRequest;
|
||||
this.onSessionStatus = options.onSessionStatus;
|
||||
this.onError = options.onError;
|
||||
this.sessionId = options.sessionId
|
||||
this.onPermissionRequest = options.onPermissionRequest
|
||||
this.onSessionStatus = options.onSessionStatus
|
||||
this.onError = options.onError
|
||||
}
|
||||
|
||||
async sendMessages({
|
||||
messages,
|
||||
abortSignal,
|
||||
}: Parameters<ChatTransport<UIMessage>["sendMessages"]>[0]): Promise<ReadableStream<UIMessageChunk>> {
|
||||
const lastMessage = messages[messages.length - 1];
|
||||
if (!lastMessage || lastMessage.role !== "user") {
|
||||
}: Parameters<ChatTransport<UIMessage>['sendMessages']>[0]): Promise<
|
||||
ReadableStream<UIMessageChunk>
|
||||
> {
|
||||
const lastMessage = messages[messages.length - 1]
|
||||
if (!lastMessage || lastMessage.role !== 'user') {
|
||||
// Return empty stream if no user message
|
||||
return new ReadableStream({ start: (c) => c.close() });
|
||||
return new ReadableStream({ start: c => c.close() })
|
||||
}
|
||||
|
||||
// Extract text from the user message parts
|
||||
const text = lastMessage.parts
|
||||
.filter((p: UIMessage["parts"][number]): p is Extract<typeof p, { type: "text" }> => p.type === "text")
|
||||
.filter(
|
||||
(
|
||||
p: UIMessage['parts'][number],
|
||||
): p is Extract<typeof p, { type: 'text' }> => p.type === 'text',
|
||||
)
|
||||
.map((p: { text: string }) => p.text)
|
||||
.join("");
|
||||
.join('')
|
||||
|
||||
if (!text.trim()) {
|
||||
return new ReadableStream({ start: (c) => c.close() });
|
||||
return new ReadableStream({ start: c => c.close() })
|
||||
}
|
||||
|
||||
// POST user message to the RCS backend
|
||||
const uuid = getUuid();
|
||||
const uuid = getUuid()
|
||||
const response = await fetch(
|
||||
`/web/sessions/${this.sessionId}/events?uuid=${encodeURIComponent(uuid)}`,
|
||||
{
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
type: "user",
|
||||
type: 'user',
|
||||
uuid: generateMessageUuid(),
|
||||
content: text,
|
||||
message: { content: text },
|
||||
}),
|
||||
signal: abortSignal,
|
||||
},
|
||||
);
|
||||
)
|
||||
|
||||
if (!response.ok) {
|
||||
const data = await response.json().catch(() => ({ error: { message: response.statusText } }));
|
||||
throw new Error(data.error?.message || "Failed to send message");
|
||||
const data = await response
|
||||
.json()
|
||||
.catch(() => ({ error: { message: response.statusText } }))
|
||||
throw new Error(data.error?.message || 'Failed to send message')
|
||||
}
|
||||
|
||||
// Create a ReadableStream from the SSE event bus
|
||||
// Collects events until the assistant turn is complete
|
||||
return new ReadableStream<UIMessageChunk>({
|
||||
start: (controller) => {
|
||||
let textId = `text-${Date.now()}`;
|
||||
let started = false;
|
||||
start: controller => {
|
||||
let textId = `text-${Date.now()}`
|
||||
let started = false
|
||||
|
||||
const ensureStarted = () => {
|
||||
if (!started) {
|
||||
started = true;
|
||||
controller.enqueue({ type: "start", messageId: `msg-${Date.now()}` });
|
||||
started = true
|
||||
controller.enqueue({
|
||||
type: 'start',
|
||||
messageId: `msg-${Date.now()}`,
|
||||
})
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
const handler = (event: SessionEvent) => {
|
||||
const type = event.type;
|
||||
const payload = event.payload || ({} as EventPayload);
|
||||
const type = event.type
|
||||
const payload = event.payload || ({} as EventPayload)
|
||||
|
||||
// Skip bridge init noise
|
||||
const serialized = JSON.stringify(event);
|
||||
if (/Remote Control connecting/i.test(serialized)) return;
|
||||
const serialized = JSON.stringify(event)
|
||||
if (/Remote Control connecting/i.test(serialized)) return
|
||||
|
||||
switch (type) {
|
||||
// ---- Assistant text ----
|
||||
case "assistant": {
|
||||
case 'assistant': {
|
||||
const content =
|
||||
typeof payload.content === "string"
|
||||
? payload.content
|
||||
: "";
|
||||
typeof payload.content === 'string' ? payload.content : ''
|
||||
if (content && content.trim()) {
|
||||
ensureStarted();
|
||||
controller.enqueue({ type: "text-start", id: textId });
|
||||
controller.enqueue({ type: "text-delta", id: textId, delta: content });
|
||||
controller.enqueue({ type: "text-end", id: textId });
|
||||
ensureStarted()
|
||||
controller.enqueue({ type: 'text-start', id: textId })
|
||||
controller.enqueue({
|
||||
type: 'text-delta',
|
||||
id: textId,
|
||||
delta: content,
|
||||
})
|
||||
controller.enqueue({ type: 'text-end', id: textId })
|
||||
}
|
||||
|
||||
// Check for embedded tool_use blocks
|
||||
const msg = payload.message as Record<string, unknown> | undefined;
|
||||
if (msg && typeof msg === "object" && Array.isArray(msg.content)) {
|
||||
const toolBlocks = (msg.content as Array<Record<string, unknown>>).filter(
|
||||
(b) => b.type === "tool_use",
|
||||
);
|
||||
const msg = payload.message as Record<string, unknown> | undefined
|
||||
if (
|
||||
msg &&
|
||||
typeof msg === 'object' &&
|
||||
Array.isArray(msg.content)
|
||||
) {
|
||||
const toolBlocks = (
|
||||
msg.content as Array<Record<string, unknown>>
|
||||
).filter(b => b.type === 'tool_use')
|
||||
for (const block of toolBlocks) {
|
||||
ensureStarted();
|
||||
const toolCallId = (block.id as string) || `call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`;
|
||||
ensureStarted()
|
||||
const toolCallId =
|
||||
(block.id as string) ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`
|
||||
controller.enqueue({
|
||||
type: "tool-input-available",
|
||||
type: 'tool-input-available',
|
||||
toolCallId,
|
||||
toolName: (block.name as string) || "tool",
|
||||
toolName: (block.name as string) || 'tool',
|
||||
input: block.input || {},
|
||||
});
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// Finish after assistant message
|
||||
ensureStarted();
|
||||
controller.enqueue({ type: "finish", finishReason: "stop" });
|
||||
controller.close();
|
||||
cleanup();
|
||||
break;
|
||||
ensureStarted()
|
||||
controller.enqueue({ type: 'finish', finishReason: 'stop' })
|
||||
controller.close()
|
||||
cleanup()
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Tool use events ----
|
||||
case "tool_use": {
|
||||
ensureStarted();
|
||||
case 'tool_use': {
|
||||
ensureStarted()
|
||||
const toolCallId =
|
||||
(payload as Record<string, unknown>).tool_call_id as string ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`;
|
||||
((payload as Record<string, unknown>).tool_call_id as string) ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`
|
||||
controller.enqueue({
|
||||
type: "tool-input-available",
|
||||
type: 'tool-input-available',
|
||||
toolCallId,
|
||||
toolName: (payload as Record<string, unknown>).tool_name as string || "tool",
|
||||
toolName:
|
||||
((payload as Record<string, unknown>).tool_name as string) ||
|
||||
'tool',
|
||||
input: (payload as Record<string, unknown>).tool_input || {},
|
||||
});
|
||||
break;
|
||||
})
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Tool result events ----
|
||||
case "tool_result": {
|
||||
ensureStarted();
|
||||
case 'tool_result': {
|
||||
ensureStarted()
|
||||
const resultCallId =
|
||||
(payload as Record<string, unknown>).tool_call_id as string ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`;
|
||||
((payload as Record<string, unknown>).tool_call_id as string) ||
|
||||
`call-${Date.now()}-${Math.random().toString(16).slice(2, 8)}`
|
||||
const output =
|
||||
typeof (payload as Record<string, unknown>).output === "string"
|
||||
typeof (payload as Record<string, unknown>).output === 'string'
|
||||
? (payload as Record<string, unknown>).output
|
||||
: (payload as Record<string, unknown>).content || "";
|
||||
: (payload as Record<string, unknown>).content || ''
|
||||
controller.enqueue({
|
||||
type: "tool-output-available",
|
||||
type: 'tool-output-available',
|
||||
toolCallId: resultCallId,
|
||||
output: output as string,
|
||||
});
|
||||
break;
|
||||
})
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Permission / control requests ----
|
||||
case "control_request":
|
||||
case "permission_request": {
|
||||
const req = payload.request as Record<string, unknown> | undefined;
|
||||
if (req && req.subtype === "can_use_tool") {
|
||||
case 'control_request':
|
||||
case 'permission_request': {
|
||||
const req = payload.request as Record<string, unknown> | undefined
|
||||
if (req && req.subtype === 'can_use_tool') {
|
||||
// Forward to the UI layer for handling
|
||||
this.onPermissionRequest?.(event);
|
||||
this.onPermissionRequest?.(event)
|
||||
}
|
||||
// Don't close the stream — wait for the response
|
||||
break;
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Status events ----
|
||||
case "status": {
|
||||
case 'status': {
|
||||
const msg =
|
||||
(typeof payload.message === "string" ? payload.message : "") ||
|
||||
(typeof payload.message === 'string' ? payload.message : '') ||
|
||||
payload.content ||
|
||||
"";
|
||||
if (/connecting|waiting|initializing|Remote Control/i.test(msg)) return;
|
||||
break;
|
||||
''
|
||||
if (/connecting|waiting|initializing|Remote Control/i.test(msg))
|
||||
return
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Session status ----
|
||||
case "session_status": {
|
||||
if (typeof payload.status === "string") {
|
||||
this.onSessionStatus?.(payload.status);
|
||||
case 'session_status': {
|
||||
if (typeof payload.status === 'string') {
|
||||
this.onSessionStatus?.(payload.status)
|
||||
if (
|
||||
payload.status === "archived" ||
|
||||
payload.status === "inactive"
|
||||
payload.status === 'archived' ||
|
||||
payload.status === 'inactive'
|
||||
) {
|
||||
ensureStarted();
|
||||
controller.enqueue({ type: "finish", finishReason: "stop" });
|
||||
controller.close();
|
||||
cleanup();
|
||||
ensureStarted()
|
||||
controller.enqueue({ type: 'finish', finishReason: 'stop' })
|
||||
controller.close()
|
||||
cleanup()
|
||||
}
|
||||
}
|
||||
break;
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Errors ----
|
||||
case "error": {
|
||||
ensureStarted();
|
||||
case 'error': {
|
||||
ensureStarted()
|
||||
controller.enqueue({
|
||||
type: "error",
|
||||
errorText: String(payload.message || payload.content || "Unknown error"),
|
||||
});
|
||||
controller.enqueue({ type: "finish", finishReason: "error" });
|
||||
controller.close();
|
||||
cleanup();
|
||||
break;
|
||||
type: 'error',
|
||||
errorText: String(
|
||||
payload.message || payload.content || 'Unknown error',
|
||||
),
|
||||
})
|
||||
controller.enqueue({ type: 'finish', finishReason: 'error' })
|
||||
controller.close()
|
||||
cleanup()
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Interrupt ----
|
||||
case "interrupt": {
|
||||
ensureStarted();
|
||||
controller.enqueue({ type: "abort", reason: "Session interrupted" });
|
||||
controller.close();
|
||||
cleanup();
|
||||
break;
|
||||
case 'interrupt': {
|
||||
ensureStarted()
|
||||
controller.enqueue({
|
||||
type: 'abort',
|
||||
reason: 'Session interrupted',
|
||||
})
|
||||
controller.close()
|
||||
cleanup()
|
||||
break
|
||||
}
|
||||
|
||||
// ---- Skip noise ----
|
||||
case "partial_assistant":
|
||||
case "result":
|
||||
case "result_success":
|
||||
case "control_response":
|
||||
case "permission_response":
|
||||
case "system":
|
||||
case "task_state":
|
||||
case "automation_state":
|
||||
return;
|
||||
case 'partial_assistant':
|
||||
case 'result':
|
||||
case 'result_success':
|
||||
case 'control_response':
|
||||
case 'permission_response':
|
||||
case 'system':
|
||||
case 'task_state':
|
||||
case 'automation_state':
|
||||
return
|
||||
|
||||
default:
|
||||
return;
|
||||
return
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
const cleanup = () => {
|
||||
if (this.unsub) {
|
||||
this.unsub();
|
||||
this.unsub = null;
|
||||
this.unsub()
|
||||
this.unsub = null
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
this.unsub = sseBus.onEvent(handler);
|
||||
this.unsub = sseBus.onEvent(handler)
|
||||
|
||||
// Handle abort
|
||||
if (abortSignal) {
|
||||
const onAbort = () => {
|
||||
controller.enqueue({ type: "abort", reason: "Aborted" });
|
||||
controller.close();
|
||||
cleanup();
|
||||
abortSignal.removeEventListener("abort", onAbort);
|
||||
};
|
||||
abortSignal.addEventListener("abort", onAbort);
|
||||
controller.enqueue({ type: 'abort', reason: 'Aborted' })
|
||||
controller.close()
|
||||
cleanup()
|
||||
abortSignal.removeEventListener('abort', onAbort)
|
||||
}
|
||||
abortSignal.addEventListener('abort', onAbort)
|
||||
}
|
||||
},
|
||||
});
|
||||
})
|
||||
}
|
||||
|
||||
/** Not supported — RCS doesn't have stream resumption */
|
||||
reconnectToStream(): Promise<ReadableStream<UIMessageChunk> | null> {
|
||||
return Promise.resolve(null);
|
||||
return Promise.resolve(null)
|
||||
}
|
||||
|
||||
/** Clean up listeners */
|
||||
destroy(): void {
|
||||
if (this.unsub) {
|
||||
this.unsub();
|
||||
this.unsub = null;
|
||||
this.unsub()
|
||||
this.unsub = null
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,97 +1,110 @@
|
||||
import React, { createContext, useContext, useEffect, useState, useCallback } from "react";
|
||||
import React, {
|
||||
createContext,
|
||||
useContext,
|
||||
useEffect,
|
||||
useState,
|
||||
useCallback,
|
||||
} from 'react'
|
||||
|
||||
export type Theme = "light" | "dark" | "system";
|
||||
export type Theme = 'light' | 'dark' | 'system'
|
||||
|
||||
interface ThemeContextValue {
|
||||
theme: Theme;
|
||||
resolvedTheme: "light" | "dark";
|
||||
setTheme: (theme: Theme) => void;
|
||||
theme: Theme
|
||||
resolvedTheme: 'light' | 'dark'
|
||||
setTheme: (theme: Theme) => void
|
||||
}
|
||||
|
||||
const ThemeContext = createContext<ThemeContextValue | undefined>(undefined);
|
||||
const ThemeContext = createContext<ThemeContextValue | undefined>(undefined)
|
||||
|
||||
const STORAGE_KEY = "theme";
|
||||
const STORAGE_KEY = 'theme'
|
||||
|
||||
function getSystemTheme(): "light" | "dark" {
|
||||
if (typeof window === "undefined") return "light";
|
||||
return window.matchMedia("(prefers-color-scheme: dark)").matches ? "dark" : "light";
|
||||
function getSystemTheme(): 'light' | 'dark' {
|
||||
if (typeof window === 'undefined') return 'light'
|
||||
return window.matchMedia('(prefers-color-scheme: dark)').matches
|
||||
? 'dark'
|
||||
: 'light'
|
||||
}
|
||||
|
||||
function getStoredTheme(): Theme {
|
||||
if (typeof window === "undefined") return "system";
|
||||
if (typeof window === 'undefined') return 'system'
|
||||
try {
|
||||
const stored = localStorage.getItem(STORAGE_KEY);
|
||||
if (stored === "light" || stored === "dark" || stored === "system") {
|
||||
return stored;
|
||||
const stored = localStorage.getItem(STORAGE_KEY)
|
||||
if (stored === 'light' || stored === 'dark' || stored === 'system') {
|
||||
return stored
|
||||
}
|
||||
} catch {
|
||||
// localStorage not available
|
||||
}
|
||||
return "system";
|
||||
return 'system'
|
||||
}
|
||||
|
||||
function applyTheme(theme: "light" | "dark") {
|
||||
const root = document.documentElement;
|
||||
root.classList.remove("light", "dark");
|
||||
root.classList.add(theme);
|
||||
function applyTheme(theme: 'light' | 'dark') {
|
||||
const root = document.documentElement
|
||||
root.classList.remove('light', 'dark')
|
||||
root.classList.add(theme)
|
||||
}
|
||||
|
||||
export function useTheme() {
|
||||
const context = useContext(ThemeContext);
|
||||
const context = useContext(ThemeContext)
|
||||
if (!context) {
|
||||
throw new Error("useTheme must be used within a ThemeProvider");
|
||||
throw new Error('useTheme must be used within a ThemeProvider')
|
||||
}
|
||||
return context;
|
||||
return context
|
||||
}
|
||||
|
||||
interface ThemeProviderProps {
|
||||
children: React.ReactNode;
|
||||
defaultTheme?: Theme;
|
||||
children: React.ReactNode
|
||||
defaultTheme?: Theme
|
||||
}
|
||||
|
||||
export function ThemeProvider({ children, defaultTheme = "system" }: ThemeProviderProps) {
|
||||
const [theme, setThemeState] = useState<Theme>(() => getStoredTheme() || defaultTheme);
|
||||
const [resolvedTheme, setResolvedTheme] = useState<"light" | "dark">(() => {
|
||||
const stored = getStoredTheme() || defaultTheme;
|
||||
return stored === "system" ? getSystemTheme() : stored;
|
||||
});
|
||||
export function ThemeProvider({
|
||||
children,
|
||||
defaultTheme = 'system',
|
||||
}: ThemeProviderProps) {
|
||||
const [theme, setThemeState] = useState<Theme>(
|
||||
() => getStoredTheme() || defaultTheme,
|
||||
)
|
||||
const [resolvedTheme, setResolvedTheme] = useState<'light' | 'dark'>(() => {
|
||||
const stored = getStoredTheme() || defaultTheme
|
||||
return stored === 'system' ? getSystemTheme() : stored
|
||||
})
|
||||
|
||||
const setTheme = useCallback((newTheme: Theme) => {
|
||||
setThemeState(newTheme);
|
||||
setThemeState(newTheme)
|
||||
try {
|
||||
localStorage.setItem(STORAGE_KEY, newTheme);
|
||||
localStorage.setItem(STORAGE_KEY, newTheme)
|
||||
} catch {
|
||||
// localStorage not available
|
||||
}
|
||||
}, []);
|
||||
}, [])
|
||||
|
||||
// Apply theme on mount and when theme changes
|
||||
useEffect(() => {
|
||||
const resolved = theme === "system" ? getSystemTheme() : theme;
|
||||
setResolvedTheme(resolved);
|
||||
applyTheme(resolved);
|
||||
}, [theme]);
|
||||
const resolved = theme === 'system' ? getSystemTheme() : theme
|
||||
setResolvedTheme(resolved)
|
||||
applyTheme(resolved)
|
||||
}, [theme])
|
||||
|
||||
// Listen for system theme changes
|
||||
useEffect(() => {
|
||||
if (theme !== "system") return;
|
||||
if (theme !== 'system') return
|
||||
|
||||
const mediaQuery = window.matchMedia("(prefers-color-scheme: dark)");
|
||||
const mediaQuery = window.matchMedia('(prefers-color-scheme: dark)')
|
||||
const handleChange = (e: MediaQueryListEvent) => {
|
||||
const newTheme = e.matches ? "dark" : "light";
|
||||
setResolvedTheme(newTheme);
|
||||
applyTheme(newTheme);
|
||||
};
|
||||
const newTheme = e.matches ? 'dark' : 'light'
|
||||
setResolvedTheme(newTheme)
|
||||
applyTheme(newTheme)
|
||||
}
|
||||
|
||||
mediaQuery.addEventListener("change", handleChange);
|
||||
return () => mediaQuery.removeEventListener("change", handleChange);
|
||||
}, [theme]);
|
||||
mediaQuery.addEventListener('change', handleChange)
|
||||
return () => mediaQuery.removeEventListener('change', handleChange)
|
||||
}, [theme])
|
||||
|
||||
const value: ThemeContextValue = {
|
||||
theme,
|
||||
resolvedTheme,
|
||||
setTheme,
|
||||
};
|
||||
}
|
||||
|
||||
return React.createElement(ThemeContext.Provider, { value }, children);
|
||||
return React.createElement(ThemeContext.Provider, { value }, children)
|
||||
}
|
||||
|
||||
@@ -2,71 +2,71 @@
|
||||
// Unified Chat Data Model — shared between ACP and RCS chat interfaces
|
||||
// =============================================================================
|
||||
|
||||
import type { ToolCallContent, PermissionOption, PlanEntry } from "../acp/types";
|
||||
import type { ToolCallContent, PermissionOption, PlanEntry } from '../acp/types'
|
||||
|
||||
// 工具调用状态
|
||||
export type ToolCallStatus =
|
||||
| "running"
|
||||
| "complete"
|
||||
| "error"
|
||||
| "waiting_for_confirmation"
|
||||
| "rejected"
|
||||
| "canceled";
|
||||
| 'running'
|
||||
| 'complete'
|
||||
| 'error'
|
||||
| 'waiting_for_confirmation'
|
||||
| 'rejected'
|
||||
| 'canceled'
|
||||
|
||||
// 工具调用数据
|
||||
export interface ToolCallData {
|
||||
id: string;
|
||||
title: string;
|
||||
status: ToolCallStatus;
|
||||
content?: ToolCallContent[];
|
||||
rawInput?: Record<string, unknown>;
|
||||
rawOutput?: Record<string, unknown>;
|
||||
id: string
|
||||
title: string
|
||||
status: ToolCallStatus
|
||||
content?: ToolCallContent[]
|
||||
rawInput?: Record<string, unknown>
|
||||
rawOutput?: Record<string, unknown>
|
||||
// 权限请求（仅当 status === "waiting_for_confirmation"）
|
||||
permissionRequest?: {
|
||||
requestId: string;
|
||||
options: PermissionOption[];
|
||||
};
|
||||
requestId: string
|
||||
options: PermissionOption[]
|
||||
}
|
||||
// 独立权限请求（无匹配工具调用时创建）
|
||||
isStandalonePermission?: boolean;
|
||||
isStandalonePermission?: boolean
|
||||
}
|
||||
|
||||
// 助手消息块 — 普通消息或思考过程
|
||||
export type AssistantChunk =
|
||||
| { type: "message"; text: string }
|
||||
| { type: "thought"; text: string };
|
||||
| { type: 'message'; text: string }
|
||||
| { type: 'thought'; text: string }
|
||||
|
||||
// 用户消息中的图片
|
||||
export interface UserMessageImage {
|
||||
mimeType: string;
|
||||
data: string; // base64 encoded
|
||||
mimeType: string
|
||||
data: string // base64 encoded
|
||||
}
|
||||
|
||||
// 用户消息条目
|
||||
export interface UserMessageEntry {
|
||||
type: "user_message";
|
||||
id: string;
|
||||
content: string;
|
||||
images?: UserMessageImage[];
|
||||
type: 'user_message'
|
||||
id: string
|
||||
content: string
|
||||
images?: UserMessageImage[]
|
||||
}
|
||||
|
||||
// 助手消息条目
|
||||
export interface AssistantMessageEntry {
|
||||
type: "assistant_message";
|
||||
id: string;
|
||||
chunks: AssistantChunk[];
|
||||
type: 'assistant_message'
|
||||
id: string
|
||||
chunks: AssistantChunk[]
|
||||
}
|
||||
|
||||
// 工具调用条目
|
||||
export interface ToolCallEntry {
|
||||
type: "tool_call";
|
||||
toolCall: ToolCallData;
|
||||
type: 'tool_call'
|
||||
toolCall: ToolCallData
|
||||
}
|
||||
|
||||
// Plan 展示条目（Agent 执行计划）
|
||||
export interface PlanDisplayEntry {
|
||||
type: "plan";
|
||||
id: string;
|
||||
entries: PlanEntry[];
|
||||
type: 'plan'
|
||||
id: string
|
||||
entries: PlanEntry[]
|
||||
}
|
||||
|
||||
// 统一聊天条目类型
|
||||
@@ -74,7 +74,7 @@ export type ThreadEntry =
|
||||
| UserMessageEntry
|
||||
| AssistantMessageEntry
|
||||
| ToolCallEntry
|
||||
| PlanDisplayEntry;
|
||||
| PlanDisplayEntry
|
||||
|
||||
// =============================================================================
|
||||
// Chat 组件 Props 类型
|
||||
@@ -82,23 +82,23 @@ export type ThreadEntry =
|
||||
|
||||
// ChatInput 提交消息
|
||||
export interface ChatInputMessage {
|
||||
text: string;
|
||||
images?: UserMessageImage[];
|
||||
text: string
|
||||
images?: UserMessageImage[]
|
||||
}
|
||||
|
||||
// 权限请求条目（用于 PermissionPanel）
|
||||
export interface PendingPermission {
|
||||
requestId: string;
|
||||
toolName: string;
|
||||
toolInput: Record<string, unknown>;
|
||||
description?: string;
|
||||
options?: PermissionOption[];
|
||||
requestId: string
|
||||
toolName: string
|
||||
toolInput: Record<string, unknown>
|
||||
description?: string
|
||||
options?: PermissionOption[]
|
||||
}
|
||||
|
||||
// 会话列表条目（用于 SessionSidebar）
|
||||
export interface SessionListItem {
|
||||
id: string;
|
||||
title?: string | null;
|
||||
updatedAt?: string | null;
|
||||
isActive?: boolean;
|
||||
id: string
|
||||
title?: string | null
|
||||
updatedAt?: string | null
|
||||
isActive?: boolean
|
||||
}
|
||||
|
||||
@@ -1,91 +1,102 @@
|
||||
import { type ClassValue, clsx } from "clsx";
|
||||
import { twMerge } from "tailwind-merge";
|
||||
import { type ClassValue, clsx } from 'clsx'
|
||||
import { twMerge } from 'tailwind-merge'
|
||||
|
||||
export function cn(...inputs: ClassValue[]) {
|
||||
return twMerge(clsx(inputs));
|
||||
return twMerge(clsx(inputs))
|
||||
}
|
||||
|
||||
export function esc(str: string | null | undefined): string {
|
||||
if (!str) return "";
|
||||
const value = String(str);
|
||||
const div = document.createElement("div");
|
||||
div.textContent = value;
|
||||
return div.innerHTML;
|
||||
if (!str) return ''
|
||||
const value = String(str)
|
||||
const div = document.createElement('div')
|
||||
div.textContent = value
|
||||
return div.innerHTML
|
||||
}
|
||||
|
||||
export function formatTime(ts: number | null | undefined): string {
|
||||
if (!ts) return "";
|
||||
return new Date(ts * 1000).toLocaleString();
|
||||
if (!ts) return ''
|
||||
return new Date(ts * 1000).toLocaleString()
|
||||
}
|
||||
|
||||
export function statusClass(status: string | null | undefined): string {
|
||||
const map: Record<string, string> = {
|
||||
active: "active",
|
||||
running: "running",
|
||||
idle: "idle",
|
||||
inactive: "inactive",
|
||||
requires_action: "requires_action",
|
||||
archived: "archived",
|
||||
error: "error",
|
||||
};
|
||||
return map[status || ""] || "default";
|
||||
active: 'active',
|
||||
running: 'running',
|
||||
idle: 'idle',
|
||||
inactive: 'inactive',
|
||||
requires_action: 'requires_action',
|
||||
archived: 'archived',
|
||||
error: 'error',
|
||||
}
|
||||
return map[status || ''] || 'default'
|
||||
}
|
||||
|
||||
export function isClosedSessionStatus(status: string | null | undefined): boolean {
|
||||
return status === "archived" || status === "inactive";
|
||||
export function isClosedSessionStatus(
|
||||
status: string | null | undefined,
|
||||
): boolean {
|
||||
return status === 'archived' || status === 'inactive'
|
||||
}
|
||||
|
||||
export function truncate(str: string | null | undefined, max: number): string {
|
||||
if (!str) return "";
|
||||
const s = String(str);
|
||||
return s.length > max ? s.slice(0, max) + "..." : s;
|
||||
if (!str) return ''
|
||||
const s = String(str)
|
||||
return s.length > max ? s.slice(0, max) + '...' : s
|
||||
}
|
||||
|
||||
function formatUuidV4(bytes: Uint8Array): string {
|
||||
bytes[6] = (bytes[6] & 0x0f) | 0x40;
|
||||
bytes[8] = (bytes[8] & 0x3f) | 0x80;
|
||||
bytes[6] = (bytes[6] & 0x0f) | 0x40
|
||||
bytes[8] = (bytes[8] & 0x3f) | 0x80
|
||||
|
||||
const hex = Array.from(bytes, (byte) => byte.toString(16).padStart(2, "0"));
|
||||
const hex = Array.from(bytes, byte => byte.toString(16).padStart(2, '0'))
|
||||
return [
|
||||
hex.slice(0, 4).join(""),
|
||||
hex.slice(4, 6).join(""),
|
||||
hex.slice(6, 8).join(""),
|
||||
hex.slice(8, 10).join(""),
|
||||
hex.slice(10, 16).join(""),
|
||||
].join("-");
|
||||
hex.slice(0, 4).join(''),
|
||||
hex.slice(4, 6).join(''),
|
||||
hex.slice(6, 8).join(''),
|
||||
hex.slice(8, 10).join(''),
|
||||
hex.slice(10, 16).join(''),
|
||||
].join('-')
|
||||
}
|
||||
|
||||
export function generateMessageUuid(): string {
|
||||
const cryptoApi = globalThis.crypto;
|
||||
if (cryptoApi && typeof cryptoApi.randomUUID === "function") {
|
||||
return cryptoApi.randomUUID();
|
||||
const cryptoApi = globalThis.crypto
|
||||
if (cryptoApi && typeof cryptoApi.randomUUID === 'function') {
|
||||
return cryptoApi.randomUUID()
|
||||
}
|
||||
if (!cryptoApi || typeof cryptoApi.getRandomValues !== "function") {
|
||||
throw new Error("crypto.getRandomValues is required to generate message UUIDs");
|
||||
if (!cryptoApi || typeof cryptoApi.getRandomValues !== 'function') {
|
||||
throw new Error(
|
||||
'crypto.getRandomValues is required to generate message UUIDs',
|
||||
)
|
||||
}
|
||||
const bytes = new Uint8Array(16);
|
||||
cryptoApi.getRandomValues(bytes);
|
||||
return formatUuidV4(bytes);
|
||||
const bytes = new Uint8Array(16)
|
||||
cryptoApi.getRandomValues(bytes)
|
||||
return formatUuidV4(bytes)
|
||||
}
|
||||
|
||||
export function extractEventText(payload: Record<string, unknown> | null | undefined): string {
|
||||
if (!payload || typeof payload !== "object") return "";
|
||||
if (typeof payload.content === "string") return payload.content;
|
||||
const msg = payload.message as Record<string, unknown> | undefined;
|
||||
if (msg && typeof msg === "object" && Array.isArray(msg.content)) {
|
||||
export function extractEventText(
|
||||
payload: Record<string, unknown> | null | undefined,
|
||||
): string {
|
||||
if (!payload || typeof payload !== 'object') return ''
|
||||
if (typeof payload.content === 'string') return payload.content
|
||||
const msg = payload.message as Record<string, unknown> | undefined
|
||||
if (msg && typeof msg === 'object' && Array.isArray(msg.content)) {
|
||||
const texts = msg.content
|
||||
.filter((b: Record<string, unknown>) => b && b.type === "text" && typeof b.text === "string")
|
||||
.map((b: Record<string, unknown>) => b.text as string);
|
||||
if (texts.length > 0) return texts.join("\n");
|
||||
.filter(
|
||||
(b: Record<string, unknown>) =>
|
||||
b && b.type === 'text' && typeof b.text === 'string',
|
||||
)
|
||||
.map((b: Record<string, unknown>) => b.text as string)
|
||||
if (texts.length > 0) return texts.join('\n')
|
||||
}
|
||||
return "";
|
||||
return ''
|
||||
}
|
||||
|
||||
export function isConversationClearedStatus(
|
||||
payload: Record<string, unknown> | null | undefined,
|
||||
): boolean {
|
||||
if (!payload || typeof payload !== "object") return false;
|
||||
if (payload.status === "conversation_cleared") return true;
|
||||
const raw = payload.raw as Record<string, unknown> | undefined;
|
||||
return !!raw && typeof raw === "object" && raw.status === "conversation_cleared";
|
||||
if (!payload || typeof payload !== 'object') return false
|
||||
if (payload.status === 'conversation_cleared') return true
|
||||
const raw = payload.raw as Record<string, unknown> | undefined
|
||||
return (
|
||||
!!raw && typeof raw === 'object' && raw.status === 'conversation_cleared'
|
||||
)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user