package com.zy.acs.manager.core.service;

import com.zy.acs.framework.common.Cools;
import com.zy.acs.manager.manager.entity.Code;
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.service.AreaAgvService;
import com.zy.acs.manager.manager.service.CodeService;
import com.zy.acs.manager.manager.service.LocService;
import com.zy.acs.manager.manager.service.StaService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.util.ArrayList;
import java.util.List;

@Slf4j
@Service
public class AgvAreaDispatcher {

    @Autowired
    private LocService locService;
    @Autowired
    private StaService staService;
    @Autowired
    private AreaGovernService areaGovernService;
    @Autowired
    private AreaAgvService areaAgvService;
    @Autowired
    private CodeService codeService;

    @PostConstruct
    public void init() {
    }

    public List<Long> getAgvNosByTask(Task task) {
        if (null == task) {
            return null;
        }
        Loc oriLoc = null;
        Sta oriSta = null;
        Loc destLoc = null;
        Sta destSta = null;
        if (null != task.getOriLoc()) {
            oriLoc = locService.getById(task.getOriLoc());
            Code code = codeService.getCacheById(oriLoc.getCode());
            return this.getAgvIdsByCode(code.getData());
        }
        if (null != task.getOriSta()) {
            oriSta = staService.getById(task.getOriSta());
            Code code = codeService.getCacheById(oriSta.getCode());
            return this.getAgvIdsByCode(code.getData());
        }
        if (null != task.getDestLoc()) {
            destLoc = locService.getById(task.getDestLoc());
            Code code = codeService.getCacheById(destLoc.getCode());
            return this.getAgvIdsByCode(code.getData());
        }
        if (null != task.getDestSta()) {
            destSta = staService.getById(task.getDestSta());
            Code code = codeService.getCacheById(destSta.getCode());
            return this.getAgvIdsByCode(code.getData());
        }
        return null;
    }

    public List<Long> getAgvIdsByCode(String code) {
        if (Cools.isEmpty(code)) {
            return new ArrayList<>();
        }
        List<Long> areaIds = areaGovernService.queryAreas(code);
        List<Long> agvIds = areaAgvService.queryAgvIdsByAreaIds(areaIds);
        agvIds.addAll(areaAgvService.findAgvIdsWithoutAreaAgv());
        return agvIds;
    }

    public List<String> getAgvNosByStaNo(String staNo) {
        return null;
    }

    // 如果都没有命中，就返回全部code
    public List<String> getCodesByAgvId(Long agvId) {
        List<Long> areaIds = areaAgvService.queryAreaIdsByAgvId(agvId);
        return areaGovernService.queryCodes(areaIds);
    }

    // 判断车辆是否被area绑定，如果绑定返回 true， 如果没绑定返回 false
    public Boolean isAgvExistsInAnyArea(Long agvId) {
        List<Long> agvIdsWithoutAreaAgv = areaAgvService.findAgvIdsWithoutAreaAgv(); // 没有被绑定的车辆集合
        if (Cools.isEmpty(agvIdsWithoutAreaAgv)) {
            return true;
        }
        return !agvIdsWithoutAreaAgv.contains(agvId);
    }

}