docx_rs/documents/elements/
table_row.rs

1use serde::ser::{SerializeStruct, Serializer};
2use serde::Serialize;
3use std::io::Write;
4
5use super::{Delete, Insert, TableCell, TableRowProperty};
6use crate::xml_builder::*;
7use crate::{documents::BuildXML, HeightRule};
8
9#[derive(Debug, Clone, PartialEq, Serialize)]
10#[serde(rename_all = "camelCase")]
11pub struct TableRow {
12    pub cells: Vec<TableRowChild>,
13    pub has_numbering: bool,
14    pub property: TableRowProperty,
15}
16
17#[derive(Debug, Clone, PartialEq)]
18pub enum TableRowChild {
19    TableCell(TableCell),
20}
21
22impl BuildXML for TableRowChild {
23    fn build_to<W: Write>(
24        &self,
25        stream: xml::writer::EventWriter<W>,
26    ) -> xml::writer::Result<xml::writer::EventWriter<W>> {
27        match self {
28            TableRowChild::TableCell(v) => v.build_to(stream),
29        }
30    }
31}
32
33impl TableRow {
34    pub fn new(cells: Vec<TableCell>) -> TableRow {
35        let property = TableRowProperty::new();
36        let has_numbering = cells.iter().any(|c| c.has_numbering);
37        let cells = cells.into_iter().map(TableRowChild::TableCell).collect();
38        Self {
39            cells,
40            property,
41            has_numbering,
42        }
43    }
44
45    pub fn grid_after(mut self, grid_after: u32) -> TableRow {
46        self.property = self.property.grid_after(grid_after);
47        self
48    }
49
50    pub fn width_after(mut self, w: f32) -> TableRow {
51        self.property = self.property.width_after(w);
52        self
53    }
54
55    pub fn grid_before(mut self, grid_before: u32) -> TableRow {
56        self.property = self.property.grid_before(grid_before);
57        self
58    }
59
60    pub fn width_before(mut self, w: f32) -> TableRow {
61        self.property = self.property.width_before(w);
62        self
63    }
64
65    pub fn row_height(mut self, h: f32) -> TableRow {
66        self.property = self.property.row_height(h);
67        self
68    }
69
70    pub fn height_rule(mut self, r: HeightRule) -> TableRow {
71        self.property = self.property.height_rule(r);
72        self
73    }
74
75    pub fn delete(mut self, d: Delete) -> TableRow {
76        self.property = self.property.delete(d);
77        self
78    }
79
80    pub fn insert(mut self, i: Insert) -> TableRow {
81        self.property = self.property.insert(i);
82        self
83    }
84
85    pub fn cant_split(mut self) -> TableRow {
86        self.property = self.property.cant_split();
87        self
88    }
89}
90
91impl BuildXML for TableRow {
92    fn build_to<W: Write>(
93        &self,
94        stream: xml::writer::EventWriter<W>,
95    ) -> xml::writer::Result<xml::writer::EventWriter<W>> {
96        XMLBuilder::from(stream)
97            .open_table_row()?
98            .add_child(&self.property)?
99            .add_children(&self.cells)?
100            .close()?
101            .into_inner()
102    }
103}
104
105impl Serialize for TableRowChild {
106    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
107    where
108        S: Serializer,
109    {
110        match *self {
111            TableRowChild::TableCell(ref r) => {
112                let mut t = serializer.serialize_struct("TableCell", 2)?;
113                t.serialize_field("type", "tableCell")?;
114                t.serialize_field("data", r)?;
115                t.end()
116            }
117        }
118    }
119}
120
121#[cfg(test)]
122mod tests {
123
124    use super::*;
125    #[cfg(test)]
126    use pretty_assertions::assert_eq;
127    use std::str;
128
129    #[test]
130    fn test_row() {
131        let b = TableRow::new(vec![TableCell::new()]).build();
132        assert_eq!(
133            str::from_utf8(&b).unwrap(),
134            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>"#
135        );
136    }
137
138    #[test]
139    fn test_row_json() {
140        let r = TableRow::new(vec![TableCell::new()]);
141        assert_eq!(
142            serde_json::to_string(&r).unwrap(),
143            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}}"#
144        );
145    }
146
147    #[test]
148    fn test_row_cant_split() {
149        let b = TableRow::new(vec![TableCell::new()]).cant_split().build();
150        assert_eq!(
151            str::from_utf8(&b).unwrap(),
152            r#"<w:tr><w:trPr><w:cantSplit /></w:trPr><w:tc><w:tcPr /><w:p w14:paraId="12345678"><w:pPr><w:rPr /></w:pPr></w:p></w:tc></w:tr>"#
153        );
154    }
155}
docx_rs/documents/elements/table_row.rs

docx_rs/documents/elements/
table_row.rs
