/*
 * Copyright (C) 2014 Dominik Schürmann <dominik@dominikschuermann.de>
 *
 * 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 android.content.Context;
import android.content.Intent;
import android.os.Bundle;
import android.os.Messenger;

import org.sufficientlysecure.keychain.keyimport.HkpKeyserver;
import org.sufficientlysecure.keychain.keyimport.ImportKeysListEntry;
import org.sufficientlysecure.keychain.keyimport.Keyserver;
import org.sufficientlysecure.keychain.keyimport.ParcelableKeyRing;
import org.sufficientlysecure.keychain.service.KeychainIntentService;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Locale;
import java.util.Set;

public class EmailKeyHelper {

    public static void importContacts(Context context, Messenger messenger) {
        importAll(context, messenger, ContactHelper.getContactMails(context));
    }

    public static void importAll(Context context, Messenger messenger, List<String> mails) {
        // Collect all candidates as ImportKeysListEntry (set for deduplication)
        Set<ImportKeysListEntry> entries = new HashSet<ImportKeysListEntry>();
        for (String mail : mails) {
            entries.addAll(getEmailKeys(context, mail));
        }

        // Put them in a list and import
        ArrayList<ParcelableKeyRing> keys = new ArrayList<ParcelableKeyRing>(entries.size());
        for (ImportKeysListEntry entry : entries) {
            keys.add(new ParcelableKeyRing(entry.getFingerprintHex(), entry.getKeyIdHex(), null));
        }
        importKeys(context, messenger, keys);
    }

    public static Set<ImportKeysListEntry> getEmailKeys(Context context, String mail) {
        Set<ImportKeysListEntry> keys = new HashSet<ImportKeysListEntry>();

        // Try _hkp._tcp SRV record first
        String[] mailparts = mail.split("@");
        if (mailparts.length == 2) {
            HkpKeyserver hkp = HkpKeyserver.resolve(mailparts[1]);
            if (hkp != null) {
                keys.addAll(getEmailKeys(mail, hkp));
            }
        }

        if (keys.isEmpty()) {
            // Most users don't have the SRV record, so ask a default server as well
            String server = Preferences.getPreferences(context).getPreferredKeyserver();
            if (server != null) {
                HkpKeyserver hkp = new HkpKeyserver(server);
                keys.addAll(getEmailKeys(mail, hkp));
            }
        }
        return keys;
    }

    private static void importKeys(Context context, Messenger messenger, ArrayList<ParcelableKeyRing> keys) {
        Intent importIntent = new Intent(context, KeychainIntentService.class);
        importIntent.setAction(KeychainIntentService.ACTION_IMPORT_KEYRING);
        Bundle importData = new Bundle();
        importData.putParcelableArrayList(KeychainIntentService.IMPORT_KEY_LIST, keys);
        importIntent.putExtra(KeychainIntentService.EXTRA_DATA, importData);
        importIntent.putExtra(KeychainIntentService.EXTRA_MESSENGER, messenger);

        context.startService(importIntent);
    }

    public static List<ImportKeysListEntry> getEmailKeys(String mail, Keyserver keyServer) {
        Set<ImportKeysListEntry> keys = new HashSet<ImportKeysListEntry>();
        try {
            for (ImportKeysListEntry key : keyServer.search(mail)) {
                if (key.isRevoked() || key.isExpired()) continue;
                for (String userId : key.getUserIds()) {
                    if (userId.toLowerCase().contains(mail.toLowerCase(Locale.ENGLISH))) {
                        keys.add(key);
                    }
                }
            }
        } catch (Keyserver.QueryFailedException ignored) {
        } catch (Keyserver.QueryNeedsRepairException ignored) {
        }
        return new ArrayList<ImportKeysListEntry>(keys);
    }
}
