squiggle/packages/squiggle-lang/src/js/SqValue.ts

import * as RSValue from "../rescript/ForTS/ForTS_SquiggleValue/ForTS_SquiggleValue.gen";
import { squiggleValueTag as Tag } from "../rescript/ForTS/ForTS_SquiggleValue/ForTS_SquiggleValue_tag";
import { wrapDistribution } from "./SqDistribution";
import { SqLambda } from "./SqLambda";
import { SqLambdaDeclaration } from "./SqLambdaDeclaration";
import { SqModule } from "./SqModule";
import { SqRecord } from "./SqRecord";
import { SqArray } from "./SqArray";
import { SqType } from "./SqType";

export { Tag as SqValueTag };

type T = RSValue.squiggleValue;

export const wrapValue = (value: T): SqValue => {
  const tag = RSValue.getTag(value);

  return new tagToClass[tag](value);
};

export abstract class SqAbstractValue {
  abstract tag: Tag;
  _value: T;

  constructor(value: T) {
    this._value = value;
  }
}

const valueMethod = <IR>(
  _this: SqAbstractValue,
  rsMethod: (v: T) => IR | null | undefined
) => {
  const value = rsMethod(_this._value);
  if (!value) throw new Error("Internal casting error");
  return value;
};

export class SqArrayValue extends SqAbstractValue {
  tag = Tag.SvtArray as const;

  get value() {
    return new SqArray(valueMethod(this, RSValue.getArray));
  }
}

export class SqArrayStringValue extends SqAbstractValue {
  tag = Tag.SvtArrayString as const;

  get value() {
    return valueMethod(this, RSValue.getArrayString);
  }
}

export class SqBoolValue extends SqAbstractValue {
  tag = Tag.SvtBool as const;

  get value() {
    return valueMethod(this, RSValue.getBool);
  }
}

export class SqCallValue extends SqAbstractValue {
  tag = Tag.SvtCall as const;

  get value() {
    return valueMethod(this, RSValue.getCall);
  }
}

export class SqDateValue extends SqAbstractValue {
  tag = Tag.SvtDate as const;

  get value() {
    return valueMethod(this, RSValue.getDate);
  }
}

export class SqDeclarationValue extends SqAbstractValue {
  tag = Tag.SvtDeclaration as const;

  get value() {
    return new SqLambdaDeclaration(valueMethod(this, RSValue.getDeclaration));
  }
}

export class SqDistributionValue extends SqAbstractValue {
  tag = Tag.SvtDistribution as const;

  get value() {
    return wrapDistribution(valueMethod(this, RSValue.getDistribution));
  }
}

export class SqLambdaValue extends SqAbstractValue {
  tag = Tag.SvtLambda as const;

  get value() {
    return new SqLambda(valueMethod(this, RSValue.getLambda));
  }
}

export class SqModuleValue extends SqAbstractValue {
  tag = Tag.SvtModule as const;

  get value() {
    return new SqModule(valueMethod(this, RSValue.getModule));
  }
}

export class SqNumberValue extends SqAbstractValue {
  tag = Tag.SvtNumber as const;

  get value() {
    return valueMethod(this, RSValue.getNumber);
  }
}

export class SqRecordValue extends SqAbstractValue {
  tag = Tag.SvtRecord as const;

  get value() {
    return new SqRecord(valueMethod(this, RSValue.getRecord));
  }
}

export class SqStringValue extends SqAbstractValue {
  tag = Tag.SvtString as const;

  get value(): string {
    return valueMethod(this, RSValue.getString);
  }
}

export class SqSymbolValue extends SqAbstractValue {
  tag = Tag.SvtSymbol as const;

  get value(): string {
    return valueMethod(this, RSValue.getSymbol);
  }
}

export class SqTimeDurationValue extends SqAbstractValue {
  tag = Tag.SvtTimeDuration as const;

  get value() {
    return valueMethod(this, RSValue.getTimeDuration);
  }
}

export class SqTypeValue extends SqAbstractValue {
  tag = Tag.SvtType as const;

  get value() {
    return new SqType(valueMethod(this, RSValue.getType));
  }
}

export class SqTypeIdentifierValue extends SqAbstractValue {
  tag = Tag.SvtTypeIdentifier as const;

  get value() {
    return valueMethod(this, RSValue.getTypeIdentifier);
  }
}

export class SqVoidValue extends SqAbstractValue {
  tag = Tag.SvtVoid as const;
}

const tagToClass = {
  [Tag.SvtArray]: SqArrayValue,
  [Tag.SvtArrayString]: SqArrayStringValue,
  [Tag.SvtBool]: SqBoolValue,
  [Tag.SvtCall]: SqCallValue,
  [Tag.SvtDate]: SqDateValue,
  [Tag.SvtDeclaration]: SqDeclarationValue,
  [Tag.SvtDistribution]: SqDistributionValue,
  [Tag.SvtLambda]: SqLambdaValue,
  [Tag.SvtModule]: SqModuleValue,
  [Tag.SvtNumber]: SqNumberValue,
  [Tag.SvtRecord]: SqRecordValue,
  [Tag.SvtString]: SqStringValue,
  [Tag.SvtSymbol]: SqSymbolValue,
  [Tag.SvtTimeDuration]: SqTimeDurationValue,
  [Tag.SvtType]: SqTypeValue,
  [Tag.SvtTypeIdentifier]: SqTypeIdentifierValue,
  [Tag.SvtVoid]: SqVoidValue,
} as const;

// FIXME
// type SqValue = typeof tagToClass[keyof typeof tagToClass];
export type SqValue =
  | SqArrayValue
  | SqArrayStringValue
  | SqBoolValue
  | SqCallValue
  | SqDateValue
  | SqDeclarationValue
  | SqDistributionValue
  | SqLambdaValue
  | SqModuleValue
  | SqNumberValue
  | SqRecordValue
  | SqStringValue
  | SqSymbolValue
  | SqTimeDurationValue
  | SqTypeValue
  | SqTypeIdentifierValue
  | SqVoidValue;