@uselingu/react-native-sdk 0.1.4 → 0.1.5

package/src/hooks/useLinguChat.ts

@@ -7,6 +7,7 @@ import { useState, useEffect, useCallback, useRef } from 'react';
 import { LinguAPI } from '../services/LinguAPI';
 import { SessionManager } from '../services/SessionManager';
 import { StorageManager } from '../services/StorageManager';
+import { resolveBaseURL } from '../types';
 import type { Message, WidgetConfig, LinguConfig } from '../types';
 
 const LOG_PREFIX = '[useLinguChat]';
@@ -23,9 +24,15 @@ interface UseLinguChatReturn {
 }
 
 export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
+   const baseURL = resolveBaseURL({
+      baseURL: linguConfig.baseURL,
+      environment: linguConfig.environment,
+   });
+
    console.log(`${LOG_PREFIX} Initializing hook with config:`, {
       apiKey: linguConfig.apiKey?.substring(0, 20) + '...',
-      baseURL: linguConfig.baseURL,
+      baseURL,
+      environment: linguConfig.environment ?? 'production',
       autoOpen: linguConfig.autoOpen,
    });
 
@@ -37,12 +44,19 @@ export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
    const apiRef = useRef<LinguAPI | null>(null);
    const sessionManagerRef = useRef<SessionManager | null>(null);
    const currentSessionIdRef = useRef<string | null>(null);
+   // Mirror `messages` into a ref so callbacks created once in the init effect
+   // (onSessionEnd, cleanup) always read the latest messages instead of the
+   // empty array captured at mount.
+   const messagesRef = useRef<Message[]>([]);
+
+   // Keep the messages ref in sync with state.
+   useEffect(() => {
+      messagesRef.current = messages;
+   }, [messages]);
 
    // Initialize API client and session manager
    useEffect(() => {
       console.log(`${LOG_PREFIX} Setting up API client and session manager`);
-
-      const baseURL = linguConfig.baseURL || 'https://api.uselingu.app/api';
       console.log(`${LOG_PREFIX} API Base URL:`, baseURL);
 
       const api = new LinguAPI(linguConfig.apiKey, baseURL);
@@ -72,10 +86,11 @@ export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
          onSessionEnd: async (sessionId) => {
             console.log(`${LOG_PREFIX} Session ending:`, sessionId);
 
-            // Save messages before ending
-            if (messages.length > 0) {
-               console.log(`${LOG_PREFIX} Saving ${messages.length} messages before session end`);
-               await StorageManager.saveMessages(sessionId, messages);
+            // Save messages before ending (read from ref for latest value)
+            const currentMessages = messagesRef.current;
+            if (currentMessages.length > 0) {
+               console.log(`${LOG_PREFIX} Saving ${currentMessages.length} messages before session end`);
+               await StorageManager.saveMessages(sessionId, currentMessages);
             }
 
             currentSessionIdRef.current = null;
@@ -110,15 +125,16 @@ export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
       // Cleanup on unmount
       return () => {
          console.log(`${LOG_PREFIX} Component unmounting, cleaning up...`);
-         // Save messages before cleanup
-         if (currentSessionIdRef.current && messages.length > 0) {
+         // Save messages before cleanup (read from ref for latest value)
+         const currentMessages = messagesRef.current;
+         if (currentSessionIdRef.current && currentMessages.length > 0) {
             console.log(`${LOG_PREFIX} Saving messages on unmount`);
-            StorageManager.saveMessages(currentSessionIdRef.current, messages);
+            StorageManager.saveMessages(currentSessionIdRef.current, currentMessages);
          }
          sessionManagerRef.current?.cleanup();
          console.log(`${LOG_PREFIX} Cleanup complete`);
       };
-   }, [linguConfig.apiKey, linguConfig.baseURL]);
+   }, [linguConfig.apiKey, baseURL]);
 
    // Save messages whenever they change
    useEffect(() => {
@@ -129,18 +145,46 @@ export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
    }, [messages]);
 
    /**
-    * Try to restore previous session
+    * Try to restore previous session from backend
     */
    const restorePreviousSession = async () => {
       try {
          const previousSessionId = await StorageManager.getCurrentSessionId();
-         if (previousSessionId) {
+         if (previousSessionId && apiRef.current) {
             console.log(`${LOG_PREFIX} Found previous session:`, previousSessionId);
+
+            // Try to fetch fresh history from backend
+            try {
+               const history = await apiRef.current.getChatHistory(previousSessionId);
+
+               if (!history.isNewSession && history.messages.length > 0) {
+                  console.log(`${LOG_PREFIX} Restored ${history.messages.length} messages from backend`);
+
+                  // Convert backend format to our Message format
+                  const restoredMessages: Message[] = history.messages.map((msg, index) => ({
+                     id: `restored-${index}-${Date.now()}`,
+                     text: msg.content,
+                     sender: msg.role === 'assistant' ? 'bot' : 'user',
+                     timestamp: new Date(msg.timestamp),
+                  }));
+
+                  setMessages(restoredMessages);
+                  currentSessionIdRef.current = previousSessionId;
+                  apiRef.current.resumeSession(previousSessionId);
+                  setIsSessionActive(true);
+                  return;
+               }
+            } catch (apiError) {
+               console.warn(`${LOG_PREFIX} Failed to fetch from backend, trying local storage:`, apiError);
+            }
+
+            // Fallback to local storage
             const savedMessages = await StorageManager.loadMessages(previousSessionId);
             if (savedMessages.length > 0) {
-               console.log(`${LOG_PREFIX} ✅ Restored ${savedMessages.length} messages from previous session`);
+               console.log(`${LOG_PREFIX} Restored ${savedMessages.length} messages from local storage`);
                setMessages(savedMessages);
                currentSessionIdRef.current = previousSessionId;
+               setIsSessionActive(true);
             } else {
                console.log(`${LOG_PREFIX} Previous session had no messages`);
             }
@@ -148,7 +192,7 @@ export const useLinguChat = (linguConfig: LinguConfig): UseLinguChatReturn => {
             console.log(`${LOG_PREFIX} No previous session found`);
          }
       } catch (error) {
-         console.error(`${LOG_PREFIX} ❌ Failed to restore previous session:`, error);
+         console.error(`${LOG_PREFIX} Failed to restore previous session:`, error);
       }
    };
 

package/src/index.ts

@@ -28,9 +28,17 @@ export { StorageManager } from './services/StorageManager';
 // Utils
 export * from './utils';
 
+// Environment helpers
+export {
+   LINGU_API_URLS,
+   DEFAULT_LINGU_ENVIRONMENT,
+   resolveBaseURL,
+} from './types';
+
 // Types
 export type {
    LinguConfig,
+   LinguEnvironment,
    ChatPosition,
    ThemeMode,
    Message,

package/src/services/LinguAPI.ts

@@ -18,7 +18,7 @@ export class LinguAPI {
    private apiKey: string;
    private sessionId: string | null = null;
 
-   constructor(apiKey: string, baseURL: string = 'https://api.uselingu.app/api') {
+   constructor(apiKey: string, baseURL: string = 'https://lingu-api-prod.fly.dev/api') {
       console.log(`${LOG_PREFIX} Initializing API client:`, {
          baseURL,
          apiKeyPrefix: apiKey.substring(0, 20) + '...',
@@ -133,6 +133,49 @@ export class LinguAPI {
       }
    }
 
+   /**
+    * Resume an existing session (sets sessionId without creating new)
+    */
+   resumeSession(sessionId: string): string {
+      console.log(`${LOG_PREFIX} Resuming session:`, sessionId);
+      this.sessionId = sessionId;
+      return sessionId;
+   }
+
+   /**
+    * Get chat history for a session from the backend
+    */
+   async getChatHistory(sessionId: string): Promise<{
+      sessionId: string;
+      messages: Array<{
+         role: 'user' | 'assistant';
+         content: string;
+         timestamp: string;
+      }>;
+      isNewSession: boolean;
+   }> {
+      console.log(`${LOG_PREFIX} Fetching chat history for session:`, sessionId);
+
+      try {
+         const startTime = Date.now();
+         const response = await this.client.get(
+            `/chat/widget/history?apiKey=${encodeURIComponent(this.apiKey)}&sessionId=${encodeURIComponent(sessionId)}`
+         );
+         const duration = Date.now() - startTime;
+
+         const data = response.data.data;
+         console.log(`${LOG_PREFIX} Chat history loaded in ${duration}ms:`, {
+            messageCount: data.messages?.length || 0,
+            isNewSession: data.isNewSession,
+         });
+
+         return data;
+      } catch (error) {
+         console.error(`${LOG_PREFIX} Failed to get chat history:`, error);
+         throw new Error('Failed to load chat history');
+      }
+   }
+
    /**
     * Send a message to the AI
     */

package/src/types/index.ts

@@ -2,9 +2,47 @@
  * Core types for Lingu React Native SDK
  */
 
+/**
+ * Which Lingu backend the SDK talks to.
+ * Ignored when an explicit `baseURL` is provided.
+ */
+export type LinguEnvironment = 'production' | 'staging';
+
+/**
+ * Predefined API base URLs per environment.
+ * An explicit `baseURL` on LinguConfig always overrides these.
+ */
+export const LINGU_API_URLS: Record<LinguEnvironment, string> = {
+   production: 'https://lingu-api-prod.fly.dev/api',
+   staging: 'https://lingu-api.fly.dev/api',
+};
+
+export const DEFAULT_LINGU_ENVIRONMENT: LinguEnvironment = 'production';
+
+/**
+ * Resolve the API base URL from config.
+ * Precedence: explicit `baseURL` > `environment` mapping > production default.
+ */
+export const resolveBaseURL = (opts: {
+   baseURL?: string;
+   environment?: LinguEnvironment;
+}): string => {
+   if (opts.baseURL) return opts.baseURL;
+   return LINGU_API_URLS[opts.environment ?? DEFAULT_LINGU_ENVIRONMENT];
+};
+
 export interface LinguConfig {
    apiKey: string;
+   /**
+    * Explicit API base URL. Overrides `environment` when set.
+    * Use this for local development (e.g. http://127.0.0.1:3003/api).
+    */
    baseURL?: string;
+   /**
+    * Select which hosted backend to use: 'production' (default) or 'staging'.
+    * Ignored if `baseURL` is provided.
+    */
+   environment?: LinguEnvironment;
    autoOpen?: boolean;
    onSessionStart?: (sessionId: string) => void;
    onSessionEnd?: (sessionId: string) => void;