Filtered withdrawal assets by cctp.json (#763)

dydxprotocol · Dec 3, 2024 · 7c5b470 · 7c5b470
1 parent 629fc6f
commit 7c5b470
Show file tree

Hide file tree

Showing 7 changed files with 32 additions and 6 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -52,7 +52,7 @@ allprojects {
 }
 
 group = "exchange.dydx.abacus"
-version = "1.13.37"
+version = "1.13.38"
 
 repositories {
     google()

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/processor/input/TransferInputProcessor.kt b/src/commonMain/kotlin/exchange.dydx.abacus/processor/input/TransferInputProcessor.kt
@@ -32,7 +32,7 @@ internal class TransferInputProcessor(
     private val parser: ParserProtocol,
     private val calculator: TransferInputCalculatorV2 = TransferInputCalculatorV2(parser = parser),
     private val routerProcessor: IRouterProcessor,
-    private val environment: V4Environment?
+    private val environment: V4Environment?,
 ) {
     fun transfer(
         inputState: InternalInputState,
@@ -250,8 +250,16 @@ internal class TransferInputProcessor(
     private fun updateWithdrawalOptions(
         transfer: InternalTransferInputState,
     ) {
-        val chains: IList<SelectionOption> = transfer.chains?.toIList() ?: iListOf()
-        val assets: IList<SelectionOption> = transfer.tokens?.toIList() ?: iListOf()
+        val chains: IList<SelectionOption> = if (environment?.isMainNet == true && environment?.featureFlags?.cctpWithdrawalOnly == true) {
+            transfer.cctpChains?.toIList() ?: iListOf()
+        } else {
+            transfer.chains?.toIList() ?: iListOf()
+        }
+        val assets: IList<SelectionOption> = if (environment?.isMainNet == true && environment?.featureFlags?.cctpWithdrawalOnly == true) {
+            transfer.cctpTokens?.toIList() ?: iListOf()
+        } else {
+            transfer.tokens?.toIList() ?: iListOf()
+        }
         var exchanges: IMutableList<SelectionOption>? = null
         exchangeList?.let { data ->
             exchanges = iMutableListOf()

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/processor/router/skip/SkipProcessor.kt b/src/commonMain/kotlin/exchange.dydx.abacus/processor/router/skip/SkipProcessor.kt
@@ -47,6 +47,7 @@ internal class SkipProcessor(
     ): Map<String, Any>? {
         throw NotImplementedError("receivedV2SdkInfo is not implemented in SkipProcessor!")
     }
+
     override fun receivedChains(
         existing: Map<String, Any>?,
         payload: Map<String, Any>

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/internalstate/InternalTransferInputState.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/internalstate/InternalTransferInputState.kt
@@ -10,6 +10,7 @@ import exchange.dydx.abacus.output.input.TransferInputTokenResource
 import exchange.dydx.abacus.output.input.TransferOutInputOptions
 import exchange.dydx.abacus.output.input.TransferType
 import exchange.dydx.abacus.output.input.WithdrawalInputOptions
+import exchange.dydx.abacus.state.manager.CctpConfig
 
 internal data class InternalTransferInputState(
     var chains: List<SelectionOption>? = null,
@@ -33,7 +34,19 @@ internal data class InternalTransferInputState(
     var summary: TransferInputSummary? = null,
     var resources: TransferInputResources? = null,
     var route: Map<String, Any>? = null,
-)
+) {
+    val cctpChains: List<SelectionOption>? // chains filtered by cctp.json
+        get() =
+            chains?.filter { chain ->
+                CctpConfig.cctpChainIds?.any { it.chainId == chain.type } == true
+            }
+
+    val cctpTokens: List<SelectionOption>? // tokens filtered by cctp.json
+        get() =
+            tokens?.filter { token ->
+                CctpConfig.cctpChainIds?.any { it.tokenAddress.lowercase() == token.type.lowercase() } == true
+            }
+}
 
 internal fun TransferInputSize.Companion.safeCreate(existing: TransferInputSize?): TransferInputSize {
     return existing ?: TransferInputSize(

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/manager/Environment.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/manager/Environment.kt
@@ -138,6 +138,7 @@ data class EnvironmentLinks(
 data class EnvironmentFeatureFlags(
     val withdrawalSafetyEnabled: Boolean,
     val isSlTpLimitOrdersEnabled: Boolean,
+    val cctpWithdrawalOnly: Boolean,
 ) {
     companion object {
         fun parse(
@@ -146,10 +147,12 @@ data class EnvironmentFeatureFlags(
         ): EnvironmentFeatureFlags {
             val withdrawalSafetyEnabled = parser.asBool(data?.get("withdrawalSafetyEnabled")) ?: false
             val isSlTpLimitOrdersEnabled = parser.asBool(data?.get("isSlTpLimitOrdersEnabled")) ?: false
+            val cctpWithdrawalOnly = parser.asBool(data?.get("CCTPWithdrawalOnly")) ?: false
 
             return EnvironmentFeatureFlags(
                 withdrawalSafetyEnabled,
                 isSlTpLimitOrdersEnabled,
+                cctpWithdrawalOnly,
             )
         }
     }

diff --git a/src/commonTest/kotlin/exchange.dydx.abacus/tests/payloads/AbacusMockData.kt b/src/commonTest/kotlin/exchange.dydx.abacus/tests/payloads/AbacusMockData.kt
@@ -132,6 +132,7 @@ class AbacusMockData {
         featureFlags = EnvironmentFeatureFlags(
             withdrawalSafetyEnabled = true,
             isSlTpLimitOrdersEnabled = true,
+            cctpWithdrawalOnly = true,
         ),
     )
 }
diff --git a/v4_abacus.podspec b/v4_abacus.podspec
@@ -1,6 +1,6 @@
 Pod::Spec.new do |spec|
     spec.name                     = 'v4_abacus'
-    spec.version                  = '1.13.37'
+    spec.version                  = '1.13.38'
     spec.homepage                 = 'https://github.com/dydxprotocol/v4-abacus'
     spec.source                   = { :http=> ''}
     spec.authors                  = ''