package org.jeecg.modules.eam.controller;

import java.io.File;
import java.io.InputStream;
import java.math.BigDecimal;
import java.text.ParseException;
import java.util.*;
import java.util.stream.Collectors;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.apache.commons.lang3.time.DateUtils;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.PmsUtil;
import org.jeecg.common.util.StrUtils;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.eam.constant.EquipmentPurchasePlanStatusEnum;
import org.jeecg.modules.eam.entity.EamEquipmentPurchasePlan;
import org.jeecg.modules.eam.service.IEamEquipmentPurchasePlanService;

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 org.jeecg.modules.eam.util.ExcelUtils;
import org.jeecg.modules.system.entity.MdcProduction;
import org.jeecg.modules.system.entity.SysUser;
import org.jeecg.modules.system.service.IMdcProductionService;
import org.jeecg.modules.system.service.ISysBusinessCodeRuleService;
import org.jeecg.modules.system.service.ISysUserService;
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.base.controller.JeecgController;
import org.springframework.beans.factory.annotation.Autowired;
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 com.alibaba.fastjson.JSON;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.jeecg.common.aspect.annotation.AutoLog;

 /**
 * @Description: eam_equipment_purchase_plan
 * @Author: jeecg-boot
 * @Date:   2025-06-18
 * @Version: V1.0
 */
@Api(tags="eam_equipment_purchase_plan")
@RestController
@RequestMapping("/eam/eamEquipmentPurchasePlan")
@Slf4j
public class EamEquipmentPurchasePlanController extends JeecgController<EamEquipmentPurchasePlan, IEamEquipmentPurchasePlanService> {
	@Autowired
	private IEamEquipmentPurchasePlanService eamEquipmentPurchasePlanService;
	@Autowired
	private ISysUserService sysUserService;
	@Autowired
	private IMdcProductionService mdcProductionService;
	@Autowired
	private ISysBusinessCodeRuleService sysBusinessCodeRuleService;
	
	/**
	 * 分页列表查询
	 *
	 * @param eamEquipmentPurchasePlan
	 * @param pageNo
	 * @param pageSize
	 * @param req
	 * @return
	 */
	//@AutoLog(value = "eam_equipment_purchase_plan-分页列表查询")
	@ApiOperation(value="eam_equipment_purchase_plan-分页列表查询", notes="eam_equipment_purchase_plan-分页列表查询")
	@GetMapping(value = "/list")
	public Result<IPage<EamEquipmentPurchasePlan>> queryPageList(EamEquipmentPurchasePlan eamEquipmentPurchasePlan,
								   @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
								   @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
								   HttpServletRequest req) {
		QueryWrapper<EamEquipmentPurchasePlan> queryWrapper = QueryGenerator.initQueryWrapper(eamEquipmentPurchasePlan, req.getParameterMap());
		Page<EamEquipmentPurchasePlan> page = new Page<EamEquipmentPurchasePlan>(pageNo, pageSize);
		IPage<EamEquipmentPurchasePlan> pageList = eamEquipmentPurchasePlanService.page(page, queryWrapper);
		return Result.OK(pageList);
	}
	
	/**
	 *   添加
	 *
	 * @param eamEquipmentPurchasePlan
	 * @return
	 */
	@AutoLog(value = "eam_equipment_purchase_plan-添加")
	@ApiOperation(value="eam_equipment_purchase_plan-添加", notes="eam_equipment_purchase_plan-添加")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:add")
	@PostMapping(value = "/add")
	public Result<String> add(@RequestBody EamEquipmentPurchasePlan eamEquipmentPurchasePlan) {
		eamEquipmentPurchasePlanService.savePurchasePlan(eamEquipmentPurchasePlan);
		return Result.OK("添加成功！");
	}
	
	/**
	 *  编辑
	 *
	 * @param eamEquipmentPurchasePlan
	 * @return
	 */
	@AutoLog(value = "eam_equipment_purchase_plan-编辑")
	@ApiOperation(value="eam_equipment_purchase_plan-编辑", notes="eam_equipment_purchase_plan-编辑")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:edit")
	@RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST})
	public Result<String> edit(@RequestBody EamEquipmentPurchasePlan eamEquipmentPurchasePlan) {
		eamEquipmentPurchasePlanService.updateById(eamEquipmentPurchasePlan);
		return Result.OK("编辑成功!");
	}

	@AutoLog(value = "eam_equipment_purchase_plan-提交")
	@ApiOperation(value="eam_equipment_purchase_plan-提交", notes = "eam_equipment_purchase_plan-提交")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:submit")
	@GetMapping(value = "/submit")
	public Result<String> submit(@RequestParam("id") String id) {
		EamEquipmentPurchasePlan plan = eamEquipmentPurchasePlanService.getById(id);
		if (Objects.isNull(plan) || !EquipmentPurchasePlanStatusEnum.NEWLY.name().equals(plan.getPlanStatus())) {
			return Result.error("采购计划不存在/计划已提交！");
		}
		plan.setPlanStatus(EquipmentPurchasePlanStatusEnum.PENDING_AUDIT.name());
		eamEquipmentPurchasePlanService.updateById(plan);
		return Result.OK("提交成功!");
	}

	@AutoLog(value = "eam_equipment_purchase_plan-审核")
	@ApiOperation(value="eam_equipment_purchase_plan-审核", notes = "eam_equipment_purchase_plan-审核")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:audit")
	@RequestMapping(value = "/audit", method = {RequestMethod.PUT,RequestMethod.POST})
	public Result<String> audit(@RequestBody EamEquipmentPurchasePlan request) {
		EamEquipmentPurchasePlan plan = eamEquipmentPurchasePlanService.getById(request.getId());
		if (Objects.isNull(plan)) {
			return Result.error("采购计划不存在！");
		}
		if (!EquipmentPurchasePlanStatusEnum.PENDING_AUDIT.name().equals(plan.getPlanStatus())) {
			return Result.error("计划未提交或已审核！");
		}
		if (EquipmentPurchasePlanStatusEnum.COMPLETED.name().equals(request.getPlanStatus())) {
			plan.setPlanStatus(EquipmentPurchasePlanStatusEnum.COMPLETED.name());
		}
		if (EquipmentPurchasePlanStatusEnum.REJECT.name().equals(request.getPlanStatus())) {
			//如果驳回，状态改为新建，重新提交审核
			plan.setPlanStatus(EquipmentPurchasePlanStatusEnum.NEWLY.name());
		}
		eamEquipmentPurchasePlanService.updateById(plan);
		return Result.OK("审核成功!");
	}
	
	/**
	 *   通过id删除
	 *
	 * @param id
	 * @return
	 */
	@AutoLog(value = "eam_equipment_purchase_plan-通过id删除")
	@ApiOperation(value="eam_equipment_purchase_plan-通过id删除", notes="eam_equipment_purchase_plan-通过id删除")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:delete")
	@DeleteMapping(value = "/delete")
	public Result<String> delete(@RequestParam(name="id",required=true) String id) {
		eamEquipmentPurchasePlanService.removeById(id);
		return Result.OK("删除成功!");
	}
	
	/**
	 *  批量删除
	 *
	 * @param ids
	 * @return
	 */
	@AutoLog(value = "eam_equipment_purchase_plan-批量删除")
	@ApiOperation(value="eam_equipment_purchase_plan-批量删除", notes="eam_equipment_purchase_plan-批量删除")
	//@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:deleteBatch")
	@DeleteMapping(value = "/deleteBatch")
	public Result<String> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
		this.eamEquipmentPurchasePlanService.removeByIds(Arrays.asList(ids.split(",")));
		return Result.OK("批量删除成功!");
	}
	
	/**
	 * 通过id查询
	 *
	 * @param id
	 * @return
	 */
	//@AutoLog(value = "eam_equipment_purchase_plan-通过id查询")
	@ApiOperation(value="eam_equipment_purchase_plan-通过id查询", notes="eam_equipment_purchase_plan-通过id查询")
	@GetMapping(value = "/queryById")
	public Result<EamEquipmentPurchasePlan> queryById(@RequestParam(name="id",required=true) String id) {
		EamEquipmentPurchasePlan eamEquipmentPurchasePlan = eamEquipmentPurchasePlanService.getById(id);
		if(eamEquipmentPurchasePlan==null) {
			return Result.error("未找到对应数据");
		}
		return Result.OK(eamEquipmentPurchasePlan);
	}

    /**
    * 导出excel
    *
    * @param request
    * @param eamEquipmentPurchasePlan
    */
    //@RequiresPermissions("org.jeecg.modules:eam_equipment_purchase_plan:exportXls")
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, EamEquipmentPurchasePlan eamEquipmentPurchasePlan) {
        return super.exportXls(request, eamEquipmentPurchasePlan, EamEquipmentPurchasePlan.class, "设备采购计划");
    }

    /**
      * 通过excel导入数据
    *
    * @param request
    * @param response
    * @return
    */
    //@RequiresPermissions("eam_equipment_purchase_plan:importExcel")
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
		MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
		Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
		int successLines = 0,dataLines = 0;
		List<String> errorMessage = CollectionUtil.newArrayList();
		List<EamEquipmentPurchasePlan> importList = CollectionUtil.newArrayList();
        try {
            for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
                // 获取上传文件对象
                MultipartFile file = entity.getValue();
                InputStream inputStream = file.getInputStream();
				Workbook workbook = WorkbookFactory.create(inputStream);
                Sheet sheet = workbook.getSheetAt(0);
				for (Row row : sheet) {
					EamEquipmentPurchasePlan plan = new EamEquipmentPurchasePlan();
					int rowNum = row.getRowNum();
					int currentRowNum  = rowNum + 1;
					if (rowNum < 3 || ExcelUtils.isRowEmpty(row)) {
						//跳过表头和空行
						continue;
					}
					dataLines++;
					Cell cell0 = row.getCell(0);
					if (cell0 == null || cell0.getCellType() == CellType.BLANK) {
						errorMessage.add("第" + currentRowNum + "行采购计划名称为空，无法导入");
						continue;
					} else {
						plan.setPlanName(ExcelUtils.getCellValueAsString(cell0));
					}
					Cell cell1 = row.getCell(1);
					if (cell1 == null || cell1.getCellType() == CellType.BLANK) {
						//如果计划状态为空，默认给 新建 状态
						plan.setPlanStatus(EquipmentPurchasePlanStatusEnum.NEWLY.name());
					} else {
						String statusZh = ExcelUtils.getCellValueAsString(cell1).trim();
						EquipmentPurchasePlanStatusEnum planStatusEnum = EquipmentPurchasePlanStatusEnum.fromZhName(statusZh);
						if (Objects.isNull(planStatusEnum)) {
							errorMessage.add("第 " + currentRowNum + " 行计划状态不合法，无法导入，请使用 [新建, 待审核, 已完成] 中的一个值");
							continue;
						}
						plan.setPlanStatus(planStatusEnum.name());
					}
					Cell cell2 = row.getCell(2);
					if (cell2 == null || cell2.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行型号为空，无法导入");
						continue;
					} else {
						plan.setEquipmentModel(ExcelUtils.getCellValueAsString(cell2));
					}
					Cell cell3 = row.getCell(3);
					if (cell3 == null || cell3.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行采购数量为空，无法导入");
						continue;
					} else {
						plan.setPurchaseNumber(Integer.valueOf(ExcelUtils.getCellValueAsString(cell3)));
					}
					Cell cell4 = row.getCell(4);
					if (cell4 == null || cell4.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行预计单价为空，无法导入");
						continue;
					} else {
						plan.setEstimatedUnitPrice(new BigDecimal(ExcelUtils.getCellValueAsString(cell4)));
					}
					Cell cell5 = row.getCell(5);
					if (cell5 == null || cell5.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行总价为空，无法导入");
						continue;
					} else {
						plan.setTotalPrice(new BigDecimal(ExcelUtils.getCellValueAsString(cell5)));
					}
					Cell cell6 = row.getCell(6);
					if (cell6 == null || cell6.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行合同总金额为空，无法导入");
						continue;
					} else {
						plan.setOverallContractAmount(new BigDecimal(ExcelUtils.getCellValueAsString(cell6)));
					}
					Cell cell7 = row.getCell(7);
					if (cell7 == null || cell7.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行管理分类为空，无法导入");
						continue;
					} else {
						plan.setEquipmentImportance(ExcelUtils.getCellValueAsString(cell7));
					}
					Cell cell8 = row.getCell(8);
					if (cell8 == null || cell8.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行申请人为空，无法导入");
						continue;
					} else {
						String username = ExcelUtils.getCellValueAsString(cell8);
						SysUser sysUser = sysUserService.getUserByName(username);
						if (sysUser == null) {
							errorMessage.add("第 " + currentRowNum + " 行申请人不存在，无法导入");
							continue;
						}
						plan.setPlanCategory(username);
					}
					Cell cell9 = row.getCell(9);
					if (cell9 == null || cell9.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行申请车间为空，无法导入");
						continue;
					} else {
						String productionName = ExcelUtils.getCellValueAsString(cell9);
						boolean present = mdcProductionService.list(new LambdaQueryWrapper<MdcProduction>()
								.eq(MdcProduction::getProductionName, productionName)
								.eq(MdcProduction::getDelFlag, CommonConstant.DEL_FLAG_0))
								.stream().findAny().isPresent();
						if (!present) {
							errorMessage.add("第 " + currentRowNum + " 行申请车间不存在，无法导入");
							continue;
						}
						plan.setApplicantProduction(productionName);
					}
					Cell cell10 = row.getCell(10);
					if (cell10 == null || cell10.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行计划开始日期为空，无法导入");
						continue;
					} else {
						String planStartDateStr = ExcelUtils.getCellValueAsString(cell10);
                        try {
							Date date = DateUtils.parseDate(planStartDateStr, "yyyy/MM/dd HH", "yyyy/MM/dd", "yyyy-MM-dd HH", "yyyy-MM-dd");
							plan.setPlanStartDate(date);
						} catch (ParseException e) {
							errorMessage.add("第 " + currentRowNum + " 行计划开始日期格式错误，无法导入，请使用 yyyy-MM-dd 格式");
							continue;
                        }
					}
					Cell cell11 = row.getCell(11);
					if (cell11 == null || cell11.getCellType() == CellType.BLANK) {
						errorMessage.add("第 " + currentRowNum + " 行计划完成时间为空，无法导入");
						continue;
					} else {
						String planEndDateStr = ExcelUtils.getCellValueAsString(cell11);
                        try {
							Date date = DateUtils.parseDate(planEndDateStr, "yyyy/MM/dd HH", "yyyy/MM/dd", "yyyy-MM-dd HH", "yyyy-MM-dd");
							plan.setPlanEndDate(date);
						} catch (ParseException e) {
							errorMessage.add("第 " + currentRowNum + " 行计划完成日期格式错误，无法导入，请使用 yyyy-MM-dd 格式");
							continue;
						}
					}
					Cell cell12 = row.getCell(12);
					plan.setPlanDescription(ExcelUtils.getCellValueAsString(cell12));
					Cell cell13 = row.getCell(13);
					plan.setRelatedDeparts(ExcelUtils.getCellValueAsString(cell13));
					Cell cell14 = row.getCell(14);
					plan.setRemark(ExcelUtils.getCellValueAsString(cell14));
					plan.setPlanCode(sysBusinessCodeRuleService.generateBusinessCodeSeq("EquipmentPurchasePlanRule"));
					plan.setDelFlag(CommonConstant.DEL_FLAG_0);
					importList.add(plan);
					successLines++;
				}
			}
			eamEquipmentPurchasePlanService.saveBatch(importList);
			if (successLines == dataLines) {
				return Result.OK("共" + dataLines + "行数据全部导入成功！");
			} else {
				JSONObject result = new JSONObject(5);
				result.put("successCount", successLines);
				result.put("msg", "总上传行数：" + dataLines + "，成功导入行数：" + successLines);
				result.put("errorMsg", errorMessage);
				String fileUrl = PmsUtil.saveErrorMsgByList(errorMessage, "userImportExcelErrorLog");
				int lastIndex = fileUrl.lastIndexOf(File.separator);
				String fileName = fileUrl.substring(lastIndex + 1);
				result.put("fileUrl", "/sys/common/static/" + fileUrl);
				result.put("fileName", fileName);
				Result res = Result.ok(result);
				res.setCode(201);
				res.setMessage("文件导入成功，但有错误。");
				return res;
			}
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

}