bgpfu-netconf 0.1.0

A toolset for working with IRR data
Documentation
use std::io::Write;

use quick_xml::Writer;

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

use super::{params::Required, Datastore, EmptyReply, Operation, Url, WriteXml};

#[derive(Debug, Clone)]
pub struct DeleteConfig {
    target: Target,
}

impl Operation for DeleteConfig {
    const NAME: &'static str = "delete-config";
    const REQUIRED_CAPABILITIES: Requirements = Requirements::None;

    type Builder<'a> = Builder<'a>;
    type Reply = EmptyReply;
}

impl WriteXml for DeleteConfig {
    fn write_xml<W: Write>(&self, writer: &mut Writer<W>) -> Result<(), WriteError> {
        writer
            .create_element(Self::NAME)
            .write_inner_content(|writer| {
                writer
                    .create_element("target")
                    .write_inner_content(|writer| self.target.write_xml(writer))
                    .map(|_| ())
            })
            .map(|_| ())
    }
}

#[derive(Debug, Clone)]
#[must_use]
pub struct Builder<'a> {
    ctx: &'a Context,
    target: Required<Target>,
}

impl Builder<'_> {
    pub fn target(mut self, target: Datastore) -> Result<Self, Error> {
        if matches!(target, Datastore::Running) {
            return Err(Error::DeleteRunningConfig);
        };
        target.try_as_target(self.ctx).map(|target| {
            self.target.set(Target::Datastore(target));
            self
        })
    }

    pub fn url<S: AsRef<str>>(mut self, url: S) -> Result<Self, Error> {
        Url::try_new(url, self.ctx).map(|url| {
            self.target.set(Target::Url(url));
            self
        })
    }
}

impl<'a> super::Builder<'a, DeleteConfig> for Builder<'a> {
    fn new(ctx: &'a Context) -> Self {
        Self {
            ctx,
            target: Required::init(),
        }
    }

    fn finish(self) -> Result<DeleteConfig, Error> {
        Ok(DeleteConfig {
            target: self.target.require::<DeleteConfig>("target")?,
        })
    }
}

#[derive(Debug, Clone)]
enum Target {
    Datastore(Datastore),
    Url(Url),
}

impl WriteXml for Target {
    fn write_xml<W: Write>(&self, writer: &mut Writer<W>) -> Result<(), WriteError> {
        match self {
            Self::Datastore(datastore) => datastore.write_xml(writer),
            Self::Url(url) => url.write_xml(writer),
        }
    }
}

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

    #[test]
    fn request_to_xml() {
        let req = Request {
            message_id: MessageId(101),
            operation: DeleteConfig {
                target: Target::Datastore(Datastore::Running),
            },
        };
        let expect = r#"<rpc message-id="101"><delete-config><target><running/></target></delete-config></rpc>]]>]]>"#;
        assert_eq!(req.to_xml().unwrap(), expect);
    }
}