package org.jeecg.modules.spare.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.aspect.annotation.AutoLog;
import org.jeecg.common.system.base.controller.JeecgController;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.modules.base.entity.Constructor;
import org.jeecg.modules.base.entity.Unit;
import org.jeecg.modules.base.service.IConstructorService;
import org.jeecg.modules.base.service.IUnitService;
import org.jeecg.modules.spare.entity.SparePart;
import org.jeecg.modules.spare.entity.SparesPartInventory;
import org.jeecg.modules.spare.service.ISparePartService;
import org.jeecg.modules.spare.service.ISparesPartInventoryService;
import org.jeecg.modules.spare.vo.AuxiliaryUnitVo;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.entity.ImportParams;
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 javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.math.BigDecimal;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

/**
 * @Description: mom_eam_spare_part
 * @Author: jeecg-boot
 * @Date: 2023-06-20
 * @Version: V1.0
 */
@Api(tags = "mom_eam_spare_part")
@RestController
@RequestMapping("/spare/sparePart")
@Slf4j
public class SparePartController extends JeecgController<SparePart, ISparePartService> {
    @Autowired
    private ISparePartService sparePartService;

    @Autowired
    private ISparesPartInventoryService sparesPartInventoryService;
    @Autowired
    private IConstructorService constructorService;
    @Autowired
    private IUnitService unitService;

    /**
     * 分页列表查询
     *
     * @param sparePart
     * @param pageNo
     * @param pageSize
     * @param req
     * @return
     */
    @ApiOperation(value = "mom_eam_spare_part-分页列表查询", notes = "mom_eam_spare_part-分页列表查询")
    @GetMapping(value = "/list")
    public Result<IPage<SparePart>> queryPageList(SparePart sparePart,
                                                  @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
                                                  @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
                                                  HttpServletRequest req) {
        QueryWrapper<SparePart> queryWrapper = QueryGenerator.initQueryWrapper(sparePart, req.getParameterMap());
        Page<SparePart> page = new Page<SparePart>(pageNo, pageSize);
        IPage<SparePart> pageList = sparePartService.page(page, queryWrapper);
        for (SparePart record : pageList.getRecords()) {
            String mainUnitId = record.getMainUnitId();
            List<AuxiliaryUnitVo> auxiliaryUnits = sparePartService.getAuxiliaryUnits(mainUnitId);
            record.setAuxiliaryUnits(auxiliaryUnits);
        }
        return Result.OK(pageList);
    }

    /**
     * 分页列表查询
     * 备件库存展示页
     */
    @ApiOperation(value = "mom_eam_spare_part-分页列表查询", notes = "mom_eam_spare_part-分页列表查询")
    @GetMapping(value = "/sparePartInventoryList")
    public Result<IPage<SparePart>> sparePartInventoryList(SparePart sparePart,
                                                           @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
                                                           @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
                                                           HttpServletRequest req) {
        QueryWrapper<SparePart> queryWrapper = QueryGenerator.initQueryWrapper(sparePart, req.getParameterMap());
        Page<SparePart> page = new Page<SparePart>(pageNo, pageSize);
        IPage<SparePart> pageList = sparePartService.page(page, queryWrapper);
        for (SparePart record : pageList.getRecords()) {
            String sparePartId = record.getId();

            QueryWrapper<SparesPartInventory> wrapper = new QueryWrapper<>();
            wrapper.eq("spare_part_id", sparePartId)
                    .eq("del_flag", 0);
            wrapper.select("sum(main_quantity) as totalMainQuantity");
            Map<String, Object> map = sparesPartInventoryService.getMap(wrapper);
            BigDecimal totalMainQuantity = new BigDecimal(0);
            if (map != null) {
                totalMainQuantity = (BigDecimal) map.get("totalMainQuantity");
            }
            record.setTotalMainQuantity(totalMainQuantity);

            wrapper.eq("spare_part_id", sparePartId)
                    .eq("del_flag", 0);
            wrapper.select("sum(auxiliary_quantity) as totalAuxiliaryQuantity");
            Map<String, Object> map2 = sparesPartInventoryService.getMap(wrapper);
            BigDecimal totalAuxiliaryQuantity = new BigDecimal(0);
            if (map != null) {
                totalAuxiliaryQuantity = (BigDecimal) map2.get("totalAuxiliaryQuantity");
            }
            record.setTotalAuxiliaryQuantity(totalAuxiliaryQuantity);

        }
        return Result.OK(pageList);
    }


    /**
     * 添加
     *
     * @param sparePart
     * @return
     */
    @AutoLog(value = "mom_eam_spare_part-添加")
    @ApiOperation(value = "mom_eam_spare_part-添加", notes = "mom_eam_spare_part-添加")
    @PostMapping(value = "/add")
    public Result<String> add(@RequestBody SparePart sparePart) {
        sparePartService.save(sparePart);
        return Result.OK("添加成功！");
    }

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

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

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

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

    /**
     * 导出excel
     *
     * @param request
     * @param sparePart
     */
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, SparePart sparePart) {
        return super.exportXls(request, sparePart, SparePart.class, "mom_eam_spare_part");
    }

    /**
     * 通过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<SparePart> list = ExcelImportUtil.importExcel(file.getInputStream(), SparePart.class, params);
                long start = System.currentTimeMillis();
                for (int i = 0; i < list.size(); i++) {
                    SparePart sparePart = list.get(i);

                    String constructorName = sparePart.getConstructorId();
                    String mainUnitName = sparePart.getMainUnitId();
                    String validityPeriodUnitName = sparePart.getValidityPeriodUnitId();
                    Constructor constructor = constructorService.getByName(constructorName);
                    Unit unit = unitService.getUnitByName(mainUnitName);
                    Unit validityPeriodUnit = unitService.getUnitByName(validityPeriodUnitName);


                    if (unit != null) {
                        if (!StringUtils.isNotEmpty(validityPeriodUnit.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行有效期单位不存在");
                        list.get(i).setValidityPeriodUnitId(validityPeriodUnit.getId());
                    }
                    if (constructor != null) {
                        if (!StringUtils.isNotEmpty(constructor.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行制造商不存在");
                        list.get(i).setConstructorId(constructor.getId());
                    }
                    if (unit != null) {
                        if (!StringUtils.isNotEmpty(unit.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行单位不存在");
                        list.get(i).setMainUnitId(unit.getId());
                    }
                    if (!StringUtils.isNotEmpty(sparePart.getName()))
                        return Result.error("文件导入失败: 第" + (i + 1) + "行备件名称不能为空");
                    if (!StringUtils.isNotEmpty(sparePart.getNum()))
                        return Result.error("文件导入失败: 第" + (i + 1) + "行备件编码不能为空");
                    if (!StringUtils.isNotEmpty(sparePart.getSpecification()))
                        return Result.error("文件导入失败: 第" + (i + 1) + "行备件规格不能为空");

                    if (!StringUtils.isNotEmpty(sparePart.getModel()))
                        return Result.error("文件导入失败: 第" + (i + 1) + "行备件型号不能为空");
                }
                sparePartService.saveBatch(list);
                log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒");
                return Result.OK("文件导入成功！数据行数：" + list.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.error("文件导入失败！");
    }

    /**
     * 通过设备备件获取适配的设备
     * qsw 2023-6-21
     */
    @GetMapping("getSpareContactEquipmentList")
    public Result<?> getSpareContactEquipmentList(@RequestParam("pageNo") Integer pageNo,
                                                  @RequestParam("pageSize") Integer pageSize,
                                                  @RequestParam Map<String, Object> params) {
        IPage<Map<String, Object>> spareContactEquipmentList = sparePartService.getSpareContactEquipmentList(pageNo, pageSize, params);
        return Result.ok(spareContactEquipmentList);
    }

    /**
     * 根据主单位获取辅单位
     * qsw 2023-6-23
     */
    @GetMapping("getAuxiliaryUnits")
    public Result<?> getAuxiliaryUnits(@RequestParam(name = "mainUnitId", required = true) String mainUnitId) {
        List<AuxiliaryUnitVo> auxiliaryUnits = sparePartService.getAuxiliaryUnits(mainUnitId);
        return Result.ok(auxiliaryUnits);
    }

    /**
     * 根据主单位、辅单位获取 唯一换算比例
     * qsw 2023-6-28
     */
    @GetMapping("getConversionRatio")
    public Result<?> getConversionRatio(@RequestParam(name = "mainUnitId", required = true) String mainUnitId, @RequestParam(name = "auxiliaryUnitId", required = true) String auxiliaryUnitId) {
        List<String> conversionRatio = sparePartService.getConversionRatio(mainUnitId, auxiliaryUnitId);
        if (conversionRatio.size() > 0) {
            return Result.ok(conversionRatio.get(0));
        } else {
            return Result.ok("");
        }
    }
}