temporeest/packages/query-runtime-ts/src/QueryFactory.ts

import {
  Context,
  IModel,
  INode,
  ModelSpecWithCreate,
  NodeSpecWithCreate,
} from '@aphro/context-runtime-ts';
import { EdgeSpec } from '@aphro/schema-api';
import { assertUnreachable } from '@strut/utils';
import MemorySourceQuery from './memory/MemorySourceQuery.js';
import MemoryHopQuery from './memory/MemoryHopQuery.js';
import { DerivedQuery, HopQuery, Query } from './Query.js';
import SQLHopQuery from './sql/SQLHopQuery.js';
import SQLSourceQuery from './sql/SQLSourceQuery.js';

// Runtime factory so we can swap to `Wire` when running on a client vs
// the native platform.
const factory = {
  createSourceQueryFor<T extends IModel<{}>>(
    ctx: Context,
    spec: ModelSpecWithCreate<T, {}>,
  ): Query<T> {
    switch (spec.storage.type) {
      case 'sql':
        return new SQLSourceQuery(ctx, spec);
      case 'memory':
        return new MemorySourceQuery(ctx, spec);
      default:
        throw new Error(spec.storage.type + ' is not yet supported');
    }
  },

  // TODO: get types into the edge specs so our hop and have types?
  createHopQueryFor<TDest>(
    ctx: Context,
    priorQuery: DerivedQuery<any>,
    edge: EdgeSpec,
  ): HopQuery<any, any> {
    const type = edge.dest.storage.type;
    switch (type) {
      case 'sql':
        return SQLHopQuery.create(ctx, priorQuery, edge);
      case 'memory':
        return MemoryHopQuery.create(ctx, priorQuery, edge);
      case 'ephemeral':
        throw new Error(`Hops for ${type} are not implemented yet`);
    }
    assertUnreachable(type);
  },
};

export default factory;