gristlabs_grist-core/app/server/lib/ServerColumnGetters.ts

import { ColumnGetter, ColumnGetters } from 'app/common/ColumnGetters';
import * as gristTypes from 'app/common/gristTypes';
import { safeJsonParse } from 'app/common/gutil';
import { choiceGetter } from 'app/common/SortFunc';
import { Sort } from 'app/common/SortSpec';

/**
 *
 * An implementation of ColumnGetters for the server, currently
 * drawing on the data and metadata prepared for CSV export.
 *
 */
export class ServerColumnGetters implements ColumnGetters {
  private _rowIndices: Map<number, number>;
  private _colIndices: Map<number, string>;

  constructor(rowIds: number[], private _dataByColId: {[colId: string]: any}, private _columns: any[]) {
    this._rowIndices = new Map<number, number>(rowIds.map((rowId, index) => [rowId, index] as [number, number]));
    this._colIndices = new Map<number, string>(_columns.map(col => [col.id, col.colId] as [number, string]));
  }

  public getColGetter(colSpec: Sort.ColSpec): ColumnGetter | null {
    const colRef = Sort.getColRef(colSpec);
    const colId = this._colIndices.get(colRef);
    if (colId === undefined) {
      return null;
    }
    const col = this._dataByColId[colId];
    let getter = (rowId: number) => {
      const idx = this._rowIndices.get(rowId);
      if (idx === undefined) {
        return null;
      }
      return col[idx];
    };
    const details = Sort.specToDetails(colSpec);
    if (details.orderByChoice) {
      const rowModel = this._columns.find(c => c.id == colRef);
      if (rowModel?.type === 'Choice') {
        const choices: string[] = safeJsonParse(rowModel.widgetOptions, {}).choices || [];
        getter = choiceGetter(getter, choices);
      }
    }
    return getter;
  }

  public getManualSortGetter(): ((rowId: number) => any) | null {
    const manualSortCol = this._columns.find(c => c.colId === gristTypes.MANUALSORT);
    if (!manualSortCol) {
      return null;
    }
    return this.getColGetter(manualSortCol.id);
  }
}