From 24af918f9e8bd4cbc9dfc26caae31015c0e980be Mon Sep 17 00:00:00 2001
From: Niels Andriesse <andriesseniels@gmail.com>
Date: Thu, 28 Jan 2021 13:46:29 +1100
Subject: [PATCH] Minor refactoring

---
 .../fragment_key_pair_migration_bottom_sheet.xml       |  8 ++++++++
 .../loki/dialogs/KeyPairMigrationBottomSheet.kt        | 10 ++++++++++
 2 files changed, 18 insertions(+)

diff --git a/res/layout/fragment_key_pair_migration_bottom_sheet.xml b/res/layout/fragment_key_pair_migration_bottom_sheet.xml
index 5588c2e933..4014fed0a0 100644
--- a/res/layout/fragment_key_pair_migration_bottom_sheet.xml
+++ b/res/layout/fragment_key_pair_migration_bottom_sheet.xml
@@ -49,4 +49,12 @@
         android:layout_marginTop="@dimen/very_large_spacing"
         android:text="Upgrade Now" />
 
+    <Button
+        style="@style/Widget.Session.Button.Common.UnimportantOutline"
+        android:id="@+id/upgradeLaterButton"
+        android:layout_width="240dp"
+        android:layout_height="@dimen/medium_button_height"
+        android:layout_marginTop="@dimen/medium_spacing"
+        android:text="Upgrade Later" />
+
 </LinearLayout>
diff --git a/src/org/thoughtcrime/securesms/loki/dialogs/KeyPairMigrationBottomSheet.kt b/src/org/thoughtcrime/securesms/loki/dialogs/KeyPairMigrationBottomSheet.kt
index f50c69d493..dff4e5fa30 100644
--- a/src/org/thoughtcrime/securesms/loki/dialogs/KeyPairMigrationBottomSheet.kt
+++ b/src/org/thoughtcrime/securesms/loki/dialogs/KeyPairMigrationBottomSheet.kt
@@ -23,6 +23,7 @@ class KeyPairMigrationBottomSheet : BottomSheetDialogFragment() {
     override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
         super.onViewCreated(view, savedInstanceState)
         upgradeNowButton.setOnClickListener { upgradeNow() }
+        upgradeLaterButton.setOnClickListener { upgradeLater() }
     }
 
     override fun onCreateDialog(savedInstanceState: Bundle?): Dialog {
@@ -41,4 +42,13 @@ class KeyPairMigrationBottomSheet : BottomSheetDialogFragment() {
         val applicationContext = requireContext().applicationContext as ApplicationContext
         applicationContext.clearAllData(true)
     }
+
+    private fun upgradeLater() {
+        val dialog = AlertDialog.Builder(requireContext())
+        dialog.setMessage("You won't be able to send or receive messages until you upgrade.")
+        dialog.setPositiveButton(R.string.ok) { _, _ ->
+            dismiss()
+        }
+        dialog.create().show()
+    }
 }
\ No newline at end of file