79 lines
2.0 KiB
Rust
79 lines
2.0 KiB
Rust
use super::{Paragraph, TableCellProperty};
|
|
use crate::documents::BuildXML;
|
|
use crate::types::*;
|
|
use crate::xml_builder::*;
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct TableCell<'a> {
|
|
pub(crate) contents: Vec<TableCellContent<'a>>,
|
|
property: TableCellProperty,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum TableCellContent<'a> {
|
|
Paragraph(Paragraph<'a>),
|
|
}
|
|
|
|
impl<'a> TableCell<'a> {
|
|
pub fn new() -> TableCell<'a> {
|
|
let property = TableCellProperty::new();
|
|
let contents = vec![];
|
|
Self { property, contents }
|
|
}
|
|
|
|
pub fn add_paragraph(mut self, p: Paragraph<'a>) -> TableCell<'a> {
|
|
self.contents.push(TableCellContent::Paragraph(p));
|
|
self
|
|
}
|
|
|
|
pub fn vertical_merge(mut self, t: VMergeType) -> TableCell<'a> {
|
|
self.property = self.property.vertical_merge(t);
|
|
self
|
|
}
|
|
|
|
pub fn grid_span(mut self, v: usize) -> TableCell<'a> {
|
|
self.property = self.property.grid_span(v);
|
|
self
|
|
}
|
|
}
|
|
|
|
impl<'a> BuildXML for TableCell<'a> {
|
|
fn build(&self) -> Vec<u8> {
|
|
let b = XMLBuilder::new();
|
|
let mut b = b.open_table_cell().add_child(&self.property);
|
|
for c in &self.contents {
|
|
match c {
|
|
TableCellContent::Paragraph(p) => b = b.add_child(p),
|
|
}
|
|
}
|
|
b.close().build()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::super::*;
|
|
use super::*;
|
|
#[cfg(test)]
|
|
use pretty_assertions::assert_eq;
|
|
use std::str;
|
|
|
|
#[test]
|
|
fn test_cell() {
|
|
let b = TableCell::new().build();
|
|
assert_eq!(str::from_utf8(&b).unwrap(), r#"<w:tc><w:tcPr /></w:tc>"#);
|
|
}
|
|
|
|
#[test]
|
|
fn test_cell_add_p() {
|
|
let b = TableCell::new()
|
|
.add_paragraph(Paragraph::new().add_run(Run::new().add_text("Hello")))
|
|
.build();
|
|
assert_eq!(
|
|
str::from_utf8(&b).unwrap(),
|
|
r#"<w:tc><w:tcPr /><w:p><w:pPr><w:pStyle w:val="Normal" /><w:rPr /></w:pPr><w:r><w:rPr /><w:t xml:space="preserve">Hello</w:t></w:r></w:p></w:tc>"#
|
|
);
|
|
}
|
|
}
|