@sqlrooms/mosaic 0.29.0-rc.1 → 0.29.0-rc.3

package/dist/MosaicSlice.js

@@ -1,40 +1,47 @@
 import { createId } from '@paralleldrive/cuid2';
-import { isWasmDuckDbConnector } from '@sqlrooms/duckdb';
+import { isWasmDuckDbConnector, } from '@sqlrooms/duckdb';
 import { createSlice, useBaseRoomStore, } from '@sqlrooms/room-store';
 import { coordinator, makeClient, Selection, wasmConnector, } from '@uwdata/mosaic-core';
 import { produce } from 'immer';
 import { z } from 'zod';
+import { createMosaicTableFromArrowTable, toArrowClientResult, } from './tableInterop';
 export const MosaicSliceConfig = z.object({});
 export function createDefaultMosaicConfig(props) {
     return {
         ...props,
     };
 }
-export function createMosaicSlice(props) {
+export function createMosaicSlice(props = {}) {
     return createSlice((set, get, store) => ({
         mosaic: {
             config: createDefaultMosaicConfig(props?.config),
-            connection: {
-                status: 'idle',
-                connector: undefined,
-            },
+            connection: { status: 'idle' },
             clients: {},
             selections: {},
             async initialize() {
                 let mosaicConnector;
+                let resolvedCoordinator;
                 set((state) => produce(state, (draft) => {
                     draft.mosaic.connection = { status: 'loading' };
                 }));
                 try {
-                    const dbConnector = await get().db.getConnector();
-                    if (!isWasmDuckDbConnector(dbConnector)) {
-                        throw new Error('Only WasmDuckDbConnector is currently supported');
+                    if (props.coordinator) {
+                        resolvedCoordinator = props.coordinator;
+                        applyMosaicPreAggregateOptions(resolvedCoordinator, props.preagg);
+                    }
+                    else {
+                        const dbConnector = await get().db.getConnector();
+                        resolvedCoordinator = coordinator();
+                        mosaicConnector = isWasmDuckDbConnector(dbConnector)
+                            ? await wasmConnector({
+                                // @ts-expect-error - We install a different version of duckdb-wasm
+                                duckDb: dbConnector.getDb(),
+                                connection: dbConnector.getConnection(),
+                            })
+                            : createDuckDbMosaicConnector(dbConnector);
+                        applyMosaicPreAggregateOptions(resolvedCoordinator, props.preagg);
+                        resolvedCoordinator.databaseConnector(mosaicConnector);
                     }
-                    mosaicConnector = await coordinator().databaseConnector(wasmConnector({
-                        // @ts-expect-error - We install a different version of duckdb-wasm
-                        duckDb: dbConnector.getDb(),
-                        connection: dbConnector.getConnection(),
-                    }));
                 }
                 catch (error) {
                     set((state) => produce(state, (draft) => {
@@ -42,19 +49,22 @@ export function createMosaicSlice(props) {
                     }));
                     throw error;
                 }
-                finally {
-                    set((state) => produce(state, (draft) => {
-                        draft.mosaic.connection = {
-                            status: 'ready',
-                            connector: mosaicConnector,
-                            coordinator: coordinator(),
-                        };
-                    }));
-                }
+                set((state) => produce(state, (draft) => {
+                    draft.mosaic.connection = {
+                        status: 'ready',
+                        connector: mosaicConnector,
+                        coordinator: resolvedCoordinator,
+                    };
+                }));
             },
             async destroy() {
                 get().mosaic.destroyAllClients();
             },
+            setConfig(config) {
+                set((state) => produce(state, (draft) => {
+                    draft.mosaic.config = config;
+                }));
+            },
             getSelection(name, type = 'crossfilter') {
                 const existing = get().mosaic.selections[name];
                 if (existing)
@@ -82,22 +92,44 @@ export function createMosaicSlice(props) {
                         : undefined);
                 // Wrap queryResult to update store state AND call external callback
                 const wrappedQueryResult = (data) => {
-                    const typedData = data;
                     set((state) => produce(state, (draft) => {
                         const tracked = draft.mosaic.clients[id];
                         if (tracked) {
-                            tracked.data = typedData;
+                            tracked.data = data;
                             tracked.isLoading = false;
+                            tracked.error = undefined;
                         }
                     }));
                     // Call external callback if provided
-                    options.queryResult?.(typedData);
+                    options.queryResult?.(toArrowClientResult(data));
+                };
+                const wrappedQueryPending = () => {
+                    set((state) => produce(state, (draft) => {
+                        const tracked = draft.mosaic.clients[id];
+                        if (tracked) {
+                            tracked.isLoading = true;
+                            tracked.error = undefined;
+                        }
+                    }));
+                };
+                const wrappedQueryError = (error) => {
+                    set((state) => produce(state, (draft) => {
+                        const tracked = draft.mosaic.clients[id];
+                        if (tracked) {
+                            tracked.isLoading = false;
+                            tracked.error = error;
+                        }
+                    }));
+                    client.enabled = false;
+                    options.queryError?.(error);
                 };
                 const client = makeClient({
                     coordinator: connection.coordinator,
                     selection,
                     query: options.query,
                     queryResult: wrappedQueryResult,
+                    queryPending: wrappedQueryPending,
+                    queryError: wrappedQueryError,
                 });
                 set((state) => produce(state, (draft) => {
                     draft.mosaic.clients[id] = {
@@ -106,9 +138,10 @@ export function createMosaicSlice(props) {
                         createdAt: Date.now(),
                         isLoading: true,
                         data: null,
+                        error: undefined,
                         selection,
                         queryResultCallback: options.queryResult
-                            ? (result) => options.queryResult(result)
+                            ? (result) => options.queryResult(toArrowClientResult(result))
                             : undefined,
                     };
                 }));
@@ -138,24 +171,48 @@ export function createMosaicSlice(props) {
                 }
                 // Create new client with wrapped queryResult that calls both store update and external callback
                 const wrappedQueryResult = (data) => {
-                    const typedData = data;
                     set((state) => produce(state, (draft) => {
                         const tracked = draft.mosaic.clients[options.id];
                         if (tracked) {
-                            tracked.data = typedData;
+                            tracked.data = data;
                             tracked.isLoading = false;
+                            tracked.error = undefined;
                         }
                     }));
+                    const arrowData = toArrowClientResult(data);
                     // Call external callback if provided
-                    options.onQueryResult?.(typedData);
+                    options.onQueryResult?.(arrowData);
                     // Also call original queryResult if provided
-                    options.queryResult?.(typedData);
+                    options.queryResult?.(arrowData);
+                };
+                const wrappedQueryPending = () => {
+                    set((state) => produce(state, (draft) => {
+                        const tracked = draft.mosaic.clients[options.id];
+                        if (tracked) {
+                            tracked.isLoading = true;
+                            tracked.error = undefined;
+                        }
+                    }));
+                };
+                const wrappedQueryError = (error) => {
+                    set((state) => produce(state, (draft) => {
+                        const tracked = draft.mosaic.clients[options.id];
+                        if (tracked) {
+                            tracked.isLoading = false;
+                            tracked.error = error;
+                        }
+                    }));
+                    client.enabled = false;
+                    options.onQueryError?.(error);
+                    options.queryError?.(error);
                 };
                 const client = makeClient({
                     coordinator: connection.coordinator,
                     selection,
                     query: options.query,
                     queryResult: wrappedQueryResult,
+                    queryPending: wrappedQueryPending,
+                    queryError: wrappedQueryError,
                 });
                 set((state) => produce(state, (draft) => {
                     draft.mosaic.clients[options.id] = {
@@ -164,30 +221,29 @@ export function createMosaicSlice(props) {
                         createdAt: Date.now(),
                         isLoading: true,
                         data: null,
+                        error: undefined,
                         selection,
-                        queryResultCallback: options.onQueryResult,
+                        queryResultCallback: options.onQueryResult
+                            ? (result) => options.onQueryResult(toArrowClientResult(result))
+                            : undefined,
                     };
                 }));
             },
             destroyClient(id) {
-                const { connection, clients } = get().mosaic;
+                const { clients } = get().mosaic;
                 const tracked = clients[id];
                 if (!tracked)
                     return;
-                if (connection.status === 'ready') {
-                    connection.coordinator.disconnect(tracked.client);
-                }
+                tracked.client.destroy();
                 set((state) => produce(state, (draft) => {
                     delete draft.mosaic.clients[id];
                 }));
             },
             destroyAllClients() {
-                const { connection, clients } = get().mosaic;
-                if (connection.status === 'ready') {
-                    Object.values(clients).forEach((tracked) => {
-                        connection.coordinator.disconnect(tracked.client);
-                    });
-                }
+                const { clients } = get().mosaic;
+                Object.values(clients).forEach((tracked) => {
+                    tracked.client.destroy();
+                });
                 set((state) => produce(state, (draft) => {
                     draft.mosaic.clients = {};
                 }));
@@ -195,7 +251,49 @@ export function createMosaicSlice(props) {
         },
     }));
 }
+function applyMosaicPreAggregateOptions(mosaicCoordinator, options) {
+    if (!options) {
+        return;
+    }
+    if (options.schema !== undefined) {
+        mosaicCoordinator.preaggregator.schema = options.schema;
+    }
+    if (options.enabled !== undefined) {
+        mosaicCoordinator.preaggregator.enabled = options.enabled;
+    }
+}
 export function useStoreWithMosaic(selector) {
     return useBaseRoomStore((state) => selector(state));
 }
+/**
+ * Adapts a {@link DuckDbConnector} to the Mosaic {@link Connector} interface.
+ *
+ * For `'arrow'` queries the Apache Arrow table returned by the connector is
+ * converted via {@link createMosaicTableFromArrowTable} into a flechette
+ * `Table`, which is the shape Mosaic consumers expect (with `.toColumns()`).
+ * For `'json'` queries, rows are materialized with {@link Array.from} which
+ * may have performance/memory implications for very large result sets.
+ * The `as any` casts on the return type are an intentional adapter trade-off
+ * to satisfy the polymorphic {@link Connector['query']} signature.
+ */
+function createDuckDbMosaicConnector(connector) {
+    return {
+        query: (async (query) => {
+            const queryType = query.type ?? 'arrow';
+            if (queryType === 'exec') {
+                await connector.execute(query.sql);
+                return undefined;
+            }
+            if (queryType === 'json') {
+                const rows = await connector.queryJson(query.sql);
+                return Array.from(rows);
+            }
+            if (queryType === 'arrow') {
+                const arrowTable = await connector.query(query.sql);
+                return createMosaicTableFromArrowTable(arrowTable);
+            }
+            throw new Error(`Unsupported Mosaic query type "${queryType}".`);
+        }),
+    };
+}
 //# sourceMappingURL=MosaicSlice.js.map

package/dist/MosaicSlice.js.map

@@ -1 +1 @@
-{"version":3,"file":"MosaicSlice.js","sourceRoot":"","sources":["../src/MosaicSlice.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAmB,qBAAqB,EAAC,MAAM,kBAAkB,CAAC;AACzE,OAAO,EAEL,WAAW,EACX,gBAAgB,GAEjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAGL,WAAW,EACX,UAAU,EACV,SAAS,EACT,aAAa,GACd,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAC9B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AA6D9C,MAAM,UAAU,yBAAyB,CACvC,KAAkC;IAElC,OAAO;QACL,GAAG,KAAK;KACY,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAEjC;IACC,OAAO,WAAW,CAGhB,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACtB,MAAM,EAAE;YACN,MAAM,EAAE,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC;YAChD,UAAU,EAAE;gBACV,MAAM,EAAE,MAAM;gBACd,SAAS,EAAE,SAAS;aACrB;YACD,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,EAAE;YAEd,KAAK,CAAC,UAAU;gBACd,IAAI,eAAsC,CAAC;gBAC3C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,EAAC,MAAM,EAAE,SAAS,EAAC,CAAC;gBAChD,CAAC,CAAC,CACH,CAAC;gBACF,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;oBAClD,IAAI,CAAC,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;wBACxC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;oBACrE,CAAC;oBACD,eAAe,GAAG,MAAM,WAAW,EAAE,CAAC,iBAAiB,CACrD,aAAa,CAAC;wBACZ,mEAAmE;wBACnE,MAAM,EAAE,WAAW,CAAC,KAAK,EAAE;wBAC3B,UAAU,EAAE,WAAW,CAAC,aAAa,EAAE;qBACxC,CAAC,CACH,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,EAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAC,CAAC;oBACrD,CAAC,CAAC,CACH,CAAC;oBACF,MAAM,KAAK,CAAC;gBACd,CAAC;wBAAS,CAAC;oBACT,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG;4BACxB,MAAM,EAAE,OAAO;4BACf,SAAS,EAAE,eAAgB;4BAC3B,WAAW,EAAE,WAAW,EAAE;yBAC3B,CAAC;oBACJ,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,KAAK,CAAC,OAAO;gBACX,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YACnC,CAAC;YAED,YAAY,CACV,IAAY,EACZ,OAA2C,aAAa;gBAExD,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC/C,IAAI,QAAQ;oBAAE,OAAO,QAAQ,CAAC;gBAE9B,MAAM,SAAS,GACb,IAAI,KAAK,aAAa;oBACpB,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;oBACzB,CAAC,CAAC,IAAI,KAAK,QAAQ;wBACjB,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE;wBACpB,CAAC,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;gBAE1B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;gBAC5C,CAAC,CAAC,CACH,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,YAAY,CAAI,OAA+B;gBAC7C,MAAM,EAAC,UAAU,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAClC,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,EAAE,GAAG,OAAO,CAAC,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAEpC,mCAAmC;gBACnC,MAAM,SAAS,GACb,OAAO,CAAC,SAAS;oBACjB,CAAC,OAAO,CAAC,aAAa;wBACpB,CAAC,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC;wBAClD,CAAC,CAAC,SAAS,CAAC,CAAC;gBAEjB,oEAAoE;gBACpE,MAAM,kBAAkB,GAAG,CAAC,IAAa,EAAE,EAAE;oBAC3C,MAAM,SAAS,GAAG,IAAS,CAAC;oBAC5B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACzC,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,GAAG,SAAS,CAAC;4BACzB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,qCAAqC;oBACrC,OAAO,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,CAAC;gBACnC,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,UAAU,CAAC;oBACxB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,SAAS;oBACT,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,kBAAkB;iBAChC,CAAC,CAAC;gBAEH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;wBACzB,EAAE;wBACF,MAAM;wBACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,SAAS,EAAE,IAAI;wBACf,IAAI,EAAE,IAAI;wBACV,SAAS;wBACT,mBAAmB,EAAE,OAAO,CAAC,WAAW;4BACtC,CAAC,CAAC,CAAC,MAAe,EAAE,EAAE,CAAC,OAAO,CAAC,WAAY,CAAC,MAAW,CAAC;4BACxD,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;gBAEF,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,YAAY,CACV,OAGC;gBAED,MAAM,EAAC,UAAU,EAAE,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC3C,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,OAAO,CAAC,wDAAwD;gBAClE,CAAC;gBAED,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAErC,mCAAmC;gBACnC,MAAM,SAAS,GACb,OAAO,CAAC,SAAS;oBACjB,CAAC,OAAO,CAAC,aAAa;wBACpB,CAAC,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC;wBAClD,CAAC,CAAC,SAAS,CAAC,CAAC;gBAEjB,+CAA+C;gBAC/C,uEAAuE;gBACvE,sEAAsE;gBACtE,gDAAgD;gBAChD,IAAI,QAAQ,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBACjD,OAAO,CAAC,oDAAoD;gBAC9D,CAAC;gBAED,oDAAoD;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACzC,CAAC;gBAED,gGAAgG;gBAChG,MAAM,kBAAkB,GAAG,CAAC,IAAa,EAAE,EAAE;oBAC3C,MAAM,SAAS,GAAG,IAAS,CAAC;oBAC5B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACjD,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,GAAG,SAAS,CAAC;4BACzB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,qCAAqC;oBACrC,OAAO,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;oBACnC,6CAA6C;oBAC7C,OAAO,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,CAAC;gBACnC,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,UAAU,CAAC;oBACxB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,SAAS;oBACT,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,kBAAkB;iBAChC,CAAC,CAAC;gBAEH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;wBACjC,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,MAAM;wBACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,SAAS,EAAE,IAAI;wBACf,IAAI,EAAE,IAAI;wBACV,SAAS;wBACT,mBAAmB,EAAE,OAAO,CAAC,aAEhB;qBACd,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,aAAa,CAAC,EAAU;gBACtB,MAAM,EAAC,UAAU,EAAE,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC3C,MAAM,OAAO,GAAG,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC5B,IAAI,CAAC,OAAO;oBAAE,OAAO;gBAErB,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpD,CAAC;gBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAClC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,iBAAiB;gBACf,MAAM,EAAC,UAAU,EAAE,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAE3C,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;wBACzC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBACpD,CAAC,CAAC,CAAC;gBACL,CAAC;gBAED,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,EAAE,CAAC;gBAC5B,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;SACF;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAID,MAAM,UAAU,kBAAkB,CAChC,QAAkD;IAElD,OAAO,gBAAgB,CAAwB,CAAC,KAAK,EAAE,EAAE,CACvD,QAAQ,CAAC,KAA8C,CAAC,CACzD,CAAC;AACJ,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {DuckDbSliceState, isWasmDuckDbConnector} from '@sqlrooms/duckdb';\nimport {\n  BaseRoomStoreState,\n  createSlice,\n  useBaseRoomStore,\n  type SliceFunctions,\n} from '@sqlrooms/room-store';\nimport {\n  Connector,\n  Coordinator,\n  coordinator,\n  makeClient,\n  Selection,\n  wasmConnector,\n} from '@uwdata/mosaic-core';\nimport {Query} from '@uwdata/mosaic-sql';\nimport {produce} from 'immer';\nimport {z} from 'zod';\n\nexport const MosaicSliceConfig = z.object({});\nexport type MosaicSliceConfig = z.infer<typeof MosaicSliceConfig>;\n\n// Client configuration options\nexport type MosaicClientOptions<T = unknown> = {\n  /** Unique identifier for this client */\n  id?: string;\n  /** Selection name for cross-filtering (will create if doesn't exist) */\n  selectionName?: string;\n  /** The selection to subscribe to for cross-filtering */\n  selection?: Selection;\n  /** Query builder function that receives the current filter */\n  query: (filter: unknown) => ReturnType<typeof Query.from>;\n  /** Callback when query results are received */\n  queryResult?: (result: T) => void;\n};\n\n// Tracked client info\nexport type TrackedClient<T = unknown> = {\n  id: string;\n  client: ReturnType<typeof makeClient>;\n  createdAt: number;\n  isLoading: boolean;\n  data: T | null;\n  selection?: Selection; // Track for change detection\n  queryResultCallback?: (result: T) => void; // External callback\n};\n\nexport type MosaicSliceState = {\n  mosaic: SliceFunctions & {\n    connection:\n      | {status: 'idle' | 'loading'}\n      | {status: 'ready'; connector: Connector; coordinator: Coordinator}\n      | {status: 'error'; error: unknown};\n    config: MosaicSliceConfig;\n    /** Record of registered clients by id */\n    clients: Record<string, TrackedClient<unknown>>;\n    /** Named selections for cross-filtering (e.g., 'brush', 'hover') */\n    selections: Record<string, Selection>;\n    initialize: () => Promise<void>;\n    /** Get or create a named selection for cross-filtering */\n    getSelection: (\n      name: string,\n      type?: 'crossfilter' | 'single' | 'union',\n    ) => Selection;\n    /** Create a mosaic client and register it */\n    createClient: <T>(options: MosaicClientOptions<T>) => string;\n    /** Ensure a client exists with given options (idempotent - creates or updates as needed) */\n    ensureClient: <T>(\n      options: MosaicClientOptions<T> & {\n        id: string;\n        onQueryResult?: (result: T) => void;\n      },\n    ) => void;\n    /** Disconnect and remove a client by id */\n    destroyClient: (id: string) => void;\n    /** Disconnect all clients (useful for cleanup) */\n    destroyAllClients: () => void;\n  };\n};\n\nexport function createDefaultMosaicConfig(\n  props?: Partial<MosaicSliceConfig>,\n): MosaicSliceConfig {\n  return {\n    ...props,\n  } as MosaicSliceConfig;\n}\n\nexport function createMosaicSlice(props?: {\n  config?: Partial<MosaicSliceConfig>;\n}) {\n  return createSlice<\n    MosaicSliceState,\n    BaseRoomStoreState & DuckDbSliceState & MosaicSliceState\n  >((set, get, store) => ({\n    mosaic: {\n      config: createDefaultMosaicConfig(props?.config),\n      connection: {\n        status: 'idle',\n        connector: undefined,\n      },\n      clients: {},\n      selections: {},\n\n      async initialize() {\n        let mosaicConnector: Connector | undefined;\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.connection = {status: 'loading'};\n          }),\n        );\n        try {\n          const dbConnector = await get().db.getConnector();\n          if (!isWasmDuckDbConnector(dbConnector)) {\n            throw new Error('Only WasmDuckDbConnector is currently supported');\n          }\n          mosaicConnector = await coordinator().databaseConnector(\n            wasmConnector({\n              // @ts-expect-error - We install a different version of duckdb-wasm\n              duckDb: dbConnector.getDb(),\n              connection: dbConnector.getConnection(),\n            }),\n          );\n        } catch (error) {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.mosaic.connection = {status: 'error', error};\n            }),\n          );\n          throw error;\n        } finally {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.mosaic.connection = {\n                status: 'ready',\n                connector: mosaicConnector!,\n                coordinator: coordinator(),\n              };\n            }),\n          );\n        }\n      },\n\n      async destroy() {\n        get().mosaic.destroyAllClients();\n      },\n\n      getSelection(\n        name: string,\n        type: 'crossfilter' | 'single' | 'union' = 'crossfilter',\n      ) {\n        const existing = get().mosaic.selections[name];\n        if (existing) return existing;\n\n        const selection =\n          type === 'crossfilter'\n            ? Selection.crossfilter()\n            : type === 'single'\n              ? Selection.single()\n              : Selection.union();\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.selections[name] = selection;\n          }),\n        );\n        return selection;\n      },\n\n      createClient<T>(options: MosaicClientOptions<T>) {\n        const {connection} = get().mosaic;\n        if (connection.status !== 'ready') {\n          throw new Error('Mosaic connection not ready');\n        }\n\n        const id = options.id ?? createId();\n\n        // Determine which selection to use\n        const selection =\n          options.selection ??\n          (options.selectionName\n            ? get().mosaic.getSelection(options.selectionName)\n            : undefined);\n\n        // Wrap queryResult to update store state AND call external callback\n        const wrappedQueryResult = (data: unknown) => {\n          const typedData = data as T;\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[id];\n              if (tracked) {\n                tracked.data = typedData;\n                tracked.isLoading = false;\n              }\n            }),\n          );\n          // Call external callback if provided\n          options.queryResult?.(typedData);\n        };\n\n        const client = makeClient({\n          coordinator: connection.coordinator,\n          selection,\n          query: options.query,\n          queryResult: wrappedQueryResult,\n        });\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients[id] = {\n              id,\n              client,\n              createdAt: Date.now(),\n              isLoading: true,\n              data: null,\n              selection,\n              queryResultCallback: options.queryResult\n                ? (result: unknown) => options.queryResult!(result as T)\n                : undefined,\n            };\n          }),\n        );\n\n        return id;\n      },\n\n      ensureClient<T>(\n        options: MosaicClientOptions<T> & {\n          id: string;\n          onQueryResult?: (result: T) => void;\n        },\n      ) {\n        const {connection, clients} = get().mosaic;\n        if (connection.status !== 'ready') {\n          return; // Silently return if not ready - hook will handle retry\n        }\n\n        const existing = clients[options.id];\n\n        // Determine which selection to use\n        const selection =\n          options.selection ??\n          (options.selectionName\n            ? get().mosaic.getSelection(options.selectionName)\n            : undefined);\n\n        // Check if client exists and selection matches\n        // Note: If query or callback changes, we recreate the client to ensure\n        // the latest versions are used. This is simpler than trying to update\n        // the bound queryResult callback in makeClient.\n        if (existing && existing.selection === selection) {\n          return; // No-op - client already exists with same selection\n        }\n\n        // If exists but selection changed, destroy it first\n        if (existing) {\n          get().mosaic.destroyClient(options.id);\n        }\n\n        // Create new client with wrapped queryResult that calls both store update and external callback\n        const wrappedQueryResult = (data: unknown) => {\n          const typedData = data as T;\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[options.id];\n              if (tracked) {\n                tracked.data = typedData;\n                tracked.isLoading = false;\n              }\n            }),\n          );\n          // Call external callback if provided\n          options.onQueryResult?.(typedData);\n          // Also call original queryResult if provided\n          options.queryResult?.(typedData);\n        };\n\n        const client = makeClient({\n          coordinator: connection.coordinator,\n          selection,\n          query: options.query,\n          queryResult: wrappedQueryResult,\n        });\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients[options.id] = {\n              id: options.id,\n              client,\n              createdAt: Date.now(),\n              isLoading: true,\n              data: null,\n              selection,\n              queryResultCallback: options.onQueryResult as\n                | ((result: unknown) => void)\n                | undefined,\n            };\n          }),\n        );\n      },\n\n      destroyClient(id: string) {\n        const {connection, clients} = get().mosaic;\n        const tracked = clients[id];\n        if (!tracked) return;\n\n        if (connection.status === 'ready') {\n          connection.coordinator.disconnect(tracked.client);\n        }\n\n        set((state) =>\n          produce(state, (draft) => {\n            delete draft.mosaic.clients[id];\n          }),\n        );\n      },\n\n      destroyAllClients() {\n        const {connection, clients} = get().mosaic;\n\n        if (connection.status === 'ready') {\n          Object.values(clients).forEach((tracked) => {\n            connection.coordinator.disconnect(tracked.client);\n          });\n        }\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients = {};\n          }),\n        );\n      },\n    },\n  }));\n}\n\nexport type DuckDbSliceStateWithMosaic = DuckDbSliceState & MosaicSliceState;\n\nexport function useStoreWithMosaic<T>(\n  selector: (state: DuckDbSliceStateWithMosaic) => T,\n): T {\n  return useBaseRoomStore<BaseRoomStoreState, T>((state) =>\n    selector(state as unknown as DuckDbSliceStateWithMosaic),\n  );\n}\n"]}
+{"version":3,"file":"MosaicSlice.js","sourceRoot":"","sources":["../src/MosaicSlice.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EACL,qBAAqB,GAGtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,WAAW,EACX,gBAAgB,GAEjB,MAAM,sBAAsB,CAAC;AAM9B,OAAO,EAGL,WAAW,EACX,UAAU,EACV,SAAS,EACT,aAAa,GACd,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAC9B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACL,+BAA+B,EAC/B,mBAAmB,GACpB,MAAM,gBAAgB,CAAC;AAExB,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAwE9C,MAAM,UAAU,yBAAyB,CACvC,KAAkC;IAElC,OAAO;QACL,GAAG,KAAK;KACY,CAAC;AACzB,CAAC;AAQD,MAAM,UAAU,iBAAiB,CAAC,QAAgC,EAAE;IAClE,OAAO,WAAW,CAGhB,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACtB,MAAM,EAAE;YACN,MAAM,EAAE,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC;YAChD,UAAU,EAAE,EAAC,MAAM,EAAE,MAAM,EAAC;YAC5B,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,EAAE;YAEd,KAAK,CAAC,UAAU;gBACd,IAAI,eAAsC,CAAC;gBAC3C,IAAI,mBAAiC,CAAC;gBACtC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,EAAC,MAAM,EAAE,SAAS,EAAC,CAAC;gBAChD,CAAC,CAAC,CACH,CAAC;gBACF,IAAI,CAAC;oBACH,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;wBACtB,mBAAmB,GAAG,KAAK,CAAC,WAAW,CAAC;wBACxC,8BAA8B,CAAC,mBAAmB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;oBACpE,CAAC;yBAAM,CAAC;wBACN,MAAM,WAAW,GAAG,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;wBAClD,mBAAmB,GAAG,WAAW,EAAE,CAAC;wBACpC,eAAe,GAAG,qBAAqB,CAAC,WAAW,CAAC;4BAClD,CAAC,CAAC,MAAM,aAAa,CAAC;gCAClB,mEAAmE;gCACnE,MAAM,EAAE,WAAW,CAAC,KAAK,EAAE;gCAC3B,UAAU,EAAE,WAAW,CAAC,aAAa,EAAE;6BACxC,CAAC;4BACJ,CAAC,CAAC,2BAA2B,CAAC,WAAW,CAAC,CAAC;wBAC7C,8BAA8B,CAAC,mBAAmB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBAClE,mBAAmB,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;oBACzD,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,EAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAC,CAAC;oBACrD,CAAC,CAAC,CACH,CAAC;oBACF,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG;wBACxB,MAAM,EAAE,OAAO;wBACf,SAAS,EAAE,eAAe;wBAC1B,WAAW,EAAE,mBAAmB;qBACjC,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,KAAK,CAAC,OAAO;gBACX,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YACnC,CAAC;YAED,SAAS,CAAC,MAAyB;gBACjC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;gBAC/B,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,YAAY,CACV,IAAY,EACZ,OAA2C,aAAa;gBAExD,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC/C,IAAI,QAAQ;oBAAE,OAAO,QAAQ,CAAC;gBAE9B,MAAM,SAAS,GACb,IAAI,KAAK,aAAa;oBACpB,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;oBACzB,CAAC,CAAC,IAAI,KAAK,QAAQ;wBACjB,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE;wBACpB,CAAC,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;gBAE1B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;gBAC5C,CAAC,CAAC,CACH,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,YAAY,CAAC,OAA4B;gBACvC,MAAM,EAAC,UAAU,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAClC,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,EAAE,GAAG,OAAO,CAAC,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAEpC,mCAAmC;gBACnC,MAAM,SAAS,GACb,OAAO,CAAC,SAAS;oBACjB,CAAC,OAAO,CAAC,aAAa;wBACpB,CAAC,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC;wBAClD,CAAC,CAAC,SAAS,CAAC,CAAC;gBAEjB,oEAAoE;gBACpE,MAAM,kBAAkB,GAAG,CAAC,IAAa,EAAE,EAAE;oBAC3C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACzC,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;4BACpB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;4BAC1B,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,qCAAqC;oBACrC,OAAO,CAAC,WAAW,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACnD,CAAC,CAAC;gBACF,MAAM,mBAAmB,GAAG,GAAG,EAAE;oBAC/B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACzC,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;4BACzB,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC,CAAC;gBACF,MAAM,iBAAiB,GAAG,CAAC,KAAY,EAAE,EAAE;oBACzC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACzC,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;4BAC1B,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;wBACxB,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;oBACvB,OAAO,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,UAAU,CAAC;oBACxB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,SAAS;oBACT,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,kBAAkB;oBAC/B,YAAY,EAAE,mBAAmB;oBACjC,UAAU,EAAE,iBAAiB;iBAC9B,CAAC,CAAC;gBAEH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;wBACzB,EAAE;wBACF,MAAM;wBACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,SAAS,EAAE,IAAI;wBACf,IAAI,EAAE,IAAI;wBACV,KAAK,EAAE,SAAS;wBAChB,SAAS;wBACT,mBAAmB,EAAE,OAAO,CAAC,WAAW;4BACtC,CAAC,CAAC,CAAC,MAAe,EAAE,EAAE,CAClB,OAAO,CAAC,WAAY,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;4BACrD,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;gBAEF,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,YAAY,CACV,OAIC;gBAED,MAAM,EAAC,UAAU,EAAE,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC3C,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAClC,OAAO,CAAC,wDAAwD;gBAClE,CAAC;gBAED,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAErC,mCAAmC;gBACnC,MAAM,SAAS,GACb,OAAO,CAAC,SAAS;oBACjB,CAAC,OAAO,CAAC,aAAa;wBACpB,CAAC,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC;wBAClD,CAAC,CAAC,SAAS,CAAC,CAAC;gBAEjB,+CAA+C;gBAC/C,uEAAuE;gBACvE,sEAAsE;gBACtE,gDAAgD;gBAChD,IAAI,QAAQ,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBACjD,OAAO,CAAC,oDAAoD;gBAC9D,CAAC;gBAED,oDAAoD;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACzC,CAAC;gBAED,gGAAgG;gBAChG,MAAM,kBAAkB,GAAG,CAAC,IAAa,EAAE,EAAE;oBAC3C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACjD,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;4BACpB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;4BAC1B,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,MAAM,SAAS,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;oBAC5C,qCAAqC;oBACrC,OAAO,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;oBACnC,6CAA6C;oBAC7C,OAAO,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,CAAC;gBACnC,CAAC,CAAC;gBACF,MAAM,mBAAmB,GAAG,GAAG,EAAE;oBAC/B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACjD,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;4BACzB,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;wBAC5B,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC,CAAC;gBACF,MAAM,iBAAiB,GAAG,CAAC,KAAY,EAAE,EAAE;oBACzC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;wBACjD,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;4BAC1B,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;wBACxB,CAAC;oBACH,CAAC,CAAC,CACH,CAAC;oBACF,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;oBACvB,OAAO,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC;oBAC9B,OAAO,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,UAAU,CAAC;oBACxB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,SAAS;oBACT,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,kBAAkB;oBAC/B,YAAY,EAAE,mBAAmB;oBACjC,UAAU,EAAE,iBAAiB;iBAC9B,CAAC,CAAC;gBAEH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;wBACjC,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,MAAM;wBACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,SAAS,EAAE,IAAI;wBACf,IAAI,EAAE,IAAI;wBACV,KAAK,EAAE,SAAS;wBAChB,SAAS;wBACT,mBAAmB,EAAE,OAAO,CAAC,aAAa;4BACxC,CAAC,CAAC,CAAC,MAAe,EAAE,EAAE,CAClB,OAAO,CAAC,aAAc,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;4BACvD,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,aAAa,CAAC,EAAU;gBACtB,MAAM,EAAC,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM,OAAO,GAAG,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC5B,IAAI,CAAC,OAAO;oBAAE,OAAO;gBAErB,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAEzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAClC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,iBAAiB;gBACf,MAAM,EAAC,OAAO,EAAC,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBACzC,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC3B,CAAC,CAAC,CAAC;gBAEH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,EAAE,CAAC;gBAC5B,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;SACF;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,8BAA8B,CACrC,iBAA8B,EAC9B,OAAmC;IAEnC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;IACT,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QACjC,iBAAiB,CAAC,aAAa,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC1D,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,iBAAiB,CAAC,aAAa,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAC5D,CAAC;AACH,CAAC;AAID,MAAM,UAAU,kBAAkB,CAChC,QAAkD;IAElD,OAAO,gBAAgB,CAAwB,CAAC,KAAK,EAAE,EAAE,CACvD,QAAQ,CAAC,KAA8C,CAAC,CACzD,CAAC;AACJ,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,2BAA2B,CAAC,SAA0B;IAC7D,OAAO;QACL,KAAK,EAAE,CAAC,KAAK,EACX,KAA8D,EAC9D,EAAE;YACF,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,IAAI,OAAO,CAAC;YACxC,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;gBACzB,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACnC,OAAO,SAAgB,CAAC;YAC1B,CAAC;YACD,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,SAAS,CACpC,KAAK,CAAC,GAAG,CACV,CAAC;gBACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAQ,CAAC;YACjC,CAAC;YACD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;gBAC1B,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACpD,OAAO,+BAA+B,CAAC,UAAU,CAAQ,CAAC;YAC5D,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,kCAAkC,SAAS,IAAI,CAAC,CAAC;QACnE,CAAC,CAAuB;KACzB,CAAC;AACJ,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {\n  isWasmDuckDbConnector,\n  type DuckDbConnector,\n  type DuckDbSliceState,\n} from '@sqlrooms/duckdb';\nimport {\n  BaseRoomStoreState,\n  createSlice,\n  useBaseRoomStore,\n  type SliceFunctions,\n} from '@sqlrooms/room-store';\nimport type {\n  ArrowQueryRequest,\n  ExecQueryRequest,\n  JSONQueryRequest,\n} from '@uwdata/mosaic-core';\nimport {\n  Connector,\n  Coordinator,\n  coordinator,\n  makeClient,\n  Selection,\n  wasmConnector,\n} from '@uwdata/mosaic-core';\nimport {Query} from '@uwdata/mosaic-sql';\nimport type {Table as ArrowTable} from 'apache-arrow';\nimport {produce} from 'immer';\nimport {z} from 'zod';\nimport {\n  createMosaicTableFromArrowTable,\n  toArrowClientResult,\n} from './tableInterop';\n\nexport const MosaicSliceConfig = z.object({});\nexport type MosaicSliceConfig = z.infer<typeof MosaicSliceConfig>;\n\nexport type MosaicPreAggregateOptions = {\n  /** Database schema/namespace for Mosaic pre-aggregate tables. */\n  schema?: string;\n  /** Enable or disable Mosaic's pre-aggregation optimization. */\n  enabled?: boolean;\n};\n\n// Client configuration options\nexport type MosaicClientOptions = {\n  /** Unique identifier for this client */\n  id?: string;\n  /** Selection name for cross-filtering (will create if doesn't exist) */\n  selectionName?: string;\n  /** The selection to subscribe to for cross-filtering */\n  selection?: Selection;\n  /** Query builder function that receives the current filter */\n  query: (filter: unknown) => ReturnType<typeof Query.from>;\n  /** Callback when query results are received */\n  queryResult?: (result: ArrowTable) => void;\n  /** Callback when query execution fails */\n  queryError?: (error: Error) => void;\n};\n\n// Tracked client info\nexport type TrackedClient = {\n  id: string;\n  client: ReturnType<typeof makeClient>;\n  createdAt: number;\n  isLoading: boolean;\n  data: unknown | null;\n  error?: Error;\n  selection?: Selection; // Track for change detection\n  queryResultCallback?: (result: ArrowTable) => void; // External callback\n};\n\nexport type MosaicSliceState = {\n  mosaic: SliceFunctions & {\n    connection:\n      | {status: 'idle' | 'loading'}\n      | {status: 'ready'; connector?: Connector; coordinator: Coordinator}\n      | {status: 'error'; error: unknown};\n    config: MosaicSliceConfig;\n    /** Record of registered clients by id */\n    clients: Record<string, TrackedClient>;\n    /** Named selections for cross-filtering (e.g., 'brush', 'hover') */\n    selections: Record<string, Selection>;\n    initialize: () => Promise<void>;\n    /** Get or create a named selection for cross-filtering */\n    getSelection: (\n      name: string,\n      type?: 'crossfilter' | 'single' | 'union',\n    ) => Selection;\n    /** Create a mosaic client and register it */\n    createClient: (options: MosaicClientOptions) => string;\n    /** Ensure a client exists with given options (idempotent - creates or updates as needed) */\n    ensureClient: (\n      options: MosaicClientOptions & {\n        id: string;\n        onQueryResult?: (result: ArrowTable) => void;\n        onQueryError?: (error: Error) => void;\n      },\n    ) => void;\n    /** Disconnect and remove a client by id */\n    destroyClient: (id: string) => void;\n    /** Disconnect all clients (useful for cleanup) */\n    destroyAllClients: () => void;\n  };\n};\n\nexport function createDefaultMosaicConfig(\n  props?: Partial<MosaicSliceConfig>,\n): MosaicSliceConfig {\n  return {\n    ...props,\n  } as MosaicSliceConfig;\n}\n\nexport type CreateMosaicSliceProps = {\n  config?: Partial<MosaicSliceConfig>;\n  coordinator?: Coordinator;\n  preagg?: MosaicPreAggregateOptions;\n};\n\nexport function createMosaicSlice(props: CreateMosaicSliceProps = {}) {\n  return createSlice<\n    MosaicSliceState,\n    BaseRoomStoreState & DuckDbSliceState & MosaicSliceState\n  >((set, get, store) => ({\n    mosaic: {\n      config: createDefaultMosaicConfig(props?.config),\n      connection: {status: 'idle'},\n      clients: {},\n      selections: {},\n\n      async initialize() {\n        let mosaicConnector: Connector | undefined;\n        let resolvedCoordinator!: Coordinator;\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.connection = {status: 'loading'};\n          }),\n        );\n        try {\n          if (props.coordinator) {\n            resolvedCoordinator = props.coordinator;\n            applyMosaicPreAggregateOptions(resolvedCoordinator, props.preagg);\n          } else {\n            const dbConnector = await get().db.getConnector();\n            resolvedCoordinator = coordinator();\n            mosaicConnector = isWasmDuckDbConnector(dbConnector)\n              ? await wasmConnector({\n                  // @ts-expect-error - We install a different version of duckdb-wasm\n                  duckDb: dbConnector.getDb(),\n                  connection: dbConnector.getConnection(),\n                })\n              : createDuckDbMosaicConnector(dbConnector);\n            applyMosaicPreAggregateOptions(resolvedCoordinator, props.preagg);\n            resolvedCoordinator.databaseConnector(mosaicConnector);\n          }\n        } catch (error) {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.mosaic.connection = {status: 'error', error};\n            }),\n          );\n          throw error;\n        }\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.connection = {\n              status: 'ready',\n              connector: mosaicConnector,\n              coordinator: resolvedCoordinator,\n            };\n          }),\n        );\n      },\n\n      async destroy() {\n        get().mosaic.destroyAllClients();\n      },\n\n      setConfig(config: MosaicSliceConfig) {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.config = config;\n          }),\n        );\n      },\n\n      getSelection(\n        name: string,\n        type: 'crossfilter' | 'single' | 'union' = 'crossfilter',\n      ) {\n        const existing = get().mosaic.selections[name];\n        if (existing) return existing;\n\n        const selection =\n          type === 'crossfilter'\n            ? Selection.crossfilter()\n            : type === 'single'\n              ? Selection.single()\n              : Selection.union();\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.selections[name] = selection;\n          }),\n        );\n        return selection;\n      },\n\n      createClient(options: MosaicClientOptions) {\n        const {connection} = get().mosaic;\n        if (connection.status !== 'ready') {\n          throw new Error('Mosaic connection not ready');\n        }\n\n        const id = options.id ?? createId();\n\n        // Determine which selection to use\n        const selection =\n          options.selection ??\n          (options.selectionName\n            ? get().mosaic.getSelection(options.selectionName)\n            : undefined);\n\n        // Wrap queryResult to update store state AND call external callback\n        const wrappedQueryResult = (data: unknown) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[id];\n              if (tracked) {\n                tracked.data = data;\n                tracked.isLoading = false;\n                tracked.error = undefined;\n              }\n            }),\n          );\n          // Call external callback if provided\n          options.queryResult?.(toArrowClientResult(data));\n        };\n        const wrappedQueryPending = () => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[id];\n              if (tracked) {\n                tracked.isLoading = true;\n                tracked.error = undefined;\n              }\n            }),\n          );\n        };\n        const wrappedQueryError = (error: Error) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[id];\n              if (tracked) {\n                tracked.isLoading = false;\n                tracked.error = error;\n              }\n            }),\n          );\n          client.enabled = false;\n          options.queryError?.(error);\n        };\n\n        const client = makeClient({\n          coordinator: connection.coordinator,\n          selection,\n          query: options.query,\n          queryResult: wrappedQueryResult,\n          queryPending: wrappedQueryPending,\n          queryError: wrappedQueryError,\n        });\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients[id] = {\n              id,\n              client,\n              createdAt: Date.now(),\n              isLoading: true,\n              data: null,\n              error: undefined,\n              selection,\n              queryResultCallback: options.queryResult\n                ? (result: unknown) =>\n                    options.queryResult!(toArrowClientResult(result))\n                : undefined,\n            };\n          }),\n        );\n\n        return id;\n      },\n\n      ensureClient(\n        options: MosaicClientOptions & {\n          id: string;\n          onQueryResult?: (result: ArrowTable) => void;\n          onQueryError?: (error: Error) => void;\n        },\n      ) {\n        const {connection, clients} = get().mosaic;\n        if (connection.status !== 'ready') {\n          return; // Silently return if not ready - hook will handle retry\n        }\n\n        const existing = clients[options.id];\n\n        // Determine which selection to use\n        const selection =\n          options.selection ??\n          (options.selectionName\n            ? get().mosaic.getSelection(options.selectionName)\n            : undefined);\n\n        // Check if client exists and selection matches\n        // Note: If query or callback changes, we recreate the client to ensure\n        // the latest versions are used. This is simpler than trying to update\n        // the bound queryResult callback in makeClient.\n        if (existing && existing.selection === selection) {\n          return; // No-op - client already exists with same selection\n        }\n\n        // If exists but selection changed, destroy it first\n        if (existing) {\n          get().mosaic.destroyClient(options.id);\n        }\n\n        // Create new client with wrapped queryResult that calls both store update and external callback\n        const wrappedQueryResult = (data: unknown) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[options.id];\n              if (tracked) {\n                tracked.data = data;\n                tracked.isLoading = false;\n                tracked.error = undefined;\n              }\n            }),\n          );\n          const arrowData = toArrowClientResult(data);\n          // Call external callback if provided\n          options.onQueryResult?.(arrowData);\n          // Also call original queryResult if provided\n          options.queryResult?.(arrowData);\n        };\n        const wrappedQueryPending = () => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[options.id];\n              if (tracked) {\n                tracked.isLoading = true;\n                tracked.error = undefined;\n              }\n            }),\n          );\n        };\n        const wrappedQueryError = (error: Error) => {\n          set((state) =>\n            produce(state, (draft) => {\n              const tracked = draft.mosaic.clients[options.id];\n              if (tracked) {\n                tracked.isLoading = false;\n                tracked.error = error;\n              }\n            }),\n          );\n          client.enabled = false;\n          options.onQueryError?.(error);\n          options.queryError?.(error);\n        };\n\n        const client = makeClient({\n          coordinator: connection.coordinator,\n          selection,\n          query: options.query,\n          queryResult: wrappedQueryResult,\n          queryPending: wrappedQueryPending,\n          queryError: wrappedQueryError,\n        });\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients[options.id] = {\n              id: options.id,\n              client,\n              createdAt: Date.now(),\n              isLoading: true,\n              data: null,\n              error: undefined,\n              selection,\n              queryResultCallback: options.onQueryResult\n                ? (result: unknown) =>\n                    options.onQueryResult!(toArrowClientResult(result))\n                : undefined,\n            };\n          }),\n        );\n      },\n\n      destroyClient(id: string) {\n        const {clients} = get().mosaic;\n        const tracked = clients[id];\n        if (!tracked) return;\n\n        tracked.client.destroy();\n\n        set((state) =>\n          produce(state, (draft) => {\n            delete draft.mosaic.clients[id];\n          }),\n        );\n      },\n\n      destroyAllClients() {\n        const {clients} = get().mosaic;\n        Object.values(clients).forEach((tracked) => {\n          tracked.client.destroy();\n        });\n\n        set((state) =>\n          produce(state, (draft) => {\n            draft.mosaic.clients = {};\n          }),\n        );\n      },\n    },\n  }));\n}\n\nfunction applyMosaicPreAggregateOptions(\n  mosaicCoordinator: Coordinator,\n  options?: MosaicPreAggregateOptions,\n) {\n  if (!options) {\n    return;\n  }\n  if (options.schema !== undefined) {\n    mosaicCoordinator.preaggregator.schema = options.schema;\n  }\n  if (options.enabled !== undefined) {\n    mosaicCoordinator.preaggregator.enabled = options.enabled;\n  }\n}\n\nexport type DuckDbSliceStateWithMosaic = DuckDbSliceState & MosaicSliceState;\n\nexport function useStoreWithMosaic<T>(\n  selector: (state: DuckDbSliceStateWithMosaic) => T,\n): T {\n  return useBaseRoomStore<BaseRoomStoreState, T>((state) =>\n    selector(state as unknown as DuckDbSliceStateWithMosaic),\n  );\n}\n\n/**\n * Adapts a {@link DuckDbConnector} to the Mosaic {@link Connector} interface.\n *\n * For `'arrow'` queries the Apache Arrow table returned by the connector is\n * converted via {@link createMosaicTableFromArrowTable} into a flechette\n * `Table`, which is the shape Mosaic consumers expect (with `.toColumns()`).\n * For `'json'` queries, rows are materialized with {@link Array.from} which\n * may have performance/memory implications for very large result sets.\n * The `as any` casts on the return type are an intentional adapter trade-off\n * to satisfy the polymorphic {@link Connector['query']} signature.\n */\nfunction createDuckDbMosaicConnector(connector: DuckDbConnector): Connector {\n  return {\n    query: (async (\n      query: ArrowQueryRequest | ExecQueryRequest | JSONQueryRequest,\n    ) => {\n      const queryType = query.type ?? 'arrow';\n      if (queryType === 'exec') {\n        await connector.execute(query.sql);\n        return undefined as any;\n      }\n      if (queryType === 'json') {\n        const rows = await connector.queryJson<Record<string, unknown>>(\n          query.sql,\n        );\n        return Array.from(rows) as any;\n      }\n      if (queryType === 'arrow') {\n        const arrowTable = await connector.query(query.sql);\n        return createMosaicTableFromArrowTable(arrowTable) as any;\n      }\n      throw new Error(`Unsupported Mosaic query type \"${queryType}\".`);\n    }) as Connector['query'],\n  };\n}\n"]}

package/dist/ResponsivePlot.d.ts

@@ -0,0 +1,23 @@
+import { PropsWithChildren } from 'react';
+export type PlotSize = {
+    width: number;
+    height: number;
+};
+export type ResponsivePlotProps = PropsWithChildren<{
+    onResize: (size: PlotSize) => void;
+    debounceMs?: number;
+    className?: string;
+}>;
+/**
+ * Container component that measures its size and notifies parent via callback.
+ * Uses ResizeObserver with debouncing to reduce update frequency.
+ * Exposes ref to the container div for direct DOM access.
+ */
+export declare const ResponsivePlot: import("react").ForwardRefExoticComponent<{
+    onResize: (size: PlotSize) => void;
+    debounceMs?: number;
+    className?: string;
+} & {
+    children?: import("react").ReactNode | undefined;
+} & import("react").RefAttributes<HTMLDivElement>>;
+//# sourceMappingURL=ResponsivePlot.d.ts.map

package/dist/ResponsivePlot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ResponsivePlot.d.ts","sourceRoot":"","sources":["../src/ResponsivePlot.tsx"],"names":[],"mappings":"AACA,OAAO,EAEL,iBAAiB,EAIlB,MAAM,OAAO,CAAC;AAEf,MAAM,MAAM,QAAQ,GAAG;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;IAClD,QAAQ,EAAE,CAAC,IAAI,EAAE,QAAQ,KAAK,IAAI,CAAC;IACnC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC,CAAC;AAEH;;;;GAIG;AACH,eAAO,MAAM,cAAc;cAVf,CAAC,IAAI,EAAE,QAAQ,KAAK,IAAI;iBACrB,MAAM;gBACP,MAAM;;;kDAqDnB,CAAC"}

package/dist/ResponsivePlot.js

@@ -0,0 +1,43 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { cn, useDebouncedCallback } from '@sqlrooms/ui';
+import { forwardRef, useEffect, useImperativeHandle, useRef, } from 'react';
+/**
+ * Container component that measures its size and notifies parent via callback.
+ * Uses ResizeObserver with debouncing to reduce update frequency.
+ * Exposes ref to the container div for direct DOM access.
+ */
+export const ResponsivePlot = forwardRef(({ onResize, debounceMs = 150, className, children }, ref) => {
+    const containerRef = useRef(null);
+    const lastSizeRef = useRef(null);
+    // Expose the internal ref to parent
+    useImperativeHandle(ref, () => containerRef.current, []);
+    // Debounced callback to reduce re-render frequency during resizing
+    const onResizeDebounced = useDebouncedCallback((size) => {
+        // Only call onResize if size actually changed
+        if (!lastSizeRef.current ||
+            lastSizeRef.current.width !== size.width ||
+            lastSizeRef.current.height !== size.height) {
+            lastSizeRef.current = size;
+            onResize(size);
+        }
+    }, debounceMs);
+    // Measure container size with ResizeObserver
+    useEffect(() => {
+        const container = containerRef.current;
+        if (!container)
+            return;
+        const resizeObserver = new ResizeObserver((entries) => {
+            for (const entry of entries) {
+                const { width, height } = entry.contentRect;
+                if (width > 0 && height > 0) {
+                    onResizeDebounced({ width, height });
+                }
+            }
+        });
+        resizeObserver.observe(container);
+        return () => resizeObserver.disconnect();
+    }, [onResizeDebounced]);
+    return (_jsx("div", { ref: containerRef, className: cn('h-full w-full', className), children: children }));
+});
+ResponsivePlot.displayName = 'ResponsivePlot';
+//# sourceMappingURL=ResponsivePlot.js.map

package/dist/ResponsivePlot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ResponsivePlot.js","sourceRoot":"","sources":["../src/ResponsivePlot.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,EAAE,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AACtD,OAAO,EACL,UAAU,EAEV,SAAS,EACT,mBAAmB,EACnB,MAAM,GACP,MAAM,OAAO,CAAC;AAaf;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CACtC,CAAC,EAAC,QAAQ,EAAE,UAAU,GAAG,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAC,EAAE,GAAG,EAAE,EAAE;IACzD,MAAM,YAAY,GAAG,MAAM,CAAiB,IAAI,CAAC,CAAC;IAClD,MAAM,WAAW,GAAG,MAAM,CAAkB,IAAI,CAAC,CAAC;IAElD,oCAAoC;IACpC,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,OAAQ,EAAE,EAAE,CAAC,CAAC;IAE1D,mEAAmE;IACnE,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,CAAC,IAAc,EAAE,EAAE;QAChE,8CAA8C;QAC9C,IACE,CAAC,WAAW,CAAC,OAAO;YACpB,WAAW,CAAC,OAAO,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK;YACxC,WAAW,CAAC,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAC1C,CAAC;YACD,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC,EAAE,UAAU,CAAC,CAAC;IAEf,6CAA6C;IAC7C,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;QACvC,IAAI,CAAC,SAAS;YAAE,OAAO;QAEvB,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,CAAC,OAAO,EAAE,EAAE;YACpD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,EAAC,KAAK,EAAE,MAAM,EAAC,GAAG,KAAK,CAAC,WAAW,CAAC;gBAC1C,IAAI,KAAK,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC5B,iBAAiB,CAAC,EAAC,KAAK,EAAE,MAAM,EAAC,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAExB,OAAO,CACL,cAAK,GAAG,EAAE,YAAY,EAAE,SAAS,EAAE,EAAE,CAAC,eAAe,EAAE,SAAS,CAAC,YAC9D,QAAQ,GACL,CACP,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,cAAc,CAAC,WAAW,GAAG,gBAAgB,CAAC","sourcesContent":["import {cn, useDebouncedCallback} from '@sqlrooms/ui';\nimport {\n  forwardRef,\n  PropsWithChildren,\n  useEffect,\n  useImperativeHandle,\n  useRef,\n} from 'react';\n\nexport type PlotSize = {\n  width: number;\n  height: number;\n};\n\nexport type ResponsivePlotProps = PropsWithChildren<{\n  onResize: (size: PlotSize) => void;\n  debounceMs?: number;\n  className?: string;\n}>;\n\n/**\n * Container component that measures its size and notifies parent via callback.\n * Uses ResizeObserver with debouncing to reduce update frequency.\n * Exposes ref to the container div for direct DOM access.\n */\nexport const ResponsivePlot = forwardRef<HTMLDivElement, ResponsivePlotProps>(\n  ({onResize, debounceMs = 150, className, children}, ref) => {\n    const containerRef = useRef<HTMLDivElement>(null);\n    const lastSizeRef = useRef<PlotSize | null>(null);\n\n    // Expose the internal ref to parent\n    useImperativeHandle(ref, () => containerRef.current!, []);\n\n    // Debounced callback to reduce re-render frequency during resizing\n    const onResizeDebounced = useDebouncedCallback((size: PlotSize) => {\n      // Only call onResize if size actually changed\n      if (\n        !lastSizeRef.current ||\n        lastSizeRef.current.width !== size.width ||\n        lastSizeRef.current.height !== size.height\n      ) {\n        lastSizeRef.current = size;\n        onResize(size);\n      }\n    }, debounceMs);\n\n    // Measure container size with ResizeObserver\n    useEffect(() => {\n      const container = containerRef.current;\n      if (!container) return;\n\n      const resizeObserver = new ResizeObserver((entries) => {\n        for (const entry of entries) {\n          const {width, height} = entry.contentRect;\n          if (width > 0 && height > 0) {\n            onResizeDebounced({width, height});\n          }\n        }\n      });\n\n      resizeObserver.observe(container);\n      return () => resizeObserver.disconnect();\n    }, [onResizeDebounced]);\n\n    return (\n      <div ref={containerRef} className={cn('h-full w-full', className)}>\n        {children}\n      </div>\n    );\n  },\n);\n\nResponsivePlot.displayName = 'ResponsivePlot';\n"]}

package/dist/VgPlotChart.d.ts

@@ -1,14 +1,38 @@
+import { Param, Selection } from '@uwdata/mosaic-core';
 import { Spec } from '@uwdata/mosaic-spec';
 import { FC } from 'react';
 type SpecProps = {
     spec: Spec;
+    /**
+     * Pre-defined params/selections to inject when rendering the spec.
+     * Keys are param names (without $), values are Param or Selection instances.
+     * This allows multiple independently-rendered specs to share the same
+     * Selection objects for cross-filtering.
+     */
+    params?: Map<string, Param<any> | Selection>;
+    /**
+     * Optional retention adapter for preserving the underlying vgplot
+     * instance across temporary unmount/remount cycles, such as dashboard tab
+     * switches.
+     */
+    retention?: VgPlotChartRetention;
 };
 type PlotProps = {
     plot: HTMLElement | SVGSVGElement;
 };
 type VgPlotChartProps = SpecProps | PlotProps;
+export type RetainedVgPlotChart = {
+    element: object;
+    params?: Map<string, Param<any> | Selection>;
+    specKey: string;
+};
+export type VgPlotChartRetention = {
+    chart?: RetainedVgPlotChart;
+    setChart: (chart: RetainedVgPlotChart) => void;
+};
 export declare function isSpecProps(props: VgPlotChartProps): props is SpecProps;
 export declare function isPlotProps(props: VgPlotChartProps): props is PlotProps;
+export declare function destroyRetainedVgPlotChart(chart: RetainedVgPlotChart): void;
 /**
  * Renders a Vega-Lite chart using the Mosaic library.
  *

package/dist/VgPlotChart.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"VgPlotChart.d.ts","sourceRoot":"","sources":["../src/VgPlotChart.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAsB,IAAI,EAAC,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAC,EAAE,EAAoB,MAAM,OAAO,CAAC;AAE5C,KAAK,SAAS,GAAG;IAAC,IAAI,EAAE,IAAI,CAAA;CAAC,CAAC;AAC9B,KAAK,SAAS,GAAG;IAAC,IAAI,EAAE,WAAW,GAAG,aAAa,CAAA;CAAC,CAAC;AACrD,KAAK,gBAAgB,GAAG,SAAS,GAAG,SAAS,CAAC;AAE9C,wBAAgB,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,KAAK,IAAI,SAAS,CAEvE;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,KAAK,IAAI,SAAS,CAEvE;AAED;;;;;;GAMG;AACH,eAAO,MAAM,WAAW,EAAE,EAAE,CAAC,gBAAgB,CAqB5C,CAAC"}
+{"version":3,"file":"VgPlotChart.d.ts","sourceRoot":"","sources":["../src/VgPlotChart.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,KAAK,EAAE,SAAS,EAAC,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAsB,IAAI,EAAC,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EACL,EAAE,EAOH,MAAM,OAAO,CAAC;AAGf,KAAK,SAAS,GAAG;IACf,IAAI,EAAE,IAAI,CAAC;IACX;;;;;OAKG;IACH,MAAM,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC;IAC7C;;;;OAIG;IACH,SAAS,CAAC,EAAE,oBAAoB,CAAC;CAClC,CAAC;AACF,KAAK,SAAS,GAAG;IAAC,IAAI,EAAE,WAAW,GAAG,aAAa,CAAA;CAAC,CAAC;AACrD,KAAK,gBAAgB,GAAG,SAAS,GAAG,SAAS,CAAC;AAG9C,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC;IAC7C,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,CAAC,EAAE,mBAAmB,CAAC;IAC5B,QAAQ,EAAE,CAAC,KAAK,EAAE,mBAAmB,KAAK,IAAI,CAAC;CAChD,CAAC;AA0BF,wBAAgB,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,KAAK,IAAI,SAAS,CAEvE;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,KAAK,IAAI,SAAS,CAEvE;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,mBAAmB,QAKpE;AA4CD;;;;;;GAMG;AACH,eAAO,MAAM,WAAW,EAAE,EAAE,CAAC,gBAAgB,CA4G5C,CAAC"}