@magemetrics/core 0.1.2 → 0.1.4

package/dist/index.d.ts

@@ -484,9 +484,11 @@ declare const inputSchema_7: z.ZodObject<{
 declare type InternalApiClient = Simplify<Client<NoAuthPaths, `${string}/${string}`>>;
 
 export declare class MageMetricsChatTransport extends DefaultChatTransport<MMChatUIMessage> {
-    constructor(apiUrl: string, flowId: string, options?: Omit<HttpChatTransportInitOptions<MMChatUIMessage>, "api" | "headers" | "prepareSendMessagesRequest">);
+    constructor(apiUrl: string, flowId: string, options: MageMetricsChatTransportOptions);
 }
 
+declare type MageMetricsChatTransportOptions = Omit<HttpChatTransportInitOptions<MMChatUIMessage>, "api" | "prepareSendMessagesRequest">;
+
 /**
  * Core MageMetrics client that handles authentication lifecycle and provides
  * a promise-based API that automatically waits for authentication to complete.
@@ -503,6 +505,7 @@ export declare class MageMetricsClient {
     private authApiResponse;
     private authPromise;
     private processedJwt;
+    private userId;
     private storageAdapter;
     private events;
     constructor(config: MageMetricsClientConfig);
@@ -544,22 +547,15 @@ export declare class MageMetricsClient {
             data: Blob | string;
         }>;
         generateContextualInsight: (payload: GenerateInsightParam) => Promise<ExtractInsightResponse>;
+        getChat: (flowId: string, options?: MageMetricsChatTransportOptions) => MageMetricsChatTransport;
     };
-    /**
-     * Initialize authentication flow
-     */
-    private initializeAuth;
     /**
      * Perform the complete authentication flow
      */
     private performAuthFlow;
     private getApiInformation;
     private initializeSupabaseClient;
-    client(): Promise<InternalApiClient>;
-    /**
-     * Initialize the API client with Supabase headers
-     */
-    private initializeApiClient;
+    client(): InternalApiClient;
     /**
      * Handle the authentication logic (check session or exchange token)
      */
@@ -593,6 +589,11 @@ export declare interface MageMetricsClientConfig {
 export declare type MageMetricsEvent = {
     type: "authStateChange";
     detail: AuthState;
+} | {
+    type: "userChange";
+    detail: {
+        id: string;
+    };
 };
 
 /**
@@ -1892,71 +1893,6 @@ declare interface operations {
             };
         };
     };
-    retrieveChatTimeline: {
-        parameters: {
-            query?: never;
-            header: {
-                "sp-access-token": string;
-            };
-            path: {
-                flowId: string;
-            };
-            cookie?: never;
-        };
-        requestBody?: never;
-        responses: {
-            /** @description The timeline of generated insights for the flow */
-            200: {
-                headers: {
-                    [name: string]: unknown;
-                };
-                content: {
-                    "application/json": ({
-                        /** @enum {string} */
-                        type: "dataReport";
-                        flowDataId: number;
-                        flowId: string;
-                        title: string;
-                        bookmarked: boolean;
-                        answer: string | null;
-                    } | {
-                        /** @enum {string} */
-                        type: "visualization";
-                        flowId: string;
-                        flowDataId: number;
-                        flowDataVisualizationId: number;
-                        title: string;
-                        bookmarked: boolean;
-                    } | {
-                        /** @enum {string} */
-                        type: "message";
-                    })[];
-                };
-            };
-            /** @description Unable to retrieve flow with this id */
-            404: {
-                headers: {
-                    [name: string]: unknown;
-                };
-                content: {
-                    "application/json": {
-                        error: string;
-                    };
-                };
-            };
-            /** @description Something wrong happened */
-            500: {
-                headers: {
-                    [name: string]: unknown;
-                };
-                content: {
-                    "application/json": {
-                        error: string;
-                    };
-                };
-            };
-        };
-    };
     retrieveChatMessages: {
         parameters: {
             query?: never;
@@ -3328,22 +3264,6 @@ declare interface paths {
         patch?: never;
         trace?: never;
     };
-    "/api/v1/chat/{flowId}/timeline": {
-        parameters: {
-            query?: never;
-            header?: never;
-            path?: never;
-            cookie?: never;
-        };
-        get: operations["retrieveChatTimeline"];
-        put?: never;
-        post?: never;
-        delete?: never;
-        options?: never;
-        head?: never;
-        patch?: never;
-        trace?: never;
-    };
     "/api/v1/chat/{flowId}/messages": {
         parameters: {
             query?: never;

package/dist/index.js

@@ -2370,7 +2370,18 @@ var hashString = (value) => {
 
 // package.json
 var package_default = {
-  version: "0.1.2"};
+  version: "0.1.4"};
+var MageMetricsChatTransport = class extends DefaultChatTransport {
+  constructor(apiUrl, flowId, options) {
+    super({
+      ...options,
+      api: `${apiUrl}/api/v1/chat/${flowId}`,
+      prepareSendMessagesRequest({ messages, id: chatId }) {
+        return { body: { message: messages.at(-1), id: chatId } };
+      }
+    });
+  }
+};
 
 // src/core/MageMetricsEventEmitter.ts
 var MageMetricsEventEmitter = class {
@@ -2414,26 +2425,20 @@ var MageMetricsEventEmitter = class {
   }
 };
 
+// src/core/resolvable.ts
+var resolve = (value) => {
+  if (typeof value === "function") {
+    return Promise.resolve(value());
+  }
+  return Promise.resolve(value);
+};
+
 // src/core/types.ts
 var TOKEN_STORAGE_KEY = "mm-ai-sp-token";
 var CHECK_KEY = "mm-ai-check-key";
 
 // src/core/MageMetricsClient.ts
 var MM_CLIENT_VERSION = package_default.version;
-var MageMetricsChatTransport = class extends DefaultChatTransport {
-  constructor(apiUrl, flowId, options) {
-    super({
-      ...options,
-      headers: {
-        [HEADER_CLIENT_VERSION]: MM_CLIENT_VERSION
-      },
-      api: `${apiUrl}/api/v1/chat/${flowId}`,
-      prepareSendMessagesRequest({ messages, id: chatId }) {
-        return { body: { message: messages.at(-1), id: chatId } };
-      }
-    });
-  }
-};
 var getPublicApiClient = (apiUrl, apiKey) => {
   const client = createApiClient({ baseUrl: apiUrl });
   client.use(throwOnError);
@@ -2445,17 +2450,23 @@ var MageMetricsClient = class {
   config;
   authState = "initializing";
   supabaseClient = null;
-  internalApiClient = null;
+  internalApiClient;
   noAuthApiClient = null;
   authApiResponse = null;
   authPromise = null;
   processedJwt = Symbol("initial");
+  userId = null;
   storageAdapter;
   events = new MageMetricsEventEmitter();
   constructor(config) {
     this.config = config;
     this.storageAdapter = config.storageAdapter || createStorageAdapter();
-    void this.initializeAuth();
+    this.internalApiClient = createApiClient({
+      baseUrl: this.config.apiUrl
+    });
+    this.internalApiClient.use(throwOnError);
+    this.internalApiClient.use(this.createSupabaseHeaderMiddleware());
+    this.internalApiClient.use(addVersionHeader(MM_CLIENT_VERSION));
   }
   /**
    * Add an event listener for authentication state changes
@@ -2482,10 +2493,10 @@ var MageMetricsClient = class {
     if (this.authState === "error") {
       throw new Error("Authentication failed");
     }
-    if (this.authPromise) {
-      return this.authPromise;
+    if (!this.authPromise) {
+      this.authPromise = this.performAuthFlow();
     }
-    throw new Error("Authentication not initialized");
+    return this.authPromise;
   }
   async updateExternalJwt(jwt) {
     if (this.config.externalJwt === jwt) return;
@@ -2506,11 +2517,9 @@ var MageMetricsClient = class {
       return {};
     }
     const { access_token } = session.data.session;
-    const headers = {};
-    if (access_token) {
-      headers["sp-access-token"] = access_token;
-    }
-    return headers;
+    return {
+      "sp-access-token": access_token
+    };
   }
   async clearStorage() {
     try {
@@ -2531,7 +2540,6 @@ var MageMetricsClient = class {
   api = {
     getRecentFlows: async (params = {}) => {
       await this.waitForAuth();
-      if (!this.internalApiClient) throw new Error("API client not ready");
       const { data, error, response } = await this.internalApiClient.GET(
         RetrieveRecentFlows.path,
         {
@@ -2554,7 +2562,6 @@ var MageMetricsClient = class {
     },
     createFlow: async (request) => {
       await this.waitForAuth();
-      if (!this.internalApiClient) throw new Error("API client not ready");
       if ("query" in request) {
         const { data, error, response } = await this.internalApiClient.POST(
           CreateFlow.path,
@@ -2601,7 +2608,6 @@ var MageMetricsClient = class {
     },
     getDashboard: async (dashboardId) => {
       await this.waitForAuth();
-      if (!this.internalApiClient) throw new Error("API client not ready");
       const { data, error, response } = await this.internalApiClient.GET(
         RetrieveDashboard.path,
         {
@@ -2624,7 +2630,6 @@ var MageMetricsClient = class {
     },
     exportReportData: async (reportId, format = "blob") => {
       await this.waitForAuth();
-      if (!this.internalApiClient) throw new Error("API client not ready");
       const { data, error, response } = await this.internalApiClient.GET(
         ExportReportData.path,
         {
@@ -2650,7 +2655,6 @@ var MageMetricsClient = class {
     },
     generateContextualInsight: async (payload) => {
       await this.waitForAuth();
-      if (!this.internalApiClient) throw new Error("API client not ready");
       const { data, error, response } = await this.internalApiClient.POST(
         GenerateInsight.path,
         {
@@ -2669,19 +2673,22 @@ var MageMetricsClient = class {
         });
       }
       return data;
+    },
+    getChat: (flowId, options) => {
+      return new MageMetricsChatTransport(this.config.apiUrl, flowId, {
+        ...options,
+        headers: async () => {
+          const authHeaders = await this.getAuthHeaders();
+          const headers = await resolve(options?.headers);
+          return {
+            ...headers,
+            ...authHeaders,
+            [HEADER_CLIENT_VERSION]: MM_CLIENT_VERSION
+          };
+        }
+      });
     }
   };
-  /**
-   * Initialize authentication flow
-   */
-  async initializeAuth() {
-    this.authPromise = this.performAuthFlow();
-    try {
-      await this.authPromise;
-    } catch (error) {
-      console.error("Failed to initialize authentication:", error);
-    }
-  }
   /**
    * Perform the complete authentication flow
    */
@@ -2690,7 +2697,6 @@ var MageMetricsClient = class {
       this.setState("initializing");
       const apiInfo = await this.getApiInformation();
       this.initializeSupabaseClient(apiInfo.anonKey, apiInfo.apiUrl);
-      this.initializeApiClient();
       await this.handleAuthentication();
       if (this.config.externalJwt) {
         await this.saveCheckKey(this.config.externalJwt, this.config.apiKey);
@@ -2733,6 +2739,12 @@ var MageMetricsClient = class {
       });
       this.supabaseClient.onAuthStateChange((event, session) => {
         console.debug("Supabase auth state change:", event, !!session);
+        if (event === "SIGNED_IN" && session?.user) {
+          if (this.userId !== session.user.id && this.userId !== null) {
+            this.events.dispatch("userChange", { id: session.user.id });
+          }
+          this.userId = session.user.id;
+        }
         if (event === "TOKEN_REFRESHED" && session) {
           console.debug("Token refreshed successfully");
           this.setState("ready");
@@ -2744,27 +2756,8 @@ var MageMetricsClient = class {
       });
     }
   }
-  /* This method always waits for authentication to complete before returning the client */
-  async client() {
-    await this.waitForAuth();
-    if (this.authState === "ready") {
-      if (!this.internalApiClient) {
-        throw new Error(
-          "Client is ready but API client not available. This should never happen"
-        );
-      }
-      return this.internalApiClient;
-    }
-    throw new Error(`Client is not available in auth state: ${this.authState}`);
-  }
-  /**
-   * Initialize the API client with Supabase headers
-   */
-  initializeApiClient() {
-    this.internalApiClient = createApiClient({ baseUrl: this.config.apiUrl });
-    this.internalApiClient.use(throwOnError);
-    this.internalApiClient.use(this.createSupabaseHeaderMiddleware());
-    this.internalApiClient.use(addVersionHeader(MM_CLIENT_VERSION));
+  client() {
+    return this.internalApiClient;
   }
   /**
    * Handle the authentication logic (check session or exchange token)
@@ -2834,19 +2827,18 @@ var MageMetricsClient = class {
     console.debug("Token exchange successful, session set");
   }
   createSupabaseHeaderMiddleware() {
-    const supabaseClient = this.supabaseClient;
     return {
-      async onRequest({ request }) {
-        if (!supabaseClient) {
+      onRequest: async ({ request }) => {
+        const headers = await this.getAuthHeaders();
+        if (!this.supabaseClient) {
           return request;
         }
-        const session = await supabaseClient.getSession();
+        const session = await this.supabaseClient.getSession();
         if (!session.data.session) {
           return request;
         }
-        const { access_token } = session.data.session;
-        if (access_token) {
-          request.headers.set("sp-access-token", access_token);
+        for (const [key, value] of Object.entries(headers)) {
+          request.headers.set(key, value);
         }
         return request;
       }
@@ -2879,6 +2871,6 @@ var MageMetricsClient = class {
   }
 };
 
-export { BrowserStorageAdapter, CHECK_KEY, MageMetricsChatTransport, MageMetricsClient, MageMetricsEventEmitter, MemoryStorageAdapter, TOKEN_STORAGE_KEY, getPublicApiClient };
+export { BrowserStorageAdapter, CHECK_KEY, MageMetricsClient, MageMetricsEventEmitter, MemoryStorageAdapter, TOKEN_STORAGE_KEY, getPublicApiClient };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map