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

package/dist/profiler/utils.js

@@ -0,0 +1,347 @@
+import { asc, column, count, desc, Query, sql, sum } from '@uwdata/mosaic-sql';
+import * as arrow from 'apache-arrow';
+export function isProfilerHistogramType(type) {
+    return (arrow.DataType.isDate(type) ||
+        arrow.DataType.isTimestamp(type) ||
+        arrow.DataType.isDecimal(type) ||
+        arrow.DataType.isFloat(type) ||
+        arrow.DataType.isInt(type));
+}
+export function isProfilerUnsupportedSummaryType(type) {
+    return (arrow.DataType.isBinary(type) ||
+        type.toString().toLowerCase().includes('geometry'));
+}
+export function getProfilerValueType(type) {
+    if (arrow.DataType.isDate(type) || arrow.DataType.isTimestamp(type)) {
+        return 'date';
+    }
+    if (arrow.DataType.isDecimal(type) ||
+        arrow.DataType.isFloat(type) ||
+        arrow.DataType.isInt(type)) {
+        return 'number';
+    }
+    return 'string';
+}
+export function buildSchemaQuery(tableName, columns) {
+    return Query.from(tableName)
+        .select(columns?.length ? columns.map((name) => column(name)) : ['*'])
+        .limit(1);
+}
+function createProfilerArrowType(sqlType) {
+    const type = sqlType.toLowerCase();
+    if (/^bool(ean)?/.test(type)) {
+        return new arrow.Bool();
+    }
+    if (/^date$/.test(type)) {
+        return new arrow.DateDay();
+    }
+    if (/^time$|^timestamp|^timestamptz/.test(type)) {
+        return new arrow.TimestampMillisecond();
+    }
+    if (/^(tinyint|smallint|integer|bigint|hugeint|utinyint|usmallint|uinteger|ubigint|uhugeint)/.test(type)) {
+        return new arrow.Int64();
+    }
+    if (/^(decimal|numeric)/.test(type)) {
+        return new arrow.Decimal(38, 9);
+    }
+    if (/^(double|float|real)/.test(type)) {
+        return new arrow.Float64();
+    }
+    if (/^(blob|bytea|binary|varbinary)/.test(type)) {
+        return new arrow.Binary();
+    }
+    if (/^geometry/.test(type)) {
+        return {
+            toString() {
+                return sqlType;
+            },
+        };
+    }
+    return new arrow.Utf8();
+}
+export function fieldInfoToProfilerField(info) {
+    return new arrow.Field(info.column, createProfilerArrowType(info.sqlType), info.nullable);
+}
+export function buildProfilerBaseQuery(args) {
+    const { columns, filter, sorting, tableName } = args;
+    const query = Query.from(tableName)
+        .select(columns?.length ? columns.map((name) => column(name)) : ['*'])
+        .where(filter ?? []);
+    if (sorting?.length) {
+        query.orderby(sorting.map((entry) => (entry.desc ? desc(entry.id) : asc(entry.id))));
+    }
+    return query;
+}
+export function buildProfilerPageQuery(baseQuery, pagination) {
+    const { pageIndex, pageSize } = normalizeProfilerPagination(pagination);
+    return baseQuery
+        .clone()
+        .limit(pageSize)
+        .offset(pageIndex * pageSize);
+}
+export function normalizeProfilerPagination(pagination) {
+    const pageSize = Math.min(1000, Math.max(1, Math.trunc(Number(pagination?.pageSize) || 0) || 100));
+    const pageIndex = Math.max(0, Math.trunc(Number(pagination?.pageIndex) || 0));
+    return { pageIndex, pageSize };
+}
+export function buildCountQuery(args) {
+    return Query.from(args.tableName)
+        .select({ count: count() })
+        .where(args.filter ?? []);
+}
+export function buildDistinctCountQuery(args) {
+    return Query.from(args.tableName)
+        .select({
+        count: count(column(args.fieldName)).distinct(),
+    })
+        .where(args.filter ?? []);
+}
+export function readCountData(data) {
+    if (!data || typeof data !== 'object') {
+        return undefined;
+    }
+    if ('toArray' in data && typeof data.toArray === 'function') {
+        return data.toArray()[0]?.count;
+    }
+    if ('get' in data && typeof data.get === 'function') {
+        return data.get(0)?.count;
+    }
+    return undefined;
+}
+export function rowsFromQueryResult(data) {
+    if (!data ||
+        typeof data !== 'object' ||
+        !('toArray' in data) ||
+        typeof data.toArray !== 'function') {
+        return [];
+    }
+    return Array.from(data.toArray());
+}
+export function buildCategorySummaryQuery(tableName, fieldName, filter) {
+    const col = column(fieldName);
+    const counts = Query.from({ source: tableName })
+        .select({
+        bucket_kind: sql `CASE
+        WHEN ${col} IS NULL THEN 'null'
+        ELSE 'value'
+      END`,
+        typed_value: col,
+        count: count(),
+    })
+        .groupby([
+        sql `CASE
+        WHEN ${col} IS NULL THEN 'null'
+        ELSE 'value'
+      END`,
+        col,
+    ])
+        .where(filter ?? []);
+    return Query.with({ counts })
+        .select({
+        bucketKind: sql `CASE
+        WHEN "count" = 1 AND "bucket_kind" = 'value' THEN 'unique'
+        ELSE "bucket_kind"
+      END`,
+        typedValue: sql `CASE
+        WHEN "count" = 1 AND "bucket_kind" = 'value' THEN NULL
+        ELSE "typed_value"
+      END`,
+        total: sum('count'),
+    })
+        .from('counts')
+        .groupby([
+        sql `CASE
+        WHEN "count" = 1 AND "bucket_kind" = 'value' THEN 'unique'
+        ELSE "bucket_kind"
+      END`,
+        sql `CASE
+        WHEN "count" = 1 AND "bucket_kind" = 'value' THEN NULL
+        ELSE "typed_value"
+      END`,
+    ]);
+}
+export function splitHistogramBins(rows) {
+    let nullCount = 0;
+    const bins = [];
+    for (const row of rows) {
+        if (row.x1 == null || row.x2 == null) {
+            nullCount += row.y;
+            continue;
+        }
+        bins.push({
+            x0: row.x1,
+            x1: row.x2,
+            length: row.y,
+        });
+    }
+    bins.sort((left, right) => {
+        const leftValue = left.x0 instanceof Date ? left.x0.getTime() : left.x0;
+        const rightValue = right.x0 instanceof Date ? right.x0.getTime() : right.x0;
+        return leftValue - rightValue;
+    });
+    return { bins, nullCount };
+}
+export function buildCategoryBuckets(filteredRows, totalRows, categoryLimit, selectedKey) {
+    const totalByKey = new Map(totalRows.map((row) => [serializeCategoryBucketKey(row), row.total]));
+    const filteredByKey = new Map(filteredRows.map((row) => [serializeCategoryBucketKey(row), row.total]));
+    const baseRows = totalRows
+        .filter((row) => row.bucketKind === 'value')
+        .slice()
+        .sort((left, right) => right.total - left.total);
+    const visibleRows = baseRows.slice(0, categoryLimit);
+    const overflowRows = baseRows.slice(categoryLimit);
+    const buckets = visibleRows.map((row) => ({
+        filteredCount: filteredByKey.get(serializeCategoryBucketKey(row)) ?? 0,
+        key: serializeCategoryBucketKey(row),
+        kind: 'value',
+        label: String(row.typedValue),
+        selectable: true,
+        totalCount: row.total,
+    }));
+    const overflowTotalCount = overflowRows.reduce((acc, row) => acc + row.total, 0);
+    const overflowFilteredCount = overflowRows.reduce((acc, row) => acc + (filteredByKey.get(serializeCategoryBucketKey(row)) ?? 0), 0);
+    if (overflowTotalCount > 0) {
+        buckets.push({
+            filteredCount: overflowFilteredCount,
+            key: serializeCategoryBucketKey({
+                bucketKind: 'unique',
+                typedValue: '__sqlrooms_overflow__',
+            }),
+            kind: 'overflow',
+            label: `${overflowRows.length} more`,
+            selectable: false,
+            totalCount: overflowTotalCount,
+        });
+    }
+    const uniqueKey = serializeCategoryBucketKey({
+        bucketKind: 'unique',
+        typedValue: null,
+    });
+    const uniqueTotalCount = totalByKey.get(uniqueKey) ?? 0;
+    if (uniqueTotalCount > 0) {
+        buckets.push({
+            filteredCount: filteredByKey.get(uniqueKey) ?? 0,
+            key: uniqueKey,
+            kind: 'unique',
+            label: 'unique',
+            selectable: false,
+            totalCount: uniqueTotalCount,
+        });
+    }
+    const nullKey = serializeCategoryBucketKey({
+        bucketKind: 'null',
+        typedValue: null,
+    });
+    const nullTotalCount = totalByKey.get(nullKey) ?? 0;
+    if (nullTotalCount > 0) {
+        buckets.push({
+            filteredCount: filteredByKey.get(nullKey) ?? 0,
+            key: nullKey,
+            kind: 'null',
+            label: 'null',
+            selectable: true,
+            totalCount: nullTotalCount,
+        });
+    }
+    return {
+        bucketCount: filteredRows.length,
+        buckets,
+        selectedKey,
+    };
+}
+export function isSelectableCategoryKey(key) {
+    const parsedKey = parseCategoryBucketKey(key);
+    return (parsedKey?.bucketKind !== 'unique' &&
+        parsedKey?.typedValue !== '__sqlrooms_overflow__');
+}
+export function categoryKeyToSelectionValue(key) {
+    if (key === undefined)
+        return undefined;
+    const parsedKey = parseCategoryBucketKey(key);
+    if (!parsedKey || parsedKey.bucketKind === 'unique') {
+        return undefined;
+    }
+    return parsedKey.bucketKind === 'null' ? null : parsedKey.typedValue;
+}
+function normalizeCategoryBucketValue(value) {
+    if (value instanceof Date) {
+        return {
+            type: 'date',
+            value: value.toISOString(),
+        };
+    }
+    if (typeof value === 'bigint') {
+        return {
+            type: 'bigint',
+            value: value.toString(),
+        };
+    }
+    return value;
+}
+function denormalizeCategoryBucketValue(value) {
+    if (!value || typeof value !== 'object') {
+        return value;
+    }
+    if ('type' in value &&
+        'value' in value &&
+        value.type === 'date' &&
+        typeof value.value === 'string') {
+        return new Date(value.value);
+    }
+    if ('type' in value &&
+        'value' in value &&
+        value.type === 'bigint' &&
+        typeof value.value === 'string') {
+        return BigInt(value.value);
+    }
+    return value;
+}
+export function serializeCategoryBucketKey(row) {
+    return JSON.stringify([
+        row.bucketKind,
+        normalizeCategoryBucketValue(row.typedValue),
+    ]);
+}
+export function parseCategoryBucketKey(key) {
+    try {
+        const parsed = JSON.parse(key);
+        if (!Array.isArray(parsed) || parsed.length !== 2) {
+            return undefined;
+        }
+        return {
+            bucketKind: parsed[0],
+            typedValue: denormalizeCategoryBucketValue(parsed[1]),
+        };
+    }
+    catch {
+        return undefined;
+    }
+}
+export function createEmptySummaryState(field) {
+    if (isProfilerUnsupportedSummaryType(field.type)) {
+        return {
+            isLoading: false,
+            kind: 'unsupported',
+            label: 'No summary',
+        };
+    }
+    return isProfilerHistogramType(field.type)
+        ? {
+            filteredBins: [],
+            filteredNullCount: 0,
+            interactor: null,
+            isLoading: true,
+            kind: 'histogram',
+            totalBins: [],
+            totalNullCount: 0,
+            valueType: getProfilerValueType(field.type) === 'date' ? 'date' : 'number',
+        }
+        : {
+            bucketCount: 0,
+            buckets: [],
+            isLoading: true,
+            kind: 'category',
+            toggleValue: () => { },
+        };
+}
+//# sourceMappingURL=utils.js.map

package/dist/profiler/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/profiler/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAC,MAAM,oBAAoB,CAAC;AAC7E,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAiBtC,MAAM,UAAU,uBAAuB,CAAC,IAAoB;IAC1D,OAAO,CACL,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC;QAC3B,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC;QAChC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC;QAC5B,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAC3B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,IAAoB;IAEpB,OAAO,CACL,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC7B,IAAI,CAAC,QAAQ,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CACnD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,IAAoB;IAEpB,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;QACpE,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IACE,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC;QAC5B,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAC1B,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,OAAkB;IAElB,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;SACzB,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;SACrE,KAAK,CAAC,CAAC,CAAC,CAAC;AACd,CAAC;AAED,SAAS,uBAAuB,CAAC,OAAe;IAC9C,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAEnC,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED,IAAI,gCAAgC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,KAAK,CAAC,oBAAoB,EAAE,CAAC;IAC1C,CAAC;IAED,IACE,yFAAyF,CAAC,IAAI,CAC5F,IAAI,CACL,EACD,CAAC;QACD,OAAO,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,IAAI,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACpC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAClC,CAAC;IAED,IAAI,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACtC,OAAO,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED,IAAI,gCAAgC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,OAAO;YACL,QAAQ;gBACN,OAAO,OAAO,CAAC;YACjB,CAAC;SACgB,CAAC;IACtB,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,IAAe;IACtD,OAAO,IAAI,KAAK,CAAC,KAAK,CACpB,IAAI,CAAC,MAAM,EACX,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC,EACrC,IAAI,CAAC,QAAQ,CACd,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,IAKtC;IACC,MAAM,EAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAC,GAAG,IAAI,CAAC;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;SAChC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;SACrE,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IAEvB,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,KAAK,CAAC,OAAO,CACX,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CACtE,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,SAAoD,EACpD,UAAyC;IAEzC,MAAM,EAAC,SAAS,EAAE,QAAQ,EAAC,GAAG,2BAA2B,CAAC,UAAU,CAAC,CAAC;IAEtE,OAAO,SAAS;SACb,KAAK,EAAE;SACP,KAAK,CAAC,QAAQ,CAAC;SACf,MAAM,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,UAA8D;IAE9D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CACvB,IAAI,EACJ,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAClE,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE9E,OAAO,EAAC,SAAS,EAAE,QAAQ,EAAC,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,IAG/B;IACC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;SAC9B,MAAM,CAAC,EAAC,KAAK,EAAE,KAAK,EAAE,EAAC,CAAC;SACxB,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,IAIvC;IACC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;SAC9B,MAAM,CAAC;QACN,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE;KAChD,CAAC;SACD,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAa;IACzC,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,SAAS,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;QAC5D,OAAQ,IAAI,CAAC,OAAO,EAA8B,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;IAC/D,CAAC;IAED,IAAI,KAAK,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,GAAG,KAAK,UAAU,EAAE,CAAC;QACpD,OAAQ,IAAI,CAAC,GAAG,CAAC,CAAC,CAAkC,EAAE,KAAK,CAAC;IAC9D,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAI,IAAa;IAClD,IACE,CAAC,IAAI;QACL,OAAO,IAAI,KAAK,QAAQ;QACxB,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC;QACpB,OAAQ,IAA4B,CAAC,OAAO,KAAK,UAAU,EAC3D,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAE,IAAyB,CAAC,OAAO,EAAE,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,SAAiB,EACjB,SAAiB,EACjB,MAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAC,MAAM,EAAE,SAAS,EAAC,CAAC;SAC3C,MAAM,CAAC;QACN,WAAW,EAAE,GAAG,CAAA;eACP,GAAG;;UAER;QACJ,WAAW,EAAE,GAAG;QAChB,KAAK,EAAE,KAAK,EAAE;KACf,CAAC;SACD,OAAO,CAAC;QACP,GAAG,CAAA;eACM,GAAG;;UAER;QACJ,GAAG;KACJ,CAAC;SACD,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IAEvB,OAAO,KAAK,CAAC,IAAI,CAAC,EAAC,MAAM,EAAC,CAAC;SACxB,MAAM,CAAC;QACN,UAAU,EAAE,GAAG,CAAA;;;UAGX;QACJ,UAAU,EAAE,GAAG,CAAA;;;UAGX;QACJ,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC;KACpB,CAAC;SACD,IAAI,CAAC,QAAQ,CAAC;SACd,OAAO,CAAC;QACP,GAAG,CAAA;;;UAGC;QACJ,GAAG,CAAA;;;UAGC;KACL,CAAC,CAAC;AACP,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,IAIE;IAEF,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,IAAI,GAAwB,EAAE,CAAC;IAErC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YACrC,SAAS,IAAI,GAAG,CAAC,CAAC,CAAC;YACnB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,IAAI,CAAC;YACR,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,CAAC;SACd,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,UAAU,GAAG,KAAK,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5E,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,OAAO,EAAC,IAAI,EAAE,SAAS,EAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,YAAgC,EAChC,SAA6B,EAC7B,aAAqB,EACrB,WAAoB;IAEpB,MAAM,UAAU,GAAG,IAAI,GAAG,CACxB,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CACrE,CAAC;IACF,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CACxE,CAAC;IAEF,MAAM,QAAQ,GAAG,SAAS;SACvB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,KAAK,OAAO,CAAC;SAC3C,KAAK,EAAE;SACP,IAAI,CACH,CAAC,IAAsB,EAAE,KAAuB,EAAE,EAAE,CAClD,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAC3B,CAAC;IAEJ,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;IACrD,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACnD,MAAM,OAAO,GAAmC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxE,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;QACtE,GAAG,EAAE,0BAA0B,CAAC,GAAG,CAAC;QACpC,IAAI,EAAE,OAAO;QACb,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC;QAC7B,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,GAAG,CAAC,KAAK;KACtB,CAAC,CAAC,CAAC;IAEJ,MAAM,kBAAkB,GAAG,YAAY,CAAC,MAAM,CAC5C,CAAC,GAAW,EAAE,GAAqB,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,EACvD,CAAC,CACF,CAAC;IACF,MAAM,qBAAqB,GAAG,YAAY,CAAC,MAAM,CAC/C,CAAC,GAAW,EAAE,GAAqB,EAAE,EAAE,CACrC,GAAG,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,EACjE,CAAC,CACF,CAAC;IACF,IAAI,kBAAkB,GAAG,CAAC,EAAE,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC;YACX,aAAa,EAAE,qBAAqB;YACpC,GAAG,EAAE,0BAA0B,CAAC;gBAC9B,UAAU,EAAE,QAAQ;gBACpB,UAAU,EAAE,uBAAuB;aACpC,CAAC;YACF,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,GAAG,YAAY,CAAC,MAAM,OAAO;YACpC,UAAU,EAAE,KAAK;YACjB,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,SAAS,GAAG,0BAA0B,CAAC;QAC3C,UAAU,EAAE,QAAQ;QACpB,UAAU,EAAE,IAAI;KACjB,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxD,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC;YACX,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YAChD,GAAG,EAAE,SAAS;YACd,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,QAAQ;YACf,UAAU,EAAE,KAAK;YACjB,UAAU,EAAE,gBAAgB;SAC7B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,0BAA0B,CAAC;QACzC,UAAU,EAAE,MAAM;QAClB,UAAU,EAAE,IAAI;KACjB,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACpD,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;QACvB,OAAO,CAAC,IAAI,CAAC;YACX,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;YAC9C,GAAG,EAAE,OAAO;YACZ,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,IAAI;YAChB,UAAU,EAAE,cAAc;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,WAAW,EAAE,YAAY,CAAC,MAAM;QAChC,OAAO;QACP,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,GAAW;IACjD,MAAM,SAAS,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;IAC9C,OAAO,CACL,SAAS,EAAE,UAAU,KAAK,QAAQ;QAClC,SAAS,EAAE,UAAU,KAAK,uBAAuB,CAClD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,GAAY;IACtD,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACxC,MAAM,SAAS,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;IAC9C,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QACpD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,SAAS,CAAC,UAAU,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC;AACvE,CAAC;AAED,SAAS,4BAA4B,CAAC,KAAc;IAClD,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;QAC1B,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,KAAK,CAAC,WAAW,EAAE;SAC3B,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE;SACxB,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,8BAA8B,CAAC,KAAc;IACpD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IACE,MAAM,IAAI,KAAK;QACf,OAAO,IAAI,KAAK;QAChB,KAAK,CAAC,IAAI,KAAK,MAAM;QACrB,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,EAC/B,CAAC;QACD,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,IACE,MAAM,IAAI,KAAK;QACf,OAAO,IAAI,KAAK;QAChB,KAAK,CAAC,IAAI,KAAK,QAAQ;QACvB,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,EAC/B,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,GAG1C;IACC,OAAO,IAAI,CAAC,SAAS,CAAC;QACpB,GAAG,CAAC,UAAU;QACd,4BAA4B,CAAC,GAAG,CAAC,UAAU,CAAC;KAC7C,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAMhD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA8C,CAAC;QAC5E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YACrB,UAAU,EAAE,8BAA8B,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACtD,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAkB;IAElB,IAAI,gCAAgC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACjD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,YAAY;SACpB,CAAC;IACJ,CAAC;IAED,OAAO,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC;QACxC,CAAC,CAAC;YACE,YAAY,EAAE,EAAE;YAChB,iBAAiB,EAAE,CAAC;YACpB,UAAU,EAAE,IAAI;YAChB,SAAS,EAAE,IAAI;YACf,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,EAAE;YACb,cAAc,EAAE,CAAC;YACjB,SAAS,EACP,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ;SAClE;QACH,CAAC,CAAC;YACE,WAAW,EAAE,CAAC;YACd,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,IAAI;YACf,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,GAAG,EAAE,GAAE,CAAC;SACtB,CAAC;AACR,CAAC","sourcesContent":["import type {FieldInfo} from '@uwdata/mosaic-core';\nimport {asc, column, count, desc, Query, sql, sum} from '@uwdata/mosaic-sql';\nimport * as arrow from 'apache-arrow';\nimport type {\n  MosaicProfilerBin,\n  MosaicProfilerCategoryBucket,\n  MosaicProfilerPaginationState,\n  MosaicProfilerSorting,\n  MosaicProfilerSummaryState,\n} from './types';\n\nexport type CategoryCountRow = {\n  bucketKind: 'null' | 'unique' | 'value';\n  total: number;\n  typedValue: unknown;\n};\n\ntype QueryWhereInput = Parameters<ReturnType<typeof Query.from>['where']>[0];\n\nexport function isProfilerHistogramType(type: arrow.DataType): boolean {\n  return (\n    arrow.DataType.isDate(type) ||\n    arrow.DataType.isTimestamp(type) ||\n    arrow.DataType.isDecimal(type) ||\n    arrow.DataType.isFloat(type) ||\n    arrow.DataType.isInt(type)\n  );\n}\n\nexport function isProfilerUnsupportedSummaryType(\n  type: arrow.DataType,\n): boolean {\n  return (\n    arrow.DataType.isBinary(type) ||\n    type.toString().toLowerCase().includes('geometry')\n  );\n}\n\nexport function getProfilerValueType(\n  type: arrow.DataType,\n): 'date' | 'number' | 'string' {\n  if (arrow.DataType.isDate(type) || arrow.DataType.isTimestamp(type)) {\n    return 'date';\n  }\n  if (\n    arrow.DataType.isDecimal(type) ||\n    arrow.DataType.isFloat(type) ||\n    arrow.DataType.isInt(type)\n  ) {\n    return 'number';\n  }\n  return 'string';\n}\n\nexport function buildSchemaQuery(\n  tableName: string,\n  columns?: string[],\n): ReturnType<typeof Query.from> {\n  return Query.from(tableName)\n    .select(columns?.length ? columns.map((name) => column(name)) : ['*'])\n    .limit(1);\n}\n\nfunction createProfilerArrowType(sqlType: string): arrow.DataType {\n  const type = sqlType.toLowerCase();\n\n  if (/^bool(ean)?/.test(type)) {\n    return new arrow.Bool();\n  }\n\n  if (/^date$/.test(type)) {\n    return new arrow.DateDay();\n  }\n\n  if (/^time$|^timestamp|^timestamptz/.test(type)) {\n    return new arrow.TimestampMillisecond();\n  }\n\n  if (\n    /^(tinyint|smallint|integer|bigint|hugeint|utinyint|usmallint|uinteger|ubigint|uhugeint)/.test(\n      type,\n    )\n  ) {\n    return new arrow.Int64();\n  }\n\n  if (/^(decimal|numeric)/.test(type)) {\n    return new arrow.Decimal(38, 9);\n  }\n\n  if (/^(double|float|real)/.test(type)) {\n    return new arrow.Float64();\n  }\n\n  if (/^(blob|bytea|binary|varbinary)/.test(type)) {\n    return new arrow.Binary();\n  }\n\n  if (/^geometry/.test(type)) {\n    return {\n      toString() {\n        return sqlType;\n      },\n    } as arrow.DataType;\n  }\n\n  return new arrow.Utf8();\n}\n\nexport function fieldInfoToProfilerField(info: FieldInfo): arrow.Field {\n  return new arrow.Field(\n    info.column,\n    createProfilerArrowType(info.sqlType),\n    info.nullable,\n  );\n}\n\nexport function buildProfilerBaseQuery(args: {\n  columns?: string[];\n  filter?: QueryWhereInput;\n  sorting?: MosaicProfilerSorting;\n  tableName: string;\n}) {\n  const {columns, filter, sorting, tableName} = args;\n  const query = Query.from(tableName)\n    .select(columns?.length ? columns.map((name) => column(name)) : ['*'])\n    .where(filter ?? []);\n\n  if (sorting?.length) {\n    query.orderby(\n      sorting.map((entry) => (entry.desc ? desc(entry.id) : asc(entry.id))),\n    );\n  }\n\n  return query;\n}\n\nexport function buildProfilerPageQuery(\n  baseQuery: ReturnType<typeof buildProfilerBaseQuery>,\n  pagination: MosaicProfilerPaginationState,\n) {\n  const {pageIndex, pageSize} = normalizeProfilerPagination(pagination);\n\n  return baseQuery\n    .clone()\n    .limit(pageSize)\n    .offset(pageIndex * pageSize);\n}\n\nexport function normalizeProfilerPagination(\n  pagination: Partial<MosaicProfilerPaginationState> | undefined,\n): MosaicProfilerPaginationState {\n  const pageSize = Math.min(\n    1000,\n    Math.max(1, Math.trunc(Number(pagination?.pageSize) || 0) || 100),\n  );\n  const pageIndex = Math.max(0, Math.trunc(Number(pagination?.pageIndex) || 0));\n\n  return {pageIndex, pageSize};\n}\n\nexport function buildCountQuery(args: {\n  filter?: QueryWhereInput;\n  tableName: string;\n}) {\n  return Query.from(args.tableName)\n    .select({count: count()})\n    .where(args.filter ?? []);\n}\n\nexport function buildDistinctCountQuery(args: {\n  filter?: QueryWhereInput;\n  fieldName: string;\n  tableName: string;\n}) {\n  return Query.from(args.tableName)\n    .select({\n      count: count(column(args.fieldName)).distinct(),\n    })\n    .where(args.filter ?? []);\n}\n\nexport function readCountData(data: unknown): number | undefined {\n  if (!data || typeof data !== 'object') {\n    return undefined;\n  }\n\n  if ('toArray' in data && typeof data.toArray === 'function') {\n    return (data.toArray() as Array<{count?: number}>)[0]?.count;\n  }\n\n  if ('get' in data && typeof data.get === 'function') {\n    return (data.get(0) as {count?: number} | undefined)?.count;\n  }\n\n  return undefined;\n}\n\nexport function rowsFromQueryResult<T>(data: unknown): T[] {\n  if (\n    !data ||\n    typeof data !== 'object' ||\n    !('toArray' in data) ||\n    typeof (data as {toArray?: unknown}).toArray !== 'function'\n  ) {\n    return [];\n  }\n\n  return Array.from((data as {toArray(): T[]}).toArray());\n}\n\nexport function buildCategorySummaryQuery(\n  tableName: string,\n  fieldName: string,\n  filter?: QueryWhereInput,\n) {\n  const col = column(fieldName);\n  const counts = Query.from({source: tableName})\n    .select({\n      bucket_kind: sql`CASE\n        WHEN ${col} IS NULL THEN 'null'\n        ELSE 'value'\n      END`,\n      typed_value: col,\n      count: count(),\n    })\n    .groupby([\n      sql`CASE\n        WHEN ${col} IS NULL THEN 'null'\n        ELSE 'value'\n      END`,\n      col,\n    ])\n    .where(filter ?? []);\n\n  return Query.with({counts})\n    .select({\n      bucketKind: sql`CASE\n        WHEN \"count\" = 1 AND \"bucket_kind\" = 'value' THEN 'unique'\n        ELSE \"bucket_kind\"\n      END`,\n      typedValue: sql`CASE\n        WHEN \"count\" = 1 AND \"bucket_kind\" = 'value' THEN NULL\n        ELSE \"typed_value\"\n      END`,\n      total: sum('count'),\n    })\n    .from('counts')\n    .groupby([\n      sql`CASE\n        WHEN \"count\" = 1 AND \"bucket_kind\" = 'value' THEN 'unique'\n        ELSE \"bucket_kind\"\n      END`,\n      sql`CASE\n        WHEN \"count\" = 1 AND \"bucket_kind\" = 'value' THEN NULL\n        ELSE \"typed_value\"\n      END`,\n    ]);\n}\n\nexport function splitHistogramBins(\n  rows: Array<{\n    x1: MosaicProfilerBin['x0'] | null;\n    x2: MosaicProfilerBin['x1'] | null;\n    y: number;\n  }>,\n) {\n  let nullCount = 0;\n  const bins: MosaicProfilerBin[] = [];\n\n  for (const row of rows) {\n    if (row.x1 == null || row.x2 == null) {\n      nullCount += row.y;\n      continue;\n    }\n\n    bins.push({\n      x0: row.x1,\n      x1: row.x2,\n      length: row.y,\n    });\n  }\n\n  bins.sort((left, right) => {\n    const leftValue = left.x0 instanceof Date ? left.x0.getTime() : left.x0;\n    const rightValue = right.x0 instanceof Date ? right.x0.getTime() : right.x0;\n    return leftValue - rightValue;\n  });\n\n  return {bins, nullCount};\n}\n\nexport function buildCategoryBuckets(\n  filteredRows: CategoryCountRow[],\n  totalRows: CategoryCountRow[],\n  categoryLimit: number,\n  selectedKey?: string,\n) {\n  const totalByKey = new Map(\n    totalRows.map((row) => [serializeCategoryBucketKey(row), row.total]),\n  );\n  const filteredByKey = new Map(\n    filteredRows.map((row) => [serializeCategoryBucketKey(row), row.total]),\n  );\n\n  const baseRows = totalRows\n    .filter((row) => row.bucketKind === 'value')\n    .slice()\n    .sort(\n      (left: CategoryCountRow, right: CategoryCountRow) =>\n        right.total - left.total,\n    );\n\n  const visibleRows = baseRows.slice(0, categoryLimit);\n  const overflowRows = baseRows.slice(categoryLimit);\n  const buckets: MosaicProfilerCategoryBucket[] = visibleRows.map((row) => ({\n    filteredCount: filteredByKey.get(serializeCategoryBucketKey(row)) ?? 0,\n    key: serializeCategoryBucketKey(row),\n    kind: 'value',\n    label: String(row.typedValue),\n    selectable: true,\n    totalCount: row.total,\n  }));\n\n  const overflowTotalCount = overflowRows.reduce(\n    (acc: number, row: CategoryCountRow) => acc + row.total,\n    0,\n  );\n  const overflowFilteredCount = overflowRows.reduce(\n    (acc: number, row: CategoryCountRow) =>\n      acc + (filteredByKey.get(serializeCategoryBucketKey(row)) ?? 0),\n    0,\n  );\n  if (overflowTotalCount > 0) {\n    buckets.push({\n      filteredCount: overflowFilteredCount,\n      key: serializeCategoryBucketKey({\n        bucketKind: 'unique',\n        typedValue: '__sqlrooms_overflow__',\n      }),\n      kind: 'overflow',\n      label: `${overflowRows.length} more`,\n      selectable: false,\n      totalCount: overflowTotalCount,\n    });\n  }\n\n  const uniqueKey = serializeCategoryBucketKey({\n    bucketKind: 'unique',\n    typedValue: null,\n  });\n  const uniqueTotalCount = totalByKey.get(uniqueKey) ?? 0;\n  if (uniqueTotalCount > 0) {\n    buckets.push({\n      filteredCount: filteredByKey.get(uniqueKey) ?? 0,\n      key: uniqueKey,\n      kind: 'unique',\n      label: 'unique',\n      selectable: false,\n      totalCount: uniqueTotalCount,\n    });\n  }\n\n  const nullKey = serializeCategoryBucketKey({\n    bucketKind: 'null',\n    typedValue: null,\n  });\n  const nullTotalCount = totalByKey.get(nullKey) ?? 0;\n  if (nullTotalCount > 0) {\n    buckets.push({\n      filteredCount: filteredByKey.get(nullKey) ?? 0,\n      key: nullKey,\n      kind: 'null',\n      label: 'null',\n      selectable: true,\n      totalCount: nullTotalCount,\n    });\n  }\n\n  return {\n    bucketCount: filteredRows.length,\n    buckets,\n    selectedKey,\n  };\n}\n\nexport function isSelectableCategoryKey(key: string) {\n  const parsedKey = parseCategoryBucketKey(key);\n  return (\n    parsedKey?.bucketKind !== 'unique' &&\n    parsedKey?.typedValue !== '__sqlrooms_overflow__'\n  );\n}\n\nexport function categoryKeyToSelectionValue(key?: string) {\n  if (key === undefined) return undefined;\n  const parsedKey = parseCategoryBucketKey(key);\n  if (!parsedKey || parsedKey.bucketKind === 'unique') {\n    return undefined;\n  }\n  return parsedKey.bucketKind === 'null' ? null : parsedKey.typedValue;\n}\n\nfunction normalizeCategoryBucketValue(value: unknown): unknown {\n  if (value instanceof Date) {\n    return {\n      type: 'date',\n      value: value.toISOString(),\n    };\n  }\n  if (typeof value === 'bigint') {\n    return {\n      type: 'bigint',\n      value: value.toString(),\n    };\n  }\n  return value;\n}\n\nfunction denormalizeCategoryBucketValue(value: unknown): unknown {\n  if (!value || typeof value !== 'object') {\n    return value;\n  }\n\n  if (\n    'type' in value &&\n    'value' in value &&\n    value.type === 'date' &&\n    typeof value.value === 'string'\n  ) {\n    return new Date(value.value);\n  }\n\n  if (\n    'type' in value &&\n    'value' in value &&\n    value.type === 'bigint' &&\n    typeof value.value === 'string'\n  ) {\n    return BigInt(value.value);\n  }\n\n  return value;\n}\n\nexport function serializeCategoryBucketKey(row: {\n  bucketKind: CategoryCountRow['bucketKind'];\n  typedValue: unknown;\n}) {\n  return JSON.stringify([\n    row.bucketKind,\n    normalizeCategoryBucketValue(row.typedValue),\n  ]);\n}\n\nexport function parseCategoryBucketKey(key: string):\n  | {\n      bucketKind: CategoryCountRow['bucketKind'];\n      typedValue: unknown;\n    }\n  | undefined {\n  try {\n    const parsed = JSON.parse(key) as [CategoryCountRow['bucketKind'], unknown];\n    if (!Array.isArray(parsed) || parsed.length !== 2) {\n      return undefined;\n    }\n\n    return {\n      bucketKind: parsed[0],\n      typedValue: denormalizeCategoryBucketValue(parsed[1]),\n    };\n  } catch {\n    return undefined;\n  }\n}\n\nexport function createEmptySummaryState(\n  field: arrow.Field,\n): MosaicProfilerSummaryState {\n  if (isProfilerUnsupportedSummaryType(field.type)) {\n    return {\n      isLoading: false,\n      kind: 'unsupported',\n      label: 'No summary',\n    };\n  }\n\n  return isProfilerHistogramType(field.type)\n    ? {\n        filteredBins: [],\n        filteredNullCount: 0,\n        interactor: null,\n        isLoading: true,\n        kind: 'histogram',\n        totalBins: [],\n        totalNullCount: 0,\n        valueType:\n          getProfilerValueType(field.type) === 'date' ? 'date' : 'number',\n      }\n    : {\n        bucketCount: 0,\n        buckets: [],\n        isLoading: true,\n        kind: 'category',\n        toggleValue: () => {},\n      };\n}\n"]}

package/dist/tableInterop.d.ts

@@ -0,0 +1,30 @@
+import { type Table as MosaicTable } from '@uwdata/flechette';
+import { type Table as ArrowTable } from 'apache-arrow';
+type TableInterop = {
+    ipcBytes: Uint8Array;
+    getArrowTable: () => ArrowTable;
+    getMosaicTable: () => object;
+};
+export declare function getTableInterop(value: unknown): TableInterop | undefined;
+export declare function getArrowTableFromInterop(value: unknown): ArrowTable | undefined;
+/**
+ * Convert an Apache Arrow table into the Mosaic-native table shape while
+ * attaching Mosaic-internal IPC-backed interop metadata.
+ *
+ * Mosaic core expects Flechette tables, while SQLRooms public client APIs
+ * expose Apache Arrow. IPC remains the durable bridge. The original Arrow
+ * table is referenced weakly so Arrow-native callers can often reuse it
+ * without forcing Mosaic-only consumers to retain both table runtimes.
+ */
+export declare function createMosaicTableFromArrowTable(arrowTable: ArrowTable): MosaicTable;
+/**
+ * Convert a raw Mosaic client result to the Apache Arrow shape exposed by
+ * SQLRooms public hooks.
+ *
+ * Results produced by {@link createMosaicTableFromArrowTable} use the
+ * interop fast path. Plain third-party Flechette tables are still accepted as
+ * a compatibility fallback, but that path has to re-encode through IPC.
+ */
+export declare function toArrowClientResult(value: unknown): ArrowTable;
+export {};
+//# sourceMappingURL=tableInterop.d.ts.map

package/dist/tableInterop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tableInterop.d.ts","sourceRoot":"","sources":["../src/tableInterop.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,KAAK,KAAK,IAAI,WAAW,EAC1B,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAA2B,KAAK,KAAK,IAAI,UAAU,EAAC,MAAM,cAAc,CAAC;AAIhF,KAAK,YAAY,GAAG;IAClB,QAAQ,EAAE,UAAU,CAAC;IACrB,aAAa,EAAE,MAAM,UAAU,CAAC;IAChC,cAAc,EAAE,MAAM,MAAM,CAAC;CAC9B,CAAC;AAmCF,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,YAAY,GAAG,SAAS,CAMxE;AAED,wBAAgB,wBAAwB,CACtC,KAAK,EAAE,OAAO,GACb,UAAU,GAAG,SAAS,CAExB;AAmBD;;;;;;;;GAQG;AACH,wBAAgB,+BAA+B,CAC7C,UAAU,EAAE,UAAU,GACrB,WAAW,CASb;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,OAAO,GAAG,UAAU,CAiB9D"}

package/dist/tableInterop.js

@@ -0,0 +1,85 @@
+import { decodeIPC } from '@uwdata/mosaic-core';
+import { tableToIPC as mosaicTableToIPC, } from '@uwdata/flechette';
+import { tableFromIPC, tableToIPC } from 'apache-arrow';
+const TABLE_INTEROP = Symbol.for('@sqlrooms/mosaic-table-interop');
+function attachTableInterop(table, options) {
+    if (getTableInterop(table)) {
+        return table;
+    }
+    const { ipcBytes, decodeArrowTable = (bytes) => tableFromIPC(bytes) } = options;
+    let arrowTable;
+    const interop = {
+        ipcBytes,
+        getArrowTable: () => {
+            arrowTable ??= decodeArrowTable(ipcBytes);
+            return arrowTable;
+        },
+        getMosaicTable: () => table,
+    };
+    Object.defineProperty(table, TABLE_INTEROP, {
+        value: interop,
+        enumerable: false,
+        configurable: false,
+        writable: false,
+    });
+    return table;
+}
+export function getTableInterop(value) {
+    if (!value || typeof value !== 'object') {
+        return undefined;
+    }
+    return value[TABLE_INTEROP];
+}
+export function getArrowTableFromInterop(value) {
+    return getTableInterop(value)?.getArrowTable();
+}
+function isArrowTable(value) {
+    return Boolean(value &&
+        typeof value === 'object' &&
+        Array.isArray(value.batches));
+}
+function isMosaicTable(value) {
+    return Boolean(value &&
+        typeof value === 'object' &&
+        typeof value.toColumns === 'function' &&
+        typeof value.getChild === 'function');
+}
+/**
+ * Convert an Apache Arrow table into the Mosaic-native table shape while
+ * attaching Mosaic-internal IPC-backed interop metadata.
+ *
+ * Mosaic core expects Flechette tables, while SQLRooms public client APIs
+ * expose Apache Arrow. IPC remains the durable bridge. The original Arrow
+ * table is referenced weakly so Arrow-native callers can often reuse it
+ * without forcing Mosaic-only consumers to retain both table runtimes.
+ */
+export function createMosaicTableFromArrowTable(arrowTable) {
+    const ipcBytes = tableToIPC(arrowTable, 'stream');
+    const arrowTableRef = typeof WeakRef === 'undefined' ? undefined : new WeakRef(arrowTable);
+    return attachTableInterop(decodeIPC(ipcBytes), {
+        ipcBytes,
+        decodeArrowTable: (bytes) => arrowTableRef?.deref() ?? tableFromIPC(bytes),
+    });
+}
+/**
+ * Convert a raw Mosaic client result to the Apache Arrow shape exposed by
+ * SQLRooms public hooks.
+ *
+ * Results produced by {@link createMosaicTableFromArrowTable} use the
+ * interop fast path. Plain third-party Flechette tables are still accepted as
+ * a compatibility fallback, but that path has to re-encode through IPC.
+ */
+export function toArrowClientResult(value) {
+    if (isArrowTable(value)) {
+        return value;
+    }
+    const interopArrowTable = getArrowTableFromInterop(value);
+    if (interopArrowTable) {
+        return interopArrowTable;
+    }
+    if (isMosaicTable(value)) {
+        return tableFromIPC(mosaicTableToIPC(value, { format: 'stream' }));
+    }
+    throw new Error('useMosaicClient expected a Mosaic table result that can be converted to Apache Arrow.');
+}
+//# sourceMappingURL=tableInterop.js.map

package/dist/tableInterop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tableInterop.js","sourceRoot":"","sources":["../src/tableInterop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,SAAS,EAAC,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EACL,UAAU,IAAI,gBAAgB,GAE/B,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAC,YAAY,EAAE,UAAU,EAA2B,MAAM,cAAc,CAAC;AAEhF,MAAM,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;AAQnE,SAAS,kBAAkB,CACzB,KAAQ,EACR,OAGC;IAED,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,EAAC,QAAQ,EAAE,gBAAgB,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,EAAC,GAAG,OAAO,CAAC;IAC9E,IAAI,UAAkC,CAAC;IAEvC,MAAM,OAAO,GAAiB;QAC5B,QAAQ;QACR,aAAa,EAAE,GAAG,EAAE;YAClB,UAAU,KAAK,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC1C,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;KAC5B,CAAC;IAEF,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,aAAa,EAAE;QAC1C,KAAK,EAAE,OAAO;QACd,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;QACnB,QAAQ,EAAE,KAAK;KAChB,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAc;IAC5C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAQ,KAA0C,CAAC,aAAa,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,KAAc;IAEd,OAAO,eAAe,CAAC,KAAK,CAAC,EAAE,aAAa,EAAE,CAAC;AACjD,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,OAAO,CACZ,KAAK;QACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,CAAC,OAAO,CAAE,KAA6B,CAAC,OAAO,CAAC,CACtD,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,OAAO,CACZ,KAAK;QACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAQ,KAA+B,CAAC,SAAS,KAAK,UAAU;QAChE,OAAQ,KAA8B,CAAC,QAAQ,KAAK,UAAU,CAC/D,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,+BAA+B,CAC7C,UAAsB;IAEtB,MAAM,QAAQ,GAAG,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAClD,MAAM,aAAa,GACjB,OAAO,OAAO,KAAK,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,CAAC;IAEvE,OAAO,kBAAkB,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;QAC7C,QAAQ;QACR,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,EAAE,KAAK,EAAE,IAAI,YAAY,CAAC,KAAK,CAAC;KAC3E,CAAgB,CAAC;AACpB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;IAC1D,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,YAAY,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAC,MAAM,EAAE,QAAQ,EAAC,CAAC,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,IAAI,KAAK,CACb,uFAAuF,CACxF,CAAC;AACJ,CAAC","sourcesContent":["import {decodeIPC} from '@uwdata/mosaic-core';\nimport {\n  tableToIPC as mosaicTableToIPC,\n  type Table as MosaicTable,\n} from '@uwdata/flechette';\nimport {tableFromIPC, tableToIPC, type Table as ArrowTable} from 'apache-arrow';\n\nconst TABLE_INTEROP = Symbol.for('@sqlrooms/mosaic-table-interop');\n\ntype TableInterop = {\n  ipcBytes: Uint8Array;\n  getArrowTable: () => ArrowTable;\n  getMosaicTable: () => object;\n};\n\nfunction attachTableInterop<T extends object>(\n  table: T,\n  options: {\n    ipcBytes: Uint8Array;\n    decodeArrowTable?: (ipcBytes: Uint8Array) => ArrowTable;\n  },\n): T {\n  if (getTableInterop(table)) {\n    return table;\n  }\n\n  const {ipcBytes, decodeArrowTable = (bytes) => tableFromIPC(bytes)} = options;\n  let arrowTable: ArrowTable | undefined;\n\n  const interop: TableInterop = {\n    ipcBytes,\n    getArrowTable: () => {\n      arrowTable ??= decodeArrowTable(ipcBytes);\n      return arrowTable;\n    },\n    getMosaicTable: () => table,\n  };\n\n  Object.defineProperty(table, TABLE_INTEROP, {\n    value: interop,\n    enumerable: false,\n    configurable: false,\n    writable: false,\n  });\n\n  return table;\n}\n\nexport function getTableInterop(value: unknown): TableInterop | undefined {\n  if (!value || typeof value !== 'object') {\n    return undefined;\n  }\n\n  return (value as {[TABLE_INTEROP]?: TableInterop})[TABLE_INTEROP];\n}\n\nexport function getArrowTableFromInterop(\n  value: unknown,\n): ArrowTable | undefined {\n  return getTableInterop(value)?.getArrowTable();\n}\n\nfunction isArrowTable(value: unknown): value is ArrowTable {\n  return Boolean(\n    value &&\n    typeof value === 'object' &&\n    Array.isArray((value as {batches?: unknown}).batches),\n  );\n}\n\nfunction isMosaicTable(value: unknown): value is MosaicTable {\n  return Boolean(\n    value &&\n    typeof value === 'object' &&\n    typeof (value as {toColumns?: unknown}).toColumns === 'function' &&\n    typeof (value as {getChild?: unknown}).getChild === 'function',\n  );\n}\n\n/**\n * Convert an Apache Arrow table into the Mosaic-native table shape while\n * attaching Mosaic-internal IPC-backed interop metadata.\n *\n * Mosaic core expects Flechette tables, while SQLRooms public client APIs\n * expose Apache Arrow. IPC remains the durable bridge. The original Arrow\n * table is referenced weakly so Arrow-native callers can often reuse it\n * without forcing Mosaic-only consumers to retain both table runtimes.\n */\nexport function createMosaicTableFromArrowTable(\n  arrowTable: ArrowTable,\n): MosaicTable {\n  const ipcBytes = tableToIPC(arrowTable, 'stream');\n  const arrowTableRef =\n    typeof WeakRef === 'undefined' ? undefined : new WeakRef(arrowTable);\n\n  return attachTableInterop(decodeIPC(ipcBytes), {\n    ipcBytes,\n    decodeArrowTable: (bytes) => arrowTableRef?.deref() ?? tableFromIPC(bytes),\n  }) as MosaicTable;\n}\n\n/**\n * Convert a raw Mosaic client result to the Apache Arrow shape exposed by\n * SQLRooms public hooks.\n *\n * Results produced by {@link createMosaicTableFromArrowTable} use the\n * interop fast path. Plain third-party Flechette tables are still accepted as\n * a compatibility fallback, but that path has to re-encode through IPC.\n */\nexport function toArrowClientResult(value: unknown): ArrowTable {\n  if (isArrowTable(value)) {\n    return value;\n  }\n\n  const interopArrowTable = getArrowTableFromInterop(value);\n  if (interopArrowTable) {\n    return interopArrowTable;\n  }\n\n  if (isMosaicTable(value)) {\n    return tableFromIPC(mosaicTableToIPC(value, {format: 'stream'}));\n  }\n\n  throw new Error(\n    'useMosaicClient expected a Mosaic table result that can be converted to Apache Arrow.',\n  );\n}\n"]}

package/dist/use-mosaic.d.ts

@@ -0,0 +1,11 @@
+import { Connector } from '@uwdata/mosaic-core';
+/**
+ * Hook to manage the Mosaic connector.
+ *
+ * @returns {Object} An object containing the Mosaic connector and a loading state.
+ */
+export declare function useMosaic(): {
+    isMosaicLoading: boolean;
+    mosaicConnector: Connector | undefined;
+};
+//# sourceMappingURL=use-mosaic.d.ts.map

package/dist/use-mosaic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-mosaic.d.ts","sourceRoot":"","sources":["../src/use-mosaic.ts"],"names":[],"mappings":"AAKA,OAAO,EAAC,SAAS,EAA6B,MAAM,qBAAqB,CAAC;AA8B1E;;;;GAIG;AACH,wBAAgB,SAAS;;;EAUxB"}

package/dist/use-mosaic.js

@@ -0,0 +1,42 @@
+import { useDuckDb, isWasmDuckDbConnector, } from '@sqlrooms/duckdb';
+import { coordinator, wasmConnector } from '@uwdata/mosaic-core';
+import { useEffect, useState } from 'react';
+let mosaicConnector;
+// TODO: Create MosaicSlice and keep the connector in the store
+/**
+ * Retrieves a Mosaic connector for the DuckDB database.
+ *
+ * @returns {Promise<MosaicConnector>} The Mosaic connector for the DuckDB database.
+ */
+async function getMosaicConnector(duckDb) {
+    if (mosaicConnector) {
+        return mosaicConnector;
+    }
+    if (!isWasmDuckDbConnector(duckDb)) {
+        throw new Error('Only WasmDuckDbConnector is currently supported');
+    }
+    await duckDb.initialize();
+    mosaicConnector = await coordinator().databaseConnector(wasmConnector({
+        // @ts-expect-error - We install a different version of duckdb-wasm
+        duckDb: duckDb.getDb(),
+        connection: duckDb.getConnection(),
+    }));
+    return mosaicConnector;
+}
+/**
+ * Hook to manage the Mosaic connector.
+ *
+ * @returns {Object} An object containing the Mosaic connector and a loading state.
+ */
+export function useMosaic() {
+    const [isLoading, setIsLoading] = useState(true);
+    const [connector, setConnector] = useState();
+    const duckDb = useDuckDb();
+    useEffect(() => {
+        getMosaicConnector(duckDb)
+            .then(setConnector)
+            .finally(() => setIsLoading(false));
+    }, [duckDb]);
+    return { isMosaicLoading: isLoading, mosaicConnector: connector };
+}
+//# sourceMappingURL=use-mosaic.js.map

package/dist/use-mosaic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-mosaic.js","sourceRoot":"","sources":["../src/use-mosaic.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,SAAS,EACT,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAY,WAAW,EAAE,aAAa,EAAC,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EAAC,SAAS,EAAE,QAAQ,EAAC,MAAM,OAAO,CAAC;AAE1C,IAAI,eAA0B,CAAC;AAE/B,+DAA+D;AAE/D;;;;GAIG;AACH,KAAK,UAAU,kBAAkB,CAAC,MAAuB;IACvD,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,eAAe,CAAC;IACzB,CAAC;IACD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;IAC1B,eAAe,GAAG,MAAM,WAAW,EAAE,CAAC,iBAAiB,CACrD,aAAa,CAAC;QACZ,mEAAmE;QACnE,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE;QACtB,UAAU,EAAE,MAAM,CAAC,aAAa,EAAE;KACnC,CAAC,CACH,CAAC;IACF,OAAO,eAAe,CAAC;AACzB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS;IACvB,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,EAAa,CAAC;IACxD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,SAAS,CAAC,GAAG,EAAE;QACb,kBAAkB,CAAC,MAAM,CAAC;aACvB,IAAI,CAAC,YAAY,CAAC;aAClB,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;IACxC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IACb,OAAO,EAAC,eAAe,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAC,CAAC;AAClE,CAAC","sourcesContent":["import {\n  DuckDbConnector,\n  useDuckDb,\n  isWasmDuckDbConnector,\n} from '@sqlrooms/duckdb';\nimport {Connector, coordinator, wasmConnector} from '@uwdata/mosaic-core';\nimport {useEffect, useState} from 'react';\n\nlet mosaicConnector: Connector;\n\n// TODO: Create MosaicSlice and keep the connector in the store\n\n/**\n * Retrieves a Mosaic connector for the DuckDB database.\n *\n * @returns {Promise<MosaicConnector>} The Mosaic connector for the DuckDB database.\n */\nasync function getMosaicConnector(duckDb: DuckDbConnector) {\n  if (mosaicConnector) {\n    return mosaicConnector;\n  }\n  if (!isWasmDuckDbConnector(duckDb)) {\n    throw new Error('Only WasmDuckDbConnector is currently supported');\n  }\n  await duckDb.initialize();\n  mosaicConnector = await coordinator().databaseConnector(\n    wasmConnector({\n      // @ts-expect-error - We install a different version of duckdb-wasm\n      duckDb: duckDb.getDb(),\n      connection: duckDb.getConnection(),\n    }),\n  );\n  return mosaicConnector;\n}\n\n/**\n * Hook to manage the Mosaic connector.\n *\n * @returns {Object} An object containing the Mosaic connector and a loading state.\n */\nexport function useMosaic() {\n  const [isLoading, setIsLoading] = useState(true);\n  const [connector, setConnector] = useState<Connector>();\n  const duckDb = useDuckDb();\n  useEffect(() => {\n    getMosaicConnector(duckDb)\n      .then(setConnector)\n      .finally(() => setIsLoading(false));\n  }, [duckDb]);\n  return {isMosaicLoading: isLoading, mosaicConnector: connector};\n}\n"]}

package/dist/useMosaicClient.d.ts

@@ -1,6 +1,7 @@
 import { Selection } from '@uwdata/mosaic-core';
 import { Query } from '@uwdata/mosaic-sql';
-export type UseMosaicClientOptions<T = unknown> = {
+import type { Table as ArrowTable } from 'apache-arrow';
+export type UseMosaicClientOptions = {
     /** Unique id for this client (auto-generated if not provided) */
     id?: string;
     /** Selection name for cross-filtering (will create if doesn't exist) */
@@ -10,12 +11,12 @@ export type UseMosaicClientOptions<T = unknown> = {
     /** Query builder - receives current filter predicate */
     query: (filter: unknown) => ReturnType<typeof Query.from>;
     /** Callback when query results are received */
-    queryResult?: (result: T) => void;
+    queryResult?: (result: ArrowTable) => void;
     /** Whether to automatically connect when mosaic is ready */
     enabled?: boolean;
 };
-export declare function useMosaicClient<T = unknown>(options: UseMosaicClientOptions<T>): {
-    data: NonNullable<T> | null;
+export declare function useMosaicClient(options: UseMosaicClientOptions): {
+    data: ArrowTable<any> | null;
     isLoading: boolean;
     client: import("@uwdata/mosaic-core").MosaicClient | null;
 };

package/dist/useMosaicClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useMosaicClient.d.ts","sourceRoot":"","sources":["../src/useMosaicClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,SAAS,EAAC,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAC,KAAK,EAAC,MAAM,oBAAoB,CAAC;AAKzC,MAAM,MAAM,sBAAsB,CAAC,CAAC,GAAG,OAAO,IAAI;IAChD,iEAAiE;IACjE,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,wEAAwE;IACxE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,mCAAmC;IACnC,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,wDAAwD;IACxD,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,UAAU,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1D,+CAA+C;IAC/C,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,IAAI,CAAC;IAClC,4DAA4D;IAC5D,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,CAAC;AAEF,wBAAgB,eAAe,CAAC,CAAC,GAAG,OAAO,EACzC,OAAO,EAAE,sBAAsB,CAAC,CAAC,CAAC;;;;EA+DnC"}
+{"version":3,"file":"useMosaicClient.d.ts","sourceRoot":"","sources":["../src/useMosaicClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,SAAS,EAAC,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAC,KAAK,EAAC,MAAM,oBAAoB,CAAC;AACzC,OAAO,KAAK,EAAC,KAAK,IAAI,UAAU,EAAC,MAAM,cAAc,CAAC;AAMtD,MAAM,MAAM,sBAAsB,GAAG;IACnC,iEAAiE;IACjE,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,wEAAwE;IACxE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,mCAAmC;IACnC,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,wDAAwD;IACxD,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,UAAU,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1D,+CAA+C;IAC/C,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,IAAI,CAAC;IAC3C,4DAA4D;IAC5D,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,CAAC;AAEF,wBAAgB,eAAe,CAAC,OAAO,EAAE,sBAAsB;;;;EAwE9D"}