temporeest/packages/query-runtime-ts/src/sql/SQLSourceExpression.ts

import { SourceExpression } from '../Expression.js';
import SQLSourceChunkIterable from './SQLSourceChunkIterable.js';
import Plan from '../Plan.js';
import { ChunkIterable } from '../ChunkIterable.js';
import HopPlan from '../HopPlan.js';
import { specToDatasetKey } from '@aphro/model-runtime-ts';
import { Context, IModel, INode } from '@aphro/context-runtime-ts';
import SQLExpression, { HoistedOperations } from './SQLExpression.js';
import { JunctionEdgeSpec, NodeSpec } from '@aphro/schema-api';

export interface SQLResult {}

export default class SQLSourceExpression<T extends IModel<Object>>
  extends SQLExpression<T>
  implements SourceExpression<T>
{
  constructor(
    ctx: Context,
    // we should take a schema instead of db
    // we'd need the schema to know if we can hoist certain fields or not
    public readonly spec: NodeSpec | JunctionEdgeSpec,
    ops: HoistedOperations,
  ) {
    super(ctx, ops);
  }

  optimize(plan: Plan, nextHop?: HopPlan): Plan {
    const [hoistedExpressions, remainingExpressions] = this.hoist(plan, nextHop);
    return new Plan(
      new SQLSourceExpression(this.ctx, this.spec, hoistedExpressions),
      remainingExpressions,
    );
  }

  get iterable(): ChunkIterable<T> {
    return new SQLSourceChunkIterable(this.ctx, this.spec, this.ops);
  }

  implicatedDataset(): string {
    return specToDatasetKey(this.spec);
  }
}