package org.jeecg.modules.tms.controller;

import java.io.File;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
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 com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
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.oConvertUtils;
import org.jeecg.modules.tms.entity.BaseTools;
import org.jeecg.modules.tms.entity.ParaHoleTools;
import org.jeecg.modules.tms.entity.ParaThreadingTool;
import org.jeecg.modules.tms.entity.ToolsClassify;
import org.jeecg.modules.tms.service.IBaseToolsService;
import org.jeecg.modules.tms.service.IParaThreadingToolService;

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.tms.service.IToolsClassifyService;
import org.jeecg.modules.tms.utils.ExcelUtils;
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: tms_para_threading_tool
 * @Author: jeecg-boot
 * @Date: 2025-05-12
 * @Version: V1.0
 */
@Api(tags = "tms_para_threading_tool")
@RestController
@RequestMapping("/tms/paraThreadingTool")
@Slf4j
public class ParaThreadingToolController extends JeecgController<ParaThreadingTool, IParaThreadingToolService> {
    @Autowired
    private IParaThreadingToolService paraThreadingToolService;
    @Autowired
    private IToolsClassifyService toolsClassifyService;
    @Autowired
    private IBaseToolsService baseToolsService;

    /**
     * 分页列表查询
     *
     * @param paraThreadingTool
     * @param pageNo
     * @param pageSize
     * @param req
     * @return
     */
    //@AutoLog(value = "tms_para_threading_tool-分页列表查询")
    @ApiOperation(value = "tms_para_threading_tool-分页列表查询", notes = "tms_para_threading_tool-分页列表查询")
    @GetMapping(value = "/list")
    public Result<IPage<ParaThreadingTool>> queryPageList(ParaThreadingTool paraThreadingTool,
                                                          @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
                                                          @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
                                                          HttpServletRequest req) {
        QueryWrapper<ParaThreadingTool> queryWrapper = QueryGenerator.initQueryWrapper(paraThreadingTool, req.getParameterMap());
        Page<ParaThreadingTool> page = new Page<ParaThreadingTool>(pageNo, pageSize);
        IPage<ParaThreadingTool> pageList = paraThreadingToolService.page(page, queryWrapper);
        return Result.OK(pageList);
    }

    /**
     * 添加
     *
     * @param paraThreadingTool
     * @return
     */
    @AutoLog(value = "tms_para_threading_tool-添加")
    @ApiOperation(value = "tms_para_threading_tool-添加", notes = "tms_para_threading_tool-添加")
    //@RequiresPermissions("org.jeecg.modules:tms_para_threading_tool:add")
    @PostMapping(value = "/add")
    public Result<String> add(@RequestBody ParaThreadingTool paraThreadingTool) {
        paraThreadingToolService.save(paraThreadingTool);
        return Result.OK("添加成功！");
    }

    /**
     * 编辑
     *
     * @param paraThreadingTool
     * @return
     */
    @AutoLog(value = "tms_para_threading_tool-编辑")
    @ApiOperation(value = "tms_para_threading_tool-编辑", notes = "tms_para_threading_tool-编辑")
    //@RequiresPermissions("org.jeecg.modules:tms_para_threading_tool:edit")
    @RequestMapping(value = "/edit", method = {RequestMethod.PUT, RequestMethod.POST})
    public Result<String> edit(@RequestBody ParaThreadingTool paraThreadingTool) {
        paraThreadingToolService.updateById(paraThreadingTool);
        return Result.OK("编辑成功!");
    }

    /**
     * 通过id删除
     *
     * @param id
     * @return
     */
    @AutoLog(value = "tms_para_threading_tool-通过id删除")
    @ApiOperation(value = "tms_para_threading_tool-通过id删除", notes = "tms_para_threading_tool-通过id删除")
    //@RequiresPermissions("org.jeecg.modules:tms_para_threading_tool:delete")
    @DeleteMapping(value = "/delete")
    public Result<String> delete(@RequestParam(name = "id", required = true) String id) {
        paraThreadingToolService.removeById(id);
        return Result.OK("删除成功!");
    }

    /**
     * 批量删除
     *
     * @param ids
     * @return
     */
    @AutoLog(value = "tms_para_threading_tool-批量删除")
    @ApiOperation(value = "tms_para_threading_tool-批量删除", notes = "tms_para_threading_tool-批量删除")
    //@RequiresPermissions("org.jeecg.modules:tms_para_threading_tool:deleteBatch")
    @DeleteMapping(value = "/deleteBatch")
    public Result<String> deleteBatch(@RequestParam(name = "ids", required = true) String ids) {
        this.paraThreadingToolService.removeByIds(Arrays.asList(ids.split(",")));
        return Result.OK("批量删除成功!");
    }

    /**
     * 通过id查询
     *
     * @param id
     * @return
     */
    //@AutoLog(value = "tms_para_threading_tool-通过id查询")
    @ApiOperation(value = "tms_para_threading_tool-通过id查询", notes = "tms_para_threading_tool-通过id查询")
    @GetMapping(value = "/queryById")
    public Result<ParaThreadingTool> queryById(@RequestParam(name = "id", required = true) String id) {
        ParaThreadingTool paraThreadingTool = paraThreadingToolService.getById(id);
        if (paraThreadingTool == null) {
            return Result.error("未找到对应数据");
        }
        return Result.OK(paraThreadingTool);
    }

    /**
     * 导出excel
     *
     * @param request
     * @param paraThreadingTool
     */
    //@RequiresPermissions("org.jeecg.modules:tms_para_threading_tool:exportXls")
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, ParaThreadingTool paraThreadingTool) {
        return super.exportXls(request, paraThreadingTool, ParaThreadingTool.class, "tms_para_threading_tool");
    }

    /**
     * 通过excel导入数据
     *
     * @param request
     * @param response
     * @return
     */
    //@RequiresPermissions("tms_para_threading_tool:importExcel")
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
        Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
        // 错误信息
        List<String> errorMessage = new ArrayList<>();
        int successLines = 0, errorLines = 0, dataLines = 0;
        List<ParaThreadingTool> paraThreadingToolImportList = new ArrayList<>();
        try {
            for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
                // 获取上传文件对象
                MultipartFile file = entity.getValue();
                InputStream inputStream = file.getInputStream();
                Workbook workbook = new XSSFWorkbook(inputStream);
                Sheet sheet = workbook.getSheetAt(0);
                for (Row row : sheet) {
                    ParaThreadingTool paraThreadingTool = new ParaThreadingTool();
                    int rowNum = row.getRowNum();
                    int currentRowNum  = rowNum + 1;
                    //表头行和空行略过
                    if (rowNum == 0 || ExcelUtils.isRowEmpty(row)) {
                        continue;
                    }
                    dataLines++;
                    Cell cell1 = row.getCell(1);
                    if (cell1 == null || cell1.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,一级分类名称为空! ");
                        errorLines++;
                        continue;
                    }
                    Cell cell2 = row.getCell(2);
                    if (cell2 == null || cell2.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,二级分类名称为空! ");
                        errorLines++;
                        continue;
                    }
                    Cell cell3 = row.getCell(3);
                    if (cell3 == null || cell3.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,三级分类名称为空! ");
                        errorLines++;
                        continue;
                    }
                    Cell cell4 = row.getCell(4);
                    if (cell4 == null || cell4.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,四级分类名称为空! ");
                        errorLines++;
                        continue;
                    }
                    Cell cell9 = row.getCell(9);
                    if (cell9 == null || cell9.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,型号/图号为空! ");
                        errorLines++;
                        continue;
                    }
                    //根据四级分类名称和型号/图号查询工具信息
                    List<ToolsClassify> toolClassifyList = toolsClassifyService.list(new LambdaQueryWrapper<ToolsClassify>()
                            .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell4))
                            .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1));
                    if (CollectionUtil.isEmpty(toolClassifyList)) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在该四级分类名称! ");
                        errorLines++;
                        continue;
                    } else {
                        ToolsClassify toolsClassify = toolClassifyList.get(0);
                        List<BaseTools> baseToolsList = baseToolsService.list(new LambdaQueryWrapper<BaseTools>()
                                .eq(BaseTools::getClassifyId, toolsClassify.getId())
                                .eq(BaseTools::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
                        if (CollectionUtil.isEmpty(baseToolsList)) {
                            errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在对应的工具信息! ");
                            errorLines++;
                            continue;
                        } else {
                            BaseTools baseTools = baseToolsList.get(0);
                            List<ParaThreadingTool> paraThreadingToolList = paraThreadingToolService.list(new LambdaQueryWrapper<ParaThreadingTool>()
                                    .eq(ParaThreadingTool::getToolCode, baseTools.getId())
                                    .eq(ParaThreadingTool::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
                            if (CollectionUtil.isEmpty(paraThreadingToolList)) {
                                paraThreadingTool.setClassifyId(toolsClassify.getId());
                                paraThreadingTool.setToolCode(baseTools.getId());
                                paraThreadingTool.setToolModel(ExcelUtils.getCellValueAsString(cell9));
                            } else {
                                paraThreadingTool = paraThreadingToolList.get(0);
                            }
                        }
                    }

                    Cell cell0 = row.getCell(0);//序号(或企业内原代码)
                    paraThreadingTool.setSignCode(ExcelUtils.getCellValueAsString(cell0));
                    Cell cell5 = row.getCell(5);//物品中文名称
                    if (cell5 == null || cell5.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,物品中文名称为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setChineseName(ExcelUtils.getCellValueAsString(cell5));
                    }
                    Cell cell6 = row.getCell(6);//物品外文名称
                    paraThreadingTool.setForeignLanguageName(ExcelUtils.getCellValueAsString(cell6));
                    Cell cell7 = row.getCell(7);//标准级别
                    paraThreadingTool.setStandardLevel(ExcelUtils.getCellValueAsString(cell7));
                    Cell cell8 = row.getCell(8);//标准号
                    paraThreadingTool.setStandardCode(ExcelUtils.getCellValueAsString(cell8));
                    Cell cell10 = row.getCell(10);//螺纹代号
                    if (cell10 == null || cell10.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,螺纹代号为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setThreadCode(ExcelUtils.getCellValueAsString(cell10));
                    }
                    Cell cell11 = row.getCell(11);//螺距
                    if (cell11 == null || cell11.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,螺距为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setPitch(ExcelUtils.getCellValueAsString(cell11));
                    }

                    Cell cell12 = row.getCell(12);//螺纹旋向
                    if (cell12 == null || cell12.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,螺纹旋向为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setRotationDirection(ExcelUtils.getCellValueAsString(cell12));
                    }
                    Cell cell13 = row.getCell(13);//螺纹公差带代号/精度等级
                    if (cell13 == null || cell13.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,螺纹公差带代号/精度等级为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setTolerancezoneLevel(ExcelUtils.getCellValueAsString(cell13));
                    }
                    Cell cell14 = row.getCell(14);//切削刃长
                    if (cell14 == null || cell14.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,切削刃长为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setEdgeLength(ExcelUtils.getCellValueAsString(cell14));
                    }
                    Cell cell15 = row.getCell(15);//刀具总长
                    if (cell15 == null || cell15.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,刀具总长为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setTotalLength(ExcelUtils.getCellValueAsString(cell15));
                    }

                    Cell cell16 = row.getCell(16);//刀具材料
                    if (cell16 == null || cell16.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,刀具材料为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setToolMaterial(ExcelUtils.getCellValueAsString(cell16));
                    }
                    Cell cell17 = row.getCell(17);//零件材料
                    paraThreadingTool.setPartMaterial(ExcelUtils.getCellValueAsString(cell17));
                    Cell cell18 = row.getCell(18);//是否涂层
                    if (cell18 == null || cell18.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,是否涂层为空! ");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setPaintcoatFlag(ExcelUtils.getCellValueAsString(cell18));
                    }
                    Cell cell19 = row.getCell(19);//外形尺寸
                    paraThreadingTool.setExternalDimensions(ExcelUtils.getCellValueAsString(cell19));
                    Cell cell20 = row.getCell(20);//柄部规格
                    if (cell20 == null || cell20.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,柄部规格为空！");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setHandleSpecifications(ExcelUtils.getCellValueAsString(cell20));
                    }
                    Cell cell21 = row.getCell(21);//附加技术条件
                    if (cell21 == null || cell21.getCellType() == CellType.BLANK) {
                        errorMessage.add("第" + currentRowNum + "行，数据导入失败,附加技术条件为空！");
                        errorLines++;
                        continue;
                    } else {
                        paraThreadingTool.setTechnicalConditions(ExcelUtils.getCellValueAsString(cell21));
                    }

                    Cell cell22 = row.getCell(22);//附加技术条件说明
                    paraThreadingTool.setConditionsInfo(ExcelUtils.getCellValueAsString(cell22));
                    Cell cell23 = row.getCell(23);//品牌
                    paraThreadingTool.setBrand(ExcelUtils.getCellValueAsString(cell23));
                    Cell cell24 = row.getCell(24);//制式
                    paraThreadingTool.setTypes(ExcelUtils.getCellValueAsString(cell24));
                    Cell cell25 = row.getCell(25);//螺孔类型
                    paraThreadingTool.setScrewHoleType(ExcelUtils.getCellValueAsString(cell25));
                    Cell cell26 = row.getCell(26);//冷却方式
                    paraThreadingTool.setCoolingMethod(ExcelUtils.getCellValueAsString(cell26));
                    Cell cell27 = row.getCell(27);//螺纹标准
                    paraThreadingTool.setThreadStandard(ExcelUtils.getCellValueAsString(cell27));
                    Cell cell28 = row.getCell(28);//排屑槽型
                    paraThreadingTool.setFluteSoltType(ExcelUtils.getCellValueAsString(cell28));
                    Cell cell29 = row.getCell(29);//螺纹类型
                    paraThreadingTool.setThreadType(ExcelUtils.getCellValueAsString(cell29));
                    Cell cell30 = row.getCell(30);//导向尺寸
                    paraThreadingTool.setGuidingSize(ExcelUtils.getCellValueAsString(cell30));
                    Cell cell31 = row.getCell(31);//连接孔径
                    paraThreadingTool.setConnectionAperture(ExcelUtils.getCellValueAsString(cell31));
                    Cell cell32 = row.getCell(32);//连接键槽
                    paraThreadingTool.setConnectingKeyway(ExcelUtils.getCellValueAsString(cell32));

                    paraThreadingToolImportList.add(paraThreadingTool);
                    successLines++;
                }
            }
            paraThreadingToolService.saveOrUpdateBatch(paraThreadingToolImportList);
            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);
        }
    }
    
}