package com.zy.asrs.controller;

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.plugins.Page;
import com.core.annotations.ManagerAuth;
import com.core.common.BaseRes;
import com.core.common.Cools;
import com.core.common.DateUtils;
import com.core.common.R;
import com.zy.asrs.entity.AgvLocDetl;
import com.zy.asrs.entity.AgvLocMast;
import com.zy.asrs.entity.param.EmptyPlateOutParam;
import com.zy.asrs.entity.param.StockOutParam;
import com.zy.asrs.service.AgvBasDevpService;
import com.zy.asrs.service.AgvLocDetlService;
import com.zy.asrs.service.AgvLocMastService;
import com.zy.asrs.service.AgvWorkService;
import com.zy.common.web.BaseController;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@RestController
@RequestMapping("/agv")
public class AgvIoWorkController extends BaseController {

    @Autowired
    AgvLocDetlService agvLocDetlService;
    @Autowired
    AgvLocMastService agvLocMastService;
    @Autowired
    AgvWorkService agvWorkService;
    @Autowired
    AgvBasDevpService agvBasDevpService;

    @RequestMapping(value = "/stock/out/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(required = false)String condition,
                  @RequestParam Map<String, Object> param,
                  @RequestParam(required = false)Boolean unreason){
        if (!Cools.isEmpty(unreason) && unreason) {

            return R.ok(agvLocDetlService.selectPage(new Page<>(curr, limit), new EntityWrapper<AgvLocDetl>()
                    .where(" DATALENGTH( batch ) != 11 or\n" +
                            "            batch LIKE '%[a-z]%'")));
        }
        param.remove("unreason");
        String row = "";
        EntityWrapper<AgvLocDetl> wrapper = new EntityWrapper<>();
        if (param.get("row") != null) {
            String chooseRow = (String) param.get("row");
            if (chooseRow.length() == 1) {
                row = "0" + chooseRow;
                param.remove("row");
            }else {
                row = chooseRow;
                param.remove("row");
            }
        }
        excludeTrash(param);
        convert(param, wrapper);
        allLike(AgvLocDetl.class, param.keySet(), wrapper, condition);
        if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));}
        if (!row.equals("")){
            wrapper.and()
                    .where("loc_no like '" +row +"%'");
        }
        Page<AgvLocDetl> agvLocDetlPage = agvLocDetlService.selectPage(new Page<>(curr, limit), wrapper);
        List<AgvLocDetl> collect = agvLocDetlPage.getRecords().stream().filter(agvLocDetl -> {
            AgvLocMast agvLocMast = agvLocMastService.selectById(agvLocDetl.getLocNo());
            return agvLocMast.getLocSts().equals("F");
        }).collect(Collectors.toList());
        agvLocDetlPage.setRecords(collect);
        return R.ok(agvLocDetlPage);
    }

    @RequestMapping(value = "/locDetl/auth")
    @ManagerAuth
    public R stockOutList(@RequestParam(value = "locNos[]") List<String> locNos){
        if (!locNos.isEmpty()) {
            List<AgvLocDetl> res = new ArrayList<>();
            for (String locNo : new HashSet<>(locNos)) {
                List<AgvLocDetl> locDetls = agvLocDetlService.selectList(new EntityWrapper<AgvLocDetl>().eq("loc_no", locNo));
                if (!locDetls.isEmpty()) {
                    res.addAll(locDetls);
                }
            }
            return R.ok().add(res);
        }
        return R.parse(BaseRes.EMPTY);
    }

    @RequestMapping("/locCheck/out/start")
    @ManagerAuth(memo = "AGV盘点出库")
    public R locCheckOutStart(@RequestBody StockOutParam param) {
        agvWorkService.locCheckOut(param, getUserId());
        return R.ok("出库启动成功");
    }

    @PostMapping(value = "/group/empty/stock")
    @ManagerAuth(memo = "获取同一楼层的空库位")
    public R getGroupEmptyStock(@RequestParam(required = false) String sourceLocNo) {

        if(Cools.isEmpty(sourceLocNo)){
            return null;
        }
        AgvLocMast agvLocMast = agvLocMastService.selectById(sourceLocNo);
        if(Cools.isEmpty(agvLocMast)){
            return null;
        }

        int floor =  Integer.parseInt(sourceLocNo.split("@")[1]);

        List<String> locNoList = agvLocMastService.queryGroupEmptyStock(floor);

        return R.ok().add(locNoList);
    }

    @RequestMapping("/loc/move/start")
    @ManagerAuth(memo = "库位移转")
    public R locMoveStart(@RequestParam String sourceLocNo,
                          @RequestParam String targetLocNo) {

        agvWorkService.locMove(sourceLocNo, targetLocNo, getUserId());
        return R.ok("移库启动成功");
    }

    @RequestMapping("/available/empty/put/site")
    @ManagerAuth(memo = "可用站点选择")
    public R availableEmptyPutSite(){
        return R.ok().add(agvBasDevpService.getAvailableEmptyInSite());
    }

    @RequestMapping("/empty/plate/in/start")
    @ManagerAuth(memo = "空板入库")
    public R emptyPlateInStart(@RequestParam String sourceStaNo) {
        return R.ok("入库启动成功").add(agvWorkService.emptyPlateIn(sourceStaNo, getUserId()));
    }

    @RequestMapping("/empty/plate/out/start")
    @ManagerAuth(memo = "空板出库")
    public R emptyPlateOutStart(EmptyPlateOutParam param) {
        agvWorkService.emptyPlateOut(param, getUserId());
        return R.ok("出库启动成功");
    }

    private <T> void convert(Map<String, Object> map, EntityWrapper<T> 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 {
                if (entry.getKey().equals("locNo")) {
                    wrapper.eq("loc_no", String.valueOf(entry.getValue()));
                } else {
                    wrapper.like(entry.getKey(), String.valueOf(entry.getValue()));
                }
            }
        }
    }

}