@unif/react-native-chat-sdk 0.1.0

package/src/tools/XStream.ts

@@ -0,0 +1,112 @@
+/**
+ * XStream — SSE 流适配器（React Native 版本）
+ *
+ * 基于 react-native-sse 封装，支持 POST 方法 + 自定义请求体
+ */
+
+import type EventSource from 'react-native-sse';
+import type {SSEOutput} from '../types/provider';
+
+export interface XStreamConfig {
+  url: string;
+  method?: 'GET' | 'POST';
+  headers?: Record<string, string>;
+  body?: Record<string, unknown>;
+  timeout?: number;
+  transformStream?: (event: { data: string; event?: string }) => SSEOutput;
+}
+
+export interface XStreamCallbacks {
+  onMessage: (data: SSEOutput) => void;
+  onError?: (error: Error) => void;
+  onComplete?: () => void;
+  onOpen?: () => void;
+}
+
+const defaultTransform = (event: {
+  data: string;
+  event?: string;
+}): SSEOutput => ({
+  data: event.data,
+  event: event.event,
+});
+
+export class XStream {
+  private config: XStreamConfig;
+  private es: EventSource | null = null;
+  private timeoutTimer: ReturnType<typeof setTimeout> | null = null;
+
+  constructor(config: XStreamConfig) {
+    this.config = config;
+  }
+
+  connect(callbacks: XStreamCallbacks): void {
+    this.abort();
+
+    const {
+      url,
+      method = 'POST',
+      headers,
+      body,
+      timeout = 120000,
+      transformStream = defaultTransform,
+    } = this.config;
+
+    // react-native-sse 动态 require，避免非 RN 环境报错
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const RNEventSource = require('react-native-sse').default;
+
+    const es = new RNEventSource(url, {
+      method,
+      headers: {
+        'Content-Type': 'application/json',
+        ...headers,
+      },
+      body: body ? JSON.stringify(body) : undefined,
+    });
+
+    this.es = es;
+
+    this.timeoutTimer = setTimeout(() => {
+      this.abort();
+      callbacks.onError?.(new Error('SSE 连接超时'));
+    }, timeout);
+
+    es.addEventListener('open', () => {
+      callbacks.onOpen?.();
+    });
+
+    es.addEventListener('message', (event: { data?: string }) => {
+      if (event.data) {
+        const output = transformStream({ data: event.data });
+        callbacks.onMessage(output);
+      }
+    });
+
+    es.addEventListener('error', (event: { type?: string; message?: string }) => {
+      this.clearTimer();
+      callbacks.onError?.(
+        new Error(event.message || 'SSE 连接错误')
+      );
+    });
+  }
+
+  abort(): void {
+    this.clearTimer();
+    if (this.es) {
+      this.es.close();
+      this.es = null;
+    }
+  }
+
+  dispose(): void {
+    this.abort();
+  }
+
+  private clearTimer(): void {
+    if (this.timeoutTimer) {
+      clearTimeout(this.timeoutTimer);
+      this.timeoutTimer = null;
+    }
+  }
+}

package/src/types/message.ts

@@ -0,0 +1,26 @@
+/**
+ * Chat 消息类型定义
+ */
+
+export type MessageRole = 'user' | 'assistant' | 'system';
+export type MessageType = 'text' | 'card' | 'system';
+export type MessageStatus =
+  | 'local'
+  | 'loading'
+  | 'updating'
+  | 'success'
+  | 'error'
+  | 'abort';
+
+export interface ChatMessage {
+  id: string;
+  text: string;
+  role: MessageRole;
+  createdAt: Date;
+  messageType: MessageType;
+  status: MessageStatus;
+  cardType?: string;
+  cardData?: { data: Record<string, unknown>; actions: string[] };
+  turnId: string;
+  extra?: Record<string, unknown>;
+}

package/src/types/provider.ts

@@ -0,0 +1,24 @@
+/**
+ * Provider 协议类型
+ * 定义 ChatProvider ↔ useXChat 之间的数据契约
+ */
+
+import type {ChatMessage} from './message';
+
+export interface RequestParams<T = ChatMessage> {
+  message: T;
+  messages?: T[];
+  model?: string;
+  extra?: Record<string, unknown>;
+}
+
+export interface SSEOutput {
+  data: string;
+  event?: string;
+}
+
+export interface ProviderCallbacks<Message = ChatMessage> {
+  onUpdate: (message: Message) => void;
+  onSuccess: (message: Message) => void;
+  onError: (error: Error) => void;
+}

package/src/types/sse.ts

@@ -0,0 +1,93 @@
+/**
+ * SSEEnvelope 协议类型定义
+ */
+
+export type SSEEventType =
+  | 'text'
+  | 'card_data'
+  | 'suggestion'
+  | 'error'
+  | 'done';
+
+export interface SSEEnvelopeBase {
+  event_id: string;
+  turn_id: string;
+  seq: number;
+  ts: string;
+  session_id: string;
+}
+
+export interface SSETextEvent extends SSEEnvelopeBase {
+  type: 'text';
+  payload: { content: string };
+}
+
+export interface SSECardDataEvent extends SSEEnvelopeBase {
+  type: 'card_data';
+  payload: {
+    cardType: string;
+    payload: CardPayload;
+  };
+}
+
+export interface SSESuggestionEvent extends SSEEnvelopeBase {
+  type: 'suggestion';
+  payload: {
+    items: SuggestionItem[];
+  };
+}
+
+export interface SSEErrorEvent extends SSEEnvelopeBase {
+  type: 'error';
+  payload: {
+    code: string;
+    message: string;
+    retryable: boolean;
+    recover_actions?: ActionPayload[];
+  };
+}
+
+export interface SSEDoneEvent extends SSEEnvelopeBase {
+  type: 'done';
+  payload: {
+    messageId: string;
+    reason: 'ok' | 'abort' | 'error';
+  };
+}
+
+export type SSEEnvelope =
+  | SSETextEvent
+  | SSECardDataEvent
+  | SSESuggestionEvent
+  | SSEErrorEvent
+  | SSEDoneEvent;
+
+export interface SuggestionItem {
+  id: string;
+  label: string;
+  action?: ActionPayload;
+  mode?: 'action' | 'input_focus';
+}
+
+export type ActionType =
+  | 'confirm_order'
+  | 'select_product'
+  | 'reorder'
+  | 'add_product'
+  | 'clear_draft'
+  | 'push_dealer'
+  | 'new_order'
+  | 'view_detail'
+  | 'add_to_order'
+  | 'browse_products'
+  | 'retry_order';
+
+export interface ActionPayload {
+  type: ActionType;
+  payload?: Record<string, unknown>;
+}
+
+export interface CardPayload {
+  data: Record<string, unknown>;
+  actions: string[];
+}