forked from sdowns/gooxml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
marshal.go
55 lines (46 loc) · 1.66 KB
/
marshal.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Copyright 2017 Baliance. All rights reserved.
//
// Use of this source code is governed by the terms of the Affero GNU General
// Public License version 3.0 as published by the Free Software Foundation and
// appearing in the file LICENSE included in the packaging of this file. A
// commercial license can be purchased by contacting sales@baliance.com.
package zippkg
import (
"archive/zip"
"encoding/xml"
"fmt"
"time"
"github.com/pevecyan/gooxml"
)
// XMLHeader is a header that MarshalXML uses to prefix the XML files it creates.
const XMLHeader = `<?xml version="1.0" encoding="UTF-8" standalone="yes"?>` + "\n"
var nl = []byte{'\r', '\n'}
func MarshalXMLByTypeIndex(z *zip.Writer, dt gooxml.DocType, typ string, idx int, v interface{}) error {
fn := gooxml.AbsoluteFilename(dt, typ, idx)
return MarshalXML(z, fn, v)
}
func MarshalXMLByType(z *zip.Writer, dt gooxml.DocType, typ string, v interface{}) error {
fn := gooxml.AbsoluteFilename(dt, typ, 0)
return MarshalXML(z, fn, v)
}
// MarshalXML creates a file inside of a zip and marshals an object as xml, prefixing it
// with a standard XML header.
func MarshalXML(z *zip.Writer, filename string, v interface{}) error {
fh := &zip.FileHeader{}
fh.Method = zip.Deflate
fh.Name = filename
fh.SetModTime(time.Now())
w, err := z.CreateHeader(fh)
if err != nil {
return fmt.Errorf("creating %s in zip: %s", filename, err)
}
_, err = w.Write([]byte(XMLHeader))
if err != nil {
return fmt.Errorf("creating xml header to %s: %s", filename, err)
}
if err = xml.NewEncoder(SelfClosingWriter{w}).Encode(v); err != nil {
return fmt.Errorf("marshaling %s: %s", filename, err)
}
_, err = w.Write(nl)
return err
}