package zy.cloud.wms.manager.controller;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.mapper.Wrapper;
import com.baomidou.mybatisplus.plugins.Page;
import com.core.annotations.ManagerAuth;
import com.core.common.*;
import com.core.exception.CoolException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;
import zy.cloud.wms.common.entity.SafeStoExport;
import zy.cloud.wms.common.model.SafeStoDto;
import zy.cloud.wms.common.web.BaseController;
import zy.cloud.wms.manager.entity.Mat;
import zy.cloud.wms.manager.entity.Node;
import zy.cloud.wms.manager.entity.Prior;
import zy.cloud.wms.manager.entity.SafeStoDo;
import zy.cloud.wms.manager.service.MatService;
import zy.cloud.wms.manager.service.NodeService;
import zy.cloud.wms.manager.service.PriorService;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.*;

@RestController
public class PriorController extends BaseController {

    @Autowired
    private PriorService priorService;
    @Autowired
    private MatService matService;
    @Autowired
    private NodeService nodeService;
    @Autowired
    private SnowflakeIdWorker snowflakeIdWorker;

    /**
     * 安全库存警告分页
     * @param curr
     * @param limit
     * @param param
     * @return
     */
    @RequestMapping(value = "/safeSto/list/auth")
    @ManagerAuth
    public R safeStoList(@RequestParam(defaultValue = "1")Integer curr,
                  @RequestParam(defaultValue = "10")Integer limit,
                  @RequestParam Map<String, Object> param){
        Long hostId = getHostId();
        if (hostId != null) {
            param.put("host_id", hostId);
        }
        return R.ok(priorService.getSafeQtyPage(toPage(curr, limit, param, SafeStoDo.class)));
    }

    @RequestMapping(value = "/safeSto/delete/auth")
    @ManagerAuth(memo = "删除库存预警")
    @Transactional
    public R del(@RequestBody List<SafeStoDto> dtos){
        if (Cools.isEmpty(dtos)) {
            return R.parse(BaseRes.PARAM);
        }
        for (SafeStoDto dto : dtos) {
            if (!priorService.delete(new EntityWrapper<Prior>().eq("node_id", dto.getNodeId()).eq("matnr", dto.getMatnr()))) {
                throw new CoolException("删除安全库存失败，请重新尝试");
            }
        }
        return R.ok();
    }

    @RequestMapping(value = "/safeSto/add/auth")
    @ManagerAuth(memo = "添加安全库存")
    public R del(@RequestBody SafeStoDto dto){
        if (Cools.isEmpty(dto.getNodeId(), dto.getMatId(), dto.getSafeQua())) {
            return R.parse(BaseRes.PARAM);
        }
        if (dto.getSafeQua() <= 0.0D) {
            return R.error("安全库存量必须大于零");
        }
        Node node = nodeService.selectById(dto.getNodeId());
        Mat mat = matService.selectById(dto.getMatId());
        if (node != null && mat != null) {
            Prior prior = new Prior();
            prior.setHostId(getHostId());
            prior.setUuid("SP" + snowflakeIdWorker.nextId());
            prior.setMatId(mat.getId());
            prior.setMatnr(mat.getMatnr());
            prior.setMaktx(mat.getMaktx());
            prior.setNodeId(node.getId());
            prior.setNodeName(node.getName());
            prior.setSafeQua(dto.getSafeQua());
            prior.setCreateBy(getUserId());
            prior.setCreateTime(new Date());
            prior.setUpdateBy(getUserId());
            prior.setUpdateTime(new Date());
            prior.setStatus(1);
            prior.setMemo(dto.getMemo());
            if (!priorService.insert(prior)) {
                throw new CoolException("数据保存失败");
            }
        } else {
            throw new CoolException("服务器内部错误");
        }
        return R.ok();
    }

    /**
     * 安全库存excel导出
     */
    @RequestMapping(value = "/safeSto/excel/export/auth")
    public void matExcelImportMould(HttpServletResponse response, @RequestParam String token) throws IOException {
        Long hostId = getHostId(token);
        List<SafeStoExport> excels = priorService.getSafeExportExcel(hostId);
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        String fileName = URLEncoder.encode("安全库存明细", "UTF-8");
        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), SafeStoExport.class)
                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
                .sheet("sheet1")
                .doWrite(excels);
    }

    /***************************************************/

    @RequestMapping(value = "/prior/{id}/auth")
    @ManagerAuth
    public R get(@PathVariable("id") String id) {
        return R.ok(priorService.selectById(String.valueOf(id)));
    }

    @RequestMapping(value = "/prior/list/auth")
    @ManagerAuth
    public R list(@RequestParam(defaultValue = "1")Integer curr,
                  @RequestParam(defaultValue = "10")Integer limit,
                  @RequestParam(required = false)String orderByField,
                  @RequestParam(required = false)String orderByType,
                  @RequestParam Map<String, Object> param){
        EntityWrapper<Prior> wrapper = new EntityWrapper<>();
        excludeTrash(param);
        convert(param, wrapper);
        hostEq(wrapper);
        if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));}
        return R.ok(priorService.selectPage(new Page<>(curr, limit), wrapper));
    }

    private void convert(Map<String, Object> map, EntityWrapper wrapper){
        for (Map.Entry<String, Object> entry : map.entrySet()){
            String val = String.valueOf(entry.getValue());
            if (val.contains(RANGE_TIME_LINK)){
                String[] dates = val.split(RANGE_TIME_LINK);
                wrapper.ge(entry.getKey(), DateUtils.convert(dates[0]));
                wrapper.le(entry.getKey(), DateUtils.convert(dates[1]));
            } else {
                wrapper.like(entry.getKey(), val);
            }
        }
    }

    @RequestMapping(value = "/prior/add/auth")
    @ManagerAuth
    public R add(Prior prior) {
        if (prior.getSafeQua() <= 0.0D) {
            return R.error("安全库存量必须大于零");
        }
        if (prior.getMatId() != null) {
            Mat mat = matService.selectById(prior.getMatId());
            if (mat != null) {
                prior.setMatnr(mat.getMatnr());
                prior.setMaktx(mat.getMaktx());
            }
        }
        if (prior.getNodeId() != null) {
            Node node = nodeService.selectById(prior.getNodeId());
            if (node != null) {
                prior.setNodeName(node.getName());
            }
        }
        prior.setHostId(getHostId());
        prior.setCreateBy(getUserId());
        prior.setCreateTime(new Date());
        prior.setUpdateBy(getUserId());
        prior.setUpdateTime(new Date());
        prior.setStatus(1);
        priorService.insert(prior);
        return R.ok();
    }

	@RequestMapping(value = "/prior/update/auth")
	@ManagerAuth
    public R update(Prior prior){
        if (Cools.isEmpty(prior) || null==prior.getId()){
            return R.error();
        }
        if (prior.getSafeQua() <= 0.0D) {
            return R.error("安全库存量必须大于零");
        }
        if (prior.getMatId() != null) {
            Mat mat = matService.selectById(prior.getMatId());
            if (mat != null) {
                prior.setMatnr(mat.getMatnr());
                prior.setMaktx(mat.getMaktx());
            }
        }
        if (prior.getNodeId() != null) {
            Node node = nodeService.selectById(prior.getNodeId());
            if (node != null) {
                prior.setNodeName(node.getName());
            }
        }
        prior.setUpdateBy(getUserId());
        prior.setUpdateTime(new Date());
        priorService.updateById(prior);
        return R.ok();
    }

    @RequestMapping(value = "/prior/delete/auth")
    @ManagerAuth
    public R delete(@RequestParam String param){
        List<Prior> list = JSONArray.parseArray(param, Prior.class);
        if (Cools.isEmpty(list)){
            return R.error();
        }
        for (Prior entity : list){
            priorService.delete(new EntityWrapper<>(entity));
        }
        return R.ok();
    }

    @RequestMapping(value = "/prior/export/auth")
    @ManagerAuth
    public R export(@RequestBody JSONObject param){
        EntityWrapper<Prior> wrapper = new EntityWrapper<>();
        hostEq(wrapper);
        List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class);
        Map<String, Object> map = excludeTrash(param.getJSONObject("prior"));
        convert(map, wrapper);
        List<Prior> list = priorService.selectList(wrapper);
        return R.ok(exportSupport(list, fields));
    }

    @RequestMapping(value = "/priorQuery/auth")
    @ManagerAuth
    public R query(String condition) {
        EntityWrapper<Prior> wrapper = new EntityWrapper<>();
        wrapper.like("id", condition);
        hostEq(wrapper);
        Page<Prior> page = priorService.selectPage(new Page<>(0, 10), wrapper);
        List<Map<String, Object>> result = new ArrayList<>();
        for (Prior prior : page.getRecords()){
            Map<String, Object> map = new HashMap<>();
            map.put("id", prior.getId());
            map.put("value", prior.getId());
            result.add(map);
        }
        return R.ok(result);
    }

    @RequestMapping(value = "/prior/check/column/auth")
    @ManagerAuth
    public R query(@RequestBody JSONObject param) {
        Wrapper<Prior> wrapper = new EntityWrapper<Prior>().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val"));
        if (null != priorService.selectOne(wrapper)){
            return R.parse(BaseRes.REPEAT).add(getComment(Prior.class, String.valueOf(param.get("key"))));
        }
        return R.ok();
    }

    @Value("${super.pwd}")
    private String superPassword;

    /************************* 数据初始化 **************************/
    @RequestMapping(value = "/prior/init/auth")
    @Transactional
    public R init(@RequestParam String password) {
        if (Cools.isEmpty(password)) {
            return R.error();
        }
        if (!password.equals(superPassword)) {
            return R.error();
        }
        List<Mat> allMat = matService.selectList(new EntityWrapper<>());
        Node topNode = getOriginNode();
        Date now = new Date();
        for (Mat mat : allMat) {
            Prior prior = new Prior(
                    getHostId(),
                    String.valueOf(snowflakeIdWorker.nextId()),    // 编号
                    null,    // 推荐名
                    mat.getId(),    // 关联商品[非空]
                    mat.getMatnr(),    // 商品编号
                    mat.getMaktx(),    // 商品名称
                    topNode.getId(),    // 关联货位[非空]
                    topNode.getUuid(),    // 货位名称
                    50.0D,    // 安全库存
                    0,    // 优先级
                    null,    // 条码
                    1,    // 状态
                    9527L,    // 添加人员
                    now,    // 添加时间
                    9527L,    // 修改人员
                    now,    // 修改时间
                    null    // 备注
            );
            if (!priorService.insert(prior)) {
                throw new CoolException("服务器异常");
            }
        }
        return R.ok();
    }

}