feat: 增加page margin和number (#48)

structsect.go

@@ -28,6 +28,9 @@ import (
 type SectPr struct {
 	XMLName xml.Name `xml:"w:sectPr,omitempty"` // properties of the document, including paper size
 	PgSz    *PgSz    `xml:"w:pgSz,omitempty"`
+	PgMar   *PgMar   `xml:"w:pgMar,omitempty"`
+	Cols    *Cols    `xml:"w:cols,omitempty"`
+	DocGrid *DocGrid `xml:"w:docGrid,omitempty"`
 }
 
 // PgSz show the paper size
@@ -36,6 +39,28 @@ type PgSz struct {
 	H int `xml:"w:h,attr"` // high of paper
 }
 
+// PgMar show the page margin
+type PgMar struct {
+	Top    int `xml:"w:top,attr"`
+	Left   int `xml:"w:left,attr"`
+	Bottom int `xml:"w:bottom,attr"`
+	Right  int `xml:"w:right,attr"`
+	Header int `xml:"w:header,attr"`
+	Footer int `xml:"w:footer,attr"`
+	Gutter int `xml:"w:gutter,attr"`
+}
+
+// Cols show the number of columns
+type Cols struct {
+	Space int `xml:"w:space,attr"`
+}
+
+// DocGrid show the document grid
+type DocGrid struct {
+	Type      string `xml:"w:type,attr"`
+	LinePitch int    `xml:"w:linePitch,attr"`
+}
+
 // UnmarshalXML ...
 func (sect *SectPr) UnmarshalXML(d *xml.Decoder, _ xml.StartElement) error {
 	for {
@@ -55,6 +80,27 @@ func (sect *SectPr) UnmarshalXML(d *xml.Decoder, _ xml.StartElement) error {
 					return err
 				}
 				sect.PgSz = &value
+			case "pgMar":
+				var value PgMar
+				err = d.DecodeElement(&value, &tt)
+				if err != nil && !strings.HasPrefix(err.Error(), "expected") {
+					return err
+				}
+				sect.PgMar = &value
+			case "cols":
+				var value Cols
+				err = d.DecodeElement(&value, &tt)
+				if err != nil && !strings.HasPrefix(err.Error(), "expected") {
+					return err
+				}
+				sect.Cols = &value
+			case "docGrid":
+				var value DocGrid
+				err = d.DecodeElement(&value, &tt)
+				if err != nil && !strings.HasPrefix(err.Error(), "expected") {
+					return err
+				}
+				sect.DocGrid = &value
 			default:
 				err = d.Skip() // skip unsupported tags
 				if err != nil {
@@ -90,3 +136,95 @@ func (pgsz *PgSz) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
 	_, err = d.Token()
 	return err
 }
+
+// UnmarshalXML ...
+func (pgmar *PgMar) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
+	var err error
+
+	for _, attr := range start.Attr {
+		switch attr.Name.Local {
+		case "top":
+			pgmar.Top, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "left":
+			pgmar.Left, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "bottom":
+			pgmar.Bottom, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "right":
+			pgmar.Right, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "header":
+			pgmar.Header, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "footer":
+			pgmar.Footer, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "gutter":
+			pgmar.Gutter, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		default:
+			// ignore other attributes now
+		}
+	}
+	// Consume the end element
+	_, err = d.Token()
+	return err
+}
+
+// UnmarshalXML ...
+func (cols *Cols) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
+	var err error
+
+	for _, attr := range start.Attr {
+		switch attr.Name.Local {
+		case "space":
+			cols.Space, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		default:
+			// ignore other attributes now
+		}
+	}
+	// Consume the end element
+	_, err = d.Token()
+	return err
+}
+
+// UnmarshalXML ...
+func (dg *DocGrid) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
+	var err error
+
+	for _, attr := range start.Attr {
+		switch attr.Name.Local {
+		case "linePitch":
+			dg.LinePitch, err = strconv.Atoi(attr.Value)
+			if err != nil {
+				return err
+			}
+		case "type":
+			dg.Type = attr.Value
+		default:
+			// ignore other attributes now
+		}
+	}
+	// Consume the end element
+	_, err = d.Token()
+	return err
+}