package de.pixart.messenger.http;

import android.os.PowerManager;
import android.util.Log;
import android.util.Pair;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;

import javax.net.ssl.HttpsURLConnection;

import de.pixart.messenger.Config;
import de.pixart.messenger.entities.Account;
import de.pixart.messenger.entities.DownloadableFile;
import de.pixart.messenger.entities.Message;
import de.pixart.messenger.entities.Transferable;
import de.pixart.messenger.parser.IqParser;
import de.pixart.messenger.persistance.FileBackend;
import de.pixart.messenger.services.AbstractConnectionManager;
import de.pixart.messenger.services.XmppConnectionService;
import de.pixart.messenger.utils.CryptoHelper;
import de.pixart.messenger.utils.Namespace;
import de.pixart.messenger.xml.Element;
import de.pixart.messenger.xmpp.OnIqPacketReceived;
import de.pixart.messenger.xmpp.jid.Jid;
import de.pixart.messenger.xmpp.stanzas.IqPacket;

public class HttpUploadConnection implements Transferable {

    private HttpConnectionManager mHttpConnectionManager;
    private XmppConnectionService mXmppConnectionService;

    private boolean canceled = false;
    private boolean delayed = false;
    private Account account;
    private DownloadableFile file;
    private Message message;
    private String mime;
    private URL mGetUrl;
    private URL mPutUrl;
    private boolean mUseTor = false;

    private byte[] key = null;

    private long transmitted = 0;

    private InputStream mFileInputStream;

    public HttpUploadConnection(HttpConnectionManager httpConnectionManager) {
        this.mHttpConnectionManager = httpConnectionManager;
        this.mXmppConnectionService = httpConnectionManager.getXmppConnectionService();
        this.mUseTor = mXmppConnectionService.useTorToConnect();
    }

    @Override
    public boolean start() {
        return false;
    }

    @Override
    public int getStatus() {
        return STATUS_UPLOADING;
    }

    @Override
    public long getFileSize() {
        return file == null ? 0 : file.getExpectedSize();
    }

    @Override
    public int getProgress() {
        if (file == null) {
            return 0;
        }
        return (int) ((((double) transmitted) / file.getExpectedSize()) * 100);
    }

    @Override
    public void cancel() {
        this.canceled = true;
    }

    private void fail(String errorMessage) {
        mHttpConnectionManager.finishUploadConnection(this);
        message.setTransferable(null);
        mXmppConnectionService.markMessage(message, Message.STATUS_SEND_FAILED, errorMessage);
        FileBackend.close(mFileInputStream);
    }

    public void init(Message message, boolean delay) {
        this.message = message;
        this.account = message.getConversation().getAccount();
        this.file = mXmppConnectionService.getFileBackend().getFile(message, false);
        if (message.getEncryption() == Message.ENCRYPTION_PGP || message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
            this.mime = "application/pgp-encrypted";
        } else {
            this.mime = this.file.getMimeType();
        }
        this.delayed = delay;
        if (Config.ENCRYPT_ON_HTTP_UPLOADED
                || message.getEncryption() == Message.ENCRYPTION_AXOLOTL
                || message.getEncryption() == Message.ENCRYPTION_OTR) {
            this.key = new byte[48]; // todo: change this to 44 for 12-byte IV instead of 16-byte at some point in future
            mXmppConnectionService.getRNG().nextBytes(this.key);
            this.file.setKeyAndIv(this.key);
        }
        Pair<InputStream, Integer> pair;
        try {
            pair = AbstractConnectionManager.createInputStream(file, true);
        } catch (FileNotFoundException e) {
            Log.d(Config.LOGTAG, account.getJid().toBareJid() + ": could not find file to upload - " + e.getMessage());
            fail(e.getMessage());
            return;
        }
        this.file.setExpectedSize(pair.second);
        message.resetFileParams();
        this.mFileInputStream = pair.first;
        Jid host = account.getXmppConnection().findDiscoItemByFeature(Namespace.HTTP_UPLOAD);
        IqPacket request = mXmppConnectionService.getIqGenerator().requestHttpUploadSlot(host, file, mime);
        mXmppConnectionService.sendIqPacket(account, request, new OnIqPacketReceived() {
            @Override
            public void onIqPacketReceived(Account account, IqPacket packet) {
                if (packet.getType() == IqPacket.TYPE.RESULT) {
                    Element slot = packet.findChild("slot", Namespace.HTTP_UPLOAD);
                    if (slot != null) {
                        try {
                            mGetUrl = new URL(slot.findChildContent("get"));
                            mPutUrl = new URL(slot.findChildContent("put"));
                            if (!canceled) {
                                new Thread(new FileUploader()).start();
                            }
                            return;
                        } catch (MalformedURLException e) {
                            //fall through
                        }
                    }
                }
                Log.d(Config.LOGTAG, account.getJid().toString() + ": invalid response to slot request " + packet);
                fail(IqParser.extractErrorMessage(packet));
            }
        });
        message.setTransferable(this);
        mXmppConnectionService.markMessage(message, Message.STATUS_UNSEND);
    }

    private class FileUploader implements Runnable {

        @Override
        public void run() {
            this.upload();
        }

        private void upload() {
            OutputStream os = null;
            HttpURLConnection connection = null;
            PowerManager.WakeLock wakeLock = mHttpConnectionManager.createWakeLock("http_upload_" + message.getUuid());
            try {
                wakeLock.acquire();
                final int expectedFileSize = (int) file.getExpectedSize();
                final int readTimeout = (expectedFileSize / 2048) + Config.SOCKET_TIMEOUT * 1000; //assuming a minimum transfer speed of 16kbit/s
                Log.d(Config.LOGTAG, "uploading to " + mPutUrl.toString() + " w/ read timeout of " + readTimeout + "s");
                if (mUseTor) {
                    connection = (HttpURLConnection) mPutUrl.openConnection(mHttpConnectionManager.getProxy());
                } else {
                    connection = (HttpURLConnection) mPutUrl.openConnection();
                }
                if (connection instanceof HttpsURLConnection) {
                    mHttpConnectionManager.setupTrustManager((HttpsURLConnection) connection, true);
                }
                connection.setRequestMethod("PUT");
                connection.setFixedLengthStreamingMode(expectedFileSize);
                connection.setRequestProperty("Content-Type", mime == null ? "application/octet-stream" : mime);
                connection.setRequestProperty("User-Agent", mXmppConnectionService.getIqGenerator().getIdentityName());
                connection.setDoOutput(true);
                connection.setConnectTimeout(Config.SOCKET_TIMEOUT * 1000);
                connection.setReadTimeout(readTimeout * 1000);
                connection.connect();
                os = connection.getOutputStream();
                transmitted = 0;
                int count;
                byte[] buffer = new byte[4096];
                while (((count = mFileInputStream.read(buffer)) != -1) && !canceled) {
                    transmitted += count;
                    os.write(buffer, 0, count);
                    mHttpConnectionManager.updateConversationUi(false);
                }
                os.flush();
                os.close();
                mFileInputStream.close();
                int code = connection.getResponseCode();
                if (code == 200 || code == 201) {
                    Log.d(Config.LOGTAG, "finished uploading file");
                    if (key != null) {
                        mGetUrl = CryptoHelper.toAesGcmUrl(new URL(mGetUrl.toString() + "#" + CryptoHelper.bytesToHex(key)));
                    }
                    mXmppConnectionService.getFileBackend().updateFileParams(message, mGetUrl);
                    mXmppConnectionService.getFileBackend().updateMediaScanner(file);
                    message.setTransferable(null);
                    message.setCounterpart(message.getConversation().getJid().toBareJid());
                    mXmppConnectionService.resendMessage(message, delayed);
                } else {
                    Log.d(Config.LOGTAG, "http upload failed because response code was " + code);
                    fail("http upload failed because response code was " + code);
                }
            } catch (IOException e) {
                e.printStackTrace();
                Log.d(Config.LOGTAG, "http upload failed " + e.getMessage());
                fail(e.getMessage());
            } finally {
                FileBackend.close(mFileInputStream);
                FileBackend.close(os);
                if (connection != null) {
                    connection.disconnect();
                }
                wakeLock.release();
            }
        }
    }
}