package com.qianwen.smartman.common.websocket.fms;

import cn.hutool.core.collection.ListUtil;
import com.alibaba.fastjson.JSONObject;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import com.qianwen.smartman.common.cache.ParamCache;
import com.qianwen.smartman.common.constant.FmsConstant;
import com.qianwen.core.tool.utils.Func;
import com.qianwen.core.websocket.config.WebSocketMessageSender;
import com.qianwen.core.websocket.distribute.MessageDO;
import com.qianwen.core.websocket.distribute.RedisMessageDistributor;
import com.qianwen.core.websocket.handler.JsonMessageHandler;
import com.qianwen.smartman.modules.cps.service.IWarehouseAreaService;
import com.qianwen.smartman.modules.cps.service.IWorkstationService;
import com.qianwen.smartman.modules.fms.entity.FmsRealTimeTray;
import com.qianwen.smartman.modules.fms.service.IFmsRealTimeTrayService;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;

@Component
public class FmsVisualDataMessageHandler implements JsonMessageHandler<FmsVisualDataRequestMessage> {
    private static String fmsCarryTest = " {\"id\": 333, \"code\": \"003\", \"name\": \"搬运\", \"color\": \"#3e607c\", \"workBenchList\": [{\"name\": \"10\", \"id\": \"bench3\", \"color\": \"#0FC015\"}] }";
    private static String fmsManualStationTest = "{\"id\": 444, \"code\": \"004\", \"name\": \"人工卸货站\", \"color\": \"#3e607c\", \"workBenchList\": [{\"name\": \"18\", \"id\": \"bench4\", \"color\": \"#0FC015\"}] }";
    private final IWarehouseAreaService warehouseAreaService;
    private final IWorkstationService workstationService;
    private final IFmsRealTimeTrayService realTimeTrayService;
    private final FmsVisualDataFilter fmsVisualDataFilter;
    private final RedisMessageDistributor messageDistributor;

    public FmsVisualDataMessageHandler(final IWarehouseAreaService warehouseAreaService, final IWorkstationService workstationService, final IFmsRealTimeTrayService realTimeTrayService, final FmsVisualDataFilter fmsVisualDataFilter, final RedisMessageDistributor messageDistributor) {
        this.warehouseAreaService = warehouseAreaService;
        this.workstationService = workstationService;
        this.realTimeTrayService = realTimeTrayService;
        this.fmsVisualDataFilter = fmsVisualDataFilter;
        this.messageDistributor = messageDistributor;
    }

    public void handle(WebSocketSession session, FmsVisualDataRequestMessage message) {
        WebSocketMessageSender.send(session, getResponseMessage(message));
    }

    public String type() {
        return "fmsVisualData";
    }

    public Class<FmsVisualDataRequestMessage> getMessageClass() {
        return FmsVisualDataRequestMessage.class;
    }

    public void pushData() {
        this.fmsVisualDataFilter.dataMap.forEach((sessionId, message) -> {
            MessageDO messageDO = new MessageDO();
            messageDO.setSessionIds(ListUtil.toList(new String[]{sessionId}));
            messageDO.setNeedBroadcast(false);
            messageDO.setMessageText(JSONObject.toJSONString(getResponseMessage(message)));
            this.messageDistributor.distribute(messageDO);
        });
    }

    public FmsVisualDataResponseMessage getResponseMessage(FmsVisualDataRequestMessage message) {
        List<String> fmsAreaListStr = message.getFmsAreaList();
        List<String> fmsCarryListStr = message.getFmsCarryList();
        List<String> fmsWorkstationListStr = message.getFmsWorkstationList();
        List<String> fmsManualStationList = message.getFmsManualStationList();
        FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
        List<FmsRealTimeTray> realTimeTrayList = this.realTimeTrayService.list();
        List<String> positionList = (List) realTimeTrayList.stream().map((v0) -> {
            return v0.getCurrentPosition();
        }).filter(currentPosition -> {
            return Func.isNotEmpty(currentPosition);
        }).collect(Collectors.toList());
        Map<String, FmsRealTimeTray> realTimeTrayMap = (Map) realTimeTrayList.stream().filter(c -> {
            return Func.isNotEmpty(c.getTrayCode()) && Func.isNotEmpty(c.getCurrentPosition());
        }).collect(Collectors.toMap((v0) -> {
            return v0.getTrayCode();
        }, r -> {
            return r;
        }));
        if (Func.isNotEmpty(fmsAreaListStr)) {
            responseMessage.setFmsAreaList(this.warehouseAreaService.assembleAreaData(fmsAreaListStr, realTimeTrayMap));
        }
        if (Func.isNotEmpty(fmsCarryListStr)) {
            responseMessage.setFmsCarryList(this.workstationService.assemblyWorkstationData(fmsCarryListStr, realTimeTrayMap, realTimeTrayList, positionList));
        }
        if (Func.isNotEmpty(fmsWorkstationListStr)) {
            responseMessage.setFmsWorkstationList(this.workstationService.assemblyWorkstationData(fmsWorkstationListStr, realTimeTrayMap, realTimeTrayList, positionList));
        }
        if (Func.isNotEmpty(fmsManualStationList)) {
            responseMessage.setFmsManualStationList(this.workstationService.assemblyWorkstationData(fmsManualStationList, realTimeTrayMap, realTimeTrayList, positionList));
        }
        responseMessage.setIsAuto(ParamCache.getValue(FmsConstant.PARAM_KEY));
        return responseMessage;
    }
}