package com.juick.ws;

import com.juick.User;
import com.juick.server.MessagesQueries;
import com.juick.server.UserQueries;
import org.apache.http.NameValuePair;
import org.apache.http.client.utils.URLEncodedUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.util.Assert;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

import java.io.IOException;
import java.net.URI;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Created by vitalyster on 28.06.2016.
 */
public class WebsocketComponent extends TextWebSocketHandler {
    private static final Logger logger = LoggerFactory.getLogger(WebsocketComponent.class);

    private JdbcTemplate jdbc;

    private final List<SocketSubscribed> clients = Collections.synchronizedList(new ArrayList<SocketSubscribed>());


    public WebsocketComponent(JdbcTemplate jdbc) {
        Assert.notNull(jdbc);
        this.jdbc = jdbc;
    }

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        URI hLocation;
        String hXRealIP = "";

        hLocation = session.getUri();
        HttpHeaders headers = session.getHandshakeHeaders();
        hXRealIP = headers.getOrDefault("X-Real-IP",
                Collections.singletonList(session.getRemoteAddress().toString())).get(0);

        // Auth
        User visitor = new User();
        List<NameValuePair> params = URLEncodedUtils.parse(hLocation, "UTF-8");
        for (NameValuePair param : params) {
            if (param.getName().equals("hash")) {
                String hash = param.getValue();
                if (hash.length() == 16) {
                    visitor = UserQueries.getUserByHash(jdbc, hash);
                } else {
                    try {
                        logger.info(String.format("wrong hash for %d from %s", visitor.getUid(), hXRealIP));
                        session.close(new CloseStatus(403, "Forbidden"));
                    } catch (IOException e) {
                        logger.warn("ws error", e);
                    }
                }
                break;
            }
        }
        logger.info(String.format("user %d connected to %s from %s", visitor.getUid(), hLocation.getPath(), hXRealIP));

        int MID = 0;
        SocketSubscribed sockSubscr = null;
        if (hLocation.getPath().equals("/")) {
            logger.info(String.format("user %d connected", visitor.getUid()));
            sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, false);
        } else if (hLocation.getPath().equals("/_all")) {
            logger.info(String.format("user %d connected to legacy _all (%s)", visitor.getUid(), hLocation.getPath()));
            sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true);
            sockSubscr.allMessages = true;
        } else if (hLocation.getPath().equals("/_replies")) {
            logger.info(String.format("user %d connected to legacy _replies (%s)", visitor.getUid(), hLocation.getPath()));
            sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true);
            sockSubscr.allReplies = true;
        } else if (hLocation.getPath().matches("/\\d+$")) {
            try {
                MID = Integer.parseInt(hLocation.getPath().substring(1));
            } catch (Exception e) {
            }
            if (MID > 0) {
                if (MessagesQueries.canViewThread(jdbc, MID, visitor.getUid())) {
                    logger.info(String.format("user %d connected to legacy thread (%d) from %s", visitor.getUid(), MID, hXRealIP));
                    sockSubscr = new SocketSubscribed(session, hXRealIP, visitor, true);
                    sockSubscr.MID = MID;
                } else {
                    try {
                        session.close(new CloseStatus(403, "Forbidden"));
                    } catch (IOException e) {
                        logger.warn("ws error", e);
                    }
                }
            }
        }
        if (sockSubscr != null) {
            synchronized (clients) {
                clients.add(sockSubscr);
                logger.info(clients.size() + " clients connected");
            }
        }
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        synchronized (clients) {
            clients.removeIf(c -> {
                if (c.session.getId().equals(session.getId())) {
                    logger.info(String.format("session from %s closed with status %d", c.clientName, status.getCode()));
                    return true;
                }
                return false;
            });
            logger.info(clients.size() + " clients connected");
        }

    }

    public List<SocketSubscribed> getClients() {
        return clients;
    }

    class SocketSubscribed {
        WebSocketSession session;
        String clientName;
        User visitor;
        int MID;
        boolean allMessages;
        boolean allReplies;
        long tsConnected;
        long tsLastData;
        boolean legacy;

        public SocketSubscribed(WebSocketSession session, String clientName, User visitor, boolean legacy) {
            this.session = session;
            this.clientName = clientName;
            this.visitor = visitor;
            tsConnected = tsLastData = System.currentTimeMillis();
            this.legacy = legacy;
        }
    }
}