summaryrefslogtreecommitdiff
path: root/file.go
blob: 979f5e1f9abf54a12ed464c7a877377905592691 (plain) (blame)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package excelize

import (
	"archive/zip"
	"bytes"
	"os"
)

// CreateFile provide function to create new file by default template.
// For example:
// xlsx := CreateFile()
func CreateFile() *File {
	file := make(map[string]string)
	file[`_rels/.rels`] = templateRels
	file[`docProps/app.xml`] = templateDocpropsApp
	file[`docProps/core.xml`] = templateDocpropsCore
	file[`xl/_rels/workbook.xml.rels`] = templateWorkbookRels
	file[`xl/theme/theme1.xml`] = templateTheme
	file[`xl/worksheets/sheet1.xml`] = templateSheet
	file[`xl/styles.xml`] = templateStyles
	file[`xl/workbook.xml`] = templateWorkbook
	file[`[Content_Types].xml`] = templateContentTypes
	return &File{
		XLSX: file,
	}
}

// Save provide function override the xlsx file with origin path.
func (f *File) Save() error {
	buf := new(bytes.Buffer)
	w := zip.NewWriter(buf)
	for path, content := range f.XLSX {
		f, err := w.Create(path)
		if err != nil {
			return err
		}
		_, err = f.Write([]byte(content))
		if err != nil {
			return err
		}
	}
	err := w.Close()
	if err != nil {
		return err
	}
	file, err := os.OpenFile(f.Path, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0666)
	if err != nil {
		return err
	}
	buf.WriteTo(file)
	return err
}

// WriteTo provide function create or update to an xlsx file at the provided path.
func (f *File) WriteTo(name string) error {
	buf := new(bytes.Buffer)
	w := zip.NewWriter(buf)
	for path, content := range f.XLSX {
		f, err := w.Create(path)
		if err != nil {
			return err
		}
		_, err = f.Write([]byte(content))
		if err != nil {
			return err
		}
	}
	err := w.Close()
	if err != nil {
		return err
	}
	file, err := os.OpenFile(name, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0666)
	if err != nil {
		return err
	}
	buf.WriteTo(file)
	return err
}