package de.pixart.messenger.utils;

import android.net.Uri;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

import de.pixart.messenger.xmpp.jid.InvalidJidException;
import de.pixart.messenger.xmpp.jid.Jid;

public class XmppUri {

    protected String jid;
    protected boolean muc;
    protected String fingerprint;
    protected List<Fingerprint> fingerprints = new ArrayList<>();
    private String body;
    protected boolean safeSource = true;
    public static final String OMEMO_URI_PARAM = "omemo-sid-";
    public static final String OTR_URI_PARAM = "otr-fingerprint";

    public XmppUri(String uri) {
        try {
            parse(Uri.parse(uri));
        } catch (IllegalArgumentException e) {
            try {
                jid = Jid.fromString(uri).toBareJid().toString();
            } catch (InvalidJidException e2) {
                jid = null;
            }
        }
    }

    public XmppUri(Uri uri) {
        parse(uri);
    }

    public static boolean isXmppUri(String uri) {
        String scheme = Uri.parse(uri).getScheme();
        return "xmpp".equalsIgnoreCase(scheme);
    }

    public XmppUri(Uri uri, boolean safeSource) {
        this.safeSource = safeSource;
        parse(uri);
    }

    public boolean isSafeSource() {
        return safeSource;
    }

    protected void parse(Uri uri) {
        String scheme = uri.getScheme();
        String host = uri.getHost();
        List<String> segments = uri.getPathSegments();
        if ("https".equalsIgnoreCase(scheme) && "jabber.pix-art.de".equalsIgnoreCase(host)) {
            if (segments.size() >= 2 && segments.get(1).contains("@")) {
                // sample : https://conversations.im/i/foo@bar.com
                try {
                    jid = Jid.fromString(segments.get(1)).toString();
                } catch (Exception e) {
                    jid = null;
                }
            } else if (segments.size() >= 3) {
                // sample : https://conversations.im/i/foo/bar.com
                jid = segments.get(1) + "@" + segments.get(2);
            }
            muc = segments.size() > 1 && "j".equalsIgnoreCase(segments.get(0));
            fingerprints = parseFingerprints(uri.getQuery(),'&');
        } else if ("xmpp".equalsIgnoreCase(scheme)) {
            // sample: xmpp:foo@bar.com
            muc = isMuc(uri.getQuery());
            if (uri.getAuthority() != null) {
                jid = uri.getAuthority();
            } else {
                jid = uri.getSchemeSpecificPart().split("\\?")[0];
            }
            this.fingerprints = parseFingerprints(uri.getQuery());
            this.body = parseBody(uri.getQuery());
        } else if ("imto".equalsIgnoreCase(scheme)) {
            // sample: imto://xmpp/foo@bar.com
            try {
                jid = URLDecoder.decode(uri.getEncodedPath(), "UTF-8").split("/")[1].trim();
            } catch (final UnsupportedEncodingException ignored) {
                jid = null;
            }
        } else {
            try {
                jid = Jid.fromString(uri.toString()).toBareJid().toString();
            } catch (final InvalidJidException ignored) {
                jid = null;
            }
        }
    }

    protected List<Fingerprint> parseFingerprints(String query) {
        return parseFingerprints(query, ';');
    }

    protected List<Fingerprint> parseFingerprints(String query, char seperator) {
        List<Fingerprint> fingerprints = new ArrayList<>();
        String[] pairs = query == null ? new String[0] : query.split(String.valueOf(seperator));
        for (String pair : pairs) {
            String[] parts = pair.split("=", 2);
            if (parts.length == 2) {
                String key = parts[0].toLowerCase(Locale.US);
                String value = parts[1].toLowerCase(Locale.US);
                if (OTR_URI_PARAM.equals(key)) {
                    fingerprints.add(new Fingerprint(FingerprintType.OTR, value));
                }
                if (key.startsWith(OMEMO_URI_PARAM)) {
                    try {
                        int id = Integer.parseInt(key.substring(OMEMO_URI_PARAM.length()));
                        fingerprints.add(new Fingerprint(FingerprintType.OMEMO, value, id));
                    } catch (Exception e) {
                        //ignoring invalid device id
                    }
                }
            }
        }
        return fingerprints;
    }

    protected String parseBody(String query) {
        for(String pair : query == null ? new String[0] : query.split(";")) {
            final String[] parts = pair.split("=",2);
            if (parts.length == 2 && "body".equals(parts[0].toLowerCase(Locale.US))) {
                try {
                    return URLDecoder.decode(parts[1],"UTF-8");
                } catch (UnsupportedEncodingException e) {
                    return null;
                }
            }
        }
        return null;
    }

    protected boolean isMuc(String query) {
        for(String pair : query == null ? new String[0] : query.split(";")) {
            final String[] parts = pair.split("=",2);
            if (parts.length == 1 && "join".equals(parts[0])) {
                return true;
            }
        }
        return false;
    }

    public Jid getJid() {
        try {
            return this.jid == null ? null : Jid.fromString(this.jid.toLowerCase());
        } catch (InvalidJidException e) {
            return null;
        }
    }

    public String getBody() {
        return body;
    }

    public List<Fingerprint> getFingerprints() {
        return this.fingerprints;
    }

    public boolean hasFingerprints() {
        return fingerprints.size() > 0;
    }

    public enum FingerprintType {
        OMEMO,
        OTR
    }

    public static String getFingerprintUri(String base, List<XmppUri.Fingerprint> fingerprints, char seperator) {
        StringBuilder builder = new StringBuilder(base);
        builder.append('?');
        for(int i = 0; i < fingerprints.size(); ++i) {
            XmppUri.FingerprintType type = fingerprints.get(i).type;
            if (type == XmppUri.FingerprintType.OMEMO) {
                builder.append(XmppUri.OMEMO_URI_PARAM);
                builder.append(fingerprints.get(i).deviceId);
            } else if (type == XmppUri.FingerprintType.OTR) {
                builder.append(XmppUri.OTR_URI_PARAM);
            }
            builder.append('=');
            builder.append(fingerprints.get(i).fingerprint);
            if (i != fingerprints.size() -1) {
                builder.append(seperator);
            }
        }
        return builder.toString();
    }

    public static class Fingerprint {
        public final FingerprintType type;
        public final String fingerprint;
        public final int deviceId;

        public Fingerprint(FingerprintType type, String fingerprint) {
            this(type, fingerprint, 0);
        }

        public Fingerprint(FingerprintType type, String fingerprint, int deviceId) {
            this.type = type;
            this.fingerprint = fingerprint;
            this.deviceId = deviceId;
        }

        @Override
        public String toString() {
            return type.toString() + ": " + fingerprint + (deviceId != 0 ? " " + String.valueOf(deviceId) : "");
        }
    }
}