rio-assist-widget 0.1.9 → 0.1.11

package/src/services/rioWebsocket.ts

@@ -1,52 +1,68 @@
-const WEBSOCKET_URL = 'wss://ws.volkswagen.latam-sandbox.rio.cloud';
-const DEFAULT_AGENT_MODEL = 'eu.amazon.nova-pro-v1:0';
+const WEBSOCKET_URL = 'wss://ws.volkswagen.latam-sandbox.rio.cloud';
+const HEARTBEAT_INTERVAL_MS = 5 * 60_000; // keep-alive before the 10min idle timeout
 
-export type RioIncomingMessage = {
-  text: string;
-  raw: string;
-  data: unknown;
-};
+export type RioIncomingMessage = {
+  text: string;
+  raw: string;
+  data: unknown;
+  action?: string;
+};
 
 export class RioWebsocketClient {
   readonly token: string;
 
   private socket: WebSocket | null = null;
 
-  private connectPromise: Promise<void> | null = null;
-
-  private readonly listeners = new Set<(message: RioIncomingMessage) => void>();
-
-  constructor(token: string) {
-    this.token = token;
-  }
+  private connectPromise: Promise<void> | null = null;
+
+  private readonly listeners = new Set<(message: RioIncomingMessage) => void>();
+
+  private heartbeatId: number | null = null;
+
+  constructor(token: string) {
+    this.token = token;
+  }
 
   matchesToken(value: string) {
     return this.token === value;
   }
 
-  async sendMessage(message: string) {
-    const socket = await this.ensureConnection();
-
-    const payload = {
-      action: 'sendMessage',
-      message,
-      agentModel: DEFAULT_AGENT_MODEL,
-    };
-
-    socket.send(JSON.stringify(payload));
-  }
-
-  onMessage(listener: (message: RioIncomingMessage) => void) {
-    this.listeners.add(listener);
-    return () => this.listeners.delete(listener);
-  }
-
-  close() {
-    if (this.socket && this.socket.readyState === WebSocket.OPEN) {
-      this.socket.close();
-    }
-
-    this.connectPromise = null;
+  async sendMessage(message: string, conversationId?: string | null) {
+    const socket = await this.ensureConnection();
+
+    const payload = {
+      action: 'sendMessage',
+      message,
+      conversationId: conversationId ?? null,
+    };
+
+    console.info('[RioAssist][ws] enviando payload de mensagem', payload);
+    socket.send(JSON.stringify(payload));
+  }
+
+  async requestHistory(options: { conversationId?: string | null; limit?: number } = {}) {
+    const socket = await this.ensureConnection();
+    const payload: Record<string, unknown> = {
+      action: 'getHistory',
+      limit: options.limit ?? 50,
+      conversationId: options.conversationId ?? null,
+    };
+
+    socket.send(JSON.stringify(payload));
+  }
+
+  onMessage(listener: (message: RioIncomingMessage) => void) {
+    this.listeners.add(listener);
+    return () => this.listeners.delete(listener);
+  }
+
+  close() {
+    this.stopHeartbeat();
+    if (this.socket && this.socket.readyState === WebSocket.OPEN) {
+      this.socket.close();
+    }
+
+    this.connectPromise = null;
     this.socket = null;
     this.listeners.clear();
   }
@@ -65,27 +81,32 @@ export class RioWebsocketClient {
       `${WEBSOCKET_URL}?token=${encodeURIComponent(this.token)}`,
     );
 
-    this.socket.addEventListener('message', (event) => this.handleMessage(event));
-    this.socket.addEventListener('close', () => {
-      this.connectPromise = null;
-      this.socket = null;
-    });
+    this.socket.addEventListener('message', (event) => this.handleMessage(event));
+    this.socket.addEventListener('close', () => {
+      this.connectPromise = null;
+      this.socket = null;
+      this.stopHeartbeat();
+    });
 
     this.connectPromise = new Promise((resolve, reject) => {
       if (!this.socket) {
         reject(new Error('Falha ao criar conexão WebSocket.'));
         return;
       }
-
-      const handleOpen = () => {
-        cleanup();
-        resolve();
-      };
-
-      const handleError = () => {
-        cleanup();
-        this.socket?.close();
-        this.socket = null;
+
+      const handleOpen = () => {
+        cleanup();
+        if (this.socket) {
+          this.startHeartbeat(this.socket);
+        }
+        resolve();
+      };
+
+      const handleError = () => {
+        cleanup();
+        this.stopHeartbeat();
+        this.socket?.close();
+        this.socket = null;
         this.connectPromise = null;
         reject(
           new Error(
@@ -109,33 +130,57 @@ export class RioWebsocketClient {
       throw new Error('Conexão WebSocket do Rio Insight não está pronta.');
     }
 
-    return this.socket;
-  }
-
-  private async handleMessage(event: MessageEvent) {
-    const raw = await this.readMessage(event.data);
-    let parsed: unknown = null;
-    let text = raw;
-
-    try {
-      parsed = JSON.parse(raw);
-      if (typeof parsed === 'object' && parsed !== null) {
-        const maybeText =
-          (parsed as any).message ??
-          (parsed as any).response ??
-          (parsed as any).text ??
-          (parsed as any).content;
+    return this.socket;
+  }
+
+  private startHeartbeat(socket: WebSocket) {
+    this.stopHeartbeat();
+    this.heartbeatId = window.setInterval(() => {
+      if (socket.readyState === WebSocket.OPEN) {
+        socket.send(JSON.stringify({ action: 'ping' }));
+      }
+    }, HEARTBEAT_INTERVAL_MS);
+  }
+
+  private stopHeartbeat() {
+    if (this.heartbeatId !== null) {
+      window.clearInterval(this.heartbeatId);
+      this.heartbeatId = null;
+    }
+  }
+
+  private async handleMessage(event: MessageEvent) {
+    const raw = await this.readMessage(event.data);
+    let parsed: unknown = null;
+    let text = raw;
+    let action: string | undefined;
+
+    try {
+      parsed = JSON.parse(raw);
+      if (typeof parsed === 'object' && parsed !== null) {
+        const maybeAction =
+          (parsed as any).action ?? (parsed as any).type ?? (parsed as any).event;
+
+        if (typeof maybeAction === 'string') {
+          action = maybeAction;
+        }
+
+        const maybeText =
+          (parsed as any).message ??
+          (parsed as any).response ??
+          (parsed as any).text ??
+          (parsed as any).content;
 
         if (typeof maybeText === 'string') {
           text = maybeText;
         }
       }
-    } catch {
-      parsed = null;
-    }
-
-    this.listeners.forEach((listener) => listener({ text, raw, data: parsed }));
-  }
+    } catch {
+      parsed = null;
+    }
+
+    this.listeners.forEach((listener) => listener({ text, raw, data: parsed, action }));
+  }
 
   private async readMessage(
     data: MessageEvent['data'],