From 10c1a9ccedfca1e76317139d60083b85a36d9848 Mon Sep 17 00:00:00 2001
From: Mikunj <mikunj@live.com.au>
Date: Thu, 26 Sep 2019 11:19:36 +1000
Subject: [PATCH] Added linking in the ui.

---
 res/layout/activity_seed.xml                  | 42 ++++++++-
 res/values/strings.xml                        |  4 +
 .../securesms/jobs/PushDecryptJob.java        |  5 ++
 .../securesms/loki/SeedActivity.kt            | 86 ++++++++++++++-----
 4 files changed, 116 insertions(+), 21 deletions(-)

diff --git a/res/layout/activity_seed.xml b/res/layout/activity_seed.xml
index 6087c88c14..b027d3d40a 100644
--- a/res/layout/activity_seed.xml
+++ b/res/layout/activity_seed.xml
@@ -76,8 +76,28 @@
             app:labeledEditText_background="@color/loki_darkest_gray"
             app:labeledEditText_label="@string/activity_key_pair_mnemonic_edit_text_label"/>
 
+        <TextView
+            android:id="@+id/linkExplanationTextView"
+            style="@style/Signal.Text.Body"
+            android:layout_width="match_parent"
+            android:layout_height="wrap_content"
+            android:layout_marginTop="24dp"
+            android:visibility="gone"
+            android:text="@string/activity_key_pair_seed_explanation_3"
+            android:textAlignment="center" />
+
+        <org.thoughtcrime.securesms.components.LabeledEditText
+                android:id="@+id/publicKeyEditText"
+                android:layout_width="match_parent"
+                android:layout_height="wrap_content"
+                android:layout_marginTop="16dp"
+                android:layout_marginBottom="20dp"
+                android:visibility="gone"
+                app:labeledEditText_background="@color/loki_darkest_gray"
+                app:labeledEditText_label="@string/activity_key_pair_public_key_edit_text_label"/>
+
         <Button
-            android:id="@+id/toggleModeButton"
+            android:id="@+id/toggleRestoreModeButton"
             android:layout_width="match_parent"
             android:layout_height="50dp"
             android:background="@color/transparent"
@@ -86,6 +106,26 @@
             android:elevation="0dp"
             android:stateListAnimator="@null" />
 
+        <Button
+            android:id="@+id/toggleRegisterModeButton"
+            android:layout_width="match_parent"
+            android:layout_height="50dp"
+            android:background="@color/transparent"
+            android:textColor="@color/signal_primary"
+            android:text="@string/activity_key_pair_toggle_mode_button_title_2"
+            android:elevation="0dp"
+            android:stateListAnimator="@null" />
+
+        <Button
+            android:id="@+id/toggleLinkModeButton"
+            android:layout_width="match_parent"
+            android:layout_height="50dp"
+            android:background="@color/transparent"
+            android:textColor="@color/signal_primary"
+            android:text="@string/activity_key_pair_toggle_mode_button_title_3"
+            android:elevation="0dp"
+            android:stateListAnimator="@null" />
+
         <com.dd.CircularProgressButton
             android:id="@+id/registerOrRestoreButton"
             android:layout_width="match_parent"
diff --git a/res/values/strings.xml b/res/values/strings.xml
index 13f06cc71c..02a9c47c6e 100644
--- a/res/values/strings.xml
+++ b/res/values/strings.xml
@@ -1558,13 +1558,17 @@
     <string name="activity_key_pair_title">Create Your Loki Messenger Account</string>
     <string name="activity_key_pair_seed_explanation_1">Please save the seed below in a safe location. It can be used to restore your account if you lose access, or to migrate to a new device.</string>
     <string name="activity_key_pair_seed_explanation_2">Restore your account by entering your seed below</string>
+    <string name="activity_key_pair_seed_explanation_3">Link to an existing device by going into its in-app settings and clicking "Link Device".</string>
     <string name="activity_key_pair_copy_button_title">Copy</string>
     <string name="activity_key_pair_mnemonic_edit_text_label">Your Seed</string>
     <string name="activity_key_pair_toggle_mode_button_title_1">Restore Using Seed</string>
     <string name="activity_key_pair_toggle_mode_button_title_2">Register a New Account</string>
+    <string name="activity_key_pair_toggle_mode_button_title_3">Link Device</string>
     <string name="activity_key_pair_mnemonic_copied_message">Copied to clipboard</string>
     <string name="activity_key_pair_register_or_restore_button_title_1">Register</string>
     <string name="activity_key_pair_register_or_restore_button_title_2">Restore</string>
+    <string name="activity_key_pair_register_or_restore_button_title_3">Link</string>
+    <string name="activity_key_pair_public_key_edit_text_label">Other Device Pub Key</string>
     <!-- Conversation list activity -->
     <string name="activity_conversation_list_empty_state_message">Looks like you don\'t have any conversations yet. Get started by messaging a friend.</string>
     <!-- Settings activity -->
diff --git a/src/org/thoughtcrime/securesms/jobs/PushDecryptJob.java b/src/org/thoughtcrime/securesms/jobs/PushDecryptJob.java
index 30a1de3a30..c2dcb83ac7 100644
--- a/src/org/thoughtcrime/securesms/jobs/PushDecryptJob.java
+++ b/src/org/thoughtcrime/securesms/jobs/PushDecryptJob.java
@@ -1067,6 +1067,11 @@ public class PushDecryptJob extends BaseJob implements InjectableType {
       return;
     }
 
+    if (LokiDeviceLinkingSession.Companion.getShared().isListeningForLinkingRequest()) {
+      Log.w("Loki", "Received authorisation but device is not is listening.");
+      return;
+    }
+
     // Unimplemented for REQUEST
     if (authorisation.getType() != LokiPairingAuthorisation.Type.GRANT) { return; }
     Log.d("Loki", "Receiving pairing authorisation from: " + authorisation.getPrimaryDevicePubKey());
diff --git a/src/org/thoughtcrime/securesms/loki/SeedActivity.kt b/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
index c9b6f44954..3efc94e086 100644
--- a/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
+++ b/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
@@ -10,6 +10,7 @@ import android.view.inputmethod.InputMethodManager
 import android.widget.Toast
 import kotlinx.android.synthetic.main.activity_seed.*
 import network.loki.messenger.R
+import org.thoughtcrime.securesms.ApplicationContext
 import org.thoughtcrime.securesms.BaseActionBarActivity
 import org.thoughtcrime.securesms.crypto.IdentityKeyUtil
 import org.thoughtcrime.securesms.database.Address
@@ -21,6 +22,7 @@ import org.whispersystems.curve25519.Curve25519
 import org.whispersystems.libsignal.util.KeyHelper
 import org.whispersystems.signalservice.loki.crypto.MnemonicCodec
 import org.whispersystems.signalservice.loki.utilities.Analytics
+import org.whispersystems.signalservice.loki.utilities.PublicKeyValidation
 import org.whispersystems.signalservice.loki.utilities.hexEncodedPublicKey
 import java.io.File
 import java.io.FileOutputStream
@@ -34,8 +36,10 @@ class SeedActivity : BaseActionBarActivity() {
     private var mnemonic: String? = null
         set(newValue) { field = newValue; updateMnemonicTextView() }
 
+    private var dialog: ProgressDialog? = null
+
     // region Types
-    enum class Mode { Register, Restore }
+    enum class Mode { Register, Restore, Link }
     // endregion
 
     // region Lifecycle
@@ -44,10 +48,17 @@ class SeedActivity : BaseActionBarActivity() {
         setContentView(R.layout.activity_seed)
         setUpLanguageFileDirectory()
         updateSeed()
+        updateUI()
         copyButton.setOnClickListener { copy() }
-        toggleModeButton.setOnClickListener { toggleMode() }
+        toggleRestoreModeButton.setOnClickListener { mode = Mode.Restore }
+        toggleRegisterModeButton.setOnClickListener { mode = Mode.Register }
+        toggleLinkModeButton.setOnClickListener { mode = Mode.Link }
         registerOrRestoreButton.setOnClickListener { registerOrRestore() }
     }
+
+    override fun onDestroy() {
+        super.onDestroy()
+    }
     // endregion
 
     // region General
@@ -85,14 +96,28 @@ class SeedActivity : BaseActionBarActivity() {
     }
 
     private fun updateUI() {
-        seedExplanationTextView1.visibility = if (mode == Mode.Register) View.VISIBLE else View.GONE
-        mnemonicTextView.visibility = if (mode == Mode.Register) View.VISIBLE else View.GONE
-        copyButton.visibility = if (mode == Mode.Register) View.VISIBLE else View.GONE
-        seedExplanationTextView2.visibility = if (mode == Mode.Restore) View.VISIBLE else View.GONE
-        mnemonicEditText.visibility = if (mode == Mode.Restore) View.VISIBLE else View.GONE
-        val toggleModeButtonTitleID = if (mode == Mode.Register) R.string.activity_key_pair_toggle_mode_button_title_1 else R.string.activity_key_pair_toggle_mode_button_title_2
-        toggleModeButton.setText(toggleModeButtonTitleID)
-        val registerOrRestoreButtonTitleID = if (mode == Mode.Register) R.string.activity_key_pair_register_or_restore_button_title_1 else R.string.activity_key_pair_register_or_restore_button_title_2
+        val showOnRegister = if (mode == Mode.Register) View.VISIBLE else View.GONE
+        val showOnRestore = if (mode == Mode.Restore) View.VISIBLE else View.GONE
+        val showOnLink = if (mode == Mode.Link) View.VISIBLE else View.GONE
+
+        seedExplanationTextView1.visibility = showOnRegister
+        mnemonicTextView.visibility = showOnRegister
+        copyButton.visibility = showOnRegister
+        seedExplanationTextView2.visibility = showOnRestore
+        mnemonicEditText.visibility = showOnRestore
+        publicKeyEditText.visibility = showOnLink
+        linkExplanationTextView.visibility = showOnLink
+
+        toggleRegisterModeButton.visibility = if (mode != Mode.Register) View.VISIBLE else View.GONE
+        toggleRestoreModeButton.visibility = if (mode != Mode.Restore) View.VISIBLE else View.GONE
+        toggleLinkModeButton.visibility = if (mode != Mode.Link) View.VISIBLE else View.GONE
+
+        val registerOrRestoreButtonTitleID = when (mode) {
+            Mode.Register -> R.string.activity_key_pair_register_or_restore_button_title_1
+            Mode.Restore -> R.string.activity_key_pair_register_or_restore_button_title_2
+            Mode.Link -> R.string.activity_key_pair_register_or_restore_button_title_3
+        }
+
         registerOrRestoreButton.setText(registerOrRestoreButtonTitleID)
         if (mode == Mode.Restore) {
             mnemonicEditText.requestFocus()
@@ -101,6 +126,14 @@ class SeedActivity : BaseActionBarActivity() {
             val inputMethodManager = getSystemService(INPUT_METHOD_SERVICE) as InputMethodManager
             inputMethodManager.hideSoftInputFromWindow(mnemonicEditText.windowToken, 0)
         }
+
+        if (mode == Mode.Link) {
+            publicKeyEditText.requestFocus()
+        } else {
+            publicKeyEditText.clearFocus()
+            val inputMethodManager = getSystemService(INPUT_METHOD_SERVICE) as InputMethodManager
+            inputMethodManager.hideSoftInputFromWindow(publicKeyEditText.windowToken, 0)
+        }
     }
 
     private fun updateMnemonic() {
@@ -121,13 +154,6 @@ class SeedActivity : BaseActionBarActivity() {
         Toast.makeText(this, R.string.activity_key_pair_mnemonic_copied_message, Toast.LENGTH_SHORT).show()
     }
 
-    private fun toggleMode() {
-        mode = when (mode) {
-            Mode.Register -> Mode.Restore
-            Mode.Restore -> Mode.Register
-        }
-    }
-
     private fun registerOrRestore() {
         var seed: ByteArray
         when (mode) {
@@ -142,6 +168,12 @@ class SeedActivity : BaseActionBarActivity() {
                     return Toast.makeText(this, message, Toast.LENGTH_SHORT).show()
                 }
             }
+            Mode.Link -> {
+                if (!PublicKeyValidation.isValid(publicKeyEditText.text.trim().toString())) {
+                    return Toast.makeText(this, "Invalid public key", Toast.LENGTH_SHORT).show()
+                }
+                seed = this.seed!!
+            }
         }
         val hexEncodedSeed = Hex.toStringCondensed(seed)
         IdentityKeyUtil.save(this, IdentityKeyUtil.lokiSeedKey, hexEncodedSeed)
@@ -155,14 +187,28 @@ class SeedActivity : BaseActionBarActivity() {
         val registrationID = KeyHelper.generateRegistrationId(false)
         TextSecurePreferences.setLocalRegistrationId(this, registrationID)
         DatabaseFactory.getIdentityDatabase(this).saveIdentity(Address.fromSerialized(hexEncodedPublicKey), publicKey,
-            IdentityDatabase.VerifiedStatus.VERIFIED, true, System.currentTimeMillis(), true)
+                IdentityDatabase.VerifiedStatus.VERIFIED, true, System.currentTimeMillis(), true)
         TextSecurePreferences.setLocalNumber(this, hexEncodedPublicKey)
         when (mode) {
             Mode.Register -> Analytics.shared.track("Seed Created")
             Mode.Restore -> Analytics.shared.track("Seed Restored")
+            Mode.Link -> Analytics.shared.track("Device Linked")
+        }
+        if (mode == Mode.Link) {
+            TextSecurePreferences.setHasSeenWelcomeScreen(this, true)
+            TextSecurePreferences.setPromptedPushRegistration(this, true)
+            val application = ApplicationContext.getInstance(this)
+            application.startLongPollingIfNeeded()
+            application.setUpStorageAPIIfNeeded()
+
+            // TODO: Show activity view here?
+
+            // TODO: Also need to reset on registration
+
+        } else {
+            startActivity(Intent(this, AccountDetailsActivity::class.java))
+            finish()
         }
-        startActivity(Intent(this, AccountDetailsActivity::class.java))
-        finish()
     }
     // endregion
 }
\ No newline at end of file