import type {
  TrainRecordCatalogDetail,
  TrainRecordCatalogQuery,
  TrainRecordPageQuery,
} from '#/views/x/tms/trainRecord/types';

import { defHttp } from '#/api/request';

const prefix = '/api/tms/train-record';

async function unwrapData<T>(promise: Promise<any>): Promise<T> {
  const res = await promise;
  if (res && typeof res === 'object' && 'data' in res && ('code' in res || 'msg' in res)) {
    return res.data as T;
  }
  return res as T;
}

export function getTrainRecordList(params: TrainRecordPageQuery) {
  return unwrapData<{ list: any[]; pagination: Record<string, any> }>(
    defHttp.get({ url: `${prefix}/list`, params }),
  );
}

export function getTrainRecordCatalog(params: TrainRecordCatalogQuery) {
  const { recordId, startDate, endDate } = params;
  return unwrapData<TrainRecordCatalogDetail>(
    defHttp.get({
      url: `${prefix}/${recordId}/catalog`,
      params: { startDate, endDate },
    }),
  );
}