@smallwebco/tinypivot-react 1.0.74 → 1.0.79

package/dist/index.js

@@ -14,6 +14,7 @@ import {
   getDefaultDemoResponse,
   getDemoSchema,
   getInitialDemoData,
+  getLatestConversationData,
   getMessagesForAPI,
   setConversationDataSource,
   validateSQLSafety
@@ -55,14 +56,22 @@ function useAIAnalyst(options) {
       }
     }
   }, [storageKey]);
-  const [conversation, setConversation] = useState(() => loadFromStorage());
+  const initialConversationRef = useRef(null);
+  if (!initialConversationRef.current) {
+    initialConversationRef.current = loadFromStorage();
+  }
+  const [conversation, setConversation] = useState(initialConversationRef.current);
   const [schemas, setSchemas] = useState(/* @__PURE__ */ new Map());
   const [allSchemas, setAllSchemas] = useState([]);
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState(null);
-  const [lastLoadedData, setLastLoadedData] = useState(null);
+  const [lastLoadedData, setLastLoadedData] = useState(
+    () => getLatestConversationData(initialConversationRef.current)
+  );
   const [discoveredDataSources, setDiscoveredDataSources] = useState([]);
   const [isLoadingTables, setIsLoadingTables] = useState(false);
+  const dataSourceLoadPromisesRef = useRef(/* @__PURE__ */ new Map());
+  const hydratedPersistedSelectionRef = useRef(false);
   const effectiveDataSources = useMemo(() => {
     if (config.dataSources && config.dataSources.length > 0) {
       return config.dataSources;
@@ -207,6 +216,59 @@ function useAIAnalyst(options) {
       console.warn("Failed to fetch sample data:", err);
     }
   }, [onDataLoaded]);
+  const loadDataSourceState = useCallback(async (dataSource) => {
+    const currentConfig = configRef.current;
+    if (currentConfig.dataSourceLoader) {
+      const { data, schema } = await currentConfig.dataSourceLoader(dataSource.id);
+      if (schema) {
+        setSchemas((prev) => new Map(prev).set(dataSource.id, schema));
+      }
+      if (data && data.length > 0) {
+        setLastLoadedData(data);
+        onDataLoaded?.({
+          data,
+          query: `SELECT * FROM ${dataSource.table} LIMIT 100`,
+          dataSourceId: dataSource.id,
+          rowCount: data.length
+        });
+      }
+      return;
+    }
+    if (currentConfig.demoMode) {
+      const demoSchema = getDemoSchema(dataSource.id);
+      if (demoSchema) {
+        setSchemas((prev) => new Map(prev).set(dataSource.id, demoSchema));
+      }
+      const initialData = getInitialDemoData(dataSource.id);
+      if (initialData) {
+        setLastLoadedData(initialData);
+        onDataLoaded?.({
+          data: initialData,
+          query: `SELECT * FROM ${dataSource.table} LIMIT 10`,
+          dataSourceId: dataSource.id,
+          rowCount: initialData.length
+        });
+      }
+      return;
+    }
+    if (currentConfig.endpoint) {
+      await fetchSchema(dataSource);
+      await fetchSampleData(dataSource);
+    }
+  }, [fetchSchema, fetchSampleData, onDataLoaded]);
+  const ensureDataSourceState = useCallback(async (dataSource) => {
+    const existingLoad = dataSourceLoadPromisesRef.current.get(dataSource.id);
+    if (existingLoad) {
+      return existingLoad;
+    }
+    const loadPromise = loadDataSourceState(dataSource).catch((err) => {
+      console.warn("Failed to load data source:", err);
+    }).finally(() => {
+      dataSourceLoadPromisesRef.current.delete(dataSource.id);
+    });
+    dataSourceLoadPromisesRef.current.set(dataSource.id, loadPromise);
+    return loadPromise;
+  }, [loadDataSourceState]);
   const selectDataSource = useCallback(async (dataSourceId) => {
     const dataSource = effectiveDataSources.find((ds) => ds.id === dataSourceId);
     if (!dataSource) {
@@ -226,44 +288,30 @@ What would you like to know about this data?`
       onConversationUpdate?.({ conversation: withMessage });
       return withMessage;
     });
-    if (configRef.current.dataSourceLoader) {
-      try {
-        const { data, schema } = await configRef.current.dataSourceLoader(dataSourceId);
-        if (schema) {
-          setSchemas((prev) => new Map(prev).set(dataSourceId, schema));
-        }
-        if (data && data.length > 0) {
-          setLastLoadedData(data);
-          onDataLoaded?.({
-            data,
-            query: `SELECT * FROM ${dataSource.table} LIMIT 100`,
-            dataSourceId,
-            rowCount: data.length
-          });
-        }
-      } catch (err) {
-        console.warn("Failed to load data source:", err);
-      }
-    } else if (configRef.current.demoMode) {
-      const demoSchema = getDemoSchema(dataSourceId);
-      if (demoSchema) {
-        setSchemas((prev) => new Map(prev).set(dataSourceId, demoSchema));
-      }
-      const initialData = getInitialDemoData(dataSourceId);
-      if (initialData) {
-        setLastLoadedData(initialData);
-        onDataLoaded?.({
-          data: initialData,
-          query: `SELECT * FROM ${dataSource.table} LIMIT 10`,
-          dataSourceId,
-          rowCount: initialData.length
-        });
-      }
-    } else if (configRef.current.endpoint) {
-      await fetchSchema(dataSource);
-      await fetchSampleData(dataSource);
+    await ensureDataSourceState(dataSource);
+  }, [effectiveDataSources, ensureDataSourceState, onConversationUpdate]);
+  useEffect(() => {
+    if (hydratedPersistedSelectionRef.current) {
+      return;
+    }
+    const initialConversation = initialConversationRef.current;
+    const initialDataSourceId = initialConversation?.dataSourceId;
+    if (!initialDataSourceId) {
+      hydratedPersistedSelectionRef.current = true;
+      return;
+    }
+    const dataSource = effectiveDataSources.find((ds) => ds.id === initialDataSourceId);
+    if (!dataSource) {
+      return;
+    }
+    const hasPersistedPreviewData = !!getLatestConversationData(initialConversation);
+    const hasSchema = schemas.has(initialDataSourceId);
+    const hasPreviewData = !!lastLoadedData?.length || hasPersistedPreviewData;
+    hydratedPersistedSelectionRef.current = true;
+    if (!hasSchema || !hasPreviewData) {
+      void ensureDataSourceState(dataSource);
     }
-  }, [effectiveDataSources, fetchSchema, fetchSampleData, onConversationUpdate, onDataLoaded]);
+  }, [effectiveDataSources, ensureDataSourceState, lastLoadedData, schemas]);
   const callAIEndpoint = useCallback(async (userInput, currentConversation, currentSchemas, currentDataSources, currentAllSchemas) => {
     if (!configRef.current.endpoint) {
       throw new Error("No endpoint configured. Set `endpoint` in AI analyst config.");
@@ -666,6 +714,7 @@ What would you like to know about this data?`
     return null;
   }, [conversation.dataSourceId, effectiveDataSources, onError]);
   const clearConversation = useCallback(() => {
+    hydratedPersistedSelectionRef.current = true;
     const newConv = createConversation(configRef.current.sessionId);
     setConversation(newConv);
     setError(null);
@@ -676,7 +725,9 @@ What would you like to know about this data?`
     return { ...conversation };
   }, [conversation]);
   const importConversation = useCallback((conv) => {
+    hydratedPersistedSelectionRef.current = true;
     setConversation(conv);
+    setLastLoadedData(getLatestConversationData(conv));
     onConversationUpdate?.({ conversation: conv });
   }, [onConversationUpdate]);
   return {
@@ -4716,9 +4767,8 @@ function PivotSkeleton({
               d: "M12 15v2m-6 4h12a2 2 0 002-2v-6a2 2 0 00-2-2H6a2 2 0 00-2 2v6a2 2 0 002 2zm10-10V7a4 4 0 00-8 0v4h8z"
             }
           ) }),
-          /* @__PURE__ */ jsx8("h3", { children: "Pro Feature" }),
-          /* @__PURE__ */ jsx8("p", { children: "Pivot Table functionality requires a Pro license." }),
-          /* @__PURE__ */ jsx8("a", { href: "https://tiny-pivot.com/#pricing", target: "_blank", rel: "noopener noreferrer", className: "vpg-pro-link", children: "Get Pro License \u2192" })
+          /* @__PURE__ */ jsx8("h3", { children: "Pivot Unavailable" }),
+          /* @__PURE__ */ jsx8("p", { children: "Pivot mode could not be enabled in this session. Try reloading the page." })
         ] }) }) : /* @__PURE__ */ jsxs8(Fragment5, { children: [
           /* @__PURE__ */ jsxs8("div", { className: "vpg-config-bar", children: [
             /* @__PURE__ */ jsxs8(