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

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

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 interface TrainStatsQuery {
  startTimeFrom?: string;
  startTimeTo?: string;
  deptId?: string;
  category?: string;
  evalMode?: string;
  groupBy?: 'dept' | 'category' | 'evalMode' | string;
  /** 明细下钻维度编码 */
  dimCode?: string;
}

export interface TrainStatsRow {
  dimCode?: string;
  dimName?: string;
  sessionCount?: number;
  personCount?: number;
  signedCount?: number;
  passedCount?: number;
  signRate?: number;
  passRate?: number;
}

export interface TrainStatsDetailRow {
  recordId?: string;
  recordNo?: string;
  subject?: string;
  category?: string;
  evalMode?: string;
  startTime?: string;
  endTime?: string;
  personCount?: number;
  signedCount?: number;
  passedCount?: number;
  signRate?: number;
  passRate?: number;
}

export function getTrainStatsList(params: TrainStatsQuery) {
  return unwrapData<TrainStatsRow[]>(defHttp.get({ url: `${prefix}/list`, params }));
}

export function getTrainStatsDetail(params: TrainStatsQuery) {
  return unwrapData<TrainStatsDetailRow[]>(defHttp.get({ url: `${prefix}/detail`, params }));
}

export function exportTrainStats(data: TrainStatsQuery) {
  return defHttp.post(
    {
      url: `${prefix}/export`,
      data,
      responseType: 'blob',
    },
    {
      isReturnNativeResponse: true,
      isTransformResponse: false,
    },
  );
}