-
Notifications
You must be signed in to change notification settings - Fork 88
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
16 changed files
with
189 additions
and
37 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,147 @@ | ||
| use ahash::{HashSet, HashSetExt}; | ||
| use alloy_primitives::Address; | ||
| use serde::{Deserialize, Deserializer}; | ||
| use std::fs::read_to_string; | ||
| use std::path::PathBuf; | ||
|
|
||
| #[derive(Debug, Clone, PartialEq, Eq, Default)] | ||
| pub struct BlockList { | ||
| list: HashSet<Address>, | ||
| } | ||
|
|
||
| impl BlockList { | ||
| fn new() -> Self { | ||
| Self { | ||
| list: HashSet::new(), | ||
| } | ||
| } | ||
|
|
||
| fn from_file(path: PathBuf) -> eyre::Result<Self> { | ||
| let blocklist_file = read_to_string(path)?; | ||
| let blocklist: Vec<Address> = serde_json::from_str(&blocklist_file)?; | ||
|
|
||
| Ok(Self { | ||
| list: blocklist.into_iter().collect(), | ||
| }) | ||
| } | ||
|
|
||
| pub fn len(&self) -> usize { | ||
| self.list.len() | ||
| } | ||
|
|
||
| pub fn contains(&self, address: &Address) -> bool { | ||
| self.list.contains(address) | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| fn add(&mut self, address: Address) { | ||
| self.list.insert(address); | ||
| } | ||
| } | ||
|
|
||
| impl From<PathBuf> for BlockList { | ||
| fn from(path: PathBuf) -> Self { | ||
| Self::from_file(path).unwrap_or_else(|_| Self::new()) | ||
| } | ||
| } | ||
|
|
||
| impl From<Vec<Address>> for BlockList { | ||
| fn from(addresses: Vec<Address>) -> Self { | ||
| Self { | ||
| list: addresses.into_iter().collect(), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl<'de> Deserialize<'de> for BlockList { | ||
| fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> | ||
| where | ||
| D: Deserializer<'de>, | ||
| { | ||
| let path: Option<PathBuf> = Option::deserialize(deserializer)?; | ||
|
|
||
| match path { | ||
| Some(path) => BlockList::from_file(path).map_err(serde::de::Error::custom), | ||
| None => Ok(BlockList::new()), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| mod test { | ||
| use super::*; | ||
| use alloy_primitives::{address, Address}; | ||
| use serde::Deserialize; | ||
|
|
||
| #[test] | ||
| fn test_read_blocklist_from_file() { | ||
| let block_list = BlockList::from_file( | ||
| PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("src/blocklist/testdata/blocklist.txt"), | ||
| ) | ||
| .unwrap(); | ||
|
|
||
| let addr0 = address!("14dC79964da2C08b23698B3D3cc7Ca32193d9955"); | ||
| assert_eq!(block_list.contains(&addr0), true); | ||
|
|
||
| let addr1 = address!("f39Fd6e51aad88F6F4ce6aB8827279cffFb92266"); | ||
| assert_eq!(block_list.contains(&addr1), true); | ||
|
|
||
| let addr2 = address!("a0Ee7A142d267C1f36714E4a8F75612F20a79720"); | ||
| assert_eq!(block_list.contains(&addr2), false); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_blocklist() { | ||
| let mut blocklist = BlockList::new(); | ||
| let addr0 = Address::random(); | ||
|
|
||
| blocklist.add(addr0); | ||
| assert_eq!(blocklist.len(), 1); | ||
| assert_eq!(blocklist.contains(&addr0), true); | ||
|
|
||
| // you cannot add twice the same value | ||
| blocklist.add(addr0); | ||
| assert_eq!(blocklist.len(), 1); | ||
|
|
||
| let addr1 = Address::random(); | ||
| assert_eq!(blocklist.contains(&addr1), false); | ||
|
|
||
| blocklist.add(addr1); | ||
| assert_eq!(blocklist.len(), 2); | ||
| assert_eq!(blocklist.contains(&addr1), true); | ||
| } | ||
|
|
||
| #[derive(Deserialize)] | ||
| struct Config { | ||
| block_list: BlockList, | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_deserialize_config() { | ||
| let config_str = r#" | ||
| block_list = "src/blocklist/testdata/blocklist.txt" | ||
| "#; | ||
| let config: Config = toml::from_str(config_str).unwrap(); | ||
| assert_eq!(config.block_list.len(), 3); | ||
|
|
||
| let addr1 = address!("f39Fd6e51aad88F6F4ce6aB8827279cffFb92266"); | ||
| assert_eq!(config.block_list.contains(&addr1), true); | ||
|
|
||
| let empty_config_str = r#""#; | ||
| let config: Config = toml::from_str(empty_config_str).unwrap(); | ||
| assert_eq!(config.block_list.len(), 0); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_from_vec() { | ||
| let addr0 = address!("14dC79964da2C08b23698B3D3cc7Ca32193d9955"); | ||
| let addr1 = address!("f39Fd6e51aad88F6F4ce6aB8827279cffFb92266"); | ||
|
|
||
| let addresses = vec![addr0, addr1]; | ||
| let blocklist = BlockList::from(addresses); | ||
|
|
||
| assert_eq!(blocklist.len(), 2); | ||
| assert_eq!(blocklist.contains(&addr0), true); | ||
| assert_eq!(blocklist.contains(&addr1), true); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| [ | ||
| "0x14dC79964da2C08b23698B3D3cc7Ca32193d9955", | ||
| "0x3C44CdDdB6a900fa2b585dd299e03d12FA4293BC", | ||
| "f39Fd6e51aad88F6F4ce6aB8827279cffFb92266" | ||
| ] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.