@milaboratories/pframes-rs-wasm 0.1.0

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "@milaboratories/pframes-rs-wasm",
+  "version": "0.1.0",
+  "files": [
+    "generated",
+    "src"
+  ],
+  "type": "module",
+  "exports": {
+    ".": "./src/index.ts"
+  },
+  "devDependencies": {
+    "@bytecodealliance/jco": "^1.16.1",
+    "@bytecodealliance/preview2-shim": "0.17.8",
+    "@milaboratories/pl-model-common": "1.24.4",
+    "@milaboratories/pl-model-middle-layer": "1.11.5",
+    "@milaboratories/ts-configs": "1.2.0",
+    "rimraf": "^6.1.2",
+    "typescript": "^5.9.3"
+  },
+  "peerDependencies": {
+    "@bytecodealliance/preview2-shim": "0.17.8",
+    "@milaboratories/pl-model-common": "1.24.4",
+    "@milaboratories/pl-model-middle-layer": "1.11.5"
+  },
+  "scripts": {
+    "component-transpile": "rimraf generated && jco transpile -o generated --no-nodejs-compat --no-namespaced-exports",
+    "build:rust": "cargo build",
+    "build:ts": "node --run component-transpile -- ../../target/wasm32-wasip2/debug/pframes_rs_wasm.wasm",
+    "build": "node --run build:rust && node --run build:ts",
+    "postbuild": "rimraf package.tgz && pnpm pack --out package.tgz",
+    "ci-build:rust": "cargo build -r --locked",
+    "ci-build:ts": "node --run component-transpile -- ../../target/wasm32-wasip2/release/pframes_rs_wasm.wasm -O -- -O4",
+    "ci-build": "node --run ci-build:rust && node --run ci-build:ts",
+    "postci-build": "node --run postbuild",
+    "preci-test": "node --run ci-build",
+    "ci-test": "exit 0",
+    "pretest": "node --run build",
+    "test": "exit 0"
+  }
+}

package/src/axes.ts

@@ -0,0 +1,20 @@
+import type {
+  AxesId,
+  AxesSpec,
+  SingleAxisSelector
+} from '@milaboratories/pl-model-common';
+import { spec as bindings } from '../generated/pframes_rs_wasm.js';
+
+export function expand(spec: AxesSpec): AxesId {
+  return JSON.parse(bindings.Axes.fromJson(JSON.stringify(spec)).expand());
+}
+
+export function collapse(ids: AxesId): AxesSpec {
+  return JSON.parse(bindings.Axes.collapse(JSON.stringify(ids)));
+}
+
+export function find(spec: AxesSpec, selector: SingleAxisSelector): number {
+  return JSON.parse(
+    bindings.Axes.fromJson(JSON.stringify(spec)).find(JSON.stringify(selector))
+  );
+}

package/src/index.ts

@@ -0,0 +1,15 @@
+import type { PFrameInternal } from '@milaboratories/pl-model-middle-layer';
+import * as axes from './axes';
+import { PFrame } from './p-frame';
+
+export const {
+  createPFrame,
+  expandAxes,
+  collapseAxes,
+  findAxis
+}: PFrameInternal.PFrameWasmAPI = {
+  createPFrame: (spec) => new PFrame(spec),
+  expandAxes: axes.expand,
+  collapseAxes: axes.collapse,
+  findAxis: axes.find
+};

package/src/p-frame.ts

@@ -0,0 +1,110 @@
+import type {
+  JoinEntry,
+  PColumnSpec,
+  PColumnValue,
+  PObjectId,
+  QuerySpec
+} from '@milaboratories/pl-model-common';
+import type { PFrameInternal } from '@milaboratories/pl-model-middle-layer';
+import { spec as bindings } from '../generated/pframes_rs_wasm.js';
+
+export class PFrame implements PFrameInternal.PFrameWasm {
+  #frame: bindings.Frame;
+  #disposables: DisposableStack;
+
+  constructor(spec: Record<string, PColumnSpec>) {
+    using disposer = new DisposableStack();
+    this.#frame = bindings.Frame.fromJson(JSON.stringify(spec));
+    // Unfortunately, jco generates incorrect types
+    // actual spec.PFrame is indeed Disposable
+    disposer.use(this.#frame as unknown as Disposable);
+    this.#disposables = disposer.move();
+  }
+
+  deleteColumns(
+    request: PFrameInternal.DeleteColumnFromColumnsRequest
+  ): PFrameInternal.DeleteColumnFromColumnsResponse {
+    return JSON.parse(bindings.Frame.deleteColumns(JSON.stringify(request)));
+  }
+
+  findColumns(
+    request: PFrameInternal.FindColumnsRequest
+  ): PFrameInternal.FindColumnsResponse {
+    return JSON.parse(this.#frame.findColumns(JSON.stringify(request)));
+  }
+
+  evaluateQuery(request: QuerySpec): PFrameInternal.EvaluateQueryResponse {
+    return JSON.parse(this.#frame.evaluateQuery(JSON.stringify(request)));
+  }
+
+  rewriteLegacyQuery(request: PFrameInternal.LegacyQuery): QuerySpec {
+    const src = joinEntryToInternal(request.src);
+    return JSON.parse(
+      this.#frame.rewriteLegacyQuery(JSON.stringify({ ...request, src }))
+    );
+  }
+
+  [Symbol.dispose](): void {
+    this.#disposables.dispose();
+  }
+}
+
+function joinEntryToInternal(
+  entry: JoinEntry<PObjectId>
+): PFrameInternal.JoinEntryV4 {
+  const type = entry.type;
+  switch (type) {
+    case 'column':
+      return {
+        type: 'column',
+        columnId: entry.column
+      };
+    case 'slicedColumn':
+      return {
+        type: 'slicedColumn',
+        columnId: entry.column,
+        newId: entry.newId,
+        axisFilters: entry.axisFilters
+      };
+    case 'artificialColumn':
+      return {
+        type: 'artificialColumn',
+        columnId: entry.column,
+        newId: entry.newId,
+        axesIndices: entry.axesIndices
+      };
+    case 'inlineColumn':
+      return {
+        type: 'inlineColumn',
+        newId: entry.column.id,
+        spec: entry.column.spec,
+        dataInfo: {
+          type: 'Json',
+          keyLength: entry.column.spec.axesSpec.length,
+          data: entry.column.data.reduce(
+            (acc, row) => {
+              acc[JSON.stringify(row.key)] = row.val;
+              return acc;
+            },
+            {} as Record<string, PColumnValue>
+          )
+        }
+      };
+    case 'inner':
+    case 'full':
+      return {
+        type: entry.type,
+        entries: entry.entries.map((col) => joinEntryToInternal(col))
+      };
+    case 'outer':
+      return {
+        type: 'outer',
+        primary: joinEntryToInternal(entry.primary),
+        secondary: entry.secondary.map((col) => joinEntryToInternal(col))
+      };
+    default:
+      throw new Error(
+        `unsupported PFrame join entry type: ${type satisfies never}`
+      );
+  }
+}