unioffice/zippkg/marshal.go

// 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"

	"baliance.com/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
}