Merge pull request invariant-labs#20 from invariant-labs/include-chec…

…king-outdated-state

Add check if ticks accounts are outdated in quote

.github/workflows/test.yml

@@ -20,13 +20,13 @@ jobs:
           cargo build --verbose
           cargo test --verbose
 
-  rust-nightly:
-    runs-on: ubuntu-latest
-    container: rustlang/rust:nightly
-    steps:
-      - name: Checkout code
-        uses: actions/checkout@v2
-      - name: Build and test with Rust nightly
-        run: |
-          cargo build --verbose
-          cargo test --verbose
+  # rust-nightly:
+  #   runs-on: ubuntu-latest
+  #   container: rustlang/rust:nightly
+  #   steps:
+  #     - name: Checkout code
+  #       uses: actions/checkout@v2
+  #     - name: Build and test with Rust nightly
+  #       run: |
+  #         cargo build --verbose
+  #         cargo test --verbose

src/lib.rs

@@ -90,6 +90,14 @@ impl Amm for JupiterInvariant {
     }
 
     fn quote(&self, quote_params: &QuoteParams) -> anyhow::Result<Quote> {
+        let accounts_outdated = self.ticks_accounts_outdated();
+        if accounts_outdated {
+            return Ok(Quote {
+                not_enough_liquidity: true,
+                ..Quote::default()
+            });
+        }
+
         let invariant_simulation_params = self.quote_to_invariant_params(quote_params)?;
         let simulation_result = self.simulate_invariant_swap(&invariant_simulation_params);
 

src/swap.rs

@@ -107,7 +107,7 @@ impl JupiterInvariant {
             by_amount_in,
         } = *invariant_simulation_params;
 
-        let (mut pool, ticks, tickmap, starting_sqrt_price) = (
+        let (pool, ticks, tickmap, starting_sqrt_price) = (
             &mut self.pool.clone(),
             &self.ticks.clone(),
             &self.tickmap.clone(),