teableio/teable

import { type IBaseQueryOrderBy } from '@teable/openapi';
import type { Knex } from 'knex';
import type { IDbProvider } from '../../../../db-provider/db.provider.interface';
import type { IFieldInstance } from '../../../field/model/factory';

export class QueryOrder {
  parse(
    order: IBaseQueryOrderBy | undefined,
    content: {
      dbProvider: IDbProvider;
      queryBuilder: Knex.QueryBuilder;
      fieldMap: Record<string, IFieldInstance>;
    }
  ): {
    queryBuilder: Knex.QueryBuilder;
    fieldMap: Record<string, IFieldInstance>;
  } {
    const { queryBuilder, fieldMap, dbProvider } = content;
    if (!order) {
      return { queryBuilder, fieldMap };
    }

    dbProvider
      .sortQuery(
        queryBuilder,
        fieldMap,
        order.map((item) => ({
          fieldId: item.column,
          order: item.order,
        }))
      )
      .appendSortBuilder();
    return { queryBuilder, fieldMap };
  }
}