Updated on 2026-08-14

2024-06-21 21:33:54 +05:00 · 2024-06-21 21:33:54 +05:00 · 0b671b4ac1
commit 0b671b4ac1
parent 24848624e9
8 changed files with 184 additions and 123 deletions
--- a/core/datasource/src/main/java/com/tangem/datasource/api/common/adapter/UnknownEnumMoshiAdapter.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/api/common/adapter/UnknownEnumMoshiAdapter.kt
@ -1,8 +1,11 @@
 package com.tangem.datasource.api.common.adapter

-import com.squareup.moshi.*
+import com.squareup.moshi.JsonAdapter
+import com.squareup.moshi.Moshi
 import com.squareup.moshi.adapters.EnumJsonAdapter
-import com.tangem.datasource.api.stakekit.models.response.model.*
+import com.tangem.datasource.api.stakekit.models.response.model.BalanceDTO
+import com.tangem.datasource.api.stakekit.models.response.model.NetworkTypeDTO
+import com.tangem.datasource.api.stakekit.models.response.model.YieldDTO
 import com.tangem.datasource.api.stakekit.models.response.model.action.StakingActionStatusDTO
 import com.tangem.datasource.api.stakekit.models.response.model.action.StakingActionTypeDTO
 import com.tangem.datasource.api.stakekit.models.response.model.transaction.StakingTransactionStatusDTO
@ -24,8 +27,7 @@ fun Moshi.Builder.addStakeKitEnumFallbackAdapters(): Moshi.Builder {
        NetworkTypeDTO::class.java to NetworkTypeDTO.UNKNOWN,
        StakingActionTypeDTO::class.java to StakingActionTypeDTO.UNKNOWN,
        YieldDTO.RewardTypeDTO::class.java to YieldDTO.RewardTypeDTO.UNKNOWN,
-        YieldBalanceWrapperDTO.BalanceDTO.BalanceType::class.java to
-            YieldBalanceWrapperDTO.BalanceDTO.BalanceType.UNKNOWN,
+        BalanceDTO.BalanceType::class.java to BalanceDTO.BalanceType.UNKNOWN,
        StakingTransactionTypeDTO::class.java to StakingTransactionTypeDTO.UNKNOWN,
        StakingTransactionStatusDTO::class.java to StakingTransactionStatusDTO.UNKNOWN,
        StakingActionStatusDTO::class.java to StakingActionStatusDTO.UNKNOWN,
--- a/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/StakeKitApi.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/StakeKitApi.kt
@ -4,6 +4,7 @@ import com.tangem.datasource.api.common.response.ApiResponse
 import com.tangem.datasource.api.stakekit.models.request.*
 import com.tangem.datasource.api.stakekit.models.response.EnabledYieldsResponse
 import com.tangem.datasource.api.stakekit.models.response.EnterActionResponse
+import com.tangem.datasource.api.stakekit.models.response.model.BalanceDTO
 import com.tangem.datasource.api.stakekit.models.response.model.transaction.StakingTransactionDTO
 import com.tangem.datasource.api.stakekit.models.response.model.TokenWithYieldDTO
 import com.tangem.datasource.api.stakekit.models.response.model.YieldDTO
@ -29,16 +30,16 @@ interface StakeKitApi {
        @Query("ledgerWalletAPICompatible") ledgerWalletAPICompatible: Boolean = false,
    ): ApiResponse<YieldDTO>

-    @GET("yields/balances")
+    @POST("yields/balances")
    suspend fun getMultipleYieldBalances(
        @Body body: List<YieldBalanceRequestBody>,
    ): ApiResponse<List<YieldBalanceWrapperDTO>>

-    @GET("yields/{integrationId}/balances")
+    @POST("yields/{integrationId}/balances")
    suspend fun getSingleYieldBalance(
        @Path("integrationId") integrationId: String,
        @Body body: YieldBalanceRequestBody,
-    ): ApiResponse<YieldBalanceWrapperDTO>
+    ): ApiResponse<List<BalanceDTO>>

    @GET("tokens")
    suspend fun getTokens(): ApiResponse<List<TokenWithYieldDTO>>
--- a/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/request/EnterActionRequestBody.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/request/EnterActionRequestBody.kt
@ -1,8 +1,8 @@
 package com.tangem.datasource.api.stakekit.models.request

 import com.squareup.moshi.Json
+import com.tangem.datasource.api.stakekit.models.response.model.BalanceDTO
 import com.tangem.datasource.api.stakekit.models.response.model.TokenDTO
-import com.tangem.datasource.api.stakekit.models.response.model.YieldBalanceWrapperDTO

 data class EnterActionRequestBody(
    @Json(name = "integrationId")
@ -27,7 +27,7 @@ data class EnterActionRequestBody(
        @Json(name = "duration")
        val duration: String? = null,
        @Json(name = "nfts")
-        val nfts: List<YieldBalanceWrapperDTO.BalanceDTO.PendingAction.PendingActionArgs.Nft>? = null,
+        val nfts: List<BalanceDTO.PendingAction.PendingActionArgs.Nft>? = null,
        @Json(name = "ledgerWalletAPICompatible")
        val ledgerWalletAPICompatible: Boolean? = null,
        @Json(name = "tronResource")
--- a/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/request/YieldBalanceRequestBody.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/request/YieldBalanceRequestBody.kt
@ -5,7 +5,6 @@ import com.squareup.moshi.Json
 data class YieldBalanceRequestBody(
    @Json(name = "addresses") val addresses: Address,
    @Json(name = "args") val args: YieldBalanceRequestArgs,
-    @Json(name = "integrationId") val integrationId: String? = null,
 ) {

    data class YieldBalanceRequestArgs(
--- a/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/response/model/YieldBalanceWrapperDTO.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/api/stakekit/models/response/model/YieldBalanceWrapperDTO.kt
@ -12,130 +12,129 @@ data class YieldBalanceWrapperDTO(
    val balances: List<BalanceDTO>,
    @Json(name = "integrationId")
    val integrationId: String?,
+)
+
+@JsonClass(generateAdapter = true)
+data class BalanceDTO(
+    @Json(name = "groupId")
+    val groupId: String,
+    @Json(name = "type")
+    val type: BalanceType,
+    @Json(name = "amount")
+    val amount: BigDecimal,
+    @Json(name = "date")
+    val date: DateTime?,
+    @Json(name = "pricePerShare")
+    val pricePerShare: BigDecimal,
+    @Json(name = "pendingActions")
+    val pendingActions: List<PendingAction>,
+    @Json(name = "token")
+    val tokenDTO: TokenDTO,
+    @Json(name = "validatorAddress")
+    val validatorAddress: String?,
+    @Json(name = "validatorAddresses")
+    val validatorAddresses: List<String>?,
+    @Json(name = "providerId")
+    val providerId: String?,
 ) {

+    enum class BalanceType {
+        @Json(name = "available")
+        AVAILABLE,
+
+        @Json(name = "staked")
+        STAKED,
+
+        @Json(name = "unstaking")
+        UNSTAKING,
+
+        @Json(name = "unstaked")
+        UNSTAKED,
+
+        @Json(name = "preparing")
+        PREPARING,
+
+        @Json(name = "rewards")
+        REWARDS,
+
+        @Json(name = "locked")
+        LOCKED,
+
+        @Json(name = "unlocking")
+        UNLOCKING,
+
+        UNKNOWN,
+    }
+
    @JsonClass(generateAdapter = true)
-    data class BalanceDTO(
-        @Json(name = "groupId")
-        val groupId: String,
+    data class PendingAction(
        @Json(name = "type")
-        val type: BalanceType,
-        @Json(name = "amount")
-        val amount: BigDecimal,
-        @Json(name = "date")
-        val date: DateTime?,
-        @Json(name = "pricePerShare")
-        val pricePerShare: BigDecimal,
-        @Json(name = "pendingActions")
-        val pendingActions: List<PendingAction>,
-        @Json(name = "token")
-        val tokenDTO: TokenDTO,
-        @Json(name = "validatorAddress")
-        val validatorAddress: String?,
-        @Json(name = "validatorAddresses")
-        val validatorAddresses: List<String>?,
-        @Json(name = "providerId")
-        val providerId: String?,
+        val type: StakingActionTypeDTO,
+        @Json(name = "passthrough")
+        val passthrough: String,
+        @Json(name = "args")
+        val args: PendingActionArgs?,
    ) {
-
-        enum class BalanceType {
-            @Json(name = "available")
-            AVAILABLE,
-
-            @Json(name = "staked")
-            STAKED,
-
-            @Json(name = "unstaking")
-            UNSTAKING,
-
-            @Json(name = "unstaked")
-            UNSTAKED,
-
-            @Json(name = "preparing")
-            PREPARING,
-
-            @Json(name = "rewards")
-            REWARDS,
-
-            @Json(name = "locked")
-            LOCKED,
-
-            @Json(name = "unlocking")
-            UNLOCKING,
-
-            UNKNOWN,
-        }
-
        @JsonClass(generateAdapter = true)
-        data class PendingAction(
-            @Json(name = "type")
-            val type: StakingActionTypeDTO,
-            @Json(name = "passthrough")
-            val passthrough: String,
-            @Json(name = "args")
-            val args: PendingActionArgs?,
+        data class PendingActionArgs(
+            @Json(name = "amount")
+            val amount: Amount?,
+            @Json(name = "duration")
+            val duration: Duration?,
+            @Json(name = "validatorAddress")
+            val validatorAddress: Required?,
+            @Json(name = "validatorAddresses")
+            val validatorAddresses: Required?,
+            @Json(name = "nfts")
+            val nfts: List<Nft>?,
+            @Json(name = "tronResource")
+            val tronResource: TronResource?,
+            @Json(name = "signatureVerification")
+            val signatureVerification: Required?,
        ) {
            @JsonClass(generateAdapter = true)
-            data class PendingActionArgs(
-                @Json(name = "amount")
-                val amount: Amount?,
-                @Json(name = "duration")
-                val duration: Duration?,
-                @Json(name = "validatorAddress")
-                val validatorAddress: Required?,
-                @Json(name = "validatorAddresses")
-                val validatorAddresses: Required?,
-                @Json(name = "nfts")
-                val nfts: List<Nft>?,
-                @Json(name = "tronResource")
-                val tronResource: TronResource?,
-                @Json(name = "signatureVerification")
-                val signatureVerification: Required?,
-            ) {
-                @JsonClass(generateAdapter = true)
-                data class Amount(
-                    @Json(name = "required")
-                    val required: Boolean,
-                    @Json(name = "minimum")
-                    val minimum: BigDecimal?,
-                    @Json(name = "maximum")
-                    val maximum: BigDecimal?,
-                )
+            data class Amount(
+                @Json(name = "required")
+                val required: Boolean,
+                @Json(name = "minimum")
+                val minimum: BigDecimal?,
+                @Json(name = "maximum")
+                val maximum: BigDecimal?,
+            )

-                @JsonClass(generateAdapter = true)
-                data class Duration(
-                    @Json(name = "required")
-                    val required: Boolean,
-                    @Json(name = "minimum")
-                    val minimum: Int?,
-                    @Json(name = "maximum")
-                    val maximum: Int?,
-                )
+            @JsonClass(generateAdapter = true)
+            data class Duration(
+                @Json(name = "required")
+                val required: Boolean,
+                @Json(name = "minimum")
+                val minimum: Int?,
+                @Json(name = "maximum")
+                val maximum: Int?,
+            )

-                @JsonClass(generateAdapter = true)
-                data class Nft(
-                    @Json(name = "baycId")
-                    val baycId: Required?,
-                    @Json(name = "maycId")
-                    val maycId: Required?,
-                    @Json(name = "bakcId")
-                    val bakcId: Required?,
-                )
+            @JsonClass(generateAdapter = true)
+            data class Nft(
+                @Json(name = "baycId")
+                val baycId: Required?,
+                @Json(name = "maycId")
+                val maycId: Required?,
+                @Json(name = "bakcId")
+                val bakcId: Required?,
+            )

-                @JsonClass(generateAdapter = true)
-                data class TronResource(
-                    @Json(name = "required")
-                    val required: Boolean,
-                    @Json(name = "options")
-                    val options: List<String>,
-                )
-            }
+            @JsonClass(generateAdapter = true)
+            data class TronResource(
+                @Json(name = "required")
+                val required: Boolean,
+                @Json(name = "options")
+                val options: List<String>,
+            )
        }
-
-        @JsonClass(generateAdapter = true)
-        data class Required(
-            @Json(name = "required")
-            val required: Boolean,
-        )
    }
+
+    @JsonClass(generateAdapter = true)
+    data class Required(
+        @Json(name = "required")
+        val required: Boolean,
+    )
 }
--- a/core/datasource/src/main/java/com/tangem/datasource/di/StakingBalanceStoreModule.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/di/StakingBalanceStoreModule.kt
@ -0,0 +1,21 @@
+package com.tangem.datasource.di
+
+import com.tangem.datasource.local.datastore.RuntimeDataStore
+import com.tangem.datasource.local.token.DefaultStakingBalanceStore
+import com.tangem.datasource.local.token.StakingBalanceStore
+import dagger.Module
+import dagger.Provides
+import dagger.hilt.InstallIn
+import dagger.hilt.components.SingletonComponent
+import javax.inject.Singleton
+
+@Module
+@InstallIn(SingletonComponent::class)
+internal object StakingBalanceStoreModule {
+
+    @Provides
+    @Singleton
+    fun provideStakingBalanceStore(): StakingBalanceStore {
+        return DefaultStakingBalanceStore(dataStore = RuntimeDataStore())
+    }
+}
--- a/core/datasource/src/main/java/com/tangem/datasource/local/token/DefaultStakingBalanceStore.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/local/token/DefaultStakingBalanceStore.kt
@ -0,0 +1,26 @@
+package com.tangem.datasource.local.token
+
+import com.tangem.datasource.api.stakekit.models.response.model.BalanceDTO
+import com.tangem.datasource.local.datastore.core.StringKeyDataStore
+import kotlinx.coroutines.flow.Flow
+
+internal class DefaultStakingBalanceStore(
+    private val dataStore: StringKeyDataStore<List<BalanceDTO>>,
+) : StakingBalanceStore {
+
+    override fun get(): Flow<List<BalanceDTO>> {
+        return dataStore.get(STAKING_BALANCE_KEY)
+    }
+
+    override suspend fun getSyncOrNull(): List<BalanceDTO>? {
+        return dataStore.getSyncOrNull(STAKING_BALANCE_KEY)
+    }
+
+    override suspend fun store(items: List<BalanceDTO>) {
+        return dataStore.store(STAKING_BALANCE_KEY, items)
+    }
+
+    companion object {
+        private const val STAKING_BALANCE_KEY = "STAKING_BALANCE_KEY"
+    }
+}
--- a/core/datasource/src/main/java/com/tangem/datasource/local/token/StakingBalanceStore.kt
+++ b/core/datasource/src/main/java/com/tangem/datasource/local/token/StakingBalanceStore.kt
@ -0,0 +1,13 @@
+package com.tangem.datasource.local.token
+
+import com.tangem.datasource.api.stakekit.models.response.model.BalanceDTO
+import kotlinx.coroutines.flow.Flow
+
+interface StakingBalanceStore {
+
+    fun get(): Flow<List<BalanceDTO>>
+
+    suspend fun getSyncOrNull(): List<BalanceDTO>?
+
+    suspend fun store(items: List<BalanceDTO>)
+}