blockworks-foundation
/
autobahn
mirror of https://github.com/blockworks-foundation/autobahn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add filter for token 2022

This commit is contained in:
none00y 2024-11-06 19:26:37 +01:00
parent 55e2633329
commit ff11105459
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/dex-invariant/src/invariant_dex.rs
View File

@ -4,7 +4,13 @@ use std::{
};
use anchor_lang::{AnchorDeserialize, Id};
use anchor_spl::{token::spl_token::{self, state::AccountState}, token_2022::Token2022};
use anchor_spl::{
token::{
self,
spl_token::{self, state::AccountState},
},
token_2022::{self, spl_token_2022, Token2022},
};
use anyhow::{Context, Ok};
use async_trait::async_trait;
use invariant_types::{
@ -242,11 +248,20 @@ impl DexInterface for InvariantDex {
let vaults = rpc.get_multiple_accounts(&reserves).await?;
let banned_reserves = vaults
.iter()
.filter(|(_, reserve)| {
spl_token::state::Account::unpack(reserve.data())
.filter(|(_, reserve)| match reserve.owner {
token_2022::ID => {
spl_token_2022::state::Account::unpack(reserve.data())
.unwrap()
.state
== spl_token_2022::state::AccountState::Frozen
}
token::ID => {
spl_token::state::Account::unpack(reserve.data())
.unwrap()
.state
== AccountState::Frozen
}
_ => true,
})
.map(|(pk, _)| pk)
.collect::<HashSet<_>>();