134 lines
3.6 KiB
Rust
134 lines
3.6 KiB
Rust
use serde::ser::{SerializeStruct, Serializer};
|
|
use serde::Serialize;
|
|
|
|
use super::{Delete, Insert, TableCell, TableRowProperty};
|
|
use crate::xml_builder::*;
|
|
use crate::{documents::BuildXML, HeightRule};
|
|
|
|
#[derive(Debug, Clone, PartialEq, Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct TableRow {
|
|
pub cells: Vec<TableRowChild>,
|
|
pub has_numbering: bool,
|
|
pub property: TableRowProperty,
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub enum TableRowChild {
|
|
TableCell(TableCell),
|
|
}
|
|
|
|
impl BuildXML for TableRowChild {
|
|
fn build(&self) -> Vec<u8> {
|
|
match self {
|
|
TableRowChild::TableCell(v) => v.build(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl TableRow {
|
|
pub fn new(cells: Vec<TableCell>) -> TableRow {
|
|
let property = TableRowProperty::new();
|
|
let has_numbering = cells.iter().any(|c| c.has_numbering);
|
|
let cells = cells.into_iter().map(TableRowChild::TableCell).collect();
|
|
Self {
|
|
cells,
|
|
property,
|
|
has_numbering,
|
|
}
|
|
}
|
|
|
|
pub fn grid_after(mut self, grid_after: u32) -> TableRow {
|
|
self.property = self.property.grid_after(grid_after);
|
|
self
|
|
}
|
|
|
|
pub fn width_after(mut self, w: f32) -> TableRow {
|
|
self.property = self.property.width_after(w);
|
|
self
|
|
}
|
|
|
|
pub fn grid_before(mut self, grid_before: u32) -> TableRow {
|
|
self.property = self.property.grid_before(grid_before);
|
|
self
|
|
}
|
|
|
|
pub fn width_before(mut self, w: f32) -> TableRow {
|
|
self.property = self.property.width_before(w);
|
|
self
|
|
}
|
|
|
|
pub fn row_height(mut self, h: f32) -> TableRow {
|
|
self.property = self.property.row_height(h);
|
|
self
|
|
}
|
|
|
|
pub fn height_rule(mut self, r: HeightRule) -> TableRow {
|
|
self.property = self.property.height_rule(r);
|
|
self
|
|
}
|
|
|
|
pub fn delete(mut self, d: Delete) -> TableRow {
|
|
self.property = self.property.delete(d);
|
|
self
|
|
}
|
|
|
|
pub fn insert(mut self, i: Insert) -> TableRow {
|
|
self.property = self.property.insert(i);
|
|
self
|
|
}
|
|
}
|
|
|
|
impl BuildXML for TableRow {
|
|
fn build(&self) -> Vec<u8> {
|
|
let b = XMLBuilder::new()
|
|
.open_table_row()
|
|
.add_child(&self.property)
|
|
.add_children(&self.cells);
|
|
b.close().build()
|
|
}
|
|
}
|
|
|
|
impl Serialize for TableRowChild {
|
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
|
where
|
|
S: Serializer,
|
|
{
|
|
match *self {
|
|
TableRowChild::TableCell(ref r) => {
|
|
let mut t = serializer.serialize_struct("TableCell", 2)?;
|
|
t.serialize_field("type", "tableCell")?;
|
|
t.serialize_field("data", r)?;
|
|
t.end()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::*;
|
|
#[cfg(test)]
|
|
use pretty_assertions::assert_eq;
|
|
use std::str;
|
|
|
|
#[test]
|
|
fn test_row() {
|
|
let b = TableRow::new(vec![TableCell::new()]).build();
|
|
assert_eq!(
|
|
str::from_utf8(&b).unwrap(),
|
|
r#"<w:tr><w:trPr /><w:tc><w:tcPr /><w:p w14:paraId="12345678"><w:pPr><w:rPr /></w:pPr></w:p></w:tc></w:tr>"#
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_row_json() {
|
|
let r = TableRow::new(vec![TableCell::new()]);
|
|
assert_eq!(
|
|
serde_json::to_string(&r).unwrap(),
|
|
r#"{"cells":[{"type":"tableCell","data":{"children":[],"property":{"width":null,"borders":null,"gridSpan":null,"verticalMerge":null,"verticalAlign":null,"textDirection":null,"shading":null},"hasNumbering":false}}],"hasNumbering":false,"property":{"gridAfter":null,"widthAfter":null,"gridBefore":null,"widthBefore":null}}"#
|
|
);
|
|
}
|
|
}
|