package com.zy.acs.manager.manager.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zy.acs.framework.common.Cools;
import com.zy.acs.framework.common.R;
import com.zy.acs.manager.common.annotation.OperationLog;
import com.zy.acs.manager.common.domain.BaseParam;
import com.zy.acs.manager.common.domain.PageParam;
import com.zy.acs.manager.common.domain.TaskDto;
import com.zy.acs.manager.core.constant.LocGroupConstant;
import com.zy.acs.manager.core.service.MainService;
import com.zy.acs.manager.manager.controller.param.OpenBusSubmitParam;
import com.zy.acs.manager.manager.entity.Loc;
import com.zy.acs.manager.manager.entity.Sta;
import com.zy.acs.manager.manager.entity.Task;
import com.zy.acs.manager.manager.enums.StaStsType;
import com.zy.acs.manager.manager.service.LocService;
import com.zy.acs.manager.manager.service.StaService;
import com.zy.acs.manager.manager.service.TaskService;
import com.zy.acs.manager.system.controller.BaseController;
import com.zy.acs.manager.system.service.ConfigService;
import io.swagger.annotations.Api;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.*;

/**
 * Created by vincent on 2023/6/12
 */
@Api(tags = "Open Api")
@RestController
@RequestMapping("/api/open")
public class OpenController extends BaseController {

    @Autowired
    private MainService mainService;

    @Autowired
    private LocService locService;

    @Autowired
    private StaService staService;

    @Autowired
    private ConfigService configService;


    @Autowired
    private TaskService taskService;


    //@PreAuthorize("hasAuthority('open:bus:submit')")
    @PostMapping("/bus/submit")
    @OperationLog("generate task from open api")
    public R save(@RequestBody OpenBusSubmitParam param) {
        if (!configService.getVal("TaskAssignMode", Boolean.class)) {
            for (TaskDto taskDto : param.getTaskList()) {
                if (Cools.isEmpty(taskDto.getSeqNum())) {
                    return R.error("缺失任务号");
                }
            }
            mainService.generateBusAndTask(param, null);
            return R.ok("generate tasks success");
        }
        return R.error("generate tasks error");
    }


    @OperationLog("Cancel Task")
    @PostMapping("/task/cancel")
    public R cancel(@RequestBody List<String> taskIds) {
        if (Cools.isEmpty(taskIds)) {
            return R.error("task cancel error");
        }
        List<Boolean> data = new ArrayList<>();
        for (String id : taskIds) {
            data.add(taskService.cancelCheckTaskSts(taskService.selectBySeqNum(id).getId(), 9527L));
        }
        return R.ok().add(data);
    }


    @OperationLog("updateTaskPriority")
    @PostMapping("/task/updateTaskPriority")
    public R update(@RequestBody List<TaskDto> taskDtos) {
        if (Cools.isEmpty(taskDtos)) {
            return R.error("task update error");
        }
        List<Boolean> data = new ArrayList<>();
        for (TaskDto taskDto : taskDtos) {
            Task task = taskService.selectBySeqNum(taskDto.getSeqNum());
            task.setPriority(taskDto.getPriority());
            task.setUpdateBy(9527L);
            task.setUpdateTime(new Date());
            data.add(taskService.updateById(task));
        }
        return R.ok("Update Success").add(data);
    }


    @PostMapping("/loc/sync")
    @OperationLog("sync loc")
    public R loc(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        PageParam<Loc, BaseParam> pageParam = new PageParam<>(baseParam, Loc.class);
        return R.ok().add(locService.page(pageParam, pageParam.buildWrapperExcludeNull(true)));
    }


    @PostMapping("/sta/update")
    @OperationLog("sync sta")
    public R updateLoc(@RequestBody Map<String, Object> map) {
        if (!configService.getVal("TaskAssignMode", Boolean.class)) {
            if (configService.getVal("InAndOutMode", Boolean.class, false)) {
                String staNo = map.get("staNo").toString();
                Sta loc = staService.selectByStaNo(staNo);
                if (staNo.equals("1001")) {
                    if (loc.getStaSts() == StaStsType.STOCK.val()) {
                        loc.setStaSts(Long.parseLong(map.get("sts").toString()));
                        return staService.updateById(loc) ? R.ok() : R.error("update loc fail");
                    }
                } else if (staNo.equals("1007")) {
                    if (loc.getStaSts() == StaStsType.IDLE.val()) {
                        loc.setStaSts(Long.parseLong(map.get("sts").toString()));
                        return staService.updateById(loc) ? R.ok() : R.error("update loc fail");
                    }
                }
            }
        }
        return R.error("update loc fail");
    }


    @PostMapping("/loc/one")
    @OperationLog("one loc")
    public R emptyLoc(@RequestBody Map<String, Object> map) {
        if (!configService.getVal("TaskAssignMode", Boolean.class)) {
            if (configService.getVal("InAndOutMode", Boolean.class, false)) {
                String staNo = map.get("staNo").toString();
                Integer startRow = null;
                Integer endRow = null;
                if (staNo.equals("101-2") || staNo.equals("101-3") || staNo.equals("102-2") || staNo.equals("102-3")) {
                    startRow = Collections.min(LocGroupConstant.FAR_RIGHT_LOC_ROW_LIST);
                    endRow = Collections.max(LocGroupConstant.FAR_RIGHT_LOC_ROW_LIST);
                } else if (staNo.equals("103-2") || staNo.equals("103-3") || staNo.equals("104-2") || staNo.equals("104-3")) {
                    startRow = Collections.min(LocGroupConstant.RIGHT_LOC_ROW_LIST);
                    endRow = Collections.max(LocGroupConstant.RIGHT_LOC_ROW_LIST);
                } else if (staNo.equals("105-2") || staNo.equals("105-3") || staNo.equals("106-2") || staNo.equals("106-3")) {
                    startRow = Collections.min(LocGroupConstant.MIDDLE_LOC_ROW_LIST);
                    endRow = Collections.max(LocGroupConstant.MIDDLE_LOC_ROW_LIST);
                } else if (staNo.equals("107-2") || staNo.equals("107-3") || staNo.equals("108-2") || staNo.equals("108-3")) {
                    startRow = Collections.min(LocGroupConstant.LEFT_LOC_ROW_LIST);
                    endRow = Collections.max(LocGroupConstant.LEFT_LOC_ROW_LIST);
                } else if (staNo.equals("1007") || staNo.equals("1001")) {
                    startRow = Collections.min(LocGroupConstant.FAR_LEFT_LOC_ROW_LIST);
                    endRow = Collections.max(LocGroupConstant.FAR_LEFT_LOC_ROW_LIST);
                } else {
                    return R.error("staNo is not support");
                }
                LambdaQueryWrapper<Loc> idleWrapper = new LambdaQueryWrapper<Loc>().eq(Loc::getLocSts, Long.parseLong(map.get("sts").toString()));
                if (null != startRow) {
                    idleWrapper.ge(Loc::getRow, startRow);
                }
                if (null != endRow) {
                    idleWrapper.le(Loc::getRow, endRow);
                }
                List<Loc> idleList = locService.list(idleWrapper);
                if (Cools.isEmpty(idleList)) {
                    return R.error("没空库位");
                }
                Collections.shuffle(idleList);
                return R.ok().add(idleList.get(0));
            }
        }
        return R.error("未开启该模式");
    }

}