package org.jeecg.modules.eam.controller;

import java.io.UnsupportedEncodingException;
import java.io.IOException;
import java.net.URLDecoder;
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.util.*;
import java.util.stream.Collectors;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.util.DateUtils;
import org.jeecg.common.util.ImportExcelUtil;
import org.jeecg.modules.eam.entity.*;
import org.jeecg.modules.eam.service.*;
import org.jeecg.modules.eam.vo.OperationCertificateImportVo;
import org.jeecg.modules.system.entity.SysDict;
import org.jeecg.modules.system.entity.SysDictItem;
import org.jeecg.modules.system.entity.SysUser;
import org.jeecg.modules.system.service.ISysDictItemService;
import org.jeecg.modules.system.service.ISysDictService;
import org.jeecg.modules.system.service.ISysUserService;
import org.jeecg.modules.system.service.SysIdentityService;
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.jeecg.common.system.vo.LoginUser;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.eam.vo.OperationCertificatePage;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
import com.alibaba.fastjson.JSON;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.jeecg.common.aspect.annotation.AutoLog;

 /**
 * @Description: 操作证书管理
 * @Author: jeecg-boot
 * @Date:   2023-10-17
 * @Version: V1.0
 */
@Api(tags="操作证书管理")
@RestController
@RequestMapping("/eam/operationCertificate")
@Slf4j
public class OperationCertificateController {
	@Autowired
	private IOperationCertificateService operationCertificateService;
	@Autowired
	private IOperationCertificateDetailService operationCertificateDetailService;

	 @Autowired
	 private IdentityService identityService;
	 @Autowired
	 private ISysUserService userService;
	 @Autowired
	 private IEamEquipmentService equipmentService;
	 @Autowired
	 private IProfessionService professionService;
	 @Autowired
	 private ISysDictService sysDictService;
	 @Autowired
	 private ISysDictItemService sysDictItemService;
	
	/**
	 * 分页列表查询
	 *
	 * @param operationCertificate
	 * @param pageNo
	 * @param pageSize
	 * @param req
	 * @return
	 */
	//@AutoLog(value = "操作证书管理-分页列表查询")
	@ApiOperation(value="操作证书管理-分页列表查询", notes="操作证书管理-分页列表查询")
	@GetMapping(value = "/list")
	public Result<IPage<OperationCertificate>> queryPageList(OperationCertificate operationCertificate,
								   @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
								   @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
								   HttpServletRequest req) {
		QueryWrapper<OperationCertificate> queryWrapper = QueryGenerator.initQueryWrapper(operationCertificate, req.getParameterMap());
		Page<OperationCertificate> page = new Page<OperationCertificate>(pageNo, pageSize);
		IPage<OperationCertificate> pageList = operationCertificateService.page(page, queryWrapper);
		return Result.OK(pageList);
	}
	
	/**
	 *   添加
	 *
	 * @param operationCertificatePage
	 * @return
	 */
	@AutoLog(value = "操作证书管理-添加")
	@ApiOperation(value="操作证书管理-添加", notes="操作证书管理-添加")
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:add")
	@PostMapping(value = "/add")
	public Result<String> add(@RequestBody OperationCertificatePage operationCertificatePage) {
		OperationCertificate operationCertificate = new OperationCertificate();
		BeanUtils.copyProperties(operationCertificatePage, operationCertificate);
		operationCertificateService.saveMain(operationCertificate, operationCertificatePage.getOperationCertificateDetailList());
		return Result.OK("添加成功！");
	}

	 @PostMapping(value = "/addOperationCertificate")
	 public Result<String> addOperationCertificate(@RequestBody List<OperationCertificateApplyDetail> operationCertificateList) {
		for(OperationCertificateApplyDetail operationCertificateApplyDetail : operationCertificateList){
			if("pass".equals(operationCertificateApplyDetail.getExaminationConclusion())){
				Date currentDate = new Date();
				Calendar c = Calendar.getInstance();
				c.setTime(currentDate);
				c.add(Calendar.YEAR, 1);
				currentDate = c.getTime();
				OperationCertificate operationCertificate = new OperationCertificate();
				operationCertificate.setNum(identityService.getNumByTypeAndLength("OperationCertificate", 4));
				SysUser sysUser = userService.getById(operationCertificateApplyDetail.getUserId());
				if(sysUser != null){
					operationCertificate.setUserId(sysUser.getUsername());
				}
				operationCertificate.setEquipmentIds(operationCertificateApplyDetail.getEquipmentIds());
				operationCertificate.setCurrentCycleScore(12);
				operationCertificate.setIssueDate(new Date());
				operationCertificate.setStartTime(new Date());
				operationCertificate.setEndTime(currentDate);
				operationCertificate.setStatus("1");
				operationCertificateService.save(operationCertificate);
			}
		}
		 return Result.OK("添加成功！");
	 }
	
	/**
	 *  编辑
	 *
	 * @param operationCertificate
	 * @return
	 */
	@AutoLog(value = "操作证书管理-编辑")
	@ApiOperation(value="操作证书管理-编辑", notes="操作证书管理-编辑")
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:edit")
	@RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST})
	public Result<String> edit(@RequestBody OperationCertificate operationCertificate) {
//		OperationCertificate operationCertificate = new OperationCertificate();
//		BeanUtils.copyProperties(operationCertificatePage, operationCertificate);
		OperationCertificate operationCertificateEntity = operationCertificateService.getById(operationCertificate.getId());
		if(operationCertificateEntity==null) {
			return Result.error("未找到对应数据");
		}
		operationCertificateService.updateMain(operationCertificate, operationCertificate.getOperationCertificateDetailList());
		return Result.OK("操作证扣分成功!");
	}
	
	/**
	 *   通过id删除
	 *
	 * @param id
	 * @return
	 */
	@AutoLog(value = "操作证书管理-通过id删除")
	@ApiOperation(value="操作证书管理-通过id删除", notes="操作证书管理-通过id删除")
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:delete")
	@DeleteMapping(value = "/delete")
	public Result<String> delete(@RequestParam(name="id",required=true) String id) {
		operationCertificateService.delMain(id);
		return Result.OK("删除成功!");
	}
	
	/**
	 *  批量删除
	 *
	 * @param ids
	 * @return
	 */
	@AutoLog(value = "操作证书管理-批量删除")
	@ApiOperation(value="操作证书管理-批量删除", notes="操作证书管理-批量删除")
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:deleteBatch")
	@DeleteMapping(value = "/deleteBatch")
	public Result<String> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
		this.operationCertificateService.delBatchMain(Arrays.asList(ids.split(",")));
		return Result.OK("批量删除成功！");
	}
	
	/**
	 * 通过id查询
	 *
	 * @param id
	 * @return
	 */
	//@AutoLog(value = "操作证书管理-通过id查询")
	@ApiOperation(value="操作证书管理-通过id查询", notes="操作证书管理-通过id查询")
	@GetMapping(value = "/queryById")
	public Result<OperationCertificate> queryById(@RequestParam(name="id",required=true) String id) {
		OperationCertificate operationCertificate = operationCertificateService.getById(id);
		if(operationCertificate==null) {
			return Result.error("未找到对应数据");
		}
		return Result.OK(operationCertificate);

	}
	
	/**
	 * 通过id查询
	 *
	 * @param id
	 * @return
	 */
	//@AutoLog(value = "操作证管理明细通过主表ID查询")
	@ApiOperation(value="操作证管理明细主表ID查询", notes="操作证管理明细-通主表ID查询")
	@GetMapping(value = "/queryOperationCertificateDetailByMainId")
	public Result<List<OperationCertificateDetail>> queryOperationCertificateDetailListByMainId(@RequestParam(name="id",required=true) String id) {
		List<OperationCertificateDetail> operationCertificateDetailList = operationCertificateDetailService.selectByMainId(id);
		return Result.OK(operationCertificateDetailList);
	}

    /**
    * 导出excel
    *
    * @param request
    * @param operationCertificate
    */
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:exportXls")
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, OperationCertificate operationCertificate) {
      // Step.1 组装查询条件查询数据
      QueryWrapper<OperationCertificate> queryWrapper = QueryGenerator.initQueryWrapper(operationCertificate, request.getParameterMap());
      LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();

      //配置选中数据查询条件
      String selections = request.getParameter("selections");
      if(oConvertUtils.isNotEmpty(selections)) {
         List<String> selectionList = Arrays.asList(selections.split(","));
         queryWrapper.in("id",selectionList);
      }
      //Step.2 获取导出数据
      List<OperationCertificate> operationCertificateList = operationCertificateService.list(queryWrapper);

      // Step.3 组装pageList
      List<OperationCertificatePage> pageList = new ArrayList<OperationCertificatePage>();
      for (OperationCertificate main : operationCertificateList) {
          OperationCertificatePage vo = new OperationCertificatePage();
          BeanUtils.copyProperties(main, vo);
          List<OperationCertificateDetail> operationCertificateDetailList = operationCertificateDetailService.selectByMainId(main.getId());
          vo.setOperationCertificateDetailList(operationCertificateDetailList);
          pageList.add(vo);
      }

      // Step.4 AutoPoi 导出Excel
      ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
      mv.addObject(NormalExcelConstants.FILE_NAME, "操作证书管理列表");
      mv.addObject(NormalExcelConstants.CLASS, OperationCertificatePage.class);
      mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("操作证书管理数据", "导出人:"+sysUser.getRealname(), "操作证书管理"));
      mv.addObject(NormalExcelConstants.DATA_LIST, pageList);
      return mv;
    }

    /**
    * 通过excel导入数据
    *
    * @param request
    * @param response
    * @return
    */
    //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:importExcel")
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
	public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) throws IOException  {
		MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
		Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
		// 错误信息
		List<String> errorMessage = new ArrayList<>();
		int successLines = 0, errorLines = 0,totalLines = 0,partSuccessLines = 0;
		for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
			// 获取上传文件对象
			MultipartFile file = entity.getValue();
			ImportParams params = new ImportParams();
			params.setNeedSave(true);
			try {
					List<OperationCertificateImportVo> list = ExcelImportUtil.importExcel(file.getInputStream(), OperationCertificateImportVo.class, params);
					totalLines = list.size();
					for (int i = 0; i < list.size(); i++) {
					OperationCertificateImportVo operationCertificateImportVo = list.get(i);
					OperationCertificate operationCertificate = new OperationCertificate();
					//判空
					if(StringUtils.isBlank(operationCertificateImportVo.getName())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:人员不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getTechnicalLevel())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:技能等级不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getProfession())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:工种不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getWorkNo())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:工作证号不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getAppointmentCardNum())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:上岗证号不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getEquipmentNum())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:统一编号不能为空！");
						errorLines++;
						continue;
					}
					if(operationCertificateImportVo.getIssueDate() == null){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:发证日期不能为空！");
						errorLines++;
						continue;
					}
					if(StringUtils.isBlank(operationCertificateImportVo.getNum())){
						operationCertificate.setNum(identityService.getNumByTypeAndLength("OperationCertificate",4));
					}else {
						operationCertificate.setNum(operationCertificateImportVo.getNum());
					}
					//根据人员姓名查询人员id
					List<SysUser> userList = userService.lambdaQuery().eq(SysUser::getRealname,operationCertificateImportVo.getName().trim())
							.eq(SysUser::getWorkNo,operationCertificateImportVo.getWorkNo().trim()).eq(SysUser::getDelFlag, CommonConstant.DEL_FLAG_0).list();
					if(CollectionUtils.isEmpty(userList)){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:人员不存在或工作证号与人员姓名不匹配,请先维护人员信息;");
						errorLines++;
						continue;
					}else{
						operationCertificate.setUserId(userList.get(0).getUsername());
					}
					//工种查询
					Profession profession = professionService.lambdaQuery().eq(Profession::getName,operationCertificateImportVo.getProfession().trim()).eq(Profession::getDelFlag,CommonConstant.DEL_FLAG_0).one();
					if(profession == null){
						Profession profession1 = new Profession();
						profession1.setNum(identityService.getNumByTypeAndLength("Profession",4));
						profession1.setName(operationCertificateImportVo.getProfession().trim());
						professionService.save(profession1);
						operationCertificate.setProfession(profession1.getId());
					}else {
						operationCertificate.setProfession(profession.getId());
					}
					//技能等级判断
					if(CollectionUtils.isEmpty(sysDictItemService.lambdaQuery().eq(SysDictItem::getDictId,sysDictService.lambdaQuery().eq(SysDict::getDictCode,"technical_level").one().getId()).eq(SysDictItem::getItemText,operationCertificateImportVo.getTechnicalLevel().trim()).list())){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:技能等级不存在,请先维护技能等级;");
						errorLines++;
						continue;
					}else{
						operationCertificate.setTechnicalLevel(sysDictItemService.lambdaQuery().eq(SysDictItem::getDictId,sysDictService.lambdaQuery().eq(SysDict::getDictCode,"technical_level").one().getId()).eq(SysDictItem::getItemText,operationCertificateImportVo.getTechnicalLevel().trim()).one().getItemValue());
					}
					SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
					operationCertificate.setIssueDate(operationCertificateImportVo.getIssueDate());
					operationCertificate.setStartTime(operationCertificateImportVo.getIssueDate());
					Date currentDate = operationCertificateImportVo.getIssueDate();
					Calendar c = Calendar.getInstance();
					c.setTime(currentDate);
					c.add(Calendar.YEAR, 1);
					currentDate = c.getTime();
					operationCertificate.setEndTime(currentDate);
					operationCertificate.setCurrentCycleScore(12);
					operationCertificate.setRemark(operationCertificateImportVo.getRemark());
					List<String> equipments  = Arrays.asList(operationCertificateImportVo.getEquipmentNum().split("/"));
					List<String> equipmentIds = new ArrayList<>();
					int a = 0;
					for(int j = 0; j<equipments.size();j++){
						List<Equipment> equipmentList = equipmentService.lambdaQuery().eq(Equipment::getNum,equipments.get(j)).eq(Equipment::getDelFlag,CommonConstant.DEL_FLAG_0).list();
						if(CollectionUtils.isNotEmpty(equipmentList)){
							equipmentIds.add(equipmentList.get(0).getId());
						}else {
							errorMessage.add("第" + (i + 1) + "行统一编号为"+equipments.get(j)+"的设备导入失败,原因:设备台账中不存在该设备,请先维护设备信息;");
							a++;
							continue;
						}
					}
					if(a == equipments.size()){
						errorLines++;
						a = 0;
						continue;
					}else if(a > 0 && a < equipments.size()){
						partSuccessLines++;
						a = 0;
					}
					if(CollectionUtils.isEmpty(equipmentIds)){
						errorMessage.add("第" + (i + 1) + "行导入失败,原因:设备不存在,请先维护设备信息;");
						errorLines++;
						continue;
					}else{
						operationCertificate.setEquipmentIds(equipmentIds.stream().map(String::toString).collect(Collectors.joining(",")));
					}
					operationCertificate.setStatus("1");
					//查询当前人员是否已有操作证
					List<OperationCertificate> operationCertificateList = operationCertificateService.lambdaQuery().eq(OperationCertificate::getUserId,userList.get(0).getUsername()).eq(OperationCertificate::getDelFlag,CommonConstant.DEL_FLAG_0).list();
					if(CollectionUtils.isEmpty(operationCertificateList)){
						operationCertificateService.save(operationCertificate);
					}else {
						//将设备增加到操作证
						String newEquipmentIds = operationCertificateList.get(0).getEquipmentIds().concat(",").concat(operationCertificate.getEquipmentIds());
						operationCertificate.setId(operationCertificateList.get(0).getId());
						HashSet<String> set = new HashSet<String>(Arrays.asList(newEquipmentIds.split(",")));
						operationCertificate.setEquipmentIds(set.stream().collect(Collectors.joining(",")));
						operationCertificateService.updateById(operationCertificate);
					}
					successLines++;
				}
			} catch (Exception e) {
				errorMessage.add("发生异常：" + e.getMessage());
				log.error(e.getMessage(), e);
				return Result.error(errorMessage.toString());
				// return (Result<?>) errorMessage;
			} finally {
				try {
					file.getInputStream().close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
		}
		return ImportExcelUtil.imporReturnOperationCertificate(totalLines,partSuccessLines,errorLines, successLines, errorMessage);
	}
	 /**
	  * 通过主表ID查询
	  * @return
	  */
	 //@AutoLog(value = "操作证管理明细-通过主表ID查询")
	 @ApiOperation(value="操作证管理明细-通过主表ID查询", notes="操作证管理明细-通过主表ID查询")
	 @GetMapping(value = "/listOperationCertificateDetailByMainId")
	 public Result<IPage<OperationCertificateDetail>> listOperationCertificateDetailByMainId(OperationCertificateDetail operationCertificateDetail,
																							 @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
																							 @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
																							 HttpServletRequest req) {
		 QueryWrapper<OperationCertificateDetail> queryWrapper = QueryGenerator.initQueryWrapper(operationCertificateDetail, req.getParameterMap());
		 Page<OperationCertificateDetail> page = new Page<OperationCertificateDetail>(pageNo, pageSize);
		 IPage<OperationCertificateDetail> pageList = operationCertificateDetailService.page(page, queryWrapper);
		 return Result.OK(pageList);
	 }
}