60 lines
1.8 KiB
Rust
60 lines
1.8 KiB
Rust
use super::{DocDefaults, Style};
|
|
use crate::documents::BuildXML;
|
|
use crate::xml_builder::*;
|
|
use crate::StyleType;
|
|
|
|
pub struct Styles {
|
|
doc_defaults: DocDefaults,
|
|
styles: Vec<Style>,
|
|
}
|
|
|
|
impl Styles {
|
|
pub fn new() -> Styles {
|
|
Default::default()
|
|
}
|
|
|
|
pub fn add_style(mut self, style: Style) -> Self {
|
|
self.styles.push(style);
|
|
self
|
|
}
|
|
}
|
|
|
|
impl Default for Styles {
|
|
fn default() -> Self {
|
|
Self {
|
|
doc_defaults: DocDefaults::new(),
|
|
styles: vec![],
|
|
}
|
|
}
|
|
}
|
|
|
|
impl BuildXML for Styles {
|
|
fn build(&self) -> Vec<u8> {
|
|
let b = XMLBuilder::new();
|
|
b.open_styles()
|
|
.add_child(&self.doc_defaults)
|
|
.add_children(&self.styles)
|
|
.close()
|
|
.build()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::*;
|
|
#[cfg(test)]
|
|
use pretty_assertions::assert_eq;
|
|
use std::str;
|
|
|
|
#[test]
|
|
fn test_build() {
|
|
let c = Styles::new().add_style(Style::new("Title", "TitleName", StyleType::Paragraph));
|
|
let b = c.build();
|
|
assert_eq!(
|
|
str::from_utf8(&b).unwrap(),
|
|
r#"<w:styles xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships" xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main" xmlns:w14="http://schemas.microsoft.com/office/word/2010/wordml" xmlns:w15="http://schemas.microsoft.com/office/word/2012/wordml" mc:Ignorable="w14 w15"><w:docDefaults><w:rPrDefault><w:rPr /></w:rPrDefault></w:docDefaults><w:style w:type="paragraph" w:styleId="Title"><w:name w:val="TitleName" /><w:rPr /><w:pPr /><w:basedOn w:val="Normal" /><w:next w:val="Normal" /><w:qFormat /></w:style></w:styles>"#
|
|
);
|
|
}
|
|
}
|