@platforma-sdk/model 1.45.0 → 1.45.17

package/src/render/util/pcolumn_data.ts

@@ -1,17 +1,21 @@
 import type {
   DataInfo,
   PartitionedDataInfoEntries,
+  PColumn,
+  PColumnValues,
 } from '@milaboratories/pl-model-common';
 import {
   dataInfoToEntries,
   isDataInfo,
   isDataInfoEntries,
+  visitDataInfo,
   type BinaryChunk,
   type DataInfoEntries,
   type PColumnDataEntry,
   type PColumnKey,
 } from '@milaboratories/pl-model-common';
 import { TreeNodeAccessor } from '../accessor';
+import type { PColumnDataUniversal } from '../api';
 
 const PCD_PREFIX = 'PColumnData/';
 
@@ -507,3 +511,23 @@ DataInfoEntries<TreeNodeAccessor> | undefined {
 
   throw new Error(`Unexpected input type: ${typeof acc}`);
 }
+
+export function isPColumnReady(c: PColumn<PColumnDataUniversal>): boolean {
+  const isValues = (d: PColumnDataUniversal): d is PColumnValues => Array.isArray(d);
+  const isAccessor = (d: PColumnDataUniversal): d is TreeNodeAccessor => d instanceof TreeNodeAccessor;
+
+  let ready = true;
+  if (isAccessor(c.data)) {
+    ready &&= c.data.getIsReadyOrError();
+  } else if (isDataInfo(c.data)) {
+    visitDataInfo(c.data, (v) => ready &&= v.getIsReadyOrError());
+  } else if (!isValues(c.data)) {
+    // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
+    throw Error(`unsupported column data type: ${c.data satisfies never}`);
+  }
+  return ready;
+}
+
+export function allPColumnsReady(columns: PColumn<PColumnDataUniversal>[]): boolean {
+  return columns.every(isPColumnReady);
+}