package com.juick.ws;

import com.juick.User;
import com.juick.json.MessageSerializer;
import com.juick.server.MessagesQueries;
import com.juick.server.SubscriptionsQueries;
import com.juick.util.ThreadHelper;
import org.apache.commons.lang3.math.NumberUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.env.Environment;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.util.Assert;
import org.springframework.web.socket.TextMessage;
import rocks.xmpp.core.XmppException;
import rocks.xmpp.core.session.Extension;
import rocks.xmpp.core.session.XmppSession;
import rocks.xmpp.core.session.XmppSessionConfiguration;
import rocks.xmpp.core.stanza.model.Message;
import rocks.xmpp.extensions.component.accept.ExternalComponent;

import java.io.IOException;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.stream.Collectors;

/**
 * @author ugnich
 */
public class XMPPConnection implements InitializingBean {
    private static final Logger logger = LoggerFactory.getLogger(XMPPConnection.class);

    private final JdbcTemplate jdbc;
    private final WebsocketComponent wsHandler;
    private final String xmppPassword;
    private final MessageSerializer ms;
    private final int xmppPort;
    private final String wsJid;

    private XmppSession xmpp;


    public XMPPConnection(
            final Environment env, final WebsocketComponent wsHandler, final JdbcTemplate jdbc) {
        Assert.notNull(env);
        Assert.notNull(wsHandler);
        Assert.notNull(jdbc);

        this.wsHandler = wsHandler;
        this.jdbc = jdbc;

        xmppPassword = env.getProperty("xmpp_password");
        xmppPort = NumberUtils.toInt(env.getProperty("xmpp_port"), 5347);
        wsJid = env.getProperty("ws_jid", "ws.juick.local");

        ms = new MessageSerializer();
    }

    @Override
    public void afterPropertiesSet() {
        XmppSessionConfiguration configuration = XmppSessionConfiguration.builder()
                .extensions(Extension.of(com.juick.Message.class))
                .build();
        xmpp = ExternalComponent.create(wsJid, xmppPassword, configuration, "localhost", xmppPort);
        xmpp.addInboundMessageListener(e -> {
            Message msg = e.getMessage();
            com.juick.Message jmsg = msg.getExtension(com.juick.Message.class);
            if (jmsg != null) {
                logger.info("got msg: " + ms.serialize(jmsg).toString());
                if (jmsg.getMid() == 0) {
                    int uid_to = NumberUtils.toInt(msg.getTo().getLocal(), 0);
                    if (uid_to > 0) {
                        onJuickPM(uid_to, jmsg);
                    }
                } else if (jmsg.getRid() == 0) {
                    onJuickMessagePost(jmsg);
                } else {
                    // to get quote
                    com.juick.Message reply = MessagesQueries.getReply(jdbc, jmsg.getMid(), jmsg.getRid());
                    onJuickMessageReply(reply);
                }
            }
        });
        try {
            xmpp.connect();
        } catch (XmppException e) {
            logger.warn("xmpp extension", e);
        }
    }

    MessageSerializer messageSerializer = new MessageSerializer();

    private void onJuickPM(final int uid_to, final com.juick.Message jmsg) {
        String json = messageSerializer.serialize(jmsg).toString();
        synchronized (wsHandler.getClients()) {
            wsHandler.getClients().stream().filter(c -> !c.legacy && c.visitor.getUid() == uid_to).forEach(c -> {
                try {
                    logger.info("sending pm to " + c.visitor.getUid());
                    c.session.sendMessage(new TextMessage(json));
                } catch (IOException e) {
                    logger.warn("ws error", e);
                }
            });
        }
    }

    private void onJuickMessagePost(final com.juick.Message jmsg) {
        String json = messageSerializer.serialize(jmsg).toString();
        List<Integer> uids = SubscriptionsQueries.getSubscribedUsers(jdbc, jmsg.getUser().getUid(), jmsg.getMid())
                .stream().map(User::getUid).collect(Collectors.toList());
        synchronized (wsHandler.getClients()) {
            wsHandler.getClients().stream().filter(c ->
                    (!c.legacy && c.visitor.getUid() == 0) // anonymous users
                            || (!c.legacy && uids.contains(c.visitor.getUid()))) // subscriptions
                    .forEach(c -> {
                        try {
                            logger.info("sending message to " + c.visitor.getUid());
                            c.session.sendMessage(new TextMessage(json));
                        } catch (IOException e) {
                            logger.warn("ws error", e);
                        }
                    });
            wsHandler.getClients().stream().filter(c ->
                    c.legacy && c.allMessages) // legacy all posts
                    .forEach(c -> {
                        try {
                            logger.info("sending message to legacy client " + c.visitor.getUid());
                            c.session.sendMessage(new TextMessage(json));
                        } catch (IOException e) {
                            logger.warn("ws error", e);
                        }
                    });
        }
    }

    private void onJuickMessageReply(final com.juick.Message jmsg) {
        String json = messageSerializer.serialize(jmsg).toString();
        List<Integer> threadUsers =
                SubscriptionsQueries.getUsersSubscribedToComments(jdbc, jmsg.getMid(), jmsg.getUser().getUid())
                        .stream().map(User::getUid).collect(Collectors.toList());
        synchronized (wsHandler.getClients()) {
            wsHandler.getClients().stream().filter(c ->
                    (!c.legacy && c.visitor.getUid() == 0) // anonymous users
                            || (!c.legacy && threadUsers.contains(c.visitor.getUid()))) // subscriptions
                    .forEach(c -> {
                        try {
                            logger.info("sending reply to " + c.visitor.getUid());
                            c.session.sendMessage(new TextMessage(json));
                        } catch (IOException e) {
                            logger.warn("ws error", e);
                        }
                    });
            wsHandler.getClients().stream().filter(c ->
                    (c.legacy && c.allReplies) || (c.legacy && c.MID == jmsg.getMid())) // legacy replies
                    .forEach(c -> {
                        try {
                            logger.info("sending reply to legacy client " + c.visitor.getUid());
                            c.session.sendMessage(new TextMessage(json));
                        } catch (IOException e) {
                            logger.warn("ws error", e);
                        }
                    });
        }
    }
}