53 lines
1002 B
Go
53 lines
1002 B
Go
package xls
|
|
|
|
import (
|
|
"github.com/extrame/ole2"
|
|
"io"
|
|
"io/ioutil"
|
|
)
|
|
|
|
//Open one xls file
|
|
func Open(file string, charset string) (*WorkBook, error) {
|
|
if bts, err := ioutil.ReadFile(file); err == nil {
|
|
return parse(bts, charset)
|
|
} else {
|
|
return nil, err
|
|
}
|
|
|
|
}
|
|
|
|
//Open xls file from reader
|
|
func OpenReader(reader io.ReadCloser, charset string) (*WorkBook, error) {
|
|
if bts, err := ioutil.ReadAll(reader); err == nil {
|
|
return parse(bts, charset)
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func parse(bts []byte, charset string) (wb *WorkBook, err error) {
|
|
var ole *ole2.Ole
|
|
if ole, err = ole2.Open(bts, charset); err == nil {
|
|
var dir []*ole2.File
|
|
if dir, err = ole.ListDir(); err == nil {
|
|
var book *ole2.File
|
|
for _, file := range dir {
|
|
name := file.Name()
|
|
if name == "Workbook" {
|
|
book = file
|
|
// break
|
|
}
|
|
if name == "Book" {
|
|
book = file
|
|
// break
|
|
}
|
|
}
|
|
if book != nil {
|
|
wb = newWorkBookFromOle2(ole.OpenFile(book))
|
|
return
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|