package com.juick.xmpp.s2s;

import com.juick.xmpp.*;
import com.juick.xmpp.utils.XmlUtils;
import org.xmlpull.v1.XmlPullParser;

import java.io.EOFException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.nio.channels.AsynchronousSocketChannel;
import java.nio.channels.Channels;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 *
 * @author ugnich
 */
public class ConnectionIn extends Connection {

    private static final Logger LOGGER = Logger.getLogger(ConnectionIn.class.getName());

    final public List<String> from = new ArrayList<>();
    public long tsRemoteData = 0;
    public long packetsRemote = 0;

    JuickBot bot;
    Stream xmpp;

    public ConnectionIn(AsynchronousSocketChannel socket, JuickBot bot, Stream xmpp) {
        super();
        this.socket = socket;
        this.bot = bot;
        this.xmpp = xmpp;
        streamID = UUID.randomUUID().toString();
    }

    public void parseStream() {
        LOGGER.info("STREAM FROM ? " + streamID + " START");
        try {
            parser.setInput(new InputStreamReader(Channels.newInputStream(socket)));
            writer = new OutputStreamWriter(Channels.newOutputStream(socket));

            parser.next(); // stream:stream
            updateTsRemoteData();
            if (!parser.getName().equals("stream:stream")
                    || !parser.getAttributeValue(null, "xmlns").equals("jabber:server")
                    || !parser.getAttributeValue(null, "xmlns:stream").equals("http://etherx.jabber.org/streams")
                    || !parser.getAttributeValue(null, "xmlns:db").equals("jabber:server:dialback")) {
//                    || !parser.getAttributeValue(null, "version").equals("1.0")
//                    || !parser.getAttributeValue(null, "to").equals(Main.HOSTNAME)) {
                throw new Exception("STREAM FROM ? " + streamID + " INVALID FIRST PACKET");
            }
            boolean xmppversionnew = parser.getAttributeValue(null, "version") != null;

            String openStream = "<?xml version='1.0'?><stream:stream xmlns='jabber:server' " +
                    "xmlns:stream='http://etherx.jabber.org/streams' xmlns:db='jabber:server:dialback' from='" +
                    S2SComponent.HOSTNAME + "' id='" + streamID + "' version='1.0'>";
            if (xmppversionnew) {
                openStream += "<stream:features></stream:features>";
            }
            sendStanza(openStream);

            while (parser.next() != XmlPullParser.END_DOCUMENT) {
                updateTsRemoteData();
                if (parser.getEventType() != XmlPullParser.START_TAG) {
                    continue;
                }
                logParser();

                packetsRemote++;

                String tag = parser.getName();
                if (tag.equals("db:result")) {
                    String dfrom = parser.getAttributeValue(null, "from");
                    String to = parser.getAttributeValue(null, "to");
                    LOGGER.info("STREAM FROM " + dfrom + " TO " + to + " " + streamID + " ASKING FOR DIALBACK");
                    if (dfrom.endsWith(S2SComponent.HOSTNAME) && (dfrom.equals(S2SComponent.HOSTNAME) || dfrom.endsWith("." + S2SComponent.HOSTNAME))) {
                        break;
                    }
                    if (dfrom != null && to != null && to.equals(S2SComponent.HOSTNAME)) {
                        String dbKey = XmlUtils.getTagText(parser);
                        updateTsRemoteData();

                        ConnectionOut c = S2SComponent.getConnectionOut(dfrom, false);
                        if (c != null) {
                            c.sendDialbackVerify(streamID, dbKey);
                        } else {
                            c = new ConnectionOut(dfrom, streamID, dbKey);
                            c.parseStream();
                        }
                    } else {
                        throw new Exception("STREAM FROM " + dfrom + " " + streamID + " DIALBACK RESULT FAIL");
                    }
                } else if (tag.equals("db:verify")) {
                    String vfrom = parser.getAttributeValue(null, "from");
                    String vto = parser.getAttributeValue(null, "to");
                    String vid = parser.getAttributeValue(null, "id");
                    String vkey = XmlUtils.getTagText(parser);
                    updateTsRemoteData();
                    boolean valid = false;
                    if (vfrom != null && vto != null && vid != null && vkey != null) {
                        String vkey2 = generateDialbackKey(vfrom, vto, vid);
                        valid = vkey.equals(vkey2);
                    }
                    if (valid) {
                        sendStanza("<db:verify from='" + vto + "' to='" + vfrom + "' id='" + vid + "' type='valid'/>");
                        LOGGER.info("STREAM FROM " + vfrom + " " + streamID + " DIALBACK VERIFY VALID");
                    } else {
                        sendStanza("<db:verify from='" + vto + "' to='" + vfrom + "' id='" + vid + "' type='invalid'/>");
                        LOGGER.warning("STREAM FROM " + vfrom + " " + streamID + " DIALBACK VERIFY INVALID");
                    }
                } else if (tag.equals("presence") && checkFromTo(parser)) {
                    Presence p = Presence.parse(parser, null);
                    if (p != null && (p.type == null || !p.type.equals(Presence.Type.error))) {
                        bot.incomingPresence(p);
                    }
                } else if (tag.equals("message") && checkFromTo(parser)) {
                    updateTsRemoteData();
                    Message msg = Message.parse(parser, S2SComponent.childParsers);
                    if (msg != null && (msg.type == null || !msg.type.equals(Message.Type.error))) {
                        LOGGER.info("STREAM " + streamID + ": " + msg.toString());
                        if (!bot.incomingMessage(msg)) {
                            xmpp.send(msg);
                        }
                    }
                } else if (tag.equals("iq") && checkFromTo(parser)) {
                    updateTsRemoteData();
                    String type = parser.getAttributeValue(null, "type");
                    String xml = XmlUtils.parseToString(parser, true);
                    if (type == null || !type.equals(Iq.Type.error)) {
                        LOGGER.info("STREAM " + streamID + ": " + xml);
                        xmpp.send(xml);
                    }
                } else {
                    LOGGER.info("STREAM " + streamID + ": " + XmlUtils.parseToString(parser, true));
                }
            }
            LOGGER.warning("STREAM " + streamID + " FINISHED");
            S2SComponent.removeConnectionIn(this);
            closeConnection();
        } catch (EOFException ex) {
            LOGGER.info(String.format("STREAM %s CLOSED (dirty)", streamID));
            S2SComponent.removeConnectionIn(this);
            closeConnection();
        } catch (Exception e) {
            LOGGER.log(Level.WARNING, "STREAM " + streamID + " ERROR", e);
            S2SComponent.removeConnectionIn(this);
            closeConnection();
        }
    }

    void updateTsRemoteData() {
        tsRemoteData = System.currentTimeMillis();
    }

    public void sendDialbackResult(String sfrom, String type) {
        try {
            sendStanza("<db:result from='" + S2SComponent.HOSTNAME + "' to='" + sfrom + "' type='" + type + "'/>");
            if (type.equals("valid")) {
                from.add(sfrom);
                LOGGER.info("STREAM FROM " + sfrom + " " + streamID + " READY");
            }
        } catch (IOException e) {
            LOGGER.warning("STREAM FROM " + sfrom + " " + streamID + " ERROR: " + e.toString());
        }
    }

    boolean checkFromTo(XmlPullParser parser) throws Exception {
        String cfrom = parser.getAttributeValue(null, "from");
        String cto = parser.getAttributeValue(null, "to");
        if (cfrom != null && cto != null && !cfrom.isEmpty() && !cto.isEmpty()) {
            JID jidto = new JID(cto);
            if (jidto.Host != null && jidto.Username != null && jidto.Host.equals(S2SComponent.HOSTNAME) && jidto.Username.matches("^[a-zA-Z0-9\\-]{2,16}$")) {
                JID jidfrom = new JID(cfrom);
                int size = from.size();
                for (int i = 0; i < size; i++) {
                    if (from.get(i).equals(jidfrom.Host)) {
                        return true;
                    }
                }
            }
        }
        return false;
    }
}