docx_rs/documents/elements/
numbering_property.rs

1use serde::ser::{SerializeStruct, Serializer};
2use serde::Serialize;
3use std::io::Write;
4
5use super::{IndentLevel, NumberingId};
6use crate::documents::BuildXML;
7use crate::xml_builder::*;
8
9#[derive(Debug, Clone, PartialEq, Default)]
10pub struct NumberingProperty {
11    pub id: Option<NumberingId>,
12    pub level: Option<IndentLevel>,
13}
14
15impl NumberingProperty {
16    pub fn new() -> NumberingProperty {
17        Default::default()
18    }
19
20    pub fn id(mut self, id: NumberingId) -> NumberingProperty {
21        self.id = Some(id);
22        self
23    }
24
25    pub(crate) fn level(mut self, level: IndentLevel) -> NumberingProperty {
26        self.level = Some(level);
27        self
28    }
29
30    pub fn add_num(mut self, id: NumberingId, level: IndentLevel) -> NumberingProperty {
31        self.id = Some(id);
32        self.level = Some(level);
33        self
34    }
35}
36
37impl BuildXML for NumberingProperty {
38    fn build_to<W: Write>(
39        &self,
40        stream: xml::writer::EventWriter<W>,
41    ) -> xml::writer::Result<xml::writer::EventWriter<W>> {
42        XMLBuilder::from(stream)
43            .open_numbering_property()?
44            .add_optional_child(&self.id)?
45            .add_optional_child(&self.level)?
46            .close()?
47            .into_inner()
48    }
49}
50
51impl Serialize for NumberingProperty {
52    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
53    where
54        S: Serializer,
55    {
56        let mut t = serializer.serialize_struct("NumberProperty", 2)?;
57        let mut id: Option<usize> = None;
58        if let Some(n) = &self.id {
59            id = Some(n.id);
60        }
61        t.serialize_field("id", &id)?;
62
63        let mut level: Option<usize> = None;
64        if let Some(n) = &self.level {
65            level = Some(n.val);
66        }
67        t.serialize_field("level", &level)?;
68        t.end()
69    }
70}
71
72#[cfg(test)]
73mod tests {
74
75    use super::*;
76    #[cfg(test)]
77    use pretty_assertions::assert_eq;
78    use std::str;
79
80    #[test]
81    fn test_num_property() {
82        let c = NumberingProperty::new().add_num(NumberingId::new(0), IndentLevel::new(3));
83        let b = c.build();
84        assert_eq!(
85            str::from_utf8(&b).unwrap(),
86            r#"<w:numPr><w:numId w:val="0" /><w:ilvl w:val="3" /></w:numPr>"#
87        );
88    }
89
90    #[test]
91    fn test_empty_num_property() {
92        let c = NumberingProperty::new();
93        let b = c.build();
94        assert_eq!(str::from_utf8(&b).unwrap(), r#"<w:numPr />"#);
95    }
96
97    #[test]
98    fn test_num_property_json() {
99        let c = NumberingProperty::new().add_num(NumberingId::new(0), IndentLevel::new(3));
100        assert_eq!(serde_json::to_string(&c).unwrap(), r#"{"id":0,"level":3}"#);
101    }
102
103    #[test]
104    fn test_empty_num_property_json() {
105        let c = NumberingProperty::new();
106        assert_eq!(
107            serde_json::to_string(&c).unwrap(),
108            r#"{"id":null,"level":null}"#
109        );
110    }
111}
docx_rs/documents/elements/numbering_property.rs

docx_rs/documents/elements/
numbering_property.rs
