use serde::Serialize;
use std::io::Write;
use crate::documents::BuildXML;
use crate::types::*;
use crate::xml_builder::*;
#[derive(Serialize, Debug, Clone, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct TableBorder {
pub border_type: BorderType,
pub size: usize,
pub color: String,
position: TableBorderPosition,
space: usize,
}
impl TableBorder {
pub fn new(position: TableBorderPosition) -> TableBorder {
TableBorder {
position,
border_type: BorderType::Single,
size: 2,
space: 0,
color: "000000".to_owned(),
}
}
pub fn color(mut self, color: impl Into<String>) -> TableBorder {
self.color = color.into();
self
}
pub fn size(mut self, size: usize) -> TableBorder {
self.size = size;
self
}
pub fn border_type(mut self, border_type: BorderType) -> TableBorder {
self.border_type = border_type;
self
}
}
impl BuildXML for TableBorder {
fn build_to<W: Write>(
&self,
stream: crate::xml::writer::EventWriter<W>,
) -> crate::xml::writer::Result<crate::xml::writer::EventWriter<W>> {
let func = match self.position {
TableBorderPosition::Top => XMLBuilder::border_top,
TableBorderPosition::Left => XMLBuilder::border_left,
TableBorderPosition::Bottom => XMLBuilder::border_bottom,
TableBorderPosition::Right => XMLBuilder::border_right,
TableBorderPosition::InsideH => XMLBuilder::border_inside_h,
TableBorderPosition::InsideV => XMLBuilder::border_inside_v,
};
XMLBuilder::from(stream)
.apply(|b| func(b, self.border_type, self.size, self.space, &self.color))?
.into_inner()
}
}
#[derive(Serialize, Debug, Clone, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct TableBorders {
top: Option<TableBorder>,
left: Option<TableBorder>,
bottom: Option<TableBorder>,
right: Option<TableBorder>,
inside_h: Option<TableBorder>,
inside_v: Option<TableBorder>,
}
impl Default for TableBorders {
fn default() -> TableBorders {
TableBorders {
top: Some(TableBorder::new(TableBorderPosition::Top)),
left: Some(TableBorder::new(TableBorderPosition::Left)),
bottom: Some(TableBorder::new(TableBorderPosition::Bottom)),
right: Some(TableBorder::new(TableBorderPosition::Right)),
inside_h: Some(TableBorder::new(TableBorderPosition::InsideH)),
inside_v: Some(TableBorder::new(TableBorderPosition::InsideV)),
}
}
}
impl TableBorders {
pub fn new() -> TableBorders {
Default::default()
}
pub fn with_empty() -> TableBorders {
TableBorders {
top: None,
left: None,
bottom: None,
right: None,
inside_h: None,
inside_v: None,
}
}
pub fn set(mut self, border: TableBorder) -> Self {
match border.position {
TableBorderPosition::Top => self.top = Some(border),
TableBorderPosition::Left => self.left = Some(border),
TableBorderPosition::Bottom => self.bottom = Some(border),
TableBorderPosition::Right => self.right = Some(border),
TableBorderPosition::InsideH => self.inside_h = Some(border),
TableBorderPosition::InsideV => self.inside_v = Some(border),
};
self
}
pub fn clear(mut self, position: TableBorderPosition) -> Self {
let nil = TableBorder::new(position.clone()).border_type(BorderType::Nil);
match position {
TableBorderPosition::Top => self.top = Some(nil),
TableBorderPosition::Left => self.left = Some(nil),
TableBorderPosition::Bottom => self.bottom = Some(nil),
TableBorderPosition::Right => self.right = Some(nil),
TableBorderPosition::InsideH => self.inside_h = Some(nil),
TableBorderPosition::InsideV => self.inside_v = Some(nil),
};
self
}
pub fn clear_all(mut self) -> Self {
self.top = Some(TableBorder::new(TableBorderPosition::Top).border_type(BorderType::Nil));
self.left = Some(TableBorder::new(TableBorderPosition::Left).border_type(BorderType::Nil));
self.bottom =
Some(TableBorder::new(TableBorderPosition::Bottom).border_type(BorderType::Nil));
self.right =
Some(TableBorder::new(TableBorderPosition::Right).border_type(BorderType::Nil));
self.inside_h =
Some(TableBorder::new(TableBorderPosition::InsideH).border_type(BorderType::Nil));
self.inside_v =
Some(TableBorder::new(TableBorderPosition::InsideV).border_type(BorderType::Nil));
self
}
}
impl BuildXML for TableBorders {
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_table_borders()?
.add_optional_child(&self.top)?
.add_optional_child(&self.left)?
.add_optional_child(&self.bottom)?
.add_optional_child(&self.right)?
.add_optional_child(&self.inside_h)?
.add_optional_child(&self.inside_v)?
.close()?
.into_inner()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[cfg(test)]
use pretty_assertions::assert_eq;
use std::str;
#[test]
fn test_table_borders() {
let b = TableBorders::new().build();
assert_eq!(
str::from_utf8(&b).unwrap(),
r#"<w:tblBorders><w:top w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:left w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:bottom w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:right w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:insideH w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:insideV w:val="single" w:sz="2" w:space="0" w:color="000000" /></w:tblBorders>"#
);
}
#[test]
fn test_table_borders_set() {
let b = TableBorders::new()
.set(TableBorder::new(TableBorderPosition::Left).color("AAAAAA"))
.clear(TableBorderPosition::Top)
.build();
assert_eq!(
str::from_utf8(&b).unwrap(),
r#"<w:tblBorders><w:top w:val="nil" w:sz="2" w:space="0" w:color="000000" /><w:left w:val="single" w:sz="2" w:space="0" w:color="AAAAAA" /><w:bottom w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:right w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:insideH w:val="single" w:sz="2" w:space="0" w:color="000000" /><w:insideV w:val="single" w:sz="2" w:space="0" w:color="000000" /></w:tblBorders>"#
);
}
}