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.*;
import org.jeecg.modules.base.service.*;
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.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.util.Arrays;
import java.util.List;
import java.util.Map;

/**
 * @Description: mom_eam_spares_part_inventory
 * @Author: jeecg-boot
 * @Date: 2023-06-20
 * @Version: V1.0
 */
@Api(tags = "mom_eam_spares_part_inventory")
@RestController
@RequestMapping("/spare/SparesPartInventory")
@Slf4j
public class SparesPartInventoryController extends JeecgController<SparesPartInventory, ISparesPartInventoryService> {
    @Autowired
    private ISparesPartInventoryService sparesPartInventoryService;
    @Autowired
    private ISparePartService sparePartService;
    @Autowired
    private IUnitService unitService;
    @Autowired
    private ISupplierService supplierService;

    @Autowired
    private IWarehouseService warehouseService;

    @Autowired
    private IWarehouseAreaService warehouseAreaService;

    @Autowired
    private IWarehouseLocationService warehouseLocationService;


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

        for (SparesPartInventory record : pageList.getRecords()) {
            String sparePartId = record.getSparePartId();
            SparePart part = sparePartService.getById(sparePartId);
            record.setSparePartName(part.getName());
            record.setSparePartNum(part.getNum());
            record.setSparePartModel(part.getModel());
            record.setSparePartSpecification(part.getSpecification());
            record.setSparePartConversionRatio(part.getConversionRatio());
        }
        return Result.OK(pageList);
    }


    /**
     * 添加
     *
     * @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:add")
    @PostMapping(value = "/add")
    public Result<String> add(@RequestBody SparesPartInventory sparePart) {
        sparesPartInventoryService.save(sparePart);
        return Result.OK("添加成功！");
    }

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

    /**
     * 通过id删除
     *
     * @param id
     * @return
     */
    @AutoLog(value = "mom_eam_spares_part_inventory-通过id删除")
    @ApiOperation(value = "mom_eam_spares_part_inventory-通过id删除", notes = "mom_eam_spares_part_inventory-通过id删除")
    @DeleteMapping(value = "/delete")
    public Result<String> delete(@RequestParam(name = "id", required = true) String id) {
        sparesPartInventoryService.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.sparesPartInventoryService.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<SparesPartInventory> queryById(@RequestParam(name = "id", required = true) String id) {
        SparesPartInventory sparesPartInventory = sparesPartInventoryService.getById(id);
        if (sparesPartInventory == null) {
            return Result.error("未找到对应数据");
        }
        return Result.OK(sparesPartInventory);
    }

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

    /**
     * 通过excel导入数据
     *
     * @param request
     * @param response
     * @return
     */
    /**
     * 导入
     *
     * @return
     */
    @RequestMapping(value = "/importExcel/{mainId}")
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) {
        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<SparesPartInventory> list = ExcelImportUtil.importExcel(file.getInputStream(), SparesPartInventory.class, params);
                long start = System.currentTimeMillis();
                for (int i = 0; i < list.size(); i++) {
                    Supplier supplier = null;
                    Unit unit = null;
                    Warehouse warehouse = null;
                    WarehouseArea warehouseArea = null;
                    WarehouseLocation warehouseLocation = null;
                    list.get(i).setSparePartId(mainId);

                    String mainUnitName = list.get(i).getMainUnitId();
                    String supplierName = list.get(i).getSupplierId();
                    String warehouseName = list.get(i).getWarehouseId();
                    String warehouseAreaName = list.get(i).getWarehouseAreaId();
                    String warehouseLocationName = list.get(i).getWarehouseLocationId();
                    String  num =list.get(i).getValidityForecast();

                    if (supplierName != null) {
                        supplier = supplierService.getByName(supplierName);
                    }
                    if (mainUnitName != null) {
                        unit = unitService.getUnitByName(mainUnitName);
                    }
                    if (warehouseName != null) {
                        warehouse = warehouseService.getByName(warehouseName);
                    }
                    if (warehouseAreaName != null) {
                        warehouseArea = warehouseAreaService.getByName(warehouseAreaName);
                    }
                    if (warehouseLocationName != null) {
                        warehouseLocation = warehouseLocationService.getByName(warehouseLocationName);
                    }
                    if (!StringUtils.isNotEmpty(list.get(i).getBatchNum()))
                        return Result.error("文件导入失败: 第" + (i + 1) + "行批次号不能为空");
                    if (list.get(i).getMainQuantity() <= 0)
                        return Result.error("文件导入失败: 第" + (i + 1) + "行数量不能为小于等于0");

                    if (unit != null) {
                        if (!StringUtils.isNotEmpty(unit.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行单位不存在");
                        list.get(i).setMainUnitId(unit.getId());
                    }
                    if (supplier != null) {
                        if (!StringUtils.isNotEmpty(supplier.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行供应商不存在");
                        list.get(i).setSupplierId(supplier.getId());
                    }

                    if (warehouse != null) {
                        if (!StringUtils.isNotEmpty(warehouse.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行仓库不存在");
                        list.get(i).setWarehouseId(warehouse.getId());
                    }
                    if (warehouseArea != null) {
                        if (!StringUtils.isNotEmpty(warehouseArea.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行库区不存在");
                        list.get(i).setWarehouseAreaId(warehouseArea.getId());
                    }

                    if (warehouseLocation != null) {
                        if (!StringUtils.isNotEmpty(warehouseLocation.getId()))
                            return Result.error("文件导入失败: 第" + (i + 1) + "行库位不存在");
                        list.get(i).setWarehouseLocationId(warehouseLocation.getId());
                    }
                        list.get(i).setValidityForecast(num);

                }

                sparesPartInventoryService.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-25
     */
    @GetMapping("getSparePartInventoryList")
    public Result<?> getSparePartInventoryList(@RequestParam("pageNo") Integer pageNo,
                                               @RequestParam("pageSize") Integer pageSize,
                                               @RequestParam Map<String, Object> params) {
        IPage<Map<String, Object>> sparePartInventoryList = sparesPartInventoryService.getSparePartInventoryList(pageNo, pageSize, params);
        return Result.ok(sparePartInventoryList);
    }


    /**
     * 备件出库  获取备件库存明细信息
     * qsw 2023-6-30
     */
    @GetMapping("findSparePartInventoryList")
    public Result<?> findSparePartInventoryList(@RequestParam("pageNo") Integer pageNo,
                                                @RequestParam("pageSize") Integer pageSize,
                                                @RequestParam Map<String, Object> params) {
        IPage<Map<String, Object>> sparePartInventoryList = sparesPartInventoryService.findSparePartInventoryList(pageNo, pageSize, params);
        return Result.ok(sparePartInventoryList);
    }

}