import request from '@/utils/request'
import type { AxiosResponse } from 'axios'
import type { 
  PreprocessMethod,
  FeatureMethod,
  ProcessRequest,
  CSVRequest,
  DatasetInfo,
  MethodDetail
} from '@/types/data'

// 获取预处理方法列表
export const getPreprocessMethods = (): Promise<{
  status: string;
  methods: PreprocessMethod[];
}> => {
  return request({
    url: '/data/preprocessing/methods',
    method: 'get'
  })
}

// 获取预处理方法详情
export const getPreprocessMethodDetails = (methodName: string): Promise<{
  status: string;
  method: MethodDetail;
}> => {
  return request({
    url: `/data/preprocessing/method/${methodName}`,
    method: 'get'
  })
}

// 获取特征工程方法列表
export const getFeatureMethods = (): Promise<{
  status: string;
  methods: FeatureMethod[];
}> => {
  return request({
    url: '/data/feature/methods',
    method: 'get'
  })
}

// 获取特征工程方法详情
export const getFeatureMethodDetails = (methodName: string): Promise<{
  status: string;
  method: MethodDetail;
}> => {
  return request({
    url: `/data/feature/method/${methodName}`,
    method: 'get'
  })
}

// 处理数据集
export const processDataset = (data: ProcessRequest): Promise<{
  status: string;
  data: any;
}> => {
  return request({
    url: '/data/process',
    method: 'post',
    data
  })
}

// 获取数据集列表
export const getDatasets = (): Promise<{
  status: string;
  datasets: DatasetInfo[];
}> => {
  return request({
    url: '/data/datasets',
    method: 'get'
  })
}

// 读取CSV文件
export const readCSV = (data: CSVRequest): Promise<{
  status: string;
  data: any;
}> => {
  return request({
    url: '/data/csv',
    method: 'post',
    data
  })
}