80 lines
3.1 KiB
Rust
80 lines
3.1 KiB
Rust
use std::io::Read;
|
|
use std::str::FromStr;
|
|
|
|
use crate::reader::*;
|
|
use xml::reader::{EventReader, XmlEvent};
|
|
|
|
use super::{Paragraph, Table};
|
|
|
|
impl FromXML for Header {
|
|
fn from_xml<R: Read>(reader: R) -> Result<Self, ReaderError> {
|
|
let mut parser = EventReader::new(reader);
|
|
let mut header = Self::default();
|
|
loop {
|
|
let e = parser.next();
|
|
match e {
|
|
Ok(XmlEvent::StartElement {
|
|
attributes, name, ..
|
|
}) => {
|
|
let e = XMLElement::from_str(&name.local_name).unwrap();
|
|
match e {
|
|
XMLElement::Paragraph => {
|
|
if let Ok(p) = Paragraph::read(&mut parser, &attributes) {
|
|
header = header.add_paragraph(p);
|
|
}
|
|
continue;
|
|
}
|
|
XMLElement::Table => {
|
|
if let Ok(t) = Table::read(&mut parser, &attributes) {
|
|
header = header.add_table(t);
|
|
}
|
|
continue;
|
|
}
|
|
XMLElement::StructuredDataTag => {
|
|
if let Ok(tag) = StructuredDataTag::read(&mut parser, &attributes) {
|
|
header = header.add_structured_data_tag(tag);
|
|
}
|
|
continue;
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
Ok(XmlEvent::EndDocument) => break,
|
|
Err(_) => return Err(ReaderError::XMLReadError),
|
|
_ => {}
|
|
}
|
|
}
|
|
Ok(header)
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_header_from_xml() {
|
|
let xml = r#"<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
|
|
<w:hdr xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships"
|
|
xmlns:o="urn:schemas-microsoft-com:office:office"
|
|
xmlns:v="urn:schemas-microsoft-com:vml"
|
|
xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main"
|
|
xmlns:w10="urn:schemas-microsoft-com:office:word"
|
|
xmlns:wp="http://schemas.openxmlformats.org/drawingml/2006/wordprocessingDrawing"
|
|
xmlns:wps="http://schemas.microsoft.com/office/word/2010/wordprocessingShape"
|
|
xmlns:wpg="http://schemas.microsoft.com/office/word/2010/wordprocessingGroup"
|
|
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
|
|
xmlns:wp14="http://schemas.microsoft.com/office/word/2010/wordprocessingDrawing"
|
|
xmlns:w14="http://schemas.microsoft.com/office/word/2010/wordml" mc:Ignorable="w14 wp14">
|
|
<w:p w14:paraId="12345678">
|
|
<w:pPr>
|
|
<w:rPr />
|
|
</w:pPr>
|
|
<w:r>
|
|
<w:rPr />
|
|
<w:t xml:space="preserve">Hello Header</w:t>
|
|
</w:r>
|
|
</w:p>
|
|
</w:hdr>"#;
|
|
let h = Header::from_xml(xml.as_bytes()).unwrap();
|
|
let expected =
|
|
Header::new().add_paragraph(Paragraph::new().add_run(Run::new().add_text("Hello Header")));
|
|
assert_eq!(h, expected)
|
|
}
|