package com.gkhy.labRiskManage.api.controller.basic;

import cn.hutool.json.JSONObject;
import cn.hutool.poi.excel.ExcelUtil;
import com.gkhy.labRiskManage.api.controller.basic.dto.repDto.BasicExperimentDeviceInsertReqBO;
import com.gkhy.labRiskManage.api.controller.basic.dto.repDto.BasicExperimentDeviceUpdateReqBO;
import com.gkhy.labRiskManage.api.controller.basic.dto.repDto.OldRiskAssessQueryReqBO;
import com.gkhy.labRiskManage.api.controller.basic.dto.respDto.*;
import com.gkhy.labRiskManage.api.controller.common.BaseController;
import com.gkhy.labRiskManage.application.basic.dto.dto.*;
import com.gkhy.labRiskManage.commons.domain.Result;
import com.gkhy.labRiskManage.commons.domain.SearchResult;
import com.gkhy.labRiskManage.commons.enums.ResultCode;
import com.gkhy.labRiskManage.commons.exception.BusinessException;
import com.gkhy.labRiskManage.commons.utils.BeanCopyUtils;
import com.gkhy.labRiskManage.domain.basic.entity.OldRiskAssess;
import com.gkhy.labRiskManage.domain.basic.service.OldRiskAssessService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.FileInputStream;

/**
 * 风险评价因子
 */
@RestController
@RequestMapping("basic")
public class OldRiskAssessController extends BaseController {

    @Autowired
    private OldRiskAssessService oldRiskAssessService;


    @PostMapping("/insert/insertOldRiskAssess")
    public Result insertOldRiskAssess(@RequestBody OldRiskAssess oldRiskAssess){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("新增成功");

        int insertResult = oldRiskAssessService.insertOldRiskAssess(getCurrentUserId(), oldRiskAssess);

        result.setCount(insertResult);
        if (insertResult < 1){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("新增失败");
        }
        result.setCount(insertResult);

        return result;
    }

    @PostMapping("/select/getOldRiskAssessPage")
    public SearchResult<OldRiskAssessQueryRespDTO> getOldRiskAssessPage(@RequestBody OldRiskAssessQueryReqBO queryReqDO){

        SearchResult result = new SearchResult();
        result.setCode(ResultCode.OK);
        result.setMsg("查询成功");

        SearchResult<OldRiskAssessQueryRespDTO> queryDTO = oldRiskAssessService.getOldRiskAssessPage(getCurrentUserId(), queryReqDO);

        Object data = queryDTO.getData();

        result.setPageIndex(queryDTO.getPageIndex());
        result.setPageSize(queryDTO.getPageSize());
        result.setTotal(queryDTO.getTotal());
        result.setCount(queryDTO.getTotal().intValue());

        result.setData(BeanCopyUtils.copyDeviceQueryRespList(data, BasicExperimentDeviceQueryRespDTO.class));

        return result;
    }

    @PostMapping("/update/updateOldRiskAssess")
    public Result updateOldRiskAssess(@RequestBody OldRiskAssess oldRiskAssess){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("修改成功");

        int updateResult = oldRiskAssessService.updateOldRiskAssess(getCurrentUserId(), oldRiskAssess);

        if (updateResult < 1){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("修改失败");
        }
        result.setCount(updateResult);

        return result;
    }

    @PostMapping("/delete/deleteOldRiskAssess")
    public Result deleteOldRiskAssess(@RequestBody JSONObject jsonObject){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("删除成功");

        int deleteResult = oldRiskAssessService.deleteOldRiskAssess(getCurrentUserId(), jsonObject.getLong("id"));

        if (deleteResult < 1){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("删除失败");
        }
        result.setCount(deleteResult);

        return result;
    }

    @PostMapping("/insert/importOldRiskAssess")
    public Result importOldRiskAssess(@RequestParam("file") MultipartFile file){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("导入成功");

        if (file.isEmpty()){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("文件为空");
            return result;
        }

        int importResult = oldRiskAssessService.importOldRiskAssess(getCurrentUserId(), file);

        if (importResult < 1){
            result.setCode(ResultCode.NOT_OK);
            result.setMsg("导入失败");
        }
        result.setCount(importResult);

        return result;
    }

    @PostMapping("/select/exportOldRiskAssess")
    public Result exportOldRiskAssess(@RequestBody OldRiskAssessQueryReqBO queryReqDO, HttpServletResponse response, HttpServletRequest request){

        Result result = new Result();
        result.setCode(ResultCode.OK);
        result.setMsg("导出成功");


        Result<OldRiskAssessExportRespDTO>  importResult = oldRiskAssessService.exportOldRiskAssess(getCurrentUserId(), queryReqDO, response, request);

//        if (importResult < 1){
//            result.setCode(ResultCode.NOT_OK);
//            result.setMsg("导出失败");
//        }
        result.setCount(importResult.getCount());
        result.setData(importResult.getData());

        return result;
    }

}