use serde::ser::{SerializeStruct, Serializer};
use serde::Serialize;
use std::io::Write;
use super::{IndentLevel, NumberingId};
use crate::documents::BuildXML;
use crate::xml_builder::*;
#[derive(Debug, Clone, PartialEq, Default)]
pub struct NumberingProperty {
pub id: Option<NumberingId>,
pub level: Option<IndentLevel>,
}
impl NumberingProperty {
pub fn new() -> NumberingProperty {
Default::default()
}
pub fn id(mut self, id: NumberingId) -> NumberingProperty {
self.id = Some(id);
self
}
pub(crate) fn level(mut self, level: IndentLevel) -> NumberingProperty {
self.level = Some(level);
self
}
pub fn add_num(mut self, id: NumberingId, level: IndentLevel) -> NumberingProperty {
self.id = Some(id);
self.level = Some(level);
self
}
}
impl BuildXML for NumberingProperty {
fn build_to<W: Write>(
&self,
stream: crate::xml::writer::EventWriter<W>,
) -> crate::xml::writer::Result<crate::xml::writer::EventWriter<W>> {
XMLBuilder::from(stream)
.open_numbering_property()?
.add_optional_child(&self.id)?
.add_optional_child(&self.level)?
.close()?
.into_inner()
}
}
impl Serialize for NumberingProperty {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let mut t = serializer.serialize_struct("NumberProperty", 2)?;
let mut id: Option<usize> = None;
if let Some(n) = &self.id {
id = Some(n.id);
}
t.serialize_field("id", &id)?;
let mut level: Option<usize> = None;
if let Some(n) = &self.level {
level = Some(n.val);
}
t.serialize_field("level", &level)?;
t.end()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[cfg(test)]
use pretty_assertions::assert_eq;
use std::str;
#[test]
fn test_num_property() {
let c = NumberingProperty::new().add_num(NumberingId::new(0), IndentLevel::new(3));
let b = c.build();
assert_eq!(
str::from_utf8(&b).unwrap(),
r#"<w:numPr><w:numId w:val="0" /><w:ilvl w:val="3" /></w:numPr>"#
);
}
#[test]
fn test_empty_num_property() {
let c = NumberingProperty::new();
let b = c.build();
assert_eq!(str::from_utf8(&b).unwrap(), r#"<w:numPr />"#);
}
#[test]
fn test_num_property_json() {
let c = NumberingProperty::new().add_num(NumberingId::new(0), IndentLevel::new(3));
assert_eq!(serde_json::to_string(&c).unwrap(), r#"{"id":0,"level":3}"#);
}
#[test]
fn test_empty_num_property_json() {
let c = NumberingProperty::new();
assert_eq!(
serde_json::to_string(&c).unwrap(),
r#"{"id":null,"level":null}"#
);
}
}