BaseSegment.ts

import type { BaseQuery } from './BaseQuery';
import { CubeSymbols } from "../compiler/CubeSymbols";

export class BaseSegment {
  public readonly expression: any;

  public readonly expressionCubeName: any;

  public readonly expressionName: any;

  public readonly isMemberExpression: boolean = false;

  public readonly joinHint: Array<string> = [];

  public constructor(
    protected readonly query: BaseQuery,
    public readonly segment: string | any
  ) {
    if (segment.expression) {
      this.expression = segment.expression;
      this.expressionCubeName = segment.cubeName;
      // In case of SQL push down expressionName doesn't contain cube name. It's just a column name.
      this.expressionName = segment.expressionName || `${segment.cubeName}.${segment.name}`;
      this.isMemberExpression = !!segment.definition;
    } else {
      // TODO move this `as` to static types
      const segmentPath = segment as string;
      const { path, joinHint } = CubeSymbols.joinHintFromPath(segmentPath);
      this.segment = path;
      this.joinHint = joinHint;
    }
  }

  public filterToWhere() {
    return this.segmentSql();
  }

  public segmentSql() {
    if (this.expression) {
      return this.query.evaluateSymbolSql(this.expressionCubeName, this.expressionName, this.definition(), 'segment');
    }
    return this.query.segmentSql(this);
  }

  public filterParams() {
    return [];
  }

  public segmentDefinition() {
    return this.query.cubeEvaluator.segmentByPath(this.segment);
  }

  public isMultiStage() {
    if (this.expression) { // TODO
      return false;
    }
    return this.definition().multiStage;
  }

  public definition(): any {
    if (this.expression) {
      return {
        sql: this.expression
      };
    }
    return this.segmentDefinition();
  }

  public getMembers() {
    return [this];
  }

  public cube() {
    if (this.expression) {
      return this.query.cubeEvaluator.cubeFromPath(this.expressionCubeName);
    }
    return this.query.cubeEvaluator.cubeFromPath(this.segment);
  }

  public sqlDefinition() {
    return this.segmentDefinition().sql;
  }

  public path() {
    if (this.expression) {
      return null;
    }
    return this.query.cubeEvaluator.parsePath('segments', this.segment);
  }

  public expressionPath() {
    if (this.expression) {
      return `expr:${this.expression.expressionName}`;
    }
    return this.query.cubeEvaluator.pathFromArray(this.path());
  }
}