package com.ard.work.component.websocket.handler;

import com.ard.common.constant.CacheConstants;
import com.ard.common.core.domain.entity.SysUser;
import com.ard.common.core.redis.RedisCache;
import com.ard.work.config.RouteStateMachineService;
import com.ard.work.domain.cache.UavSignalInfoCache;
import com.ard.work.domain.cache.UavStateCache;
import com.ard.work.domain.cache.UavTelemetryDataCache;
import com.ard.work.domain.cache.UavWaylineCache;
import com.ard.work.domain.dto.signal.SignalInfo;
import com.ard.work.domain.dto.state.UavState;
import com.ard.work.domain.dto.telemetry.TelemetryData;
import com.ard.work.domain.dto.wayline.WaylineInfo;
import com.ard.work.domain.enums.RouteState;
import com.ard.work.service.control.UavControlService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@Component
public class UavAllManager extends BaseSubscriptionManager<Map<String, Object>> {

    @Resource
    private UavControlService uavControlService;
    @Autowired
    private RedisCache redisCache;
    @Autowired
    private RouteStateMachineService stateMachineService;

    @Override
    protected Map<String, Object> getData(String boxSn) {
        log.debug("获取无人机所有数据");
        Map<String, Object> map = new HashMap<>();
        map.put("boxSn", boxSn);

        WaylineInfo waylineInfo = UavWaylineCache.get(boxSn);
        if (waylineInfo != null) {
            RouteState currentState = stateMachineService.getCurrentState(boxSn);
            waylineInfo.setRouteState(currentState);
        }
        map.put("wayline", waylineInfo);

        UavState uavState = UavStateCache.get(boxSn);
        map.put("state", uavState);

        TelemetryData uavTelemetryData = UavTelemetryDataCache.get(boxSn);
        map.put("telemetry", uavTelemetryData);

        SignalInfo uavSignalInfo = UavSignalInfoCache.get(boxSn);
        map.put("signal", uavSignalInfo);
        // 获取当前控制飞行的用户ID
        Long flyUserId = uavControlService.getCurrentController(boxSn, "fly");
        SysUser flyUser = redisCache.getCacheObject(CacheConstants.SYS_USER_KEY + flyUserId);
        // 将飞行当前控制者用户ID添加到映射中
        map.put("flyController", flyUser);
        // 获取当前控制载荷的用户ID
        Long payloadUserId = uavControlService.getCurrentController(boxSn, "payload");
        SysUser payloadUser = redisCache.getCacheObject(CacheConstants.SYS_USER_KEY + payloadUserId);
        // 将载荷当前控制者用户ID添加到映射中
        map.put("payloadController", payloadUser);
        return map;
    }

    @Override
    protected String getMessageType() {
        return "all";
    }

}