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

use oura::filters::selection::{Config, Predicate};
use plutus_ledger_api::v2::{script::MintingPolicyHash, value::CurrencySymbol};
use tx_bakery::csl;
use tx_bakery::utils::pla_to_csl::TryFromPLAWithDef;

/// Interesting transaction components to look for when filtering transactions
/// relevant to the protocol.
/// Set curr_symbols to empty vectors to handle any transaction event indiscriminately.
pub struct Filter {
    pub curr_symbols: Vec<CurrencySymbol>,
}

// We only obtain Transaction events that contain the policy in the output
// NOTE: Must enable 'include_transaction_details' in oura Mapper config.
impl Filter {
    pub fn to_selection_config(self) -> Config {
        Config {
            check: Predicate::AnyOf(vec![
                Predicate::VariantIn(vec!["RollBack".to_string(), "Block".to_string()]),
                Predicate::AllOf(vec![
                    Predicate::VariantIn(vec!["Transaction".to_string()]),
                    if self.curr_symbols.is_empty() {
                        ALWAYS_TRUE
                    } else {
                        Predicate::AnyOf(
                            self.curr_symbols
                                .into_iter()
                                .map(serialize_cur_sym)
                                .map(Predicate::PolicyEquals)
                                .collect(),
                        )
                    },
                ]),
            ]),
        }
    }
}

// Filter predicate that always succeeds.
const ALWAYS_TRUE: Predicate = Predicate::AllOf(vec![]);

fn serialize_cur_sym(cur_sym: CurrencySymbol) -> String {
    match cur_sym {
        CurrencySymbol::Ada => String::new(),
        CurrencySymbol::NativeToken(MintingPolicyHash(script_hash)) => {
            csl::ScriptHash::try_from_pla(&script_hash)
                .unwrap()
                .to_hex()
        }
    }
}