bgpfu-netconf 0.1.0

A toolset for working with IRR data
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

use std::{fmt::Debug, io::Write, marker::PhantomData};

use quick_xml::Writer;

use crate::{
    capabilities::Requirements,
    message::{ReadXml, WriteError},
    session::Context,
    Error,
};

use super::{params::Required, DataReply, Datastore, Filter, Operation, WriteXml};

#[derive(Debug, Clone)]
pub struct GetConfig<D> {
    source: Datastore,
    filter: Option<Filter>,
    _reply: PhantomData<D>,
}

impl<D> Default for GetConfig<D> {
    fn default() -> Self {
        Self {
            source: Datastore::default(),
            filter: None,
            _reply: PhantomData,
        }
    }
}

impl<D> Operation for GetConfig<D>
where
    D: ReadXml + Debug + Send + Sync,
{
    const NAME: &'static str = "get-config";
    const REQUIRED_CAPABILITIES: Requirements = Requirements::None;

    type Builder<'a> = Builder<'a>;
    type Reply = DataReply<D>;
}

impl<D> WriteXml for GetConfig<D>
where
    D: ReadXml + Debug + Send + Sync,
{
    fn write_xml<W: Write>(&self, writer: &mut Writer<W>) -> Result<(), WriteError> {
        writer
            .create_element(Self::NAME)
            .write_inner_content(|writer| {
                _ = writer
                    .create_element("source")
                    .write_inner_content(|writer| self.source.write_xml(writer))?;
                if let Some(ref filter) = self.filter {
                    filter.write_xml(writer)?;
                };
                Ok(())
            })
            .map(|_| ())
    }
}

#[derive(Debug, Clone)]
#[must_use]
pub struct Builder<'a> {
    ctx: &'a Context,
    source: Required<Datastore>,
    filter: Option<Filter>,
}

impl Builder<'_> {
    pub fn source(mut self, source: Datastore) -> Result<Self, Error> {
        source.try_as_source(self.ctx).map(|source| {
            self.source.set(source);
            self
        })
    }

    pub fn filter(mut self, filter: Option<Filter>) -> Result<Self, Error> {
        self.filter = filter.map(|filter| filter.try_use(self.ctx)).transpose()?;
        Ok(self)
    }
}

impl<'a, D> super::Builder<'a, GetConfig<D>> for Builder<'a>
where
    D: ReadXml + Debug + Send + Sync,
{
    fn new(ctx: &'a Context) -> Self {
        Self {
            ctx,
            source: Required::init(),
            filter: None,
        }
    }

    fn finish(self) -> Result<GetConfig<D>, Error> {
        Ok(GetConfig {
            source: self.source.require::<GetConfig<D>>("source")?,
            filter: self.filter,
            _reply: PhantomData,
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::message::{
        rpc::{operation::Opaque, MessageId, Request},
        ClientMsg,
    };

    #[test]
    fn default_request_to_xml() {
        let req: Request<GetConfig<Opaque>> = Request {
            message_id: MessageId(101),
            operation: GetConfig::default(),
        };
        let expect = r#"<rpc message-id="101"><get-config><source><running/></source></get-config></rpc>]]>]]>"#;
        assert_eq!(req.to_xml().unwrap(), expect);
    }
}