Create `EditText.setOnEnterOrDoneAction` extension

1 files changed, 15 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/util/EditTextExt.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/util/EditTextExt.kt
new file mode 100644
index 0000000000..b90201edfe
--- /dev/null
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/util/EditTextExt.kt
@@ -0,0 +1,15 @@
+package net.mullvad.mullvadvpn.util
+
+import android.view.KeyEvent
+import android.view.inputmethod.EditorInfo
+import android.widget.EditText
+
+fun EditText.setOnEnterOrDoneAction(callback: () -> Unit) {
+    setOnEditorActionListener { _, action, event ->
+        if (action == EditorInfo.IME_ACTION_DONE || event?.keyCode == KeyEvent.KEYCODE_ENTER) {
+            callback()
+        }
+
+        false
+    }
+}