1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use std::collections::BTreeMap;
use std::vec::Vec;
use crate::message::MatchRule;
use crate::Message;
use crate::channel::Token;

pub struct Filters<F> {
    list: BTreeMap<Token, (MatchRule<'static>, F)>,
    nextid: Token,
}


impl<F> Default for Filters<F> {
    fn default() -> Self { Filters {
        list: BTreeMap::new(),
        nextid: Token(1),
    }}
}

impl<F> Filters<F> {
    pub fn add(&mut self, m: MatchRule<'static>, f: F) -> Token {
        let id = self.nextid;
        self.nextid.0 += 1;
        self.list.insert(id, (m, f));
        id
    }

    pub fn insert(&mut self, (t, m, f): (Token, MatchRule<'static>, F)) {
        self.list.insert(t, (m, f));
    }

    pub fn remove(&mut self, id: Token) -> Option<(MatchRule<'static>, F)> {
        self.list.remove(&id)
    }

    /// Removes and returns the first filter which matches the given message.
    pub fn remove_first_matching(&mut self, msg: &Message) -> Option<(Token, MatchRule<'static>, F)> {
        if let Some(k) = self.list.iter_mut().find_map(|(k, v)| if v.0.matches(&msg) { Some(*k) } else { None }) {
            let v = self.list.remove(&k).unwrap();
            Some((k, v.0, v.1))
        } else {
            None
        }
    }

    /// Removes and returns all filters which match the given message.
    pub fn remove_all_matching(&mut self, msg: &Message) -> Vec<(Token, MatchRule<'static>, F)> {
        let matching: Vec<_> = self.list.iter().filter_map(|(k, v)| if v.0.matches(&msg) { Some(*k) } else { None }).collect();
        matching
            .into_iter()
            .map(|k| {
                let v = self.list.remove(&k).unwrap();
                (k, v.0, v.1)
            })
            .collect()
    }
}