package de.pixart.messenger.http;

import org.apache.http.conn.ssl.StrictHostnameVerifier;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.X509TrustManager;

import de.pixart.messenger.entities.Message;
import de.pixart.messenger.services.AbstractConnectionManager;
import de.pixart.messenger.services.XmppConnectionService;
import de.pixart.messenger.utils.TLSSocketFactory;

public class HttpConnectionManager extends AbstractConnectionManager {

    public HttpConnectionManager(XmppConnectionService service) {
        super(service);
    }

    private List<HttpDownloadConnection> downloadConnections = new CopyOnWriteArrayList<>();
    private List<HttpUploadConnection> uploadConnections = new CopyOnWriteArrayList<>();

    public HttpDownloadConnection createNewDownloadConnection(Message message) {
        return this.createNewDownloadConnection(message, false);
    }

    public HttpDownloadConnection createNewDownloadConnection(Message message, boolean interactive) {
        HttpDownloadConnection connection = new HttpDownloadConnection(this);
        connection.init(message, interactive);
        this.downloadConnections.add(connection);
        return connection;
    }

    public HttpUploadConnection createNewUploadConnection(Message message, boolean delay) {
        HttpUploadConnection connection = new HttpUploadConnection(this);
        connection.init(message, delay);
        this.uploadConnections.add(connection);
        return connection;
    }

    public void finishConnection(HttpDownloadConnection connection) {
        this.downloadConnections.remove(connection);
    }

    public void finishUploadConnection(HttpUploadConnection httpUploadConnection) {
        this.uploadConnections.remove(httpUploadConnection);
    }

    public void setupTrustManager(final HttpsURLConnection connection, final boolean interactive) {
        final X509TrustManager trustManager;
        final HostnameVerifier hostnameVerifier;
        if (interactive) {
            trustManager = mXmppConnectionService.getMemorizingTrustManager().getInteractive();
            hostnameVerifier = mXmppConnectionService
                    .getMemorizingTrustManager().wrapHostnameVerifier(
                            new StrictHostnameVerifier());
        } else {
            trustManager = mXmppConnectionService.getMemorizingTrustManager()
                    .getNonInteractive();
            hostnameVerifier = mXmppConnectionService
                    .getMemorizingTrustManager()
                    .wrapHostnameVerifierNonInteractive(
                            new StrictHostnameVerifier());
        }
        try {
            final SSLSocketFactory sf = new TLSSocketFactory(new X509TrustManager[]{trustManager}, mXmppConnectionService.getRNG());
            connection.setSSLSocketFactory(sf);
            connection.setHostnameVerifier(hostnameVerifier);
        } catch (final KeyManagementException | NoSuchAlgorithmException ignored) {
        }
    }

    public Proxy getProxy() throws IOException {
        return new Proxy(Proxy.Type.HTTP, new InetSocketAddress(InetAddress.getByAddress(new byte[]{127, 0, 0, 1}), 8118));
    }
}