disable everything else than deposit sol when not ignoring limitation

jup-ag · Mar 14, 2024 · 7ff1a53 · 7ff1a53
1 parent 942aded
commit 7ff1a53
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stakedex_sdk/src/lib.rs b/stakedex_sdk/src/lib.rs
@@ -438,7 +438,7 @@ impl Stakedex {
     }
 
     /// Creates all possible Amms from the underlying available Stakedexes
-    pub fn get_amms(self) -> Vec<Box<dyn Amm + Send + Sync>> {
+    pub fn get_amms(self, ignore_cpi_limitation: bool) -> Vec<Box<dyn Amm + Send + Sync>> {
         #[derive(Clone)]
         enum Stakedex {
             SplStakePool(SplStakePoolStakedex),
@@ -479,6 +479,10 @@ impl Stakedex {
             }
         }
 
+        if !ignore_cpi_limitation {
+            return amms;
+        }
+
         // SplStakePool WithdrawStake + DepositStake
         // UnstakeIt DepositStake
         // Marinade DepositStake