docx_rs/documents/elements/
name.rs

1use serde::{Serialize, Serializer};
2use std::io::Write;
3
4use std::str::FromStr;
5
6use crate::documents::BuildXML;
7use crate::{escape, xml_builder::*};
8
9#[derive(Debug, Clone, PartialEq)]
10pub struct Name {
11    name: String,
12}
13
14impl Name {
15    pub fn new(name: impl Into<String>) -> Name {
16        Name {
17            name: escape::escape(&name.into()),
18        }
19    }
20
21    pub fn starts_with(&self, s: &str) -> bool {
22        self.name.starts_with(s)
23    }
24
25    pub fn is_heading(&self) -> bool {
26        self.name.to_lowercase().starts_with("heading")
27    }
28
29    pub fn get_heading_number(&self) -> Option<usize> {
30        let replaced = self.name.to_lowercase().replace("heading ", "");
31        usize::from_str(&replaced).ok()
32    }
33}
34
35impl BuildXML for Name {
36    fn build_to<W: Write>(
37        &self,
38        stream: crate::xml::writer::EventWriter<W>,
39    ) -> crate::xml::writer::Result<crate::xml::writer::EventWriter<W>> {
40        XMLBuilder::from(stream).name(&self.name)?.into_inner()
41    }
42}
43
44impl Serialize for Name {
45    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
46    where
47        S: Serializer,
48    {
49        serializer.serialize_str(&self.name)
50    }
51}
52
53#[cfg(test)]
54mod tests {
55
56    use super::*;
57    #[cfg(test)]
58    use pretty_assertions::assert_eq;
59    use std::str;
60
61    #[test]
62    fn test_build() {
63        let c = Name::new("Heading");
64        let b = c.build();
65        assert_eq!(str::from_utf8(&b).unwrap(), r#"<w:name w:val="Heading" />"#);
66    }
67}
docx_rs/documents/elements/name.rs

docx_rs/documents/elements/
name.rs
