@tanstack/db/dist/esm/query/live/utils.js

A reactive client store for building super fast apps on sync

import { MultiSet } from "@tanstack/db-ivm";
import { UnsupportedRootScalarSelectError } from "../../errors.js";
import { normalizeOrderByPaths } from "../compiler/expressions.js";
import { buildQuery } from "../builder/index.js";
import { collectCollectionSources, isExpressionLike } from "../ir.js";
import { getQueryIR } from "../builder/query-ir.js";
function extractCollectionsFromQuery(query) {
  const collections = {};
  for (const source of collectCollectionSources(query)) {
    collections[source.collection.id] = source.collection;
  }
  return collections;
}
function extractCollectionFromSource(query) {
  const from = query.from;
  if (from.type === `collectionRef`) {
    return from.collection;
  } else if (from.type === `queryRef`) {
    return extractCollectionFromSource(from.query);
  } else if (from.type === `unionFrom`) {
    return extractCollectionFromSource({ from: from.sources[0] });
  } else if (from.type === `unionAll`) {
    return extractCollectionFromSource(from.queries[0]);
  }
  throw new Error(
    `Failed to extract collection. Invalid FROM clause: ${JSON.stringify(query)}`
  );
}
function isNestedSelectObject(obj) {
  if (obj === null || typeof obj !== `object`) return false;
  if (isExpressionLike(obj)) return false;
  if (obj.__refProxy) return false;
  return true;
}
function buildQueryFromConfig(config) {
  const query = typeof config.query === `function` ? buildQuery(config.query) : getQueryIR(config.query);
  if (config.requireObjectResult && query.select && !isNestedSelectObject(query.select)) {
    throw new UnsupportedRootScalarSelectError();
  }
  return query;
}
function sendChangesToInput(input, changes) {
  const multiSetArray = [];
  for (const change of changes) {
    const key = change.key;
    if (change.type === `insert`) {
      multiSetArray.push([[key, change.value], 1]);
    } else if (change.type === `update`) {
      multiSetArray.push([[key, change.previousValue], -1]);
      multiSetArray.push([[key, change.value], 1]);
    } else {
      multiSetArray.push([[key, change.value], -1]);
    }
  }
  if (multiSetArray.length !== 0) {
    input.sendData(new MultiSet(multiSetArray));
  }
  return multiSetArray.length;
}
function* splitUpdates(changes) {
  for (const change of changes) {
    if (change.type === `update`) {
      yield { type: `delete`, key: change.key, value: change.previousValue };
      yield { type: `insert`, key: change.key, value: change.value };
    } else {
      yield change;
    }
  }
}
function reconcileChangesForD2(changes, sentRows) {
  const reconciled = [];
  for (const change of changes) {
    const previousValue = sentRows.get(change.key);
    if (change.type === `insert`) {
      if (previousValue !== void 0) continue;
      sentRows.set(change.key, change.value);
      reconciled.push(change);
    } else if (change.type === `delete`) {
      if (previousValue === void 0) continue;
      sentRows.delete(change.key);
      reconciled.push({ ...change, value: previousValue });
    } else {
      sentRows.set(change.key, change.value);
      reconciled.push(
        previousValue === void 0 ? { type: `insert`, key: change.key, value: change.value } : { ...change, previousValue }
      );
    }
  }
  return reconciled;
}
function computeSubscriptionOrderByHints(query, alias) {
  const { orderBy, limit, offset } = query;
  const effectiveLimit = limit !== void 0 && offset !== void 0 ? limit + offset : limit;
  const normalizedOrderBy = orderBy ? normalizeOrderByPaths(orderBy, alias) : void 0;
  const canPassOrderBy = normalizedOrderBy?.every((clause) => {
    const exp = clause.expression;
    if (exp.type !== `ref`) return false;
    const path = exp.path;
    return Array.isArray(path) && path.length === 1;
  }) ?? false;
  return {
    orderBy: canPassOrderBy ? normalizedOrderBy : void 0,
    limit: canPassOrderBy ? effectiveLimit : void 0
  };
}
export {
  buildQueryFromConfig,
  computeSubscriptionOrderByHints,
  extractCollectionFromSource,
  collectCollectionSources as extractCollectionSources,
  extractCollectionsFromQuery,
  reconcileChangesForD2,
  sendChangesToInput,
  splitUpdates
};
//# sourceMappingURL=utils.js.map