teliox 0.17.13

Library for the Transaction Event Log
Documentation 
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

use std::sync::Arc;

use keri_core::prefix::IdentifierPrefix;

use crate::{
    database::TelEventDatabase,
    error::Error,
    event::{verifiable_event::VerifiableEvent, Event},
    query::TelQueryRoute,
    state::{vc_state::TelState, ManagerTelState},
};

use super::TelReplyType;

pub struct TelEventStorage<D: TelEventDatabase> {
    pub db: Arc<D>,
}
impl<D: TelEventDatabase> TelEventStorage<D> {
    pub fn new(db: Arc<D>) -> Self {
        Self { db }
    }

    pub fn compute_management_tel_state(
        &self,
        id: &IdentifierPrefix,
    ) -> Result<Option<ManagerTelState>, Error> {
        self.db
            .get_management_events(id)
            .map(|events| {
                events.fold(
                    Ok(ManagerTelState::default()),
                    |state: Result<ManagerTelState, Error>,
                     ev: VerifiableEvent|
                     -> Result<ManagerTelState, Error> {
                        match ev.event {
                            Event::Management(event) => state?.apply(&event),
                            Event::Vc(_) => Err(Error::Generic("Improper event type".into())),
                        }
                    },
                )
            })
            .transpose()
    }

    pub fn compute_vc_state(&self, vc_id: &IdentifierPrefix) -> Result<Option<TelState>, Error> {
        self.db
            .get_events(vc_id)
            .map(|events| {
                events.into_iter().fold(
                    Ok(TelState::default()),
                    |state, ev| -> Result<TelState, Error> {
                        match ev.event {
                            Event::Vc(event) => state?.apply(&event),
                            _ => state,
                        }
                    },
                )
            })
            .transpose()
    }

    pub fn get_management_events(&self, id: &IdentifierPrefix) -> Result<Option<Vec<u8>>, Error> {
        match self.db.get_management_events(id) {
            Some(events) => Ok(Some(
                events
                    .map(|event| event.serialize().unwrap_or_default())
                    .fold(vec![], |mut accum, serialized_event| {
                        accum.extend(serialized_event);
                        accum
                    }),
            )),
            None => Ok(None),
        }
    }

    pub fn get_events(&self, vc_id: &IdentifierPrefix) -> Result<Vec<VerifiableEvent>, Error> {
        match self.db.get_events(vc_id) {
            Some(events) => Ok(events.collect()),
            None => Ok(vec![]),
        }
    }

    pub fn get_management_event_at_sn(
        &self,
        id: &IdentifierPrefix,
        sn: u64,
    ) -> Result<Option<VerifiableEvent>, Error> {
        match self.db.get_management_events(id) {
            Some(mut events) => Ok(events.find(|event| {
                if let Event::Management(man) = &event.event {
                    man.data.sn == sn
                } else {
                    false
                }
            })),
            None => Ok(None),
        }
    }

    pub fn add_event(&self, event: VerifiableEvent) -> Result<(), Error> {
        self.db
            .add_new_event(event.clone(), &event.get_event().get_prefix())
    }

    pub fn process_query(&self, qry: &TelQueryRoute) -> Result<TelReplyType, Error> {
        match qry {
            TelQueryRoute::Tels {
                reply_route: _,
                args,
            } => {
                let management_tel = args
                    .ri
                    .as_ref()
                    .map(|ri| self.get_management_events(ri))
                    .transpose()?
                    .unwrap_or_default()
                    .unwrap_or_default();
                if let Some(vc_id) = &args.i {
                    let vc_tel = self
                        .get_events(&vc_id)?
                        .into_iter()
                        .flat_map(|event| event.serialize().unwrap());
                    Ok(TelReplyType::Tel(
                        management_tel.into_iter().chain(vc_tel).collect(),
                    ))
                } else {
                    Ok(TelReplyType::Tel(management_tel))
                }
            }
        }
    }
}