package com.ard.gb28181.transmit.event.request.impl.message.response.cmd;

import com.ard.gb28181.api.bean.MessageResponseTask;
import com.ard.gb28181.api.bean.Preset;
import com.ard.gb28181.api.domain.Device;
import com.ard.gb28181.transmit.event.request.SIPRequestProcessorParent;
import com.ard.gb28181.transmit.event.request.impl.message.IMessageHandler;
import com.ard.gb28181.transmit.event.request.impl.message.response.ResponseMessageHandler;
import gov.nist.javax.sip.message.SIPRequest;
import lombok.extern.slf4j.Slf4j;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.sip.InvalidArgumentException;
import javax.sip.RequestEvent;
import javax.sip.SipException;
import javax.sip.message.Response;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.DelayQueue;
import java.util.concurrent.TimeUnit;

import static com.ard.gb28181.api.utils.XmlUtil.getText;

/**
 * 设备预置位查询应答
 */
@Slf4j
@Component
public class PresetQueryResponseMessageHandler extends SIPRequestProcessorParent implements InitializingBean, IMessageHandler {

    private final String cmdType = "PresetQuery";

    @Autowired
    private ResponseMessageHandler responseMessageHandler;

    private final Map<String, MessageResponseTask<Preset>> mesageMap = new ConcurrentHashMap<>();

    private final DelayQueue<MessageResponseTask<Preset>> delayQueue = new DelayQueue<>();


    @Override
    public void afterPropertiesSet() throws Exception {
        responseMessageHandler.addHandler(cmdType, this);
    }

    @Override
    public void handForDevice(RequestEvent evt, Device device, Element element) {

        SIPRequest request = (SIPRequest) evt.getRequest();

        try {
            Element rootElement = getRootElement(evt, device.getCharset());

            if (rootElement == null) {
                log.warn("[ 设备预置位查询应答 ] content cannot be null, {}", evt.getRequest());
                try {
                    responseAck(request, Response.BAD_REQUEST);
                } catch (InvalidArgumentException | ParseException | SipException e) {
                    log.error("[命令发送失败] 设备预置位查询应答处理: {}", e.getMessage());
                }
                return;
            }
            Element presetListNumElement = rootElement.element("PresetList");
            Element snElement = rootElement.element("SN");
            //该字段可能为通道或则设备的id
            if (snElement == null || presetListNumElement == null) {
                try {
                    responseAck(request, Response.BAD_REQUEST, "xml error");
                } catch (InvalidArgumentException | ParseException | SipException e) {
                    log.error("[命令发送失败] 设备预置位查询应答处理: {}", e.getMessage());
                }
                return;
            }
            int num = Integer.parseInt(presetListNumElement.attributeValue("Num"));
            List<Preset> presetQuerySipReqList = new ArrayList<>();
            if (num > 0) {
                for (Iterator<Element> presetIterator = presetListNumElement.elementIterator(); presetIterator.hasNext(); ) {
                    Element itemListElement = presetIterator.next();
                    Preset presetQuerySipReq = new Preset();
                    for (Iterator<Element> itemListIterator = itemListElement.elementIterator(); itemListIterator.hasNext(); ) {
                        // 遍历item
                        Element itemOne = itemListIterator.next();
                        String name = itemOne.getName();
                        String textTrim = itemOne.getTextTrim();
                        if ("PresetID".equalsIgnoreCase(name)) {
                            presetQuerySipReq.setPresetId(textTrim);
                        } else {
                            presetQuerySipReq.setPresetName(textTrim);
                        }
                    }
                    presetQuerySipReqList.add(presetQuerySipReq);
                }
            }
            String sn = getText(element, "SN");
            addCatch(cmdType + "_" + sn, num, rootElement, presetQuerySipReqList);
            try {
                responseAck(request, Response.OK);
            } catch (InvalidArgumentException | ParseException | SipException e) {
                log.error("[命令发送失败] 设备预置位查询应答处理: {}", e.getMessage());
            }
        } catch (DocumentException e) {
            log.error("[解析xml]失败: ", e);
        }
    }

    private void addCatch(String key, int sumNum, Element rootElement, List<Preset> presetQuerySipReqList) {
        if (presetQuerySipReqList.size() == sumNum) {
            responseMessageHandler.handMessageEvent(rootElement, presetQuerySipReqList);
            if (mesageMap.containsKey(key)) {
                MessageResponseTask<Preset> messageResponseTask = mesageMap.get(key);
                mesageMap.remove(key);
                boolean remove = delayQueue.remove(messageResponseTask);
                if (!remove) {
                    log.info("[移除预置位查询任务] 从延时队列内移除失败： {}", key);
                }
            }
        } else {
            if (mesageMap.containsKey(key)) {
                MessageResponseTask<Preset> messageResponseTask = mesageMap.get(key);
                List<Preset> data = messageResponseTask.getData();
                data.addAll(presetQuerySipReqList);
                if (data.size() == sumNum) {
                    responseMessageHandler.handMessageEvent(rootElement, data);
                    mesageMap.remove(key);
                    boolean remove = delayQueue.remove(messageResponseTask);
                    if (!remove) {
                        log.info("[移除预置位查询任务] 从延时队列内移除失败： {}", key);
                    }
                    return;
                }
                messageResponseTask.setDelayTime(System.currentTimeMillis() + 1000);
            } else {
                MessageResponseTask<Preset> messageResponseTask = new MessageResponseTask<>();
                messageResponseTask.setElement(rootElement);
                messageResponseTask.setData(presetQuerySipReqList);
                messageResponseTask.setDelayTime(System.currentTimeMillis() + 1000);
                messageResponseTask.setKey(key);
                mesageMap.put(key, messageResponseTask);
                delayQueue.offer(messageResponseTask);
            }
        }
    }

    // 处理过期的缓存
    @Scheduled(fixedDelay = 500, timeUnit = TimeUnit.MILLISECONDS)
    public void expirationCheck() {
        while (!delayQueue.isEmpty()) {
            MessageResponseTask<Preset> take = null;
            try {
                take = delayQueue.take();
                try {
                    responseMessageHandler.handMessageEvent(take.getElement(), take.getData());
                    mesageMap.remove(take.getKey());
                } catch (Exception e) {
                    log.error("[预置位查询到期] {} 到期处理时出现异常", take.getKey());
                }
            } catch (InterruptedException e) {
                log.error("[设备订阅任务] ", e);
            }
        }
    }
}