expert 'setting' to remove omemo identity

1 files changed, 61 insertions, 1 deletions

diff --git a/src/main/java/de/pixart/messenger/ui/SettingsActivity.java b/src/main/java/de/pixart/messenger/ui/SettingsActivity.java
index 150a1d779..737f10c86 100644
--- a/src/main/java/de/pixart/messenger/ui/SettingsActivity.java
+++ b/src/main/java/de/pixart/messenger/ui/SettingsActivity.java
@@ -29,6 +29,8 @@ import de.pixart.messenger.R;
 import de.pixart.messenger.entities.Account;
 import de.pixart.messenger.services.ExportLogsService;
 import de.pixart.messenger.xmpp.XmppConnection;
+import de.pixart.messenger.xmpp.jid.InvalidJidException;
+import de.pixart.messenger.xmpp.jid.Jid;
 
 public class SettingsActivity extends XmppActivity implements
 		OnSharedPreferenceChangeListener {
@@ -84,7 +86,7 @@ public class SettingsActivity extends XmppActivity implements
 					displayToast(getString(R.string.toast_no_trusted_certs));
 					return true;
 				}
-				final ArrayList selectedItems = new ArrayList<Integer>();
+                final ArrayList selectedItems = new ArrayList<>();
 				final AlertDialog.Builder dialogBuilder = new AlertDialog.Builder(SettingsActivity.this);
 				dialogBuilder.setTitle(getResources().getString(R.string.dialog_manage_certs_title));
 				dialogBuilder.setMultiChoiceItems(aliases.toArray(new CharSequence[aliases.size()]), null,
@@ -144,6 +146,64 @@ public class SettingsActivity extends XmppActivity implements
 				return true;
 			}
 		});
+
+        final Preference deleteOmemoPreference = mSettingsFragment.findPreference("delete_omemo_identities");
+        		deleteOmemoPreference.setOnPreferenceClickListener(new Preference.OnPreferenceClickListener() {
+            			@Override
+            			public boolean onPreferenceClick(Preference preference) {
+                				deleteOmemoIdentities();
+                				return true;
+                			}
+            		});
+        	}
+
+            	private void deleteOmemoIdentities() {
+        		AlertDialog.Builder builder = new AlertDialog.Builder(this);
+        		builder.setTitle(R.string.pref_delete_omemo_identities);
+        		final List<CharSequence> accounts = new ArrayList<>();
+        		for(Account account : xmppConnectionService.getAccounts()) {
+            			if (!account.isOptionSet(Account.OPTION_DISABLED)) {
+                				accounts.add(account.getJid().toBareJid().toString());
+                			}
+            		}
+        		final boolean[] checkedItems = new boolean[accounts.size()];
+        		builder.setMultiChoiceItems(accounts.toArray(new CharSequence[accounts.size()]), checkedItems, new DialogInterface.OnMultiChoiceClickListener() {
+            			@Override
+            			public void onClick(DialogInterface dialog, int which, boolean isChecked) {
+                				checkedItems[which] = isChecked;
+                				final AlertDialog alertDialog = (AlertDialog) dialog;
+                				for(boolean item : checkedItems) {
+                    					if (item) {
+                        						alertDialog.getButton(DialogInterface.BUTTON_POSITIVE).setEnabled(true);
+                        						return;
+                        					}
+                    				}
+                				alertDialog.getButton(DialogInterface.BUTTON_POSITIVE).setEnabled(false);
+                			}
+            		});
+        		builder.setNegativeButton(R.string.cancel,null);
+        		builder.setPositiveButton(R.string.delete_selected_keys, new DialogInterface.OnClickListener() {
+            			@Override
+            			public void onClick(DialogInterface dialog, int which) {
+                				for(int i = 0; i < checkedItems.length; ++i) {
+                    					if (checkedItems[i]) {
+                        						try {
+                            							Jid jid = Jid.fromString(accounts.get(i).toString());
+                            							Account account = xmppConnectionService.findAccountByJid(jid);
+                            							if (account != null) {
+                                								account.getAxolotlService().regenerateKeys(true);
+                                							}
+                            						} catch (InvalidJidException e) {
+                            							//
+                                    						}
+
+                                					}
+                    				}
+                			}
+            		});
+        		AlertDialog dialog = builder.create();
+        		dialog.show();
+        		dialog.getButton(AlertDialog.BUTTON_POSITIVE).setEnabled(false);
 	}
 
 	@Override