package com.skua.modules.ajh.controller;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.skua.core.api.vo.Result;
import com.skua.core.aspect.annotation.AutoLog;
import com.skua.core.context.BaseContextHandler;
import com.skua.core.query.QueryGenerator;
import com.skua.modules.ajh.dto.AjhRectificationInfoAnalyseQueryDto;
import com.skua.modules.ajh.dto.AjhRectificationInfoMendDto;
import com.skua.modules.ajh.entity.AjhRectificationInfo;
import com.skua.modules.ajh.service.IAjhRectificationInfoService;
import com.skua.modules.ajh.vo.RectificationInfoAnalyseVO;
import com.skua.modules.emergency.entity.EmergencyRiskEvent;
import com.skua.modules.flow.business.service.FlowBusinessService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.*;
import java.util.stream.Collectors;

/**
* 隐患检查
*/
@Slf4j
@Api(tags="隐患检查")
@RestController
@RequestMapping("/ajh/ajhRectificationInfo")
public class AjhRectificationInfoController {
   @Autowired
   private IAjhRectificationInfoService ajhRectificationInfoService;

    @Autowired
    private FlowBusinessService flowBusinessService;
   /**
     * 分页列表查询
    * @param ajhRectificationInfo
    * @param pageNo
    * @param pageSize
    * @param req
    * @return
    */
   @AutoLog(value = "隐患检查-分页列表查询")
   @ApiOperation(value="隐患检查-分页列表查询", notes="隐患检查-分页列表查询")
   @GetMapping(value = "/list")
   public Result<IPage<AjhRectificationInfo>> queryPageList(AjhRectificationInfo ajhRectificationInfo,
                                                            @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
                                                            @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
                                                            HttpServletRequest req) {
       Result<IPage<AjhRectificationInfo>> result = new Result<IPage<AjhRectificationInfo>>();
       if(StringUtils.isEmpty(ajhRectificationInfo.getDepartId())){
           ajhRectificationInfo.setDepartId(BaseContextHandler.getDeparts());
       }
       QueryWrapper<AjhRectificationInfo> queryWrapper = QueryGenerator.initQueryWrapper(ajhRectificationInfo, req.getParameterMap());
       queryWrapper.orderByDesc("rec_ord_level");//重大风险排前面
       Page<AjhRectificationInfo> page = new Page<AjhRectificationInfo>(pageNo, pageSize);
       IPage<AjhRectificationInfo> pageList = ajhRectificationInfoService.page(page, queryWrapper);

       //获取流程状态
       List<AjhRectificationInfo> records = pageList.getRecords();
       Set<String> processInstanceIdSet = records.stream().filter(data-> org.apache.commons.lang3.StringUtils.isNotBlank(data.getProcessInstanceId()))
               .map(AjhRectificationInfo::getProcessInstanceId).collect(Collectors.toSet());
       Map<String, Map<String, String>> processInstanceMap;
       try {
           processInstanceMap = flowBusinessService.getProcessState(processInstanceIdSet);
       } catch (Exception e) {
           e.printStackTrace();
           processInstanceMap = new HashMap<>();
       }
       for (AjhRectificationInfo record : records) {
           record.getFlow().setProcessDefinitionKey(processInstanceMap.getOrDefault(record.getProcessInstanceId(), new HashMap<>()).get("processDefinitionKey"));
           record.getFlow().setTaskDefinitionKey(processInstanceMap.getOrDefault(record.getProcessInstanceId(), new HashMap<>()).get("taskDefinitionKey"));
           record.getFlow().setStatus(processInstanceMap.getOrDefault(record.getProcessInstanceId(), new HashMap<>()).get("state"));
           record.getFlow().setTaskName(processInstanceMap.getOrDefault(record.getProcessInstanceId(), new HashMap<>()).get("processDefinitionName"));
       }

       result.setSuccess(true);
       result.setResult(pageList);
       return result;
   }

   /**
     *   添加
    * @param ajhRectificationInfo
    * @return
    */
   @AutoLog(value = "隐患检查-添加(开始流程)")
   @ApiOperation(value="隐患检查-添加", notes="隐患检查-添加(开始流程)")
   @PostMapping(value = "/add")
   public Result<AjhRectificationInfo> add(@RequestBody AjhRectificationInfo ajhRectificationInfo) {
       Result<AjhRectificationInfo> result = new Result<AjhRectificationInfo>();
       try {
           ajhRectificationInfoService.startProcess(ajhRectificationInfo);
           result.success("添加成功！");
       } catch (Exception e) {
           log.error(e.getMessage(),e);
           result.error500("操作失败");
       }
       return result;
   }



    /**
     *  整改
     * @return
     */
    @AutoLog(value = "隐患检查-整改")
    @ApiOperation(value="隐患检查-整改", notes="隐患检查-整改")
    @PostMapping(value = "/mend")
    public Result mend(@Valid @RequestBody AjhRectificationInfoMendDto dto) {

        Result<EmergencyRiskEvent> result = new Result<EmergencyRiskEvent>();
        ajhRectificationInfoService.mend(dto);
        result.success("处理成功!");

        return result;
    }

    /**
     *  整改
     * @return
     */
    @AutoLog(value = "隐患检查-统计分析")
    @ApiOperation(value="隐患检查-统计分析", notes="隐患检查-统计分析")
    @PostMapping(value = "/analyse")
    public Result<RectificationInfoAnalyseVO> analyse(@Valid @RequestBody AjhRectificationInfoAnalyseQueryDto dto) {
        Result<RectificationInfoAnalyseVO> result = new Result<>();
        result.setResult(ajhRectificationInfoService.analyse(dto));
        return result;
    }


    @AutoLog(value = "处理数量统计")
    @ApiOperation(value="处理数量统计", notes="处理数量统计")
    @GetMapping(value = "/getDefectHandleCount")
    public Result<List<Map<String,Object>>> getDefectHandleCount(AjhRectificationInfo ajhRectificationInfo, HttpServletRequest req){
        Result<List<Map<String,Object>>> result = new Result<>();
        QueryWrapper<AjhRectificationInfo> queryWrapper = QueryGenerator.initQueryWrapper(ajhRectificationInfo, req.getParameterMap());
        queryWrapper.select("count(id) count,handle_result handleResult");
        queryWrapper.groupBy("handle_result");
        List<Map<String,Object>> list = ajhRectificationInfoService.listMaps(queryWrapper);
        result.setSuccess(true);
        result.setResult(list);
        return result;
    }


   /**
     *  编辑
    * @param ajhRectificationInfo
    * @return
    */
   @AutoLog(value = "隐患检查-编辑")
   @ApiOperation(value="隐患检查-编辑", notes="隐患检查-编辑")
   @PutMapping(value = "/edit")
   public Result<AjhRectificationInfo> edit(@RequestBody AjhRectificationInfo ajhRectificationInfo) {
       Result<AjhRectificationInfo> result = new Result<AjhRectificationInfo>();
       AjhRectificationInfo ajhRectificationInfoEntity = ajhRectificationInfoService.getById(ajhRectificationInfo.getId());
       if(ajhRectificationInfoEntity==null) {
           result.error500("未找到对应实体");
       }else {
           boolean ok = ajhRectificationInfoService.updateById(ajhRectificationInfo);
           //TODO 返回false说明什么？
           if(ok) {
               result.success("修改成功!");
           }
       }

       return result;
   }

   /**
     *   通过id删除
    * @param id
    * @return
    */
   @AutoLog(value = "隐患检查-通过id删除")
   @ApiOperation(value="隐患检查-通过id删除", notes="隐患检查-通过id删除")
   @DeleteMapping(value = "/delete")
   public Result<?> delete(@RequestParam(name="id",required=true) String id) {
       try {
           ajhRectificationInfoService.removeById(id);
       } catch (Exception e) {
           log.error("删除失败",e.getMessage());
           return Result.error("删除失败!");
       }
       return Result.ok("删除成功!");
   }

   /**
     *  批量删除
    * @param ids
    * @return
    */
   @AutoLog(value = "隐患检查-批量删除")
   @ApiOperation(value="隐患检查-批量删除", notes="隐患检查-批量删除")
   @DeleteMapping(value = "/deleteBatch")
   public Result<AjhRectificationInfo> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
       Result<AjhRectificationInfo> result = new Result<AjhRectificationInfo>();
       if(ids==null || "".equals(ids.trim())) {
           result.error500("参数不识别！");
       }else {
           this.ajhRectificationInfoService.removeByIds(Arrays.asList(ids.split(",")));
           result.success("删除成功!");
       }
       return result;
   }

   /**
     * 通过id查询
    * @param id
    * @return
    */
   @AutoLog(value = "隐患检查-通过id查询")
   @ApiOperation(value="隐患检查-通过id查询", notes="隐患检查-通过id查询")
   @GetMapping(value = "/queryById")
   public Result<AjhRectificationInfo> queryById(@RequestParam(name="id",required=true) String id) {
       Result<AjhRectificationInfo> result = new Result<AjhRectificationInfo>();
       AjhRectificationInfo ajhRectificationInfo = ajhRectificationInfoService.getById(id);
       if(ajhRectificationInfo==null) {
           result.error500("未找到对应实体");
       }else {
           result.setResult(ajhRectificationInfo);
           result.setSuccess(true);
       }
       return result;
   }

 /**
     * 导出excel
  *
  * @param request
  * @param response
  */
 @RequestMapping(value = "/exportXls")
 public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) {
     // Step.1 组装查询条件
     QueryWrapper<AjhRectificationInfo> queryWrapper = null;
     try {
         String paramsStr = request.getParameter("paramsStr");
         if (org.apache.commons.lang3.StringUtils.isNotEmpty(paramsStr)) {
             String deString = URLDecoder.decode(paramsStr, "UTF-8");
             AjhRectificationInfo ajhRectificationInfo = JSON.parseObject(deString, AjhRectificationInfo.class);
             queryWrapper = QueryGenerator.initQueryWrapper(ajhRectificationInfo, request.getParameterMap());
         }
     } catch (UnsupportedEncodingException e) {
         e.printStackTrace();
     }

     //Step.2 AutoPoi 导出Excel
     ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
     List<AjhRectificationInfo> pageList = ajhRectificationInfoService.list(queryWrapper);
     //导出文件名称
     mv.addObject(NormalExcelConstants.FILE_NAME, "隐患检查列表");
     mv.addObject(NormalExcelConstants.CLASS, AjhRectificationInfo.class);
     mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("隐患检查列表数据", "导出人:Jeecg", "导出信息"));
     mv.addObject(NormalExcelConstants.DATA_LIST, pageList);
     return mv;
 }

 /**
     * 通过excel导入数据
  *
  * @param request
  * @param response
  * @return
  */
 @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
 public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
     MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
     Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
     for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
         MultipartFile file = entity.getValue();// 获取上传文件对象
         ImportParams params = new ImportParams();
         params.setTitleRows(2);
         params.setHeadRows(1);
         params.setNeedSave(true);
         try {
             List<AjhRectificationInfo> listAjhRectificationInfos = ExcelImportUtil.importExcel(file.getInputStream(), AjhRectificationInfo.class, params);
             ajhRectificationInfoService.saveBatch(listAjhRectificationInfos);
             return Result.ok("文件导入成功！数据行数:" + listAjhRectificationInfos.size());
         } catch (Exception e) {
             log.error(e.getMessage(),e);
             return Result.error("文件导入失败:"+e.getMessage());
         } finally {
             try {
                 file.getInputStream().close();
             } catch (IOException e) {
                 e.printStackTrace();
             }
         }
     }
     return Result.ok("文件导入失败！");
 }


    /*自定义接口*/

    /**
     * 安全态势总览数据统计
     * @param now   查询时间
     * @param paramType     查询类型
     * @return
     */
    @RequestMapping(value = "/searchCenus", method = RequestMethod.GET)
    public Map<String,Object> searchSafeCensus(@RequestParam(name="now",required=true) String now,@RequestParam(name="paramType") String paramType) {
        Map<String, Object> result = ajhRectificationInfoService.searchSafeCensus(now, paramType);
        return result;
    }
}