teableio/teable

import { CellValueType } from '../../models/field/constant';
import type { TypedValue } from '../typed-value';
import type { IFormulaContext } from './common';
import { FormulaFunc, FormulaFuncType, FunctionName } from './common';

abstract class SystemFunc extends FormulaFunc {
  readonly type = FormulaFuncType.System;
}

export class TextAll extends SystemFunc {
  name = FunctionName.TextAll;

  acceptValueType = new Set([CellValueType.String]);

  acceptMultipleValue = true;

  validateParams(params: TypedValue[]) {
    if (params.length !== 1) {
      throw new Error(`${FunctionName.TextAll} only allow 1 param`);
    }
  }

  getReturnType(params: TypedValue[]) {
    if (params[0].isMultiple) {
      return { type: CellValueType.String, isMultiple: true };
    }
    return { type: CellValueType.String };
  }

  eval(params: TypedValue[]): boolean | number | string | (string | null)[] | null {
    const param = params[0];
    if (param.isMultiple) {
      return param.value
        ? (param.value as string[]).map((p) => {
            if (Array.isArray(p)) {
              return p.join(', ');
            }
            return p;
          })
        : null;
    }

    return param.value || null;
  }
}

export class RecordId extends SystemFunc {
  name = FunctionName.RecordId;

  acceptValueType = new Set([CellValueType.String]);

  acceptMultipleValue = true;

  // eslint-disable-next-line @typescript-eslint/no-empty-function
  validateParams() {}

  getReturnType() {
    return { type: CellValueType.String };
  }

  eval(_params: TypedValue<string | null>[], context: IFormulaContext): string {
    return context.record.id;
  }
}

export class AutoNumber extends SystemFunc {
  name = FunctionName.RecordId;

  acceptValueType = new Set([CellValueType.String]);

  acceptMultipleValue = true;

  // eslint-disable-next-line @typescript-eslint/no-empty-function
  validateParams() {}

  getReturnType() {
    return { type: CellValueType.Number };
  }

  eval(_params: TypedValue<string | null>[], context: IFormulaContext): number | null {
    return context.record.autoNumber ?? null;
  }
}