hq/Noteguard
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

filter: add whitelist filter

Browse source 
Fixes: https://github.com/damus-io/noteguard/issues/3
Signed-off-by: William Casarin <jb55@jb55.com>
This commit is contained in:
William Casarin 2024-07-09 09:44:31 -07:00
parent bae87107ae
commit 74c90453b8
5 changed files with 51 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/filters/whitelist.rs
View file

@ -1,23 +1,31 @@
use crate::{Action, InputMessage, NoteFilter, OutputMessage};
use serde::Deserialize;
#[derive(Deserialize)]
#[derive(Deserialize, Default)]
pub struct Whitelist {
pub pubkeys: Vec<String>,
pub ips: Vec<String>,
pub pubkeys: Option<Vec<String>>,
pub ips: Option<Vec<String>>,
}
impl NoteFilter for Whitelist {
fn filter_note(&mut self, msg: &InputMessage) -> OutputMessage {
if self.pubkeys.contains(&msg.event.pubkey) || self.ips.contains(&msg.source_info) {
OutputMessage::new(msg.event.id.clone(), Action::Accept, None)
} else {
OutputMessage::new(
msg.event.id.clone(),
Action::Reject,
Some("blocked: pubkey not on the whitelist".to_string()),
)
if let Some(pubkeys) = &self.pubkeys {
if pubkeys.contains(&msg.event.pubkey) {
return OutputMessage::new(msg.event.id.clone(), Action::Accept, None);
}
}
if let Some(ips) = &self.ips {
if ips.contains(&msg.source_info) {
return OutputMessage::new(msg.event.id.clone(), Action::Accept, None);
}
}
OutputMessage::new(
msg.event.id.clone(),
Action::Reject,
Some("blocked: pubkey/ip not on the whitelist".to_string()),
)
}
fn name(&self) -> &'static str {