123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797 |
- package excelize
- import (
- "bytes"
- "encoding/json"
- "encoding/xml"
- "fmt"
- "io"
- "io/ioutil"
- "log"
- "os"
- "path"
- "path/filepath"
- "reflect"
- "regexp"
- "strconv"
- "strings"
- "unicode/utf8"
- "github.com/mohae/deepcopy"
- )
- func (f *File) NewSheet(name string) int {
-
- index := f.GetSheetIndex(name)
- if index != -1 {
- return index
- }
- f.DeleteSheet(name)
- f.SheetCount++
- wb := f.workbookReader()
- sheetID := 0
- for _, v := range wb.Sheets.Sheet {
- if v.SheetID > sheetID {
- sheetID = v.SheetID
- }
- }
- sheetID++
-
- f.setAppXML()
-
- f.setContentTypes("/xl/worksheets/sheet"+strconv.Itoa(sheetID)+".xml", ContentTypeSpreadSheetMLWorksheet)
-
- f.setSheet(sheetID, name)
-
- rID := f.addRels(f.getWorkbookRelsPath(), SourceRelationshipWorkSheet, fmt.Sprintf("/xl/worksheets/sheet%d.xml", sheetID), "")
-
- f.setWorkbook(name, sheetID, rID)
- return f.GetSheetIndex(name)
- }
- func (f *File) contentTypesReader() *xlsxTypes {
- var err error
- if f.ContentTypes == nil {
- f.ContentTypes = new(xlsxTypes)
- f.ContentTypes.Lock()
- defer f.ContentTypes.Unlock()
- if err = f.xmlNewDecoder(bytes.NewReader(namespaceStrictToTransitional(f.readXML("[Content_Types].xml")))).
- Decode(f.ContentTypes); err != nil && err != io.EOF {
- log.Printf("xml decode error: %s", err)
- }
- }
- return f.ContentTypes
- }
- func (f *File) contentTypesWriter() {
- if f.ContentTypes != nil {
- output, _ := xml.Marshal(f.ContentTypes)
- f.saveFileList("[Content_Types].xml", output)
- }
- }
- func (f *File) getWorkbookPath() (path string) {
- if rels := f.relsReader("_rels/.rels"); rels != nil {
- rels.Lock()
- defer rels.Unlock()
- for _, rel := range rels.Relationships {
- if rel.Type == SourceRelationshipOfficeDocument {
- path = strings.TrimPrefix(rel.Target, "/")
- return
- }
- }
- }
- return
- }
- func (f *File) getWorkbookRelsPath() (path string) {
- wbPath := f.getWorkbookPath()
- wbDir := filepath.Dir(wbPath)
- if wbDir == "." {
- path = "_rels/" + filepath.Base(wbPath) + ".rels"
- return
- }
- path = strings.TrimPrefix(filepath.Dir(wbPath)+"/_rels/"+filepath.Base(wbPath)+".rels", "/")
- return
- }
- func (f *File) workbookReader() *xlsxWorkbook {
- var err error
- if f.WorkBook == nil {
- wbPath := f.getWorkbookPath()
- f.WorkBook = new(xlsxWorkbook)
- if _, ok := f.xmlAttr[wbPath]; !ok {
- d := f.xmlNewDecoder(bytes.NewReader(namespaceStrictToTransitional(f.readXML(wbPath))))
- f.xmlAttr[wbPath] = append(f.xmlAttr[wbPath], getRootElement(d)...)
- f.addNameSpaces(wbPath, SourceRelationship)
- }
- if err = f.xmlNewDecoder(bytes.NewReader(namespaceStrictToTransitional(f.readXML(wbPath)))).
- Decode(f.WorkBook); err != nil && err != io.EOF {
- log.Printf("xml decode error: %s", err)
- }
- }
- return f.WorkBook
- }
- func (f *File) workBookWriter() {
- if f.WorkBook != nil {
- output, _ := xml.Marshal(f.WorkBook)
- f.saveFileList(f.getWorkbookPath(), replaceRelationshipsBytes(f.replaceNameSpaceBytes(f.getWorkbookPath(), output)))
- }
- }
- func (f *File) workSheetWriter() {
- var arr []byte
- buffer := bytes.NewBuffer(arr)
- encoder := xml.NewEncoder(buffer)
- f.Sheet.Range(func(p, ws interface{}) bool {
- if ws != nil {
- sheet := ws.(*xlsxWorksheet)
- for k, v := range sheet.SheetData.Row {
- sheet.SheetData.Row[k].C = trimCell(v.C)
- }
- if sheet.SheetPr != nil || sheet.Drawing != nil || sheet.Hyperlinks != nil || sheet.Picture != nil || sheet.TableParts != nil {
- f.addNameSpaces(p.(string), SourceRelationship)
- }
-
- _ = encoder.Encode(sheet)
- f.saveFileList(p.(string), replaceRelationshipsBytes(f.replaceNameSpaceBytes(p.(string), buffer.Bytes())))
- ok := f.checked[p.(string)]
- if ok {
- f.Sheet.Delete(p.(string))
- f.checked[p.(string)] = false
- }
- buffer.Reset()
- }
- return true
- })
- }
- func trimCell(column []xlsxC) []xlsxC {
- rowFull := true
- for i := range column {
- rowFull = column[i].hasValue() && rowFull
- }
- if rowFull {
- return column
- }
- col := make([]xlsxC, len(column))
- i := 0
- for _, c := range column {
- if c.hasValue() {
- col[i] = c
- i++
- }
- }
- return col[0:i]
- }
- func (f *File) setContentTypes(partName, contentType string) {
- content := f.contentTypesReader()
- content.Lock()
- defer content.Unlock()
- content.Overrides = append(content.Overrides, xlsxOverride{
- PartName: partName,
- ContentType: contentType,
- })
- }
- func (f *File) setSheet(index int, name string) {
- ws := xlsxWorksheet{
- Dimension: &xlsxDimension{Ref: "A1"},
- SheetViews: &xlsxSheetViews{
- SheetView: []xlsxSheetView{{WorkbookViewID: 0}},
- },
- }
- path := "xl/worksheets/sheet" + strconv.Itoa(index) + ".xml"
- f.sheetMap[trimSheetName(name)] = path
- f.Sheet.Store(path, &ws)
- f.xmlAttr[path] = []xml.Attr{NameSpaceSpreadSheet}
- }
- func (f *File) setWorkbook(name string, sheetID, rid int) {
- content := f.workbookReader()
- content.Sheets.Sheet = append(content.Sheets.Sheet, xlsxSheet{
- Name: trimSheetName(name),
- SheetID: sheetID,
- ID: "rId" + strconv.Itoa(rid),
- })
- }
- func (f *File) relsWriter() {
- f.Relationships.Range(func(path, rel interface{}) bool {
- if rel != nil {
- output, _ := xml.Marshal(rel.(*xlsxRelationships))
- if strings.HasPrefix(path.(string), "xl/worksheets/sheet/rels/sheet") {
- output = f.replaceNameSpaceBytes(path.(string), output)
- }
- f.saveFileList(path.(string), replaceRelationshipsBytes(output))
- }
- return true
- })
- }
- func (f *File) setAppXML() {
- f.saveFileList("docProps/app.xml", []byte(templateDocpropsApp))
- }
- func replaceRelationshipsBytes(content []byte) []byte {
- oldXmlns := []byte(`xmlns:relationships="http://schemas.openxmlformats.org/officeDocument/2006/relationships" relationships`)
- newXmlns := []byte("r")
- return bytesReplace(content, oldXmlns, newXmlns, -1)
- }
- func (f *File) SetActiveSheet(index int) {
- if index < 0 {
- index = 0
- }
- wb := f.workbookReader()
- for activeTab := range wb.Sheets.Sheet {
- if activeTab == index {
- if wb.BookViews == nil {
- wb.BookViews = &xlsxBookViews{}
- }
- if len(wb.BookViews.WorkBookView) > 0 {
- wb.BookViews.WorkBookView[0].ActiveTab = activeTab
- } else {
- wb.BookViews.WorkBookView = append(wb.BookViews.WorkBookView, xlsxWorkBookView{
- ActiveTab: activeTab,
- })
- }
- }
- }
- for idx, name := range f.GetSheetList() {
- ws, err := f.workSheetReader(name)
- if err != nil {
-
- return
- }
- if ws.SheetViews == nil {
- ws.SheetViews = &xlsxSheetViews{
- SheetView: []xlsxSheetView{{WorkbookViewID: 0}},
- }
- }
- if len(ws.SheetViews.SheetView) > 0 {
- ws.SheetViews.SheetView[0].TabSelected = false
- }
- if index == idx {
- if len(ws.SheetViews.SheetView) > 0 {
- ws.SheetViews.SheetView[0].TabSelected = true
- } else {
- ws.SheetViews.SheetView = append(ws.SheetViews.SheetView, xlsxSheetView{
- TabSelected: true,
- })
- }
- }
- }
- }
- func (f *File) GetActiveSheetIndex() (index int) {
- var sheetID = f.getActiveSheetID()
- wb := f.workbookReader()
- if wb != nil {
- for idx, sheet := range wb.Sheets.Sheet {
- if sheet.SheetID == sheetID {
- index = idx
- }
- }
- }
- return
- }
- func (f *File) getActiveSheetID() int {
- wb := f.workbookReader()
- if wb != nil {
- if wb.BookViews != nil && len(wb.BookViews.WorkBookView) > 0 {
- activeTab := wb.BookViews.WorkBookView[0].ActiveTab
- if len(wb.Sheets.Sheet) > activeTab && wb.Sheets.Sheet[activeTab].SheetID != 0 {
- return wb.Sheets.Sheet[activeTab].SheetID
- }
- }
- if len(wb.Sheets.Sheet) >= 1 {
- return wb.Sheets.Sheet[0].SheetID
- }
- }
- return 0
- }
- func (f *File) SetSheetName(oldName, newName string) {
- oldName = trimSheetName(oldName)
- newName = trimSheetName(newName)
- if newName == oldName {
- return
- }
- content := f.workbookReader()
- for k, v := range content.Sheets.Sheet {
- if v.Name == oldName {
- content.Sheets.Sheet[k].Name = newName
- f.sheetMap[newName] = f.sheetMap[oldName]
- delete(f.sheetMap, oldName)
- }
- }
- }
- func (f *File) GetSheetName(index int) (name string) {
- for idx, sheet := range f.GetSheetList() {
- if idx == index {
- name = sheet
- }
- }
- return
- }
- func (f *File) getSheetID(name string) int {
- var ID = -1
- for sheetID, sheet := range f.GetSheetMap() {
- if sheet == trimSheetName(name) {
- ID = sheetID
- }
- }
- return ID
- }
- func (f *File) GetSheetIndex(name string) int {
- var idx = -1
- for index, sheet := range f.GetSheetList() {
- if strings.EqualFold(sheet, trimSheetName(name)) {
- idx = index
- }
- }
- return idx
- }
- func (f *File) GetSheetMap() map[int]string {
- wb := f.workbookReader()
- sheetMap := map[int]string{}
- if wb != nil {
- for _, sheet := range wb.Sheets.Sheet {
- sheetMap[sheet.SheetID] = sheet.Name
- }
- }
- return sheetMap
- }
- func (f *File) GetSheetList() (list []string) {
- wb := f.workbookReader()
- if wb != nil {
- for _, sheet := range wb.Sheets.Sheet {
- list = append(list, sheet.Name)
- }
- }
- return
- }
- func (f *File) getSheetMap() map[string]string {
- maps := map[string]string{}
- for _, v := range f.workbookReader().Sheets.Sheet {
- for _, rel := range f.relsReader(f.getWorkbookRelsPath()).Relationships {
- if rel.ID == v.ID {
-
-
-
-
- path := filepath.ToSlash(strings.TrimPrefix(
- strings.Replace(filepath.Clean(fmt.Sprintf("%s/%s", filepath.Dir(f.getWorkbookPath()), rel.Target)), "\\", "/", -1), "/"))
- if strings.HasPrefix(rel.Target, "/") {
- path = filepath.ToSlash(strings.TrimPrefix(strings.Replace(filepath.Clean(rel.Target), "\\", "/", -1), "/"))
- }
- if _, ok := f.Pkg.Load(path); ok {
- maps[v.Name] = path
- }
- }
- }
- }
- return maps
- }
- func (f *File) SetSheetBackground(sheet, picture string) error {
- var err error
-
- if _, err = os.Stat(picture); os.IsNotExist(err) {
- return err
- }
- ext, ok := supportImageTypes[path.Ext(picture)]
- if !ok {
- return ErrImgExt
- }
- file, _ := ioutil.ReadFile(picture)
- name := f.addMedia(file, ext)
- sheetRels := "xl/worksheets/_rels/" + strings.TrimPrefix(f.sheetMap[trimSheetName(sheet)], "xl/worksheets/") + ".rels"
- rID := f.addRels(sheetRels, SourceRelationshipImage, strings.Replace(name, "xl", "..", 1), "")
- f.addSheetPicture(sheet, rID)
- f.addSheetNameSpace(sheet, SourceRelationship)
- f.setContentTypePartImageExtensions()
- return err
- }
- func (f *File) DeleteSheet(name string) {
- if f.SheetCount == 1 || f.GetSheetIndex(name) == -1 {
- return
- }
- sheetName := trimSheetName(name)
- wb := f.workbookReader()
- wbRels := f.relsReader(f.getWorkbookRelsPath())
- activeSheetName := f.GetSheetName(f.GetActiveSheetIndex())
- deleteLocalSheetID := f.GetSheetIndex(name)
-
- if wb.DefinedNames != nil {
- for idx := 0; idx < len(wb.DefinedNames.DefinedName); idx++ {
- dn := wb.DefinedNames.DefinedName[idx]
- if dn.LocalSheetID != nil {
- localSheetID := *dn.LocalSheetID
- if localSheetID == deleteLocalSheetID {
- wb.DefinedNames.DefinedName = append(wb.DefinedNames.DefinedName[:idx], wb.DefinedNames.DefinedName[idx+1:]...)
- idx--
- } else if localSheetID > deleteLocalSheetID {
- wb.DefinedNames.DefinedName[idx].LocalSheetID = intPtr(*dn.LocalSheetID - 1)
- }
- }
- }
- }
- for idx, sheet := range wb.Sheets.Sheet {
- if strings.EqualFold(sheet.Name, sheetName) {
- wb.Sheets.Sheet = append(wb.Sheets.Sheet[:idx], wb.Sheets.Sheet[idx+1:]...)
- var sheetXML, rels string
- if wbRels != nil {
- for _, rel := range wbRels.Relationships {
- if rel.ID == sheet.ID {
- sheetXML = rel.Target
- rels = "xl/worksheets/_rels/" + strings.TrimPrefix(f.sheetMap[sheetName], "xl/worksheets/") + ".rels"
- }
- }
- }
- target := f.deleteSheetFromWorkbookRels(sheet.ID)
- f.deleteSheetFromContentTypes(target)
- f.deleteCalcChain(sheet.SheetID, "")
- delete(f.sheetMap, sheet.Name)
- f.Pkg.Delete(sheetXML)
- f.Pkg.Delete(rels)
- f.Relationships.Delete(rels)
- f.Sheet.Delete(sheetXML)
- delete(f.xmlAttr, sheetXML)
- f.SheetCount--
- }
- }
- f.SetActiveSheet(f.GetSheetIndex(activeSheetName))
- }
- func (f *File) deleteSheetFromWorkbookRels(rID string) string {
- content := f.relsReader(f.getWorkbookRelsPath())
- content.Lock()
- defer content.Unlock()
- for k, v := range content.Relationships {
- if v.ID == rID {
- content.Relationships = append(content.Relationships[:k], content.Relationships[k+1:]...)
- return v.Target
- }
- }
- return ""
- }
- func (f *File) deleteSheetFromContentTypes(target string) {
- if !strings.HasPrefix(target, "/") {
- target = "/xl/" + target
- }
- content := f.contentTypesReader()
- content.Lock()
- defer content.Unlock()
- for k, v := range content.Overrides {
- if v.PartName == target {
- content.Overrides = append(content.Overrides[:k], content.Overrides[k+1:]...)
- }
- }
- }
- func (f *File) CopySheet(from, to int) error {
- if from < 0 || to < 0 || from == to || f.GetSheetName(from) == "" || f.GetSheetName(to) == "" {
- return ErrSheetIdx
- }
- return f.copySheet(from, to)
- }
- func (f *File) copySheet(from, to int) error {
- fromSheet := f.GetSheetName(from)
- sheet, err := f.workSheetReader(fromSheet)
- if err != nil {
- return err
- }
- worksheet := deepcopy.Copy(sheet).(*xlsxWorksheet)
- toSheetID := strconv.Itoa(f.getSheetID(f.GetSheetName(to)))
- path := "xl/worksheets/sheet" + toSheetID + ".xml"
- if len(worksheet.SheetViews.SheetView) > 0 {
- worksheet.SheetViews.SheetView[0].TabSelected = false
- }
- worksheet.Drawing = nil
- worksheet.TableParts = nil
- worksheet.PageSetUp = nil
- f.Sheet.Store(path, worksheet)
- toRels := "xl/worksheets/_rels/sheet" + toSheetID + ".xml.rels"
- fromRels := "xl/worksheets/_rels/sheet" + strconv.Itoa(f.getSheetID(fromSheet)) + ".xml.rels"
- if rels, ok := f.Pkg.Load(fromRels); ok && rels != nil {
- f.Pkg.Store(toRels, rels.([]byte))
- }
- fromSheetXMLPath := f.sheetMap[trimSheetName(fromSheet)]
- fromSheetAttr := f.xmlAttr[fromSheetXMLPath]
- f.xmlAttr[path] = fromSheetAttr
- return err
- }
- func (f *File) SetSheetVisible(name string, visible bool) error {
- name = trimSheetName(name)
- content := f.workbookReader()
- if visible {
- for k, v := range content.Sheets.Sheet {
- if v.Name == name {
- content.Sheets.Sheet[k].State = ""
- }
- }
- return nil
- }
- count := 0
- for _, v := range content.Sheets.Sheet {
- if v.State != "hidden" {
- count++
- }
- }
- for k, v := range content.Sheets.Sheet {
- xlsx, err := f.workSheetReader(v.Name)
- if err != nil {
- return err
- }
- tabSelected := false
- if len(xlsx.SheetViews.SheetView) > 0 {
- tabSelected = xlsx.SheetViews.SheetView[0].TabSelected
- }
- if v.Name == name && count > 1 && !tabSelected {
- content.Sheets.Sheet[k].State = "hidden"
- }
- }
- return nil
- }
- func parseFormatPanesSet(formatSet string) (*formatPanes, error) {
- format := formatPanes{}
- err := json.Unmarshal([]byte(formatSet), &format)
- return &format, err
- }
- func (f *File) SetPanes(sheet, panes string) error {
- fs, _ := parseFormatPanesSet(panes)
- ws, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- p := &xlsxPane{
- ActivePane: fs.ActivePane,
- TopLeftCell: fs.TopLeftCell,
- XSplit: float64(fs.XSplit),
- YSplit: float64(fs.YSplit),
- }
- if fs.Freeze {
- p.State = "frozen"
- }
- ws.SheetViews.SheetView[len(ws.SheetViews.SheetView)-1].Pane = p
- if !(fs.Freeze) && !(fs.Split) {
- if len(ws.SheetViews.SheetView) > 0 {
- ws.SheetViews.SheetView[len(ws.SheetViews.SheetView)-1].Pane = nil
- }
- }
- s := []*xlsxSelection{}
- for _, p := range fs.Panes {
- s = append(s, &xlsxSelection{
- ActiveCell: p.ActiveCell,
- Pane: p.Pane,
- SQRef: p.SQRef,
- })
- }
- ws.SheetViews.SheetView[len(ws.SheetViews.SheetView)-1].Selection = s
- return err
- }
- func (f *File) GetSheetVisible(name string) bool {
- content := f.workbookReader()
- visible := false
- for k, v := range content.Sheets.Sheet {
- if v.Name == trimSheetName(name) {
- if content.Sheets.Sheet[k].State == "" || content.Sheets.Sheet[k].State == "visible" {
- visible = true
- }
- }
- }
- return visible
- }
- func (f *File) SearchSheet(sheet, value string, reg ...bool) ([]string, error) {
- var (
- regSearch bool
- result []string
- )
- for _, r := range reg {
- regSearch = r
- }
- name, ok := f.sheetMap[trimSheetName(sheet)]
- if !ok {
- return result, ErrSheetNotExist{sheet}
- }
- if ws, ok := f.Sheet.Load(name); ok && ws != nil {
-
- output, _ := xml.Marshal(ws.(*xlsxWorksheet))
- f.saveFileList(name, f.replaceNameSpaceBytes(name, output))
- }
- return f.searchSheet(name, value, regSearch)
- }
- func (f *File) searchSheet(name, value string, regSearch bool) (result []string, err error) {
- var (
- cellName, inElement string
- cellCol, row int
- d *xlsxSST
- )
- d = f.sharedStringsReader()
- decoder := f.xmlNewDecoder(bytes.NewReader(f.readXML(name)))
- for {
- var token xml.Token
- token, err = decoder.Token()
- if err != nil || token == nil {
- if err == io.EOF {
- err = nil
- }
- break
- }
- switch xmlElement := token.(type) {
- case xml.StartElement:
- inElement = xmlElement.Name.Local
- if inElement == "row" {
- row, err = attrValToInt("r", xmlElement.Attr)
- if err != nil {
- return
- }
- }
- if inElement == "c" {
- colCell := xlsxC{}
- _ = decoder.DecodeElement(&colCell, &xmlElement)
- val, _ := colCell.getValueFrom(f, d)
- if regSearch {
- regex := regexp.MustCompile(value)
- if !regex.MatchString(val) {
- continue
- }
- } else {
- if val != value {
- continue
- }
- }
- cellCol, _, err = CellNameToCoordinates(colCell.R)
- if err != nil {
- return result, err
- }
- cellName, err = CoordinatesToCellName(cellCol, row)
- if err != nil {
- return result, err
- }
- result = append(result, cellName)
- }
- default:
- }
- }
- return
- }
- func attrValToInt(name string, attrs []xml.Attr) (val int, err error) {
- for _, attr := range attrs {
- if attr.Name.Local == name {
- val, err = strconv.Atoi(attr.Value)
- if err != nil {
- return
- }
- }
- }
- return
- }
- func (f *File) SetHeaderFooter(sheet string, settings *FormatHeaderFooter) error {
- ws, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- if settings == nil {
- ws.HeaderFooter = nil
- return err
- }
- v := reflect.ValueOf(*settings)
-
-
- for i := 4; i < v.NumField()-1; i++ {
- if v.Field(i).Len() >= 255 {
- return fmt.Errorf("field %s must be less than 255 characters", v.Type().Field(i).Name)
- }
- }
- ws.HeaderFooter = &xlsxHeaderFooter{
- AlignWithMargins: settings.AlignWithMargins,
- DifferentFirst: settings.DifferentFirst,
- DifferentOddEven: settings.DifferentOddEven,
- ScaleWithDoc: settings.ScaleWithDoc,
- OddHeader: settings.OddHeader,
- OddFooter: settings.OddFooter,
- EvenHeader: settings.EvenHeader,
- EvenFooter: settings.EvenFooter,
- FirstFooter: settings.FirstFooter,
- FirstHeader: settings.FirstHeader,
- }
- return err
- }
- func (f *File) ProtectSheet(sheet string, settings *FormatSheetProtection) error {
- ws, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- if settings == nil {
- settings = &FormatSheetProtection{
- EditObjects: true,
- EditScenarios: true,
- SelectLockedCells: true,
- }
- }
- ws.SheetProtection = &xlsxSheetProtection{
- AutoFilter: settings.AutoFilter,
- DeleteColumns: settings.DeleteColumns,
- DeleteRows: settings.DeleteRows,
- FormatCells: settings.FormatCells,
- FormatColumns: settings.FormatColumns,
- FormatRows: settings.FormatRows,
- InsertColumns: settings.InsertColumns,
- InsertHyperlinks: settings.InsertHyperlinks,
- InsertRows: settings.InsertRows,
- Objects: settings.EditObjects,
- PivotTables: settings.PivotTables,
- Scenarios: settings.EditScenarios,
- SelectLockedCells: settings.SelectLockedCells,
- SelectUnlockedCells: settings.SelectUnlockedCells,
- Sheet: true,
- Sort: settings.Sort,
- }
- if settings.Password != "" {
- ws.SheetProtection.Password = genSheetPasswd(settings.Password)
- }
- return err
- }
- func (f *File) UnprotectSheet(sheet string) error {
- ws, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- ws.SheetProtection = nil
- return err
- }
- func trimSheetName(name string) string {
- if strings.ContainsAny(name, ":\\/?*[]") || utf8.RuneCountInString(name) > 31 {
- r := make([]rune, 0, 31)
- for _, v := range name {
- switch v {
- case 58, 92, 47, 63, 42, 91, 93:
- continue
- default:
- r = append(r, v)
- }
- if len(r) == 31 {
- break
- }
- }
- name = string(r)
- }
- return name
- }
- type PageLayoutOption interface {
- setPageLayout(layout *xlsxPageSetUp)
- }
- type PageLayoutOptionPtr interface {
- PageLayoutOption
- getPageLayout(layout *xlsxPageSetUp)
- }
- type (
-
- BlackAndWhite bool
-
-
- FirstPageNumber uint
-
-
- PageLayoutOrientation string
-
- PageLayoutPaperSize int
-
- FitToHeight int
-
- FitToWidth int
-
-
-
- PageLayoutScale uint
- )
- const (
-
- OrientationPortrait = "portrait"
-
- OrientationLandscape = "landscape"
- )
- func (p BlackAndWhite) setPageLayout(ps *xlsxPageSetUp) {
- ps.BlackAndWhite = bool(p)
- }
- func (p *BlackAndWhite) getPageLayout(ps *xlsxPageSetUp) {
- if ps == nil {
- *p = false
- return
- }
- *p = BlackAndWhite(ps.BlackAndWhite)
- }
- func (p FirstPageNumber) setPageLayout(ps *xlsxPageSetUp) {
- if 0 < int(p) {
- ps.FirstPageNumber = strconv.Itoa(int(p))
- ps.UseFirstPageNumber = true
- }
- }
- func (p *FirstPageNumber) getPageLayout(ps *xlsxPageSetUp) {
- if ps != nil && ps.UseFirstPageNumber {
- if number, _ := strconv.Atoi(ps.FirstPageNumber); number != 0 {
- *p = FirstPageNumber(number)
- return
- }
- }
- *p = 1
- }
- func (o PageLayoutOrientation) setPageLayout(ps *xlsxPageSetUp) {
- ps.Orientation = string(o)
- }
- func (o *PageLayoutOrientation) getPageLayout(ps *xlsxPageSetUp) {
-
- if ps == nil || ps.Orientation == "" {
- *o = OrientationPortrait
- return
- }
- *o = PageLayoutOrientation(ps.Orientation)
- }
- func (p PageLayoutPaperSize) setPageLayout(ps *xlsxPageSetUp) {
- ps.PaperSize = int(p)
- }
- func (p *PageLayoutPaperSize) getPageLayout(ps *xlsxPageSetUp) {
-
- if ps == nil || ps.PaperSize == 0 {
- *p = 1
- return
- }
- *p = PageLayoutPaperSize(ps.PaperSize)
- }
- func (p FitToHeight) setPageLayout(ps *xlsxPageSetUp) {
- if int(p) > 0 {
- ps.FitToHeight = int(p)
- }
- }
- func (p *FitToHeight) getPageLayout(ps *xlsxPageSetUp) {
- if ps == nil || ps.FitToHeight == 0 {
- *p = 1
- return
- }
- *p = FitToHeight(ps.FitToHeight)
- }
- func (p FitToWidth) setPageLayout(ps *xlsxPageSetUp) {
- if int(p) > 0 {
- ps.FitToWidth = int(p)
- }
- }
- func (p *FitToWidth) getPageLayout(ps *xlsxPageSetUp) {
- if ps == nil || ps.FitToWidth == 0 {
- *p = 1
- return
- }
- *p = FitToWidth(ps.FitToWidth)
- }
- func (p PageLayoutScale) setPageLayout(ps *xlsxPageSetUp) {
- if 10 <= int(p) && int(p) <= 400 {
- ps.Scale = int(p)
- }
- }
- func (p *PageLayoutScale) getPageLayout(ps *xlsxPageSetUp) {
- if ps == nil || ps.Scale < 10 || ps.Scale > 400 {
- *p = 100
- return
- }
- *p = PageLayoutScale(ps.Scale)
- }
- func (f *File) SetPageLayout(sheet string, opts ...PageLayoutOption) error {
- s, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- ps := s.PageSetUp
- if ps == nil {
- ps = new(xlsxPageSetUp)
- s.PageSetUp = ps
- }
- for _, opt := range opts {
- opt.setPageLayout(ps)
- }
- return err
- }
- func (f *File) GetPageLayout(sheet string, opts ...PageLayoutOptionPtr) error {
- s, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- ps := s.PageSetUp
- for _, opt := range opts {
- opt.getPageLayout(ps)
- }
- return err
- }
- func (f *File) SetDefinedName(definedName *DefinedName) error {
- wb := f.workbookReader()
- d := xlsxDefinedName{
- Name: definedName.Name,
- Comment: definedName.Comment,
- Data: definedName.RefersTo,
- }
- if definedName.Scope != "" {
- if sheetIndex := f.GetSheetIndex(definedName.Scope); sheetIndex >= 0 {
- d.LocalSheetID = &sheetIndex
- }
- }
- if wb.DefinedNames != nil {
- for _, dn := range wb.DefinedNames.DefinedName {
- var scope string
- if dn.LocalSheetID != nil {
- scope = f.GetSheetName(*dn.LocalSheetID)
- }
- if scope == definedName.Scope && dn.Name == definedName.Name {
- return ErrDefinedNameduplicate
- }
- }
- wb.DefinedNames.DefinedName = append(wb.DefinedNames.DefinedName, d)
- return nil
- }
- wb.DefinedNames = &xlsxDefinedNames{
- DefinedName: []xlsxDefinedName{d},
- }
- return nil
- }
- func (f *File) DeleteDefinedName(definedName *DefinedName) error {
- wb := f.workbookReader()
- if wb.DefinedNames != nil {
- for idx, dn := range wb.DefinedNames.DefinedName {
- scope := "Workbook"
- deleteScope := definedName.Scope
- if deleteScope == "" {
- deleteScope = "Workbook"
- }
- if dn.LocalSheetID != nil {
- scope = f.GetSheetName(*dn.LocalSheetID)
- }
- if scope == deleteScope && dn.Name == definedName.Name {
- wb.DefinedNames.DefinedName = append(wb.DefinedNames.DefinedName[:idx], wb.DefinedNames.DefinedName[idx+1:]...)
- return nil
- }
- }
- }
- return ErrDefinedNameScope
- }
- func (f *File) GetDefinedName() []DefinedName {
- var definedNames []DefinedName
- wb := f.workbookReader()
- if wb.DefinedNames != nil {
- for _, dn := range wb.DefinedNames.DefinedName {
- definedName := DefinedName{
- Name: dn.Name,
- Comment: dn.Comment,
- RefersTo: dn.Data,
- Scope: "Workbook",
- }
- if dn.LocalSheetID != nil && *dn.LocalSheetID >= 0 {
- definedName.Scope = f.GetSheetName(*dn.LocalSheetID)
- }
- definedNames = append(definedNames, definedName)
- }
- }
- return definedNames
- }
- func (f *File) GroupSheets(sheets []string) error {
-
- var inActiveSheet bool
- activeSheet := f.GetActiveSheetIndex()
- sheetMap := f.GetSheetList()
- for idx, sheetName := range sheetMap {
- for _, s := range sheets {
- if s == sheetName && idx == activeSheet {
- inActiveSheet = true
- }
- }
- }
- if !inActiveSheet {
- return ErrGroupSheets
- }
-
- wss := []*xlsxWorksheet{}
- for _, sheet := range sheets {
- worksheet, err := f.workSheetReader(sheet)
- if err != nil {
- return err
- }
- wss = append(wss, worksheet)
- }
- for _, ws := range wss {
- sheetViews := ws.SheetViews.SheetView
- if len(sheetViews) > 0 {
- for idx := range sheetViews {
- ws.SheetViews.SheetView[idx].TabSelected = true
- }
- continue
- }
- }
- return nil
- }
- func (f *File) UngroupSheets() error {
- activeSheet := f.GetActiveSheetIndex()
- for index, sheet := range f.GetSheetList() {
- if activeSheet == index {
- continue
- }
- ws, _ := f.workSheetReader(sheet)
- sheetViews := ws.SheetViews.SheetView
- if len(sheetViews) > 0 {
- for idx := range sheetViews {
- ws.SheetViews.SheetView[idx].TabSelected = false
- }
- }
- }
- return nil
- }
- func (f *File) InsertPageBreak(sheet, cell string) (err error) {
- var ws *xlsxWorksheet
- var row, col int
- var rowBrk, colBrk = -1, -1
- if ws, err = f.workSheetReader(sheet); err != nil {
- return
- }
- if col, row, err = CellNameToCoordinates(cell); err != nil {
- return
- }
- col--
- row--
- if col == row && col == 0 {
- return
- }
- if ws.RowBreaks == nil {
- ws.RowBreaks = &xlsxBreaks{}
- }
- if ws.ColBreaks == nil {
- ws.ColBreaks = &xlsxBreaks{}
- }
- for idx, brk := range ws.RowBreaks.Brk {
- if brk.ID == row {
- rowBrk = idx
- }
- }
- for idx, brk := range ws.ColBreaks.Brk {
- if brk.ID == col {
- colBrk = idx
- }
- }
- if row != 0 && rowBrk == -1 {
- ws.RowBreaks.Brk = append(ws.RowBreaks.Brk, &xlsxBrk{
- ID: row,
- Max: 16383,
- Man: true,
- })
- ws.RowBreaks.ManualBreakCount++
- }
- if col != 0 && colBrk == -1 {
- ws.ColBreaks.Brk = append(ws.ColBreaks.Brk, &xlsxBrk{
- ID: col,
- Max: 1048575,
- Man: true,
- })
- ws.ColBreaks.ManualBreakCount++
- }
- ws.RowBreaks.Count = len(ws.RowBreaks.Brk)
- ws.ColBreaks.Count = len(ws.ColBreaks.Brk)
- return
- }
- func (f *File) RemovePageBreak(sheet, cell string) (err error) {
- var ws *xlsxWorksheet
- var row, col int
- if ws, err = f.workSheetReader(sheet); err != nil {
- return
- }
- if col, row, err = CellNameToCoordinates(cell); err != nil {
- return
- }
- col--
- row--
- if col == row && col == 0 {
- return
- }
- removeBrk := func(ID int, brks []*xlsxBrk) []*xlsxBrk {
- for i, brk := range brks {
- if brk.ID == ID {
- brks = append(brks[:i], brks[i+1:]...)
- }
- }
- return brks
- }
- if ws.RowBreaks == nil || ws.ColBreaks == nil {
- return
- }
- rowBrks := len(ws.RowBreaks.Brk)
- colBrks := len(ws.ColBreaks.Brk)
- if rowBrks > 0 && rowBrks == colBrks {
- ws.RowBreaks.Brk = removeBrk(row, ws.RowBreaks.Brk)
- ws.ColBreaks.Brk = removeBrk(col, ws.ColBreaks.Brk)
- ws.RowBreaks.Count = len(ws.RowBreaks.Brk)
- ws.ColBreaks.Count = len(ws.ColBreaks.Brk)
- ws.RowBreaks.ManualBreakCount--
- ws.ColBreaks.ManualBreakCount--
- return
- }
- if rowBrks > 0 && rowBrks > colBrks {
- ws.RowBreaks.Brk = removeBrk(row, ws.RowBreaks.Brk)
- ws.RowBreaks.Count = len(ws.RowBreaks.Brk)
- ws.RowBreaks.ManualBreakCount--
- return
- }
- if colBrks > 0 && colBrks > rowBrks {
- ws.ColBreaks.Brk = removeBrk(col, ws.ColBreaks.Brk)
- ws.ColBreaks.Count = len(ws.ColBreaks.Brk)
- ws.ColBreaks.ManualBreakCount--
- }
- return
- }
- func (f *File) relsReader(path string) *xlsxRelationships {
- var err error
- rels, _ := f.Relationships.Load(path)
- if rels == nil {
- if _, ok := f.Pkg.Load(path); ok {
- c := xlsxRelationships{}
- if err = f.xmlNewDecoder(bytes.NewReader(namespaceStrictToTransitional(f.readXML(path)))).
- Decode(&c); err != nil && err != io.EOF {
- log.Printf("xml decode error: %s", err)
- }
- f.Relationships.Store(path, &c)
- }
- }
- if rels, _ = f.Relationships.Load(path); rels != nil {
- return rels.(*xlsxRelationships)
- }
- return nil
- }
- func prepareSheetXML(ws *xlsxWorksheet, col int, row int) {
- ws.Lock()
- defer ws.Unlock()
- rowCount := len(ws.SheetData.Row)
- sizeHint := 0
- var ht float64
- var customHeight bool
- if ws.SheetFormatPr != nil && ws.SheetFormatPr.CustomHeight {
- ht = ws.SheetFormatPr.DefaultRowHeight
- customHeight = true
- }
- if rowCount > 0 {
- sizeHint = len(ws.SheetData.Row[rowCount-1].C)
- }
- if rowCount < row {
-
- for rowIdx := rowCount; rowIdx < row; rowIdx++ {
- ws.SheetData.Row = append(ws.SheetData.Row, xlsxRow{R: rowIdx + 1, CustomHeight: customHeight, Ht: ht, C: make([]xlsxC, 0, sizeHint)})
- }
- }
- rowData := &ws.SheetData.Row[row-1]
- fillColumns(rowData, col, row)
- }
- func fillColumns(rowData *xlsxRow, col, row int) {
- cellCount := len(rowData.C)
- if cellCount < col {
- for colIdx := cellCount; colIdx < col; colIdx++ {
- cellName, _ := CoordinatesToCellName(colIdx+1, row)
- rowData.C = append(rowData.C, xlsxC{R: cellName})
- }
- }
- }
- func makeContiguousColumns(ws *xlsxWorksheet, fromRow, toRow, colCount int) {
- ws.Lock()
- defer ws.Unlock()
- for ; fromRow < toRow; fromRow++ {
- rowData := &ws.SheetData.Row[fromRow-1]
- fillColumns(rowData, colCount, fromRow)
- }
- }
|