@retrivora-ai/rag-engine 2.2.7 → 2.2.9

package/src/handlers/index.ts

@@ -5,6 +5,8 @@ import { ChatMessage, ChatResponse } from '../types';
 import { DocumentParser } from '../utils/DocumentParser';
 import { UITransformer } from '../utils/UITransformer';
 import { DatabaseStorage, HistoryMessage } from '../core/DatabaseStorage';
+import { LicenseVerifier } from '../core/LicenseVerifier';
+import { SDK_VERSION } from '../version';
 
 export interface HandlerOptions {
   onAuthorize?: (req: NextRequest) => boolean | Promise<boolean> | Response | Promise<Response>;
@@ -147,8 +149,6 @@ function getOrCreatePlugin(configOrPlugin: Partial<RagConfig> | VectorPlugin | u
 
 // ─── Telemetry Reporter Helper ───────────────────────────────────────────────
 
-import { SDK_VERSION } from '../version';
-
 /**
  * Asynchronously sends telemetry payload to telemetryUrl when telemetry is enabled
  * or when a licenseKey is set. Non-blocking.
@@ -596,6 +596,64 @@ export function createHealthHandler(
   };
 }
 
+/**
+ * createLicenseHandler — factory for the license validation endpoint (/v1/license/validate).
+ */
+export function createLicenseHandler(
+  configOrPlugin?: Partial<RagConfig> | VectorPlugin,
+  options?: HandlerOptions
+) {
+  const plugin = getOrCreatePlugin(configOrPlugin);
+  const onAuthorize = options?.onAuthorize;
+
+  return async function POST(req: NextRequest) {
+    if (req) {
+      const authResult = await checkAuth(req, onAuthorize);
+      if (authResult) return authResult as any;
+    }
+
+    try {
+      const body = await req.json().catch(() => ({}));
+      const config = plugin.getConfig();
+      const licenseKey =
+        req.headers.get('x-license-key') ||
+        body?.licenseKey ||
+        req.headers.get('authorization')?.replace(/^Bearer\s+/i, '') ||
+        config.licenseKey ||
+        process.env.RETRIVORA_LICENSE_KEY ||
+        process.env.NEXT_PUBLIC_RETRIVORA_LICENSE_KEY ||
+        '';
+
+      const projectId = body?.projectId || config.projectId || 'my-rag-app';
+
+      const payload = LicenseVerifier.verify(licenseKey, projectId);
+
+      return NextResponse.json({
+        valid: true,
+        licenseStatus: 'ACTIVE',
+        minimumSupportedVersion: '2.1.0',
+        latestVersion: SDK_VERSION,
+        forceUpgrade: false,
+        expiresAt: payload.expiresAt ? new Date(payload.expiresAt * 1000).toISOString() : null,
+        message: 'License key is active and valid.',
+      });
+    } catch (err: any) {
+      return NextResponse.json(
+        {
+          valid: false,
+          licenseStatus: 'REVOKED',
+          minimumSupportedVersion: '2.1.0',
+          latestVersion: SDK_VERSION,
+          forceUpgrade: false,
+          expiresAt: null,
+          message: err?.message || 'License validation failed.',
+        },
+        { status: 400 }
+      );
+    }
+  };
+}
+
 /**
  * createUploadHandler — factory for the file upload ingestion endpoint.
  */
@@ -932,6 +990,7 @@ export function createRagHandler(
   const historyHandler = createHistoryHandler(plugin, { onAuthorize });
   const feedbackHandler = createFeedbackHandler(plugin, { onAuthorize });
   const sessionsHandler = createSessionsHandler(plugin, { onAuthorize });
+  const licenseHandler = createLicenseHandler(plugin, { onAuthorize });
 
   async function routePostRequest(req: any, segment: string) {
     switch (segment) {
@@ -950,6 +1009,9 @@ export function createRagHandler(
         return historyHandler(req);
       case 'feedback':
         return feedbackHandler(req);
+      case 'license/validate':
+      case 'v1/license/validate':
+        return licenseHandler(req);
       default:
         return NextResponse.json({ error: `Not Found: POST segment "${segment}" not supported.` }, { status: 404 });
     }
@@ -983,6 +1045,7 @@ export function createRagHandler(
     try {
       const url = new URL(req.url);
       const pathname = url.pathname;
+      if (pathname.includes('/license/validate')) return 'v1/license/validate';
       if (pathname.endsWith('/upload')) return 'upload';
       if (pathname.endsWith('/chat') || pathname.endsWith('/chat-sync')) return 'chat';
       if (pathname.endsWith('/health')) return 'health';

package/src/hooks/useRagChat.ts

@@ -129,8 +129,27 @@ export function useRagChat(
           userMessageId: userMsgId,
         });
 
-        const fetchHeaders = {
+        const getHeader = (headersObj: Record<string, string> | undefined, name: string) => {
+          if (!headersObj) return undefined;
+          const target = name.toLowerCase();
+          for (const [k, v] of Object.entries(headersObj)) {
+            if (k.toLowerCase() === target) return v;
+          }
+          return undefined;
+        };
+
+        const resolvedLicenseKey =
+          options.licenseKey ||
+          getHeader(options.headers, 'x-license-key') ||
+          getHeader(options.headers, 'authorization')?.replace(/^Bearer\s+/i, '') ||
+          (typeof process !== 'undefined'
+            ? (process.env.NEXT_PUBLIC_RETRIVORA_LICENSE_KEY || process.env.RETRIVORA_LICENSE_KEY)
+            : '') ||
+          '';
+
+        const fetchHeaders: Record<string, string> = {
           'Content-Type': 'application/json',
+          ...(resolvedLicenseKey ? { 'x-license-key': resolvedLicenseKey } : {}),
           ...(options.headers || {})
         };
 

package/src/types/chat.ts

@@ -57,6 +57,8 @@ export interface UseRagChatOptions {
   onError?: (error: string) => void;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Session ID for history and feedback tracking (default: 'default') */
   sessionId?: string;
 }

package/src/types/props.ts

@@ -57,11 +57,12 @@ export interface ChatWindowProps {
   onAddToCart?: (product: Product) => void;
   /** Optional custom API URL to send chat messages to */
   apiUrl?: string;
-  /**
-   * Base URL for Retrivora SDK history/feedback/sessions endpoints.
+  /** Base URL for Retrivora SDK history/feedback/sessions endpoints.
    * Defaults to '/api/retrivora'. Set this to match your catch-all route location.
    */
   retrivoraApiBase?: string;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
 }
@@ -78,6 +79,8 @@ export interface ChatWidgetProps {
    * Defaults to '/api/retrivora'.
    */
   retrivoraApiBase?: string;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
 }