package com.ruoyi.platform.controller.dataset;

import com.ruoyi.common.core.web.controller.BaseController;
import com.ruoyi.common.core.web.domain.GenericsAjaxResult;
import com.ruoyi.platform.domain.DatasetVersion;
import com.ruoyi.platform.service.DatasetVersionService;
import com.ruoyi.platform.vo.LabelDatasetVersionVo;
import io.swagger.annotations.ApiOperation;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;
import java.util.Map;

/**
 * (DatasetVersion)表控制层
 *
 * @author Xidaray
 * @since 2024-01-03 16:55:19
 */
@RestController
@RequestMapping("datasetVersion")
@ApiOperation(value = "数据集版本管理")
public class DatasetVersionController extends BaseController {
    /**
     * 服务对象
     */
    @Resource
    private DatasetVersionService datasetVersionService;

    /**
     * 分页查询
     *
     * @param datasetVersion 筛选条件
     * @param page           页数
     * @param size           大小
     * @return 查询结果
     */
    @GetMapping
    @ApiOperation("分页查询")
    public GenericsAjaxResult<Page<DatasetVersion>> queryByPage(DatasetVersion datasetVersion, int page, int size) {
        PageRequest pageRequest = PageRequest.of(page,size);
        return  genericsSuccess(this.datasetVersionService.queryByPage(datasetVersion, pageRequest));
    }


    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    @ApiOperation("根据id查询数据集版本")
    public GenericsAjaxResult<DatasetVersion> queryById(@PathVariable("id") Integer id) {
        return  genericsSuccess(this.datasetVersionService.queryById(id));
    }

    /**
     * 通过数据集id和version查询版本列表
     *
     * @param datasetId 数据集ID
     * @param version   数据集版本
     * @return 匹配的数据集版本记录列表
     */
    @GetMapping("/versions")
    @ApiOperation("通过数据集id和version查询版本文件列表")
    public GenericsAjaxResult<Map<String,Object>> queryByDatasetIdAndVersion(@RequestParam("dataset_id") Integer datasetId,
                                                                               @RequestParam("version") String version) {
        return genericsSuccess(this.datasetVersionService.queryByDatasetIdAndVersion(datasetId, version));
    }


    /**
     * 新增单条数据
     *
     * @param datasetVersion 实体
     * @return 新增结果
     */
    @PostMapping
    @ApiOperation("添加数据集版本")
    public GenericsAjaxResult<DatasetVersion> add(@RequestBody DatasetVersion datasetVersion) throws Exception {
        return  genericsSuccess(this.datasetVersionService.insert(datasetVersion));
    }

    /**
     * 批量新增数据集版本
     *
     * @param datasetVersions 实体
     * @return 新增结果
     */
    @PostMapping("/addDatasetVersions")
    @ApiOperation("添加数据集版本")
    public GenericsAjaxResult<String> addDatasetVersions(@RequestBody List<DatasetVersion> datasetVersions) throws Exception {
        return genericsSuccess(this.datasetVersionService.addDatasetVersions(datasetVersions));
    }


    /**
     * 编辑数据
     *
     * @param datasetVersion 实体
     * @return 编辑结果
     */
    @PutMapping
    @ApiOperation("编辑数据集版本")
    public GenericsAjaxResult<DatasetVersion> edit(@RequestBody DatasetVersion datasetVersion) {
        return  genericsSuccess(this.datasetVersionService.update(datasetVersion));
    }

    /**
     * 删除数据
     *
     * @param id 主键
     * @return 删除是否成功
     */
    @DeleteMapping({"{id}"})
    @ApiOperation("删除数据集版本")
    public GenericsAjaxResult<String> deleteById(@PathVariable("id") Integer id) throws Exception {
        return  genericsSuccess(this.datasetVersionService.removeById(id));
    }

    /**
     * 删除版本下的所有数据
     *
     * @param datasetId 模型主键
     * @param version   版本
     * @return 删除是否成功
     */
    @DeleteMapping("/deleteVersion")
    @ApiOperation(value = "逻辑删除模型版本", notes = "根据数据集ID和版本逻辑删除模型版本记录。")
    public GenericsAjaxResult<Map<Integer, String>> deleteDatasetVersion(@RequestParam("dataset_id") Integer datasetId,
                                                                         @RequestParam("version") String version) throws Exception {
        return genericsSuccess(this.datasetVersionService.deleteDatasetVersion(datasetId, version));
    }

//    @PostMapping("/addDatasetVersionsFromLabel")
//    @ApiOperation("从数据标注添加数据集版本")
//    public GenericsAjaxResult<?> addDatasetVersionsFromLabel(@RequestBody LabelDatasetVersionVo labelDatasetVersionVo) throws Exception {
//        datasetVersionService.addDatasetVersionsFromLabel(labelDatasetVersionVo);
//
//        return GenericsAjaxResult.success();
//    }
}

