/*
 * Copyright (C) 2016 Tobias Erthal
 * Copyright (C) 2014-2016 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.ui.adapter;

import android.content.Context;
import android.database.Cursor;
import android.view.View;
import android.view.ViewGroup;
import android.widget.AdapterView;
import android.widget.AdapterView.OnItemClickListener;
import android.widget.CheckBox;

import org.sufficientlysecure.keychain.Constants;
import org.sufficientlysecure.keychain.R;
import org.sufficientlysecure.keychain.util.Log;

import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

public class KeySelectableAdapter extends KeyAdapter implements OnItemClickListener {

    private HashSet<Long> mSelectedItems = new HashSet<>();

    public KeySelectableAdapter(Context context, Cursor c, int flags, Set<Long> initialChecked) {
        super(context, c, flags);
        if (initialChecked != null) {
            mSelectedItems.addAll(initialChecked);
        }
    }

    public static class KeySelectableItemViewHolder extends KeyItemViewHolder {

        public CheckBox mCheckbox;

        public KeySelectableItemViewHolder(View view) {
            super(view);
            mCheckbox = (CheckBox) view.findViewById(R.id.selected);
        }

        public void setCheckedState(boolean checked) {
            mCheckbox.setChecked(checked);
        }

    }

    @Override
    public View newView(Context context, Cursor cursor, ViewGroup parent) {
        View view = mInflater.inflate(R.layout.key_list_selectable_item, parent, false);
        KeySelectableItemViewHolder holder = new KeySelectableItemViewHolder(view);
        view.setTag(holder);
        return view;
    }

    @Override
    public void bindView(View view, Context context, Cursor cursor) {
        super.bindView(view, context, cursor);

        KeySelectableItemViewHolder h = (KeySelectableItemViewHolder) view.getTag();
        h.setCheckedState(mSelectedItems.contains(h.mDisplayedItem.mKeyId));

    }

    public void setCheckedStates(Set<Long> checked) {
        mSelectedItems.clear();
        mSelectedItems.addAll(checked);
        notifyDataSetChanged();
    }

    public Set<Long> getSelectedMasterKeyIds() {
        return Collections.unmodifiableSet(mSelectedItems);
    }

    @Override
    public void onItemClick(AdapterView<?> parent, View view, int position, long id) {
        long masterKeyId = getMasterKeyId(position);
        if (mSelectedItems.contains(masterKeyId)) {
            mSelectedItems.remove(masterKeyId);
        } else {
            mSelectedItems.add(masterKeyId);
        }
        notifyDataSetChanged();
    }

}