Class Literal

frontend/src/utils/python-poet/poet.ts:43–119

}

export class Literal implements PythonCode {
  public readonly value: unknown;
  public readonly opts: LiteralOptions;
  constructor(value: unknown, opts: LiteralOptions = {}) {
    this.value = value;
    this.opts = opts;
  }

  static from(value: unknown, opts: LiteralOptions = {}): Literal {
    return new Literal(value, opts);
  }

  toCode(): string {
    const EMPTY_VALUE = "";
    const { removeNull = false, removeUndefined = true } = this.opts;

    if (this.value === undefined) {
      if (removeUndefined) {
        return EMPTY_VALUE;
      }
      return "None";
    }
    if (this.value === null) {
      if (removeNull) {
        return EMPTY_VALUE;
      }
      return "None";
    }
    if (typeof this.value === "string") {
      return `'${this.value}'`;
    }
    if (typeof this.value === "boolean") {
      return this.value ? "True" : "False";
    }

    // If a PythonCode object
    if (typeof this.value === "object" && "toCode" in this.value) {
      return (this.value as PythonCode).toCode();
    }

    if (Array.isArray(this.value)) {
      if (this.value.length === 0) {
        return "[]";
      }
      return `[\n${indent(
        this.value
          .map((item) => new Literal(item, this.opts).toCode())
          .filter((item) => item !== EMPTY_VALUE)
          .join(",\n"),
      )}\n]`;
    }

    if (typeof this.value === "object") {
      const entries = Object.entries(this.value as Record<string, unknown>);
      if (entries.length === 0) {
        return "{}";
      }