From 362c7d2b77e62bad827c6ac5ce17a70d725817d6 Mon Sep 17 00:00:00 2001
From: Amr Hossam <amrhossam.dev@gmail.com>
Date: Wed, 25 Dec 2024 03:59:00 +0200
Subject: [PATCH] refactor picking address

Signed-off-by: Amr Hossam <amrhossam.dev@gmail.com>
---
 .../android/ui/TripPlanFragment.java          | 61 ++++++++++++++-----
 1 file changed, 45 insertions(+), 16 deletions(-)

diff --git a/onebusaway-android/src/main/java/org/onebusaway/android/ui/TripPlanFragment.java b/onebusaway-android/src/main/java/org/onebusaway/android/ui/TripPlanFragment.java
index 74e608fa4..2b408cfd3 100644
--- a/onebusaway-android/src/main/java/org/onebusaway/android/ui/TripPlanFragment.java
+++ b/onebusaway-android/src/main/java/org/onebusaway/android/ui/TripPlanFragment.java
@@ -167,28 +167,55 @@ public Intent parseResult(int i, @Nullable Intent addressIntent) {
         if (addressIntent == null) {
             return;
         }
+
         Uri addressUri = addressIntent.getData();
-        String[] projection = new String[]{ContactsContract.CommonDataKinds.StructuredPostal.FORMATTED_ADDRESS};
+        if (addressUri == null) {
+            return;
+        }
+
+        String[] projection = {ContactsContract.CommonDataKinds.StructuredPostal.FORMATTED_ADDRESS};
+
         try (Cursor cursor = getContext().getContentResolver().query(addressUri, projection, null, null, null)) {
-            if (cursor != null && cursor.moveToFirst()) {
-                int addressIndex = cursor.getColumnIndex(ContactsContract.CommonDataKinds.StructuredPostal.FORMATTED_ADDRESS);
-                String address = cursor.getString(addressIndex).replace("\n", ", ");
-                int addressInputId = addressIntent.getIntExtra(ADDRESS_INPUT_ID_KEY, -1);
-                TextView addressInput = getActivity().findViewById(addressInputId);
-                addressInput.post(() -> addressInput.setText(address));
-                if (addressInputId == mToAddressTextArea.getId() && mBuilder.ready()) {
-                    new AlertDialog.Builder(getContext())
-                            .setTitle(R.string.plan_trip)
-                            .setMessage(R.string.do_you_want_to_plan_the_trip_now)
-                            .setPositiveButton(R.string.ok, (dialog, which) -> checkRequestAndSubmit())
-                            .setNegativeButton(R.string.cancel, null)
-                            .create()
-                            .show();
-                }
+            if (cursor == null || !cursor.moveToFirst()) {
+                return;
+            }
+
+            String address = extractAddress(cursor);
+            int addressInputId = addressIntent.getIntExtra(ADDRESS_INPUT_ID_KEY, -1);
+            if (addressInputId == -1) {
+                return;
             }
+
+            updateAddressInput(address, addressInputId);
+            updateAddressData(address, addressInputId);
         }
     };
 
+    private String extractAddress(Cursor cursor) {
+        int addressIndex = cursor.getColumnIndex(ContactsContract.CommonDataKinds.StructuredPostal.FORMATTED_ADDRESS);
+        return cursor.getString(addressIndex).replace("\n", ", ");
+    }
+
+    private void updateAddressInput(String address, int addressInputId) {
+        TextView addressInput = getActivity().findViewById(addressInputId);
+        addressInput.post(() -> addressInput.setText(address));
+        addressInput.requestFocus();
+    }
+
+    private void updateAddressData(String address, int addressInputId) {
+        CustomAddress customAddress = CustomAddress.getEmptyAddress();
+        customAddress.setAddressLine(0, address);
+
+        if (addressInputId == mFromAddressTextArea.getId()) {
+            mFromAddress = customAddress;
+            mBuilder.setFrom(mFromAddress);
+        } else if (addressInputId == mToAddressTextArea.getId()) {
+            mToAddress = customAddress;
+            mBuilder.setTo(mToAddress);
+        }
+    }
+
+
     private final ActivityResultLauncher<TextView> mSelectAddressFromContactLauncher = registerForActivityResult(
             selectAddressFromContactContract,
             addressIntentActivityResultCallback
@@ -373,6 +400,8 @@ private void checkRequestAndSubmit() {
         if (mBuilder.ready() && mListener != null) {
             mFromAddressTextArea.dismissDropDown();
             mToAddressTextArea.dismissDropDown();
+            mFromAddressTextArea.clearFocus();
+            mToAddressTextArea.clearFocus();
             UIUtils.closeKeyboard(getContext(), mFromAddressTextArea);
             mListener.onTripRequestReady();
         }