Add tables loop to Paragraphs funcs of Header and Footer (unido…

…c#303)

* Add `Tables` func for `Header`, `Footer`
* Embed tables loop into `Header.Paragraphs`, `Footer.Paragraphs`

document/footer.go

@@ -48,6 +48,20 @@ func (f Footer) Index() int {
 	return -1
 }
 
+// Tables returns the tables defined in the footer.
+func (f Footer) Tables() []Table {
+	ret := []Table{}
+	if f.x == nil {
+		return nil
+	}
+	for _, c := range f.x.EG_ContentBlockContent {
+		for _, t := range f.tables(f.d, c) {
+			ret = append(ret, t)
+		}
+	}
+	return ret
+}
+
 // Paragraphs returns the paragraphs defined in a footer.
 func (f Footer) Paragraphs() []Paragraph {
 	ret := []Paragraph{}
@@ -56,6 +70,14 @@ func (f Footer) Paragraphs() []Paragraph {
 			ret = append(ret, Paragraph{f.d, p})
 		}
 	}
+
+	for _, t := range f.Tables() {
+		for _, r := range t.Rows() {
+			for _, c := range r.Cells() {
+				ret = append(ret, c.Paragraphs()...)
+			}
+		}
+	}
 	return ret
 }
 
@@ -106,3 +128,27 @@ func (f Footer) AddImage(i common.Image) (common.ImageRef, error) {
 	r.SetRelID(rel.X().IdAttr)
 	return r, nil
 }
+
+func (f Footer) tables(d *Document, bc *wml.EG_ContentBlockContent) []Table {
+	ret := []Table{}
+	for _, t := range bc.Tbl {
+		ret = append(ret, Table{d, t})
+		for _, crc := range t.EG_ContentRowContent {
+			for _, tr := range crc.Tr {
+				for _, ccc := range tr.EG_ContentCellContent {
+					for _, tc := range ccc.Tc {
+						for _, ble := range tc.EG_BlockLevelElts {
+							for _, cbc := range ble.EG_ContentBlockContent {
+								for _, tbl := range f.tables(d, cbc) {
+									ret = append(ret, tbl)
+								}
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+
+	return ret
+}

document/header.go

@@ -48,6 +48,20 @@ func (h Header) Index() int {
 	return -1
 }
 
+// Tables returns the tables defined in the header.
+func (h Header) Tables() []Table {
+	ret := []Table{}
+	if h.x == nil {
+		return nil
+	}
+	for _, c := range h.x.EG_ContentBlockContent {
+		for _, t := range h.tables(h.d, c) {
+			ret = append(ret, t)
+		}
+	}
+	return ret
+}
+
 // Paragraphs returns the paragraphs defined in a header.
 func (h Header) Paragraphs() []Paragraph {
 	ret := []Paragraph{}
@@ -56,6 +70,14 @@ func (h Header) Paragraphs() []Paragraph {
 			ret = append(ret, Paragraph{h.d, p})
 		}
 	}
+
+	for _, t := range h.Tables() {
+		for _, r := range t.Rows() {
+			for _, c := range r.Cells() {
+				ret = append(ret, c.Paragraphs()...)
+			}
+		}
+	}
 	return ret
 }
 
@@ -106,3 +128,27 @@ func (h Header) AddImage(i common.Image) (common.ImageRef, error) {
 	r.SetRelID(rel.X().IdAttr)
 	return r, nil
 }
+
+func (h Header) tables(d *Document, bc *wml.EG_ContentBlockContent) []Table {
+	ret := []Table{}
+	for _, t := range bc.Tbl {
+		ret = append(ret, Table{d, t})
+		for _, crc := range t.EG_ContentRowContent {
+			for _, tr := range crc.Tr {
+				for _, ccc := range tr.EG_ContentCellContent {
+					for _, tc := range ccc.Tc {
+						for _, ble := range tc.EG_BlockLevelElts {
+							for _, cbc := range ble.EG_ContentBlockContent {
+								for _, tbl := range h.tables(d, cbc) {
+									ret = append(ret, tbl)
+								}
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+
+	return ret
+}