Dispatch not on main thread

Author: porter-stripe

android/src/main/java/com/reactnativestripesdk/PaymentOptionDisplayDataMapper.kt

@@ -4,32 +4,34 @@ import com.reactnativestripesdk.getBase64FromBitmap
 import com.reactnativestripesdk.getBitmapFromDrawable
 import com.reactnativestripesdk.utils.mapFromPaymentSheetBillingDetails
 import com.stripe.android.paymentelement.EmbeddedPaymentElement
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import kotlinx.coroutines.withTimeout
 
 /**
  * Serialize Stripe's PaymentOptionDisplayData into a WritableMap
  * that can be sent over the RN bridge.
  */
-fun EmbeddedPaymentElement.PaymentOptionDisplayData.toWritableMap(): WritableMap =
+suspend fun EmbeddedPaymentElement.PaymentOptionDisplayData.toWritableMap(): WritableMap =
   Arguments.createMap().apply {
     putString("label", label)
     putString("paymentMethodType", paymentMethodType)
     putMap("billingDetails", mapFromPaymentSheetBillingDetails(billingDetails))
 
-    // Direct access to imageLoader and call it synchronously with timeout
+    // Load image off the main thread with a timeout
     val imageBase64 =
       try {
-        // Load synchronously with 5 second timeout
-        val drawable =
-          kotlinx.coroutines.runBlocking {
-            withTimeout(5000L) {
-              imageLoader()
+        withContext(Dispatchers.Default) {
+          val drawable =
+            withTimeout(5_000L) {
+              withContext(Dispatchers.IO) {
+                imageLoader()
+              }
             }
-          }
-
-        getBitmapFromDrawable(drawable)?.let { bitmap ->
-          getBase64FromBitmap(bitmap)
-        } ?: ""
+          getBitmapFromDrawable(drawable)?.let { bitmap ->
+            getBase64FromBitmap(bitmap)
+          } ?: ""
+        }
       } catch (e: Exception) {
         // If imageLoader fails or times out, return empty string
         ""