From a9d3ee28693adfe920b3cd047f52862d8a79a00e Mon Sep 17 00:00:00 2001 From: Ri Xu Date: Tue, 30 Aug 2016 11:51:31 +0800 Subject: Init commit. --- file.go | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100644 file.go (limited to 'file.go') diff --git a/file.go b/file.go new file mode 100644 index 0000000..bdefc3b --- /dev/null +++ b/file.go @@ -0,0 +1,54 @@ +package excelize + +import ( + "archive/zip" + "bytes" + "fmt" + "os" +) + +// Create a new xlsx file +// +// For example: +// +// xlsx := CreateFile() +// +func CreateFile() []FileList { + var file []FileList + file = saveFileList(file, `_rels/.rels`, TEMPLATE_RELS) + file = saveFileList(file, `docProps/app.xml`, TEMPLATE_DOCPROPS_APP) + file = saveFileList(file, `docProps/core.xml`, TEMPLATE_DOCPROPS_CORE) + file = saveFileList(file, `xl/_rels/workbook.xml.rels`, TEMPLATE_WORKBOOK_RELS) + file = saveFileList(file, `xl/theme/theme1.xml`, TEMPLATE_THEME) + file = saveFileList(file, `xl/worksheets/sheet1.xml`, TEMPLATE_SHEET) + file = saveFileList(file, `xl/styles.xml`, TEMPLATE_STYLES) + file = saveFileList(file, `xl/workbook.xml`, TEMPLATE_WORKBOOK) + file = saveFileList(file, `[Content_Types].xml`, TEMPLATE_CONTENT_TYPES) + return file +} + +// Save after create or update to an xlsx file at the provided path. +func Save(files []FileList, name string) error { + buf := new(bytes.Buffer) + w := zip.NewWriter(buf) + for _, file := range files { + f, err := w.Create(file.Key) + if err != nil { + fmt.Println(err) + } + _, err = f.Write([]byte(file.Value)) + if err != nil { + return err + } + } + err := w.Close() + if err != nil { + return err + } + f, err := os.OpenFile(name, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0666) + if err != nil { + return err + } + buf.WriteTo(f) + return err +} -- cgit v1.2.1