2016-08-30 11:51:31 +08:00
|
|
|
package excelize
|
|
|
|
|
|
|
|
import (
|
|
|
|
"archive/zip"
|
|
|
|
"bytes"
|
|
|
|
"io"
|
|
|
|
"log"
|
|
|
|
"math"
|
|
|
|
)
|
|
|
|
|
2017-01-18 16:05:01 +08:00
|
|
|
// ReadZipReader can be used to read an XLSX in memory without touching the
|
|
|
|
// filesystem.
|
2016-09-06 21:20:24 +08:00
|
|
|
func ReadZipReader(r *zip.Reader) (map[string]string, int, error) {
|
2016-09-05 10:44:32 +08:00
|
|
|
fileList := make(map[string]string)
|
2016-09-06 21:20:24 +08:00
|
|
|
worksheets := 0
|
2016-08-30 11:51:31 +08:00
|
|
|
for _, v := range r.File {
|
2016-09-05 10:44:32 +08:00
|
|
|
fileList[v.Name] = readFile(v)
|
2016-09-06 21:20:24 +08:00
|
|
|
if len(v.Name) > 18 {
|
2017-01-18 14:47:23 +08:00
|
|
|
if v.Name[0:19] == "xl/worksheets/sheet" {
|
2016-09-06 21:20:24 +08:00
|
|
|
worksheets++
|
|
|
|
}
|
|
|
|
}
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
2016-09-06 21:20:24 +08:00
|
|
|
return fileList, worksheets, nil
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
|
|
|
|
2017-01-24 18:29:02 +08:00
|
|
|
// readXML provides function to read XML content as string.
|
2016-09-05 16:37:15 +08:00
|
|
|
func (f *File) readXML(name string) string {
|
|
|
|
if content, ok := f.XLSX[name]; ok {
|
2016-12-26 23:55:59 +08:00
|
|
|
return content
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
2017-01-23 16:15:01 +08:00
|
|
|
return ""
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
|
|
|
|
2017-01-24 18:29:02 +08:00
|
|
|
// saveFileList provides function to update given file content in file list of
|
|
|
|
// XLSX.
|
2017-03-06 12:05:41 +08:00
|
|
|
func (f *File) saveFileList(name, content string) {
|
2016-09-05 16:37:15 +08:00
|
|
|
f.XLSX[name] = XMLHeader + content
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
|
|
|
|
2016-10-19 20:39:44 +08:00
|
|
|
// Read file content as string in a archive file.
|
2016-08-30 11:51:31 +08:00
|
|
|
func readFile(file *zip.File) string {
|
|
|
|
rc, err := file.Open()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
buff := bytes.NewBuffer(nil)
|
|
|
|
io.Copy(buff, rc)
|
|
|
|
rc.Close()
|
|
|
|
return string(buff.Bytes())
|
|
|
|
}
|
|
|
|
|
2017-01-24 18:29:02 +08:00
|
|
|
// toAlphaString provides function to convert integer to Excel sheet column
|
|
|
|
// title.
|
2016-08-30 11:51:31 +08:00
|
|
|
func toAlphaString(value int) string {
|
|
|
|
if value < 0 {
|
2017-01-23 16:15:01 +08:00
|
|
|
return ""
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
|
|
|
var ans string
|
|
|
|
i := value
|
|
|
|
for i > 0 {
|
|
|
|
ans = string((i-1)%26+65) + ans
|
|
|
|
i = (i - 1) / 26
|
|
|
|
}
|
|
|
|
return ans
|
|
|
|
}
|
|
|
|
|
2017-01-24 18:29:02 +08:00
|
|
|
// titleToNumber provides function to convert Excel sheet column title to int.
|
2016-08-30 11:51:31 +08:00
|
|
|
func titleToNumber(s string) int {
|
|
|
|
weight := 0.0
|
|
|
|
sum := 0
|
|
|
|
for i := len(s) - 1; i >= 0; i-- {
|
|
|
|
sum = sum + (int(s[i])-int('A')+1)*int(math.Pow(26, weight))
|
|
|
|
weight++
|
|
|
|
}
|
|
|
|
return sum - 1
|
|
|
|
}
|
|
|
|
|
2017-01-18 16:05:01 +08:00
|
|
|
// letterOnlyMapF is used in conjunction with strings.Map to return only the
|
|
|
|
// characters A-Z and a-z in a string.
|
2016-09-12 17:37:06 +08:00
|
|
|
func letterOnlyMapF(rune rune) rune {
|
|
|
|
switch {
|
|
|
|
case 'A' <= rune && rune <= 'Z':
|
|
|
|
return rune
|
|
|
|
case 'a' <= rune && rune <= 'z':
|
|
|
|
return rune - 32
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
2016-09-12 17:37:06 +08:00
|
|
|
return -1
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
|
|
|
|
2017-01-18 16:05:01 +08:00
|
|
|
// intOnlyMapF is used in conjunction with strings.Map to return only the
|
|
|
|
// numeric portions of a string.
|
2016-09-12 17:37:06 +08:00
|
|
|
func intOnlyMapF(rune rune) rune {
|
|
|
|
if rune >= 48 && rune < 58 {
|
|
|
|
return rune
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|
2016-09-12 17:37:06 +08:00
|
|
|
return -1
|
2016-08-30 11:51:31 +08:00
|
|
|
}
|