package com.iplatform.tcp.support;

import com.iplatform.tcp.util.ws.WebBroadCastResponse;
import com.iplatform.tcp.util.ws.WebDataResponse;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.push.AbstractPushObject;
import com.walker.push.Notification;
import com.walker.push.NotificationChannel;
import com.walker.push.PushException;
import com.walker.push.PushResult;
import com.walker.push.util.PushUtils;
import com.walker.tcp.netty.WebSocketEngine;

import java.util.ArrayList;
import java.util.List;

/**
 * WebSocket方式实现的推送对象。
 * @author 时克英
 * @date 2023-04-21
 */
public class WebSocketPush extends AbstractPushObject<WebBroadCastResponse> {

    public WebSocketPush(){
        this.setId(NotificationChannel.WebSocket.getIndex());
        this.setName(NotificationChannel.WebSocket.getName() + "推送");
    }

    @Override
    protected PushResult doPushContent(Notification notification, List<WebBroadCastResponse> data) throws PushException {
        PushResult pushResult = PushUtils.acquireSuccessPushResult();
        String error = null;
        for(WebBroadCastResponse response : data){
            if(notification.getBroadcast()){
                webSocketEngine.sendBroadcast(response);
                logger.debug("websocket广播直接发送，没有状态");
                return pushResult;
            }

            error = this.webSocketEngine.sendResponse(response);

            if(StringUtils.isNotEmpty(error)){
                if(pushResult.getCode() == 0){
                    pushResult.setCode(-1);
                }
                // 这里无需转换用户信息，因为使用的就是userId，一致的!
                pushResult.addOneFailed(response.getName());
            }
        }
        return pushResult;
    }

//    @Override
//    protected PushResult doPushContent(Notification notification) throws PushException {
//        if(notification.getBroadcast()){
//            logger.debug("websocket推送广播");
//
//            WebBroadCastResponse msg = new WebBroadCastResponse();
//            msg.setMessageId(String.valueOf(NumberGenerator.getSequenceNumber()));
//            msg.setData(notification.getContent());
//            webSocketEngine.sendBroadcast(msg);
//            return PushUtils.acquireSuccessPushResult();
//        }
//
//        // 普通推送消息
//        PushResult pushResult = new PushResult();
//        WebDataResponse response = null;
//        String error = null;
//
//        List<String> receiverList = notification.getReceiverList();
//        for(String user : receiverList){
//            response = new WebDataResponse();
//            response.setMessageId(String.valueOf(NumberGenerator.getSequenceNumber()));
//            response.setData(notification.getContent());
//            response.setName(user); // websocket用户标识为：userId
//            error = this.webSocketEngine.sendResponse(response);
//
//            if(StringUtils.isNotEmpty(error)){
//                if(pushResult.getCode() == 0){
//                    pushResult.setCode(-1);
//                }
//                pushResult.addOneFailed(user);
//            }
//        }
//        return pushResult;
//    }

    @Override
    public NotificationChannel getNotificationChannel() {
        return NotificationChannel.WebSocket;
    }

    @Override
    public List<WebBroadCastResponse> translateToTarget(Notification notification) {

        List<WebBroadCastResponse> data = new ArrayList<>(2);
        WebBroadCastResponse webBroadCastResponse = null;

        if(notification.getBroadcast()){
            logger.debug("websocket推送广播");
            webBroadCastResponse = new WebBroadCastResponse();
            webBroadCastResponse.setMessageId(notification.getId());
            webBroadCastResponse.setData(notification.getContent());
            data.add(webBroadCastResponse);
            return data;
        }

        // 非广播，则需要把所有接收人集合返回
        int i = 1;
        for(String user: notification.getReceiverList()){
            webBroadCastResponse = new WebDataResponse();
            webBroadCastResponse.setMessageId(notification.getId() + i);
            webBroadCastResponse.setData(notification.getContent());
            webBroadCastResponse.setName(user); // websocket用户标识为：userId
            data.add(webBroadCastResponse);
            i++;
        }
        return data;
    }

    @Override
    public void startup() {

    }

    public void setWebSocketEngine(WebSocketEngine webSocketEngine) {
        this.webSocketEngine = webSocketEngine;
    }

    private WebSocketEngine webSocketEngine;
}