/*
 * Copyright (C) 2017 Schürmann & Breitmoser GbR
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.sufficientlysecure.keychain.util;


import java.net.Proxy;
import java.util.ArrayList;
import java.util.ListIterator;
import java.util.UUID;

import android.annotation.SuppressLint;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.SharedPreferences.Editor;
import android.os.Parcelable;
import android.preference.PreferenceManager;
import android.support.annotation.Nullable;

import com.google.auto.value.AutoValue;
import org.sufficientlysecure.keychain.Constants;
import org.sufficientlysecure.keychain.Constants.Pref;
import org.sufficientlysecure.keychain.keyimport.HkpKeyserverAddress;
import timber.log.Timber;


/**
 * Singleton Implementation of a Preference Helper
 */
@SuppressLint("CommitPrefEdits")
public class Preferences {
    private static Preferences sPreferences;
    private SharedPreferences mSharedPreferences;

    private static String PREF_FILE_NAME = "APG.main";
    private static int PREF_FILE_MODE = Context.MODE_MULTI_PROCESS;

    public static synchronized Preferences getPreferences(Context context) {
        return getPreferences(context, false);
    }

    public static synchronized Preferences getPreferences(Context context, boolean forceNew) {
        if (sPreferences == null || forceNew) {
            sPreferences = new Preferences(context);
        } else {
            // to make it safe for multiple processes, call getSharedPreferences everytime
            sPreferences.updateSharedPreferences(context);
        }
        return sPreferences;
    }

    private Preferences(Context context) {
        updateSharedPreferences(context);
    }

    /**
     * Makes android's preference framework write to our file instead of default.
     * This allows us to use the xml "persistent" attribute to simplify code, which automatically
     * writes and reads preference values.
     */
    public static void setPreferenceManagerFileAndMode(PreferenceManager manager) {
        manager.setSharedPreferencesName(PREF_FILE_NAME);
        manager.setSharedPreferencesMode(PREF_FILE_MODE);
    }

    public void updateSharedPreferences(Context context) {
        // multi-process safe preferences
        mSharedPreferences = context.getSharedPreferences(PREF_FILE_NAME, PREF_FILE_MODE);
    }

    public SharedPreferences getSharedPreferences() {
        return mSharedPreferences;
    }

    public String getLanguage() {
        return mSharedPreferences.getString(Constants.Pref.LANGUAGE, "");
    }

    public void setLanguage(String value) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putString(Constants.Pref.LANGUAGE, value);
        editor.commit();
    }

    public boolean getPassphraseCacheSubs() {
        return mSharedPreferences.getBoolean(Pref.PASSPHRASE_CACHE_SUBS, false);
    }

    public int getCacheTtlSeconds() {
        return mSharedPreferences.getInt(Pref.PASSPHRASE_CACHE_LAST_TTL, Integer.MAX_VALUE);
    }

    public void setCacheTtlSeconds(int ttlSeconds) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putInt(Pref.PASSPHRASE_CACHE_LAST_TTL, ttlSeconds);
        editor.commit();
    }

    public boolean getCachedConsolidate() {
        return mSharedPreferences.getBoolean(Pref.CACHED_CONSOLIDATE, false);
    }

    public void setCachedConsolidate(boolean value) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.CACHED_CONSOLIDATE, value);
        editor.commit();
    }

    public boolean isFirstTime() {
        return mSharedPreferences.getBoolean(Constants.Pref.FIRST_TIME_WIZARD, true);
    }

    public void setFirstTime(boolean value) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Constants.Pref.FIRST_TIME_WIZARD, value);
        editor.commit();
    }

    public boolean isAppExecutedFirstTime() {
        return mSharedPreferences.getBoolean(Pref.FIRST_TIME_APP, true);
    }

    public void setAppExecutedFirstTime(boolean value) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Constants.Pref.FIRST_TIME_APP, value);
        editor.commit();
    }

    public ArrayList<HkpKeyserverAddress> getKeyServers() {
        String rawData = mSharedPreferences.getString(Constants.Pref.KEY_SERVERS,
                Constants.Defaults.KEY_SERVERS);
        if ("".equals(rawData)) {
            return new ArrayList<>();
        }
        ArrayList<HkpKeyserverAddress> servers = new ArrayList<>();
        String[] entries = rawData.split(",");
        for (String entry : entries) {
            String[] addresses = entry.trim().split(";");
            String url = addresses[0];
            String onion = addresses.length == 1 ? null : addresses[1];

            if (url.isEmpty()) {
                continue;
            }

            servers.add(HkpKeyserverAddress.createWithOnionProxy(url, onion));
        }
        return servers;
    }

    public HkpKeyserverAddress getPreferredKeyserver() {
        ArrayList<HkpKeyserverAddress> keyservers = getKeyServers();
        return keyservers.size() == 0 ? null : keyservers.get(0);
    }

    public void setKeyServers(ArrayList<HkpKeyserverAddress> keyservers) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        String rawData = "";
        for (HkpKeyserverAddress server : keyservers) {
            if (server.getUrl().isEmpty()) {
                continue;
            }
            rawData += server.getUrl();

            if (server.getOnion() != null && !server.getOnion().isEmpty()) {
                rawData += ";" + server.getOnion();
            }

            if (!"".equals(rawData)) {
                rawData += ",";
            }
        }
        editor.putString(Constants.Pref.KEY_SERVERS, rawData);
        editor.commit();
    }

    public void setFilesUseCompression(boolean compress) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.FILE_USE_COMPRESSION, compress);
        editor.commit();
    }

    public boolean getFilesUseCompression() {
        return mSharedPreferences.getBoolean(Pref.FILE_USE_COMPRESSION, true);
    }

    public void setFilesEncryptToSelf(boolean selfEncrypt) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.FILE_SELF_ENCRYPT, selfEncrypt);
        editor.commit();
    }

    public boolean getFilesEncryptToSelf() {
        return mSharedPreferences.getBoolean(Pref.FILE_SELF_ENCRYPT, true);
    }

    public void setTextUseCompression(boolean compress) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.TEXT_USE_COMPRESSION, compress);
        editor.commit();
    }

    public boolean getTextUseCompression() {
        return mSharedPreferences.getBoolean(Pref.TEXT_USE_COMPRESSION, true);
    }

    public void setTextSelfEncrypt(boolean selfEncrypt) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.TEXT_SELF_ENCRYPT, selfEncrypt);
        editor.commit();
    }

    public boolean getTextSelfEncrypt() {
        return mSharedPreferences.getBoolean(Pref.TEXT_SELF_ENCRYPT, true);
    }

    public String getTheme() {
        return mSharedPreferences.getString(Pref.THEME, Pref.Theme.LIGHT);
    }

    public void setTheme(String value) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putString(Constants.Pref.THEME, value);
        editor.commit();
    }

    public void setUseArmor(boolean useArmor) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.USE_ARMOR, useArmor);
        editor.commit();
    }

    public boolean useNumKeypadForSecurityTokenPin() {
        return mSharedPreferences.getBoolean(Pref.USE_NUMKEYPAD_FOR_SECURITY_TOKEN_PIN, true);
    }

    public void setUseNumKeypadForSecurityTokenPin(boolean useNumKeypad) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.USE_NUMKEYPAD_FOR_SECURITY_TOKEN_PIN, useNumKeypad);
        editor.commit();
    }

    public boolean getUseArmor() {
        return mSharedPreferences.getBoolean(Pref.USE_ARMOR, false);
    }

    public void setEncryptFilenames(boolean encryptFilenames) {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.ENCRYPT_FILENAMES, encryptFilenames);
        editor.commit();
    }

    public boolean getEncryptFilenames() {
        return mSharedPreferences.getBoolean(Pref.ENCRYPT_FILENAMES, true);
    }

    // proxy preference functions start here

    public boolean getUseNormalProxy() {
        return mSharedPreferences.getBoolean(Constants.Pref.USE_NORMAL_PROXY, false);
    }

    public boolean getUseTorProxy() {
        return mSharedPreferences.getBoolean(Constants.Pref.USE_TOR_PROXY, false);
    }

    public String getProxyHost() {
        return mSharedPreferences.getString(Constants.Pref.PROXY_HOST, null);
    }

    /**
     * we store port as String for easy interfacing with EditTextPreference, but return it as an integer
     *
     * @return port number of proxy
     */
    public int getProxyPort() {
        return Integer.parseInt(mSharedPreferences.getString(Pref.PROXY_PORT, "-1"));
    }

    Proxy.Type getProxyType() {
        final String typeHttp = Pref.ProxyType.TYPE_HTTP;
        final String typeSocks = Pref.ProxyType.TYPE_SOCKS;

        String type = mSharedPreferences.getString(Pref.PROXY_TYPE, typeHttp);

        switch (type) {
            case typeHttp:
                return Proxy.Type.HTTP;
            case typeSocks:
                return Proxy.Type.SOCKS;
            default:  // shouldn't happen
                Timber.e("Invalid Proxy Type in preferences");
                return null;
        }
    }

    public ParcelableProxy getParcelableProxy() {
        boolean useTor = getUseTorProxy();
        boolean useNormalProxy = getUseNormalProxy();

        if (useTor) {
            //TODO: Replace Constants.Orbot.PROXY_PORT with OrbotStatusReceiver.getProxyPortHttp()
            //TODO: in order to always have the actual port Orbot is offering?

            return new ParcelableProxy(Constants.Orbot.PROXY_HOST, Constants.Orbot.PROXY_PORT,
                    Constants.Orbot.PROXY_TYPE, ParcelableProxy.PROXY_MODE_TOR);
        } else if (useNormalProxy) {
            return new ParcelableProxy(getProxyHost(), getProxyPort(), getProxyType(),
                    ParcelableProxy.PROXY_MODE_NORMAL);
        } else {
            return new ParcelableProxy(null, -1, null, ParcelableProxy.PROXY_MODE_NORMAL);
        }
    }

    // cloud prefs

    public CloudSearchPrefs getCloudSearchPrefs() {
        return CloudSearchPrefs.create(mSharedPreferences.getBoolean(Pref.SEARCH_KEYSERVER, true),
                mSharedPreferences.getBoolean(Pref.SEARCH_KEYBASE, true),
                false,
                mSharedPreferences.getBoolean(Pref.SEARCH_WEB_KEY_DIRECTORY, true),
                getPreferredKeyserver());
    }

    public boolean isKeySignaturesTableInitialized() {
        return mSharedPreferences.getBoolean(Pref.KEY_SIGNATURES_TABLE_INITIALIZED, false);
    }

    public void setKeySignaturesTableInitialized() {
        Editor editor = mSharedPreferences.edit();
        editor.putBoolean(Pref.KEY_SIGNATURES_TABLE_INITIALIZED, true);
        editor.commit();
    }

    public boolean isKeyserverSyncEnabled() {
        return mSharedPreferences.getBoolean(Pref.SYNC_KEYSERVER, true);
    }

    public UUID getKeyserverSyncWorkUuid() {
        String uuidString = mSharedPreferences.getString(Pref.SYNC_WORK_UUID, null);
        return uuidString != null ? UUID.fromString(uuidString) : null;
    }

    public void setKeyserverSyncScheduled(UUID uuid) {
        String value = uuid != null ? uuid.toString() : null;
        mSharedPreferences.edit().putString(Pref.SYNC_WORK_UUID, value).apply();
    }

    public boolean isAnalyticsAskedPolitely() {
        return mSharedPreferences.getBoolean(Pref.KEY_ANALYTICS_ASKED_POLITELY, false);
    }

    public void setAnalyticsAskedPolitely() {
        mSharedPreferences.edit().putBoolean(Pref.KEY_ANALYTICS_ASKED_POLITELY, true).apply();
    }

    public boolean isAnalyticsHasConsent() {
        return mSharedPreferences.getBoolean(Pref.KEY_ANALYTICS_CONSENT, false);
    }

    public void setAnalyticsGotUserConsent(boolean hasUserConsent) {
        mSharedPreferences.edit().putBoolean(Pref.KEY_ANALYTICS_CONSENT, hasUserConsent).apply();
    }

    public void setAnalyticsLastAskedNow() {
        mSharedPreferences.edit().putLong(Pref.KEY_ANALYTICS_LAST_ASKED, System.currentTimeMillis()).apply();
    }

    public long getAnalyticsLastAsked() {
        return mSharedPreferences.getLong(Pref.KEY_ANALYTICS_LAST_ASKED, 0);
    }

    @AutoValue
    public static abstract class CloudSearchPrefs implements Parcelable {
        public abstract boolean isKeyserverEnabled();
        public abstract boolean isKeybaseEnabled();
        public abstract boolean isFacebookEnabled();
        public abstract boolean isWebKeyDirectoryEnabled();

        @Nullable
        public abstract HkpKeyserverAddress getKeyserver();

        /**
         * @param searchKeyserver       should passed keyserver be searched
         * @param searchKeybase         should keybase.io be searched
         * @param searchFacebook        should Facebook be searched
         * @param searchWebKeyDirectory should WKD be searched
         * @param keyserver             the keyserver url authority to search on
         */
        public static CloudSearchPrefs create(boolean searchKeyserver, boolean searchKeybase,
                                                       boolean searchFacebook, boolean searchWebKeyDirectory,
                                                       @Nullable HkpKeyserverAddress keyserver) {
            return new AutoValue_Preferences_CloudSearchPrefs(searchKeyserver,
                    searchKeybase,
                    searchFacebook,
                    searchWebKeyDirectory,
                    keyserver);
        }

        public static CloudSearchPrefs createWebKeyDirectoryOnly() {
            return create(false, false, false, true, null);
        }

        public static CloudSearchPrefs createKeyserverOnly(HkpKeyserverAddress keyserver) {
            return create(true, false, false, false, keyserver);
        }

        public static CloudSearchPrefs createSocialOnly() {
            return create(false, true, true, false, null);
        }
    }

    // sync preferences

    public boolean getWifiOnlySync() {
        return mSharedPreferences.getBoolean(Pref.ENABLE_WIFI_SYNC_ONLY, true);
    }

    // experimental prefs

    public boolean getExperimentalEnableKeybase() {
        return mSharedPreferences.getBoolean(Pref.EXPERIMENTAL_ENABLE_KEYBASE, false);
    }

    public boolean getExperimentalUsbAllowUntested() {
        return mSharedPreferences.getBoolean(Pref.EXPERIMENTAL_USB_ALLOW_UNTESTED, false);
    }

    public boolean getExperimentalSmartPGPAuthoritiesEnable() {
        return mSharedPreferences.getBoolean(Pref.EXPERIMENTAL_SMARTPGP_VERIFY_AUTHORITY, false);
    }

    public void setPrefVersionToCurrentVersion() {
        SharedPreferences.Editor editor = mSharedPreferences.edit();
        editor.putInt(Pref.PREF_VERSION, Constants.Defaults.PREF_CURRENT_VERSION);
        editor.commit();
    }

    public void upgradePreferences() {
        int oldVersion = mSharedPreferences.getInt(Constants.Pref.PREF_VERSION, 0);
        boolean requiresUpgrade = oldVersion < Constants.Defaults.PREF_CURRENT_VERSION;

        if (requiresUpgrade) {
            Timber.d("Upgrading preferences from %s to %s…", oldVersion, Constants.Defaults.PREF_CURRENT_VERSION);

            switch (oldVersion) {
                case 1:
                case 2:
                case 3: {
                    migrateToHkps();
                }
                case 4: {
                    setTheme(Constants.Pref.Theme.DEFAULT);
                }
                case 5:
                case 6:
                case 7: {
                    addOnionToSks();
                }
                case 8: {
                    replaceDefaultKeyserverWithUbuntu();
                }
            }

            // write new preference version
            mSharedPreferences.edit()
                    .putInt(Constants.Pref.PREF_VERSION, Constants.Defaults.PREF_CURRENT_VERSION)
                    .commit();
        }
    }

    private void migrateToHkps() {
        ArrayList<HkpKeyserverAddress> servers = getKeyServers();
        ListIterator<HkpKeyserverAddress> it = servers.listIterator();
        while (it.hasNext()) {
            HkpKeyserverAddress server = it.next();
            if (server == null) {
                continue;
            }
            switch (server.getUrl()) {
                case "pool.sks-keyservers.net": {
                    // use HKPS!
                    it.set(HkpKeyserverAddress.createFromUri("hkps://hkps.pool.sks-keyservers.net"));
                    break;
                }
                case "pgp.mit.edu": {
                    // use HKPS!
                    it.set(HkpKeyserverAddress.createFromUri("hkps://pgp.mit.edu"));
                    break;
                }
                case "subkeys.pgp.net": {
                    // remove, because often down and no HKPS!
                    it.remove();
                    break;
                }
            }

        }
        setKeyServers(servers);
    }

    private void addOnionToSks() {
        ArrayList<HkpKeyserverAddress> servers = getKeyServers();
        ListIterator<HkpKeyserverAddress> it = servers.listIterator();
        while (it.hasNext()) {
            HkpKeyserverAddress server = it.next();
            if (server == null) {
                continue;
            }
            if ("hkps://hkps.pool.sks-keyservers.net".equals(server.getUrl())) {
                it.set(HkpKeyserverAddress.createWithOnionProxy(
                        "hkps://hkps.pool.sks-keyservers.net",
                        "hkp://jirk5u4osbsr34t5.onion"));
            }

        }
        setKeyServers(servers);
    }

    private void replaceDefaultKeyserverWithUbuntu() {
        ArrayList<HkpKeyserverAddress> servers = getKeyServers();
        if (servers.isEmpty()) {
            return;
        }
        boolean oldDefaults = "hkps://hkps.pool.sks-keyservers.net".equalsIgnoreCase(servers.get(0).getUrl()) ||
                "hkps://pgp.mit.edu".equalsIgnoreCase(servers.get(0).getUrl());

        HkpKeyserverAddress ubuntuKeyserver = HkpKeyserverAddress.createFromUri("hkps://keyserver.ubuntu.com");
        if (oldDefaults) {
            servers.add(0, ubuntuKeyserver);
        } else if (!servers.contains(ubuntuKeyserver)) {
            servers.add(ubuntuKeyserver);
        }
        setKeyServers(servers);
    }

    public void clear() {
        mSharedPreferences.edit().clear().commit();
    }

}