From beeb676517a89c90e1af40c1d727fb767c656a93 Mon Sep 17 00:00:00 2001
From: Taras Smakula <tarassmakula@gmail.com>
Date: Tue, 27 Feb 2024 17:52:18 +0200
Subject: [PATCH] Update subscription ds

---
 .../SubscriptionStageDataSource.kt                | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/auth/src/main/java/org/futo/circles/auth/feature/sign_up/subscription_stage/SubscriptionStageDataSource.kt b/auth/src/main/java/org/futo/circles/auth/feature/sign_up/subscription_stage/SubscriptionStageDataSource.kt
index a755a74dc..8e6c653fc 100644
--- a/auth/src/main/java/org/futo/circles/auth/feature/sign_up/subscription_stage/SubscriptionStageDataSource.kt
+++ b/auth/src/main/java/org/futo/circles/auth/feature/sign_up/subscription_stage/SubscriptionStageDataSource.kt
@@ -1,21 +1,22 @@
 package org.futo.circles.auth.feature.sign_up.subscription_stage
 
-import org.futo.circles.auth.feature.sign_up.SignUpDataSource
-import org.futo.circles.auth.feature.sign_up.SignUpDataSource.Companion.REGISTRATION_SUBSCRIPTION_TYPE
+import org.futo.circles.auth.feature.uia.UIADataSource.Companion.REGISTRATION_SUBSCRIPTION_TYPE
+import org.futo.circles.auth.feature.uia.UIADataSource.Companion.TYPE_PARAM_KEY
+import org.futo.circles.auth.feature.uia.UIADataSourceProvider
 import org.futo.circles.auth.model.SubscriptionReceiptData
 import org.futo.circles.core.extensions.Response
 import org.matrix.android.sdk.api.auth.registration.RegistrationResult
 import org.matrix.android.sdk.api.auth.registration.Stage
 import javax.inject.Inject
 
-class SubscriptionStageDataSource @Inject constructor(
-    private val signUpDataSource: SignUpDataSource
-) {
+class SubscriptionStageDataSource @Inject constructor() {
+
+    private val uiaDataSource = UIADataSourceProvider.getDataSourceOrThrow()
 
     suspend fun validateSubscription(
         subscriptionReceiptData: SubscriptionReceiptData
     ): Response<RegistrationResult> =
-        signUpDataSource.performRegistrationStage(
+        uiaDataSource.performUIAStage(
             mapOf(
                 TYPE_PARAM_KEY to REGISTRATION_SUBSCRIPTION_TYPE,
                 ORDER_ID_KEY to subscriptionReceiptData.orderId,
@@ -25,7 +26,7 @@ class SubscriptionStageDataSource @Inject constructor(
             )
         )
 
-    fun getProductIdsList() = ((signUpDataSource.currentStage as? Stage.Other)
+    fun getProductIdsList() = ((uiaDataSource.currentStage as? Stage.Other)
         ?.params?.get(SUBSCRIPTION_IDS_PARAMS_KEY) as? List<*>)
         ?.map { it.toString() }
         ?: emptyList()
-- 
GitLab