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

use serde::Serialize;

use crate::documents::BuildXML;
use crate::xml_builder::*;

#[derive(Debug, Clone, PartialEq, Serialize, Default)]
#[serde(rename_all = "camelCase")]
pub struct CustomProps {
    pub properties: std::collections::HashMap<String, String>,
}

impl CustomProps {
    pub(crate) fn new() -> Self {
        Self::default()
    }

    pub fn add_custom_property(mut self, name: impl Into<String>, item: impl Into<String>) -> Self {
        self.properties.insert(name.into(), item.into());
        self
    }
}

impl BuildXML for CustomProps {
    fn build(&self) -> Vec<u8> {
        let b = XMLBuilder::new();
        let mut base = b.declaration(Some(true)).open_custom_properties(
            "http://schemas.openxmlformats.org/officeDocument/2006/custom-properties",
            "http://schemas.openxmlformats.org/officeDocument/2006/docPropsVTypes",
        );

        for (i, (key, item)) in self.properties.iter().enumerate() {
            base = base
                .open_property(
                    "{D5CDD505-2E9C-101B-9397-08002B2CF9AE}",
                    // I can not found spec about this id.
                    // It is invalid if pid started by 1....
                    &format!("{}", i + 2),
                    key,
                )
                .lpwstr(item)
                .close()
        }

        base.close().build()
    }
}