cx-chat 0.0.2 → 0.0.4

package/src/utils/chatUtils.ts

@@ -1,524 +0,0 @@
-import type { Message, Thread } from '@/types';
-
-/**
- * 聊天工具：后端 Result / VO → 前端 Thread、Message 等模型（纯函数）。
- */
-
-/** 解开 Result 包裹：仅当业务 code === 200 时取 data */
-export function unwrapChatApiBody(raw: unknown): unknown {
-  if (raw && typeof raw === 'object' && 'code' in raw) {
-    const o = raw as { code?: number; data?: unknown }
-    if (o.code === 200 && o.data !== undefined) return o.data
-  }
-  return raw
-}
-
-export function parseNewSessionId(raw: unknown): string {
-  const body = unwrapChatApiBody(raw)
-  if (typeof body === 'string' && body.trim()) return body.trim()
-  if (body && typeof body === 'object') {
-    const o = body as Record<string, unknown>
-    const id = o.sessionId ?? o.session_id ?? o.id
-    if (typeof id === 'string' && id) return id
-    if (id != null && id !== '') return String(id)
-  }
-  throw new Error('Invalid session-id response')
-}
-
-/** 从路由 `/chat/:sessionId` 解析会话 id（与 `encodeURIComponent` 后的路径一致）；非会话页返回 null */
-export function parseChatSessionIdFromPathname(pathname: string): string | null {
-  const m = pathname.match(/^\/chat\/([^/]+)\/?$/)
-  if (!m?.[1]) return null
-  try {
-    return decodeURIComponent(m[1])
-  } catch {
-    return m[1]
-  }
-}
-
-function coerceChatPromptItemsToStrings(items: unknown[]): string[] {
-  const out: string[] = []
-  for (const x of items) {
-    if (typeof x === 'string') {
-      const s = x.trim()
-      if (s) out.push(s)
-      continue
-    }
-    if (x && typeof x === 'object') {
-      const o = x as Record<string, unknown>
-      const t =
-        o.text ?? o.title ?? o.label ?? o.prompt ?? o.content ?? o.name
-      if (typeof t === 'string' && t.trim()) out.push(t.trim())
-    }
-  }
-  return out
-}
-
-/**
- * 解析「会话搜索建议」接口体：`data` 常为 JSON 字符串，解析失败或非数组则返回 []。
- * 支持数组、或对象内 items / suggestions / list / prompts / options 等数组字段。
- */
-export function parseChatSessionPromptSuggestions(raw: unknown): string[] {
-  let node: unknown = unwrapChatApiBody(raw)
-
-  if (node && typeof node === 'object') {
-    const o = node as Record<string, unknown>
-    const rawData = o.data
-    if (typeof rawData === 'string') {
-      const s = rawData.trim()
-      if (!s) return []
-      try {
-        node = JSON.parse(s) as unknown
-      } catch {
-        return []
-      }
-    } else if (rawData !== undefined && rawData !== null) {
-      node = rawData
-    }
-  }
-
-  if (node == null) return []
-  if (Array.isArray(node)) {
-    return coerceChatPromptItemsToStrings(node)
-  }
-  if (typeof node === 'string') {
-    const s = node.trim()
-    if (!s) return []
-    try {
-      const parsed = JSON.parse(s) as unknown
-      return parseChatSessionPromptSuggestions(parsed)
-    } catch {
-      return [s]
-    }
-  }
-  if (typeof node === 'object') {
-    const o = node as Record<string, unknown>
-    const keys = [
-      'items',
-      'suggestions',
-      'list',
-      'prompts',
-      'options',
-    ] as const
-    for (const k of keys) {
-      const v = o[k]
-      if (Array.isArray(v)) return coerceChatPromptItemsToStrings(v)
-    }
-    if (typeof o.prompt === 'string' && o.prompt.trim()) {
-      return [o.prompt.trim()]
-    }
-  }
-  return []
-}
-
-/** 从后往前第一条非 system 消息 */
-export function findLastNonSystemMessage(messages: Message[]): Message | null {
-  for (let i = messages.length - 1; i >= 0; i -= 1) {
-    const m = messages[i]
-    if (m.role !== 'system') return m
-  }
-  return null
-}
-
-/** 从后往前第一条助手消息 */
-export function findLastAssistantMessage(messages: Message[]): Message | null {
-  for (let i = messages.length - 1; i >= 0; i -= 1) {
-    const m = messages[i]
-    if (m.role === 'assistant') return m
-  }
-  return null
-}
-
-/**
- * 把某次「搜索建议」接口结果绑到指定助手 messageId 的 options（与 suggested_replies 同槽展示）；
- * 仅当该条仍是当前列表中的最后一条助手、且线程末尾为该助手时写入，避免挂在新提问/流式占位上。
- * 不与已有 suggested_replies 叠加：该条已有非空 options 时保持原样，避免两类建议合并成一长串。
- */
-export function applySessionPromptBinding(
-  messages: Message[],
-  binding: { messageId: string; items: string[] } | null,
-): Message[] {
-  if (!binding?.items.length) return messages
-  const tail = findLastNonSystemMessage(messages)
-  if (tail?.role !== 'assistant' || tail.id !== binding.messageId) {
-    return messages
-  }
-  const lastAsst = findLastAssistantMessage(messages)
-  if (!lastAsst || lastAsst.id !== binding.messageId) return messages
-  return messages.map((m) => {
-    if (m.id !== binding.messageId) return m
-    const existing = m.options ?? []
-    if (existing.length > 0) return m
-    return { ...m, options: [...binding.items] }
-  })
-}
-
-/** 流式新会话：接口列表尚未含该会话时，将乐观占位行插回列表（与 ChatContext.streamingPlaceholderThread 配合） */
-export function mergeStreamingPlaceholderThread(
-  threads: Thread[],
-  opts: {
-    placeholder: Thread | null
-    activeThreadId: string | null
-    isStreaming: boolean
-  },
-): Thread[] {
-  const { placeholder, activeThreadId, isStreaming } = opts
-  const base = [...threads]
-  if (
-    placeholder &&
-    activeThreadId &&
-    placeholder.id === activeThreadId &&
-    isStreaming &&
-    !base.some((x) => x.id === placeholder.id)
-  ) {
-    return [placeholder, ...base]
-  }
-  return base
-}
-
-/** ChatSessionVO → Thread */
-export function normalizeSessionsToThreads(raw: unknown): Thread[] {
-  const body = unwrapChatApiBody(raw)
-  let list: unknown[] = []
-  if (Array.isArray(body)) list = body
-  else if (body && typeof body === 'object') {
-    const o = body as Record<string, unknown>
-    if (Array.isArray(o.list)) list = o.list
-    else if (Array.isArray(o.items)) list = o.items
-    else if (Array.isArray(o.sessions)) list = o.sessions
-    else if (Array.isArray(o.data)) list = o.data
-  }
-  return list
-    .map((item): Thread | null => {
-      if (!item || typeof item !== 'object') return null
-      const s = item as Record<string, unknown>
-      const id = s.sessionId ?? s.session_id ?? s.id
-      if (id == null || id === '') return null
-      const createTime = String(s.createTime ?? s.create_time ?? '')
-      const topN = Number(s.topStatus ?? s.top_status)
-      const top_status = topN === 1 ? 1 : 0
-      return {
-        id: String(id),
-        title: String(s.title ?? ''),
-        created_at: createTime,
-        updated_at: String(s.updateTime ?? s.updated_at ?? createTime),
-        user_id: String(s.userId ?? s.user_id ?? ''),
-        top_status,
-      }
-    })
-    .filter((x): x is Thread => x !== null)
-}
-
-/** 列表展示：是否已赞 */
-export function messageLikeActive(m: Message): boolean {
-  if (m.evaluation_type === null) return false
-  if (m.evaluation_type === 10) return true
-  if (m.evaluation_type === undefined && m.feedback_score === 1) return true
-  return false
-}
-
-/** 列表展示：是否已踩 */
-export function messageDislikeActive(m: Message): boolean {
-  if (m.evaluation_type === null) return false
-  if (m.evaluation_type === 20 || m.evaluation_type === 30) return true
-  if (m.evaluation_type === undefined && m.feedback_score === -1) return true
-  return false
-}
-
-export type MessageEvaluationDisplay = 'none' | 'like' | 'dislike'
-
-export function getMessageEvaluationDisplay(m: Message): MessageEvaluationDisplay {
-  if (messageLikeActive(m)) return 'like'
-  if (messageDislikeActive(m)) return 'dislike'
-  return 'none'
-}
-
-/**
- * 提交 /chat/feedback/review：10=点赞、20=取消点赞、30=点踩、40=取消点踩
- */
-export function buildChatFeedbackEvaluationType(
-  display: MessageEvaluationDisplay,
-  clickedLike: boolean,
-): number {
-  if (clickedLike) {
-    return display === 'like' ? 20 : 10
-  }
-  return display === 'dislike' ? 40 : 30
-}
-
-/** 请求成功后与列表 VO 对齐的本地 evaluation_type（null=未评价） */
-export function evaluationTypeAfterFeedbackRequest(sent: number): number | null {
-  switch (sent) {
-    case 10:
-      return 10
-    case 20:
-      return null
-    case 30:
-      return 20
-    case 40:
-      return null
-    default:
-      return null
-  }
-}
-
-/** 点赞点踩接口用 turnId；优先 turn_id，否则从合成 id（`{turnId}-10|20`）还原 */
-export function resolveMessageTurnIdForFeedback(m: Message): string {
-  const tid = m.turn_id != null ? String(m.turn_id).trim() : ''
-  if (tid) return tid
-  const id = String(m.id ?? '')
-  return id.replace(/-(?:10|20)$/, '') || id
-}
-
-/** type：10=用户，20=智能体 */
-function chatTypeToRole(type: unknown): 'user' | 'assistant' | 'system' {
-  const n = Number(type)
-  if (n === 10) return 'user'
-  if (n === 20) return 'assistant'
-  return 'assistant'
-}
-
-/**
- * 解析历史消息里的 referenceMaterials 字段，转换为 references_knowledge_items。
- */
-function parseReferenceMaterials(
-  referenceMaterials: unknown,
-): NonNullable<Message['references_knowledge_items']> | undefined {
-  if (!referenceMaterials) return undefined
-  if (Array.isArray(referenceMaterials)) {
-    return referenceMaterials as NonNullable<Message['references_knowledge_items']>
-  }
-  if (typeof referenceMaterials === 'string') {
-    const jsonText = referenceMaterials.trim()
-    if (!jsonText) return undefined
-    try {
-      const parsed = JSON.parse(jsonText) as unknown
-      if (Array.isArray(parsed)) {
-        return parsed as NonNullable<Message['references_knowledge_items']>
-      }
-      if (parsed && typeof parsed === 'object') {
-        const parsedObj = parsed as Record<string, unknown>
-        if (Array.isArray(parsedObj.items)) {
-          return parsedObj.items as NonNullable<Message['references_knowledge_items']>
-        }
-      }
-      return undefined
-    } catch {
-      return undefined
-    }
-  }
-  if (referenceMaterials && typeof referenceMaterials === 'object') {
-    const rawObj = referenceMaterials as Record<string, unknown>
-    if (Array.isArray(rawObj.items)) {
-      return rawObj.items as NonNullable<Message['references_knowledge_items']>
-    }
-  }
-  return undefined
-}
-
-/** ChatMessageVO → Message，按 createTime 升序（上旧下新） */
-export function normalizeMessagesList(raw: unknown): Message[] {
-  const body = unwrapChatApiBody(raw)
-  let list: unknown[] = []
-  if (Array.isArray(body)) list = body
-  else if (body && typeof body === 'object') {
-    const o = body as Record<string, unknown>
-    if (Array.isArray(o.messages)) list = o.messages
-    else if (Array.isArray(o.items)) list = o.items
-    else if (Array.isArray(o.list)) list = o.list
-    else if (Array.isArray(o.data)) list = o.data
-  }
-  const mapped = list
-    .map((item): Message | null => {
-      if (!item || typeof item !== 'object') return null
-      const m = item as Record<string, unknown>
-      const turnId = m.turnId ?? m.turn_id
-      const typeNum = Number(m.type)
-      const turnKey =
-        turnId != null && String(turnId) !== '' ? String(turnId) : null
-      let id: string | null = null
-      if (turnKey && (typeNum === 10 || typeNum === 20)) {
-        id = `${turnKey}-${typeNum}`
-      } else if (m.id != null && String(m.id) !== '') {
-        id = String(m.id)
-      }
-      if (!id) return null
-      const createTime = String(
-        m.createTime ?? m.create_time ?? new Date().toISOString(),
-      )
-      return {
-        id,
-        turn_id: turnKey ?? undefined,
-        role: chatTypeToRole(m.type),
-        content: String(m.content ?? m.text ?? ''),
-        created_at: createTime,
-        thoughts: Array.isArray(m.thoughts)
-          ? (m.thoughts as string[])
-          : undefined,
-        options: Array.isArray(m.options)
-          ? (m.options as string[])
-          : Array.isArray(m.suggested_replies)
-            ? (m.suggested_replies as string[])
-            : undefined,
-        references_knowledge_items: (() => {
-          const refMaterials = parseReferenceMaterials(m.referenceMaterials)
-          if (refMaterials && refMaterials.length > 0) return refMaterials
-          const rawRefs =
-            (m.references_knowledge as unknown) ??
-            (m.referencesKnowledge as unknown) ??
-            undefined
-          if (Array.isArray(rawRefs)) return rawRefs as any[]
-          if (rawRefs && typeof rawRefs === 'object') {
-            const o = rawRefs as Record<string, unknown>
-            if (Array.isArray(o.items)) return o.items as any[]
-          }
-          return undefined
-        })(),
-        ...(() => {
-          const etRaw = m.evaluationType ?? m.evaluation_type
-          let evaluation_type: number | null | undefined
-          let feedback_score: number | undefined
-
-          if (etRaw === null) {
-            evaluation_type = null
-            feedback_score = undefined
-          } else if (
-            etRaw !== undefined &&
-            etRaw !== '' &&
-            Number.isFinite(Number(etRaw))
-          ) {
-            evaluation_type = Number(etRaw)
-            if (evaluation_type === 10) feedback_score = 1
-            else if (evaluation_type === 20 || evaluation_type === 30)
-              feedback_score = -1
-            else feedback_score = undefined
-          } else {
-            evaluation_type = undefined
-            feedback_score =
-              typeof m.feedback_score === 'number'
-                ? m.feedback_score
-                : typeof m.feedbackScore === 'number'
-                  ? m.feedbackScore
-                  : undefined
-            if (feedback_score === 1) evaluation_type = 10
-            else if (feedback_score === -1) evaluation_type = 20
-          }
-          return { evaluation_type, feedback_score }
-        })(),
-      }
-    })
-    .filter((x): x is Message => x !== null)
-
-  // 保持后端返回的原始顺序，不做重排序
-  return mapped
-}
-
-/** 会话标题：有非空字符串用原标题，否则用 fallback（如「新会话」） */
-export function resolveThreadDisplayTitle(
-    title: string | undefined | null,
-    fallback: string,
-): string {
-    const v = title != null ? String(title).trim() : ''
-    return v || fallback
-}
-
-/** 与 mergeMessagePreparationDelta 一致：以 code + name 唯一定位一行（如多条 tool_call） */
-function findPreparationItemIndex(
-    list: NonNullable<Message['preparation_items']>,
-    code: string,
-    nameForMatch: string,
-    status: string,
-): number {
-    const byCodeName = () =>
-        list.findIndex((x) => x.code === code && x.name === nameForMatch)
-    if (nameForMatch !== code) return byCodeName()
-    if (status === 'end') {
-        for (let i = list.length - 1; i >= 0; i--) {
-            if (list[i].code === code && list[i].phase === 'running') return i
-        }
-    }
-    return byCodeName()
-}
-
-/**
- * 合并单条 SSE preparation.delta（code/name/status，可选 description）；约定 start→running、end→done，其它 status 原样写入 phase。
- * **同一 code + name 更新原位置**（如多条 tool_call 仅 code 相同）；首次出现排在数组末尾。end 未带 name 时回退为同 code 的最后一条 running。
- */
-export function mergeMessagePreparationDelta(
-    prev: Message['preparation_items'] | undefined,
-    delta: Record<string, unknown> | undefined,
-): NonNullable<Message['preparation_items']> {
-    if (!delta || typeof delta !== 'object') return [...(prev ?? [])]
-    const codeRaw = delta.code
-    const code = typeof codeRaw === 'string' ? codeRaw.trim() : ''
-    if (!code) return [...(prev ?? [])]
-    const nameTrim =
-        typeof delta.name === 'string' && delta.name.trim()
-            ? delta.name.trim()
-            : ''
-    const nameForMatch = nameTrim || code
-    const nameForRow = nameTrim || code
-    const status =
-        typeof delta.status === 'string' ? delta.status.trim() : ''
-    const list = [...(prev ?? [])]
-    const idx = findPreparationItemIndex(list, code, nameForMatch, status)
-    const prevRow = idx >= 0 ? list[idx] : undefined
-    let description: string | undefined = prevRow?.description
-    if ('description' in delta) {
-        const v = delta.description
-        if (typeof v === 'string') {
-            description = v.trim() || undefined
-        }
-    }
-    const descPatch = description ? { description } : {}
-
-    if (status === 'start') {
-        const next = { code, name: nameForRow, phase: 'running', ...descPatch }
-        if (idx >= 0) list[idx] = next
-        else list.push(next)
-        return list
-    }
-    if (status === 'end') {
-        const next = {
-            code,
-            name: nameTrim || prevRow?.name || code,
-            phase: 'done',
-            ...descPatch,
-        }
-        if (idx >= 0) list[idx] = next
-        else list.push(next)
-        return list
-    }
-    // 非 start/end 的 status 原样记入 phase，便于后端新增状态时保留语义
-    if (status) {
-        const next = {
-            code,
-            name: nameTrim || prevRow?.name || code,
-            phase: status,
-            ...descPatch,
-        }
-        if (idx >= 0) list[idx] = next
-        else list.push(next)
-        return list
-    }
-    return [...(prev ?? [])]
-}
-
-/** 流结束或块结束时，将准备步骤全部标为完成（避免缺少 end 时一直转圈） */
-export function markMessagePreparationDone(msg: Message): Message {
-    if (!msg.preparation_items?.length) return msg
-    return {
-        ...msg,
-        preparation_items: msg.preparation_items.map((x) => ({
-            ...x,
-            phase: 'done',
-        })),
-    }
-}
-
-/** 写入消息列表缓存前去掉 preparation（仅流式 UI 使用） */
-export function omitMessagePreparationForPersist(msg: Message): Message {
-    if (!msg.preparation_items) return msg
-    const { preparation_items: _drop, ...rest } = msg
-    return rest
-}

package/src/utils/cn.ts

@@ -1,6 +0,0 @@
-import { type ClassValue, clsx } from "clsx"
-import { twMerge } from "tailwind-merge"
-
-export function cn(...inputs: ClassValue[]) {
-    return twMerge(clsx(inputs))
-}

package/src/utils/crypto.ts

@@ -1,164 +0,0 @@
-import forge from 'node-forge';
-
-class EncryptionManager {
-    private publicKey: forge.pki.rsa.PublicKey | null = null;
-    private encryptionLevel: 'none' | 'partial' | 'full' = 'none';
-    private initialized = false;
-
-    async init() {
-        if (this.initialized) return;
-        try {
-            // We use fetch directly to avoid circular dependency with api client interceptors
-            const response = await fetch('/api/v1/auth/public-key', { method: 'POST' });
-            if (response.ok) {
-                const data = await response.json();
-                this.encryptionLevel = data.encryption_level;
-                if (data.public_key) {
-                    try {
-                        this.publicKey = forge.pki.publicKeyFromPem(data.public_key);
-                    } catch (e) {
-                        console.error('Failed to parse public key:', e);
-                    }
-                }
-            }
-            this.initialized = true;
-            console.log(`Encryption initialized. Level: ${this.encryptionLevel}`);
-        } catch (error) {
-            console.error('Failed to initialize encryption:', error);
-            // Fallback to none if failed
-            this.encryptionLevel = 'none';
-        }
-    }
-
-    get level() {
-        return this.encryptionLevel;
-    }
-
-    encrypt(data: any, force: boolean = false): { encryptedData: any; headers: Record<string, string>; aesKey: string } | null {
-        if ((this.encryptionLevel === 'none' && !force) || !this.publicKey) {
-            return null;
-        }
-
-        try {
-            // 1. Generate AES Key (32 bytes)
-            const aesKey = forge.random.getBytesSync(32);
-            const iv = forge.random.getBytesSync(12);
-
-            // 2. Encrypt Data with AES-GCM
-            const jsonString = JSON.stringify(data);
-            const cipher = forge.cipher.createCipher('AES-GCM', aesKey);
-            cipher.start({ iv: iv });
-            cipher.update(forge.util.createBuffer(jsonString, 'utf8'));
-            cipher.finish();
-            const encryptedData = cipher.output.getBytes();
-            const tag = cipher.mode.tag.getBytes();
-
-            // Combine ciphertext + tag
-            const finalEncryptedData = encryptedData + tag;
-
-            // 3. Encrypt AES Key with RSA Public Key
-            const encryptedAesKey = this.publicKey.encrypt(aesKey, 'RSA-OAEP', {
-                md: forge.md.sha256.create(),
-                mgf1: {
-                    md: forge.md.sha256.create()
-                }
-            });
-
-            // 4. Encode to Base64
-            const encryptedAesKeyB64 = forge.util.encode64(encryptedAesKey);
-            const ivB64 = forge.util.encode64(iv);
-            const encryptedDataB64 = forge.util.encode64(finalEncryptedData);
-
-            return {
-                encryptedData: encryptedDataB64,
-                aesKey: forge.util.bytesToHex(aesKey), // Return hex string of key
-                headers: {
-                    'X-Encrypted-Key': encryptedAesKeyB64,
-                    'X-IV': ivB64,
-                    'Content-Type': 'text/plain'
-                }
-            };
-        } catch (error) {
-            console.error('Encryption failed:', error);
-            throw error;
-        }
-    }
-
-    prepareEncryption(force: boolean = false): { headers: Record<string, string>; aesKey: string } | null {
-        if ((this.encryptionLevel === 'none' && !force) || !this.publicKey) {
-            return null;
-        }
-
-        try {
-            // 1. Generate AES Key (32 bytes)
-            const aesKey = forge.random.getBytesSync(32);
-
-            // 2. Encrypt AES Key with RSA Public Key
-            const encryptedAesKey = this.publicKey.encrypt(aesKey, 'RSA-OAEP', {
-                md: forge.md.sha256.create(),
-                mgf1: {
-                    md: forge.md.sha256.create()
-                }
-            });
-
-            // 3. Encode to Base64
-            const encryptedAesKeyB64 = forge.util.encode64(encryptedAesKey);
-
-            return {
-                aesKey: forge.util.bytesToHex(aesKey),
-                headers: {
-                    'X-Encrypted-Key': encryptedAesKeyB64
-                }
-            };
-        } catch (error) {
-            console.error('Failed to prepare encryption headers:', error);
-            return null;
-        }
-    }
-
-    decrypt(encryptedData: { iv: string; data: string }, aesKey: string): any {
-        try {
-            const iv = forge.util.decode64(encryptedData.iv);
-            const dataWithTag = forge.util.decode64(encryptedData.data);
-
-            // Extract tag (last 16 bytes)
-            const tagLength = 16;
-            const dataLength = dataWithTag.length - tagLength;
-            const ciphertext = dataWithTag.substring(0, dataLength);
-            const tag = dataWithTag.substring(dataLength);
-
-            // Convert Hex key back to bytes
-            const keyBytes = forge.util.hexToBytes(aesKey);
-
-            const decipher = forge.cipher.createDecipher('AES-GCM', keyBytes);
-            decipher.start({
-                iv: iv,
-                tag: forge.util.createBuffer(tag)
-            });
-            decipher.update(forge.util.createBuffer(ciphertext));
-
-            const success = decipher.finish();
-
-            if (success) {
-                const decryptedString = decipher.output.toString();
-                try {
-                    return JSON.parse(decryptedString);
-                } catch {
-                    return decryptedString;
-                }
-            } else {
-                throw new Error('Decryption authentication failed');
-            }
-        } catch (error) {
-            console.error('Decryption failed:', error);
-            throw error;
-        }
-    }
-
-    // Helper to generate raw AES key string for storage/passing
-    generateKey(): string {
-        return forge.random.getBytesSync(32);
-    }
-}
-
-export const encryptionManager = new EncryptionManager();