preserve state in CertifyKeyFragment

OpenKeychain/src/main/java/org/sufficientlysecure/keychain/ui/CertifyKeyFragment.java

@@ -69,6 +69,9 @@ import java.util.ArrayList;
 public class CertifyKeyFragment extends CryptoOperationFragment
         implements LoaderManager.LoaderCallbacks<Cursor> {
 
+    public static final String ARG_CHECK_STATES = "check_states";
+    public static final String ARG_CACHED_ACTIONS = "cached_actions";
+
     private CheckBox mUploadKeyCheckbox;
     ListView mUserIds;
 
@@ -76,6 +79,8 @@ public class CertifyKeyFragment extends CryptoOperationFragment
 
     private long[] mPubMasterKeyIds;
 
+    private CertifyActionsParcel mCachedActionsParcel;
+
     public static final String[] USER_IDS_PROJECTION = new String[]{
             UserPackets._ID,
             UserPackets.MASTER_KEY_ID,
@@ -104,10 +109,22 @@ public class CertifyKeyFragment extends CryptoOperationFragment
 
         mPassthroughMessenger = getActivity().getIntent().getParcelableExtra(
                 KeychainIntentService.EXTRA_MESSENGER);
-        mPassthroughMessenger = null; // TODO remove, development hack
+        mPassthroughMessenger = null; // TODO doesn't work with CryptoOperationFragment, disabled for now
+
+        ArrayList<Boolean> checkedStates;
+        if (savedInstanceState != null) {
+            mCachedActionsParcel = savedInstanceState.getParcelable(ARG_CACHED_ACTIONS);
+            checkedStates = (ArrayList<Boolean>) savedInstanceState.getSerializable(ARG_CHECK_STATES);
+
+            // key spinner and the checkbox keep their own state
+
+        } else {
+            mCachedActionsParcel = null;
+            checkedStates = null;
 
             // preselect certify key id if given
-        long certifyKeyId = getActivity().getIntent().getLongExtra(CertifyKeyActivity.EXTRA_CERTIFY_KEY_ID, Constants.key.none);
+            long certifyKeyId = getActivity().getIntent()
+                    .getLongExtra(CertifyKeyActivity.EXTRA_CERTIFY_KEY_ID, Constants.key.none);
             if (certifyKeyId != Constants.key.none) {
                 try {
                     CachedPublicKeyRing key = (new ProviderHelper(getActivity())).getCachedPublicKeyRing(certifyKeyId);
@@ -119,7 +136,9 @@ public class CertifyKeyFragment extends CryptoOperationFragment
                 }
             }
 
-        mUserIdsAdapter = new MultiUserIdsAdapter(getActivity(), null, 0);
+        }
+
+        mUserIdsAdapter = new MultiUserIdsAdapter(getActivity(), null, 0, checkedStates);
         mUserIds.setAdapter(mUserIdsAdapter);
         mUserIds.setDividerHeight(0);
 
@@ -132,6 +151,16 @@ public class CertifyKeyFragment extends CryptoOperationFragment
         }
     }
 
+    @Override
+    public void onSaveInstanceState(Bundle outState) {
+        super.onSaveInstanceState(outState);
+
+        ArrayList<Boolean> states = mUserIdsAdapter.getCheckStates();
+        // no proper parceling method available :(
+        outState.putSerializable(ARG_CHECK_STATES, states);
+        outState.putParcelable(ARG_CACHED_ACTIONS, mCachedActionsParcel);
+    }
+
     @Override
     public View onCreateView(LayoutInflater inflater, ViewGroup superContainer, Bundle savedInstanceState) {
         View view = inflater.inflate(R.layout.certify_key_fragment, null);
@@ -288,6 +317,10 @@ public class CertifyKeyFragment extends CryptoOperationFragment
 
     @Override
     protected void cryptoOperation(CryptoInputParcel cryptoInput) {
+        Bundle data = new Bundle();
+        {
+
+            if (mCachedActionsParcel == null) {
                 // Bail out if there is not at least one user id selected
                 ArrayList<CertifyAction> certifyActions = mUserIdsAdapter.getSelectedCertifyActions();
                 if (certifyActions.isEmpty()) {
@@ -296,16 +329,16 @@ public class CertifyKeyFragment extends CryptoOperationFragment
                     return;
                 }
 
-        Bundle data = new Bundle();
-        {
                 long selectedKeyId = mCertifyKeySpinner.getSelectedKeyId();
 
                 // fill values for this action
-            CertifyActionsParcel parcel = new CertifyActionsParcel(selectedKeyId);
+                mCachedActionsParcel = new CertifyActionsParcel(selectedKeyId);
-            parcel.mCertifyActions.addAll(certifyActions);
+                mCachedActionsParcel.mCertifyActions.addAll(certifyActions);
+            }
 
             data.putParcelable(KeychainIntentService.EXTRA_CRYPTO_INPUT, cryptoInput);
-            data.putParcelable(KeychainIntentService.CERTIFY_PARCEL, parcel);
+            data.putParcelable(KeychainIntentService.CERTIFY_PARCEL, mCachedActionsParcel);
+
             if (mUploadKeyCheckbox.isChecked()) {
                 String keyserver = Preferences.getPreferences(getActivity()).getPreferredKeyserver();
                 data.putString(KeychainIntentService.UPLOAD_KEY_SERVER, keyserver);
@@ -368,4 +401,11 @@ public class CertifyKeyFragment extends CryptoOperationFragment
 
     }
 
+    @Override
+    protected void onCryptoOperationCancelled() {
+        super.onCryptoOperationCancelled();
+
+        // forget this ever happened
+        mCachedActionsParcel = null;
+    }
 }

OpenKeychain/src/main/java/org/sufficientlysecure/keychain/ui/adapter/MultiUserIdsAdapter.java

@@ -40,20 +40,19 @@ public class MultiUserIdsAdapter extends CursorAdapter {
     private LayoutInflater mInflater;
     private final ArrayList<Boolean> mCheckStates;
 
-    public MultiUserIdsAdapter(Context context, Cursor c, int flags) {
+    public MultiUserIdsAdapter(Context context, Cursor c, int flags, ArrayList<Boolean> preselectStates) {
         super(context, c, flags);
         mInflater = LayoutInflater.from(context);
-        mCheckStates = new ArrayList<>();
+        mCheckStates = preselectStates == null ? new ArrayList<Boolean>() : preselectStates;
     }
 
     @Override
     public Cursor swapCursor(Cursor newCursor) {
-        mCheckStates.clear();
         if (newCursor != null) {
             int count = newCursor.getCount();
             mCheckStates.ensureCapacity(count);
-            // initialize to true (use case knowledge: we usually want to sign all uids)
+            // initialize new fields to true (use case knowledge: we usually want to sign all uids)
-            for (int i = 0; i < count; i++) {
+            for (int i = mCheckStates.size(); i < count; i++) {
                 mCheckStates.add(true);
             }
         }
@@ -151,6 +150,10 @@ public class MultiUserIdsAdapter extends CursorAdapter {
 
     }
 
+    public ArrayList<Boolean> getCheckStates() {
+        return mCheckStates;
+    }
+
     public ArrayList<CertifyAction> getSelectedCertifyActions() {
         LongSparseArray<CertifyAction> actions = new LongSparseArray<>();
         for (int i = 0; i < mCheckStates.size(); i++) {