use serde::{Deserialize, Serialize};
use std::io::Write;
use crate::documents::BuildXML;
use crate::xml_builder::*;
#[derive(Serialize, Deserialize, Debug, PartialEq, Clone, Default)]
pub struct Rels {
pub rels: Vec<(String, String, String)>,
}
impl Rels {
pub fn new() -> Rels {
Default::default()
}
pub fn set_default(mut self) -> Self {
self.rels.push((
"http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties"
.to_owned(),
"rId1".to_owned(),
"docProps/core.xml".to_owned(),
));
self.rels.push(
("http://schemas.openxmlformats.org/officeDocument/2006/relationships/extended-properties".to_owned(),
"rId2".to_owned(), "docProps/app.xml".to_owned()),
);
self.rels.push((
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"
.to_owned(),
"rId3".to_owned(),
"word/document.xml".to_owned(),
));
self.rels.push((
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/custom-properties"
.to_owned(),
"rId4".to_owned(),
"docProps/custom.xml".to_owned(),
));
self
}
pub fn add_taskpanes_rel(mut self) -> Self {
self = self.add_rel(
"http://schemas.microsoft.com/office/2011/relationships/webextensiontaskpanes",
"word/webextensions/taskpanes.xml",
);
self
}
pub fn add_rel(mut self, rel_type: impl Into<String>, target: impl Into<String>) -> Self {
self.rels.push((
rel_type.into(),
format!("rId{}", self.rels.len() + 1),
target.into(),
));
self
}
pub fn find_target(&self, rel_type: &str) -> Option<&(String, String, String)> {
self.rels.iter().find(|rel| rel.0 == rel_type)
}
}
impl BuildXML for Rels {
fn build_to<W: Write>(
&self,
stream: crate::xml::writer::EventWriter<W>,
) -> crate::xml::writer::Result<crate::xml::writer::EventWriter<W>> {
XMLBuilder::from(stream)
.declaration(None)?
.open_relationships("http://schemas.openxmlformats.org/package/2006/relationships")?
.apply_each(&self.rels, |(k, id, v), b| b.relationship(id, k, v))?
.close()?
.into_inner()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[cfg(test)]
use pretty_assertions::assert_eq;
use std::str;
#[test]
fn test_build() {
let c = Rels::new().set_default();
let b = c.build();
assert_eq!(
str::from_utf8(&b).unwrap(),
r#"<?xml version="1.0" encoding="UTF-8"?><Relationships xmlns="http://schemas.openxmlformats.org/package/2006/relationships"><Relationship Id="rId1" Type="http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties" Target="docProps/core.xml" /><Relationship Id="rId2" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/extended-properties" Target="docProps/app.xml" /><Relationship Id="rId3" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument" Target="word/document.xml" /><Relationship Id="rId4" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/custom-properties" Target="docProps/custom.xml" /></Relationships>"#
);
}
}