mirror of
https://github.com/unidoc/unioffice.git
synced 2025-04-25 13:48:53 +08:00
9090e4f93c
This seems like the better choice, a lot of the logic is shared between the document types, and it allows generating filenames in a single place. The only downside is that you must pass in the document type as some content types have different typical names depending on the document type (e.g. an 'office document' is the main document.xml, workbook.xml and presentation.xml
56 lines
1.7 KiB
Go
56 lines
1.7 KiB
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
|
|
}
|