sheet.go 28.1 KB
Newer Older
xurime's avatar
xurime 已提交
1 2 3 4 5 6 7 8
// Copyright 2016 - 2018 The excelize Authors. All rights reserved. Use of
// this source code is governed by a BSD-style license that can be found in
// the LICENSE file.
//
// Package excelize providing a set of functions that allow you to write to
// and read from XLSX files. Support reads and writes XLSX file generated by
// Microsoft Excel™ 2007 and later. Support save file without losing original
// charts of XLSX. This library needs Go version 1.8 or later.
xurime's avatar
xurime 已提交
9

xurime's avatar
xurime 已提交
10 11 12
package excelize

import (
13
	"bytes"
xurime's avatar
xurime 已提交
14
	"encoding/json"
xurime's avatar
xurime 已提交
15
	"encoding/xml"
16
	"errors"
17
	"fmt"
18
	"io/ioutil"
19 20
	"os"
	"path"
R
r-uchino 已提交
21
	"regexp"
xurime's avatar
xurime 已提交
22 23
	"strconv"
	"strings"
24
	"unicode/utf8"
25 26

	"github.com/mohae/deepcopy"
xurime's avatar
xurime 已提交
27 28
)

S
Sebastian Willing 已提交
29
// NewSheet provides function to create a new sheet by given worksheet name.
xurime's avatar
xurime 已提交
30 31
// When creating a new XLSX file, the default sheet will be created. Returns
// the number of sheets in the workbook (file) after appending the new sheet.
32
func (f *File) NewSheet(name string) int {
33 34 35 36
	// Check if the worksheet already exists
	if f.GetSheetIndex(name) != 0 {
		return f.SheetCount
	}
37
	f.DeleteSheet(name)
38
	f.SheetCount++
39 40 41 42 43 44 45 46
	wb := f.workbookReader()
	sheetID := 0
	for _, v := range wb.Sheets.Sheet {
		if v.SheetID > sheetID {
			sheetID = v.SheetID
		}
	}
	sheetID++
xurime's avatar
xurime 已提交
47
	// Update docProps/app.xml
48
	f.setAppXML()
xurime's avatar
xurime 已提交
49
	// Update [Content_Types].xml
50
	f.setContentTypes(sheetID)
xurime's avatar
xurime 已提交
51
	// Create new sheet /xl/worksheets/sheet%d.xml
52
	f.setSheet(sheetID, name)
xurime's avatar
xurime 已提交
53
	// Update xl/_rels/workbook.xml.rels
54
	rID := f.addXlsxWorkbookRels(sheetID)
xurime's avatar
xurime 已提交
55
	// Update xl/workbook.xml
56 57
	f.setWorkbook(name, sheetID, rID)
	return sheetID
xurime's avatar
xurime 已提交
58 59
}

xurime's avatar
xurime 已提交
60
// contentTypesReader provides a function to get the pointer to the
xurime's avatar
xurime 已提交
61 62 63 64
// [Content_Types].xml structure after deserialization.
func (f *File) contentTypesReader() *xlsxTypes {
	if f.ContentTypes == nil {
		var content xlsxTypes
65
		_ = xml.Unmarshal(namespaceStrictToTransitional(f.readXML("[Content_Types].xml")), &content)
xurime's avatar
xurime 已提交
66 67 68 69 70
		f.ContentTypes = &content
	}
	return f.ContentTypes
}

xurime's avatar
xurime 已提交
71
// contentTypesWriter provides a function to save [Content_Types].xml after
xurime's avatar
xurime 已提交
72 73 74 75
// serialize structure.
func (f *File) contentTypesWriter() {
	if f.ContentTypes != nil {
		output, _ := xml.Marshal(f.ContentTypes)
76
		f.saveFileList("[Content_Types].xml", output)
xurime's avatar
xurime 已提交
77 78 79
	}
}

xurime's avatar
xurime 已提交
80
// workbookReader provides a function to get the pointer to the xl/workbook.xml
xurime's avatar
xurime 已提交
81 82 83 84
// structure after deserialization.
func (f *File) workbookReader() *xlsxWorkbook {
	if f.WorkBook == nil {
		var content xlsxWorkbook
85
		_ = xml.Unmarshal(namespaceStrictToTransitional(f.readXML("xl/workbook.xml")), &content)
xurime's avatar
xurime 已提交
86 87 88 89 90
		f.WorkBook = &content
	}
	return f.WorkBook
}

xurime's avatar
xurime 已提交
91
// workbookWriter provides a function to save xl/workbook.xml after serialize
xurime's avatar
xurime 已提交
92 93 94 95
// structure.
func (f *File) workbookWriter() {
	if f.WorkBook != nil {
		output, _ := xml.Marshal(f.WorkBook)
96
		f.saveFileList("xl/workbook.xml", replaceRelationshipsNameSpaceBytes(output))
xurime's avatar
xurime 已提交
97 98 99
	}
}

xurime's avatar
xurime 已提交
100
// worksheetWriter provides a function to save xl/worksheets/sheet%d.xml after
xurime's avatar
xurime 已提交
101 102 103 104
// serialize structure.
func (f *File) worksheetWriter() {
	for path, sheet := range f.Sheet {
		if sheet != nil {
105 106 107
			for k, v := range sheet.SheetData.Row {
				f.Sheet[path].SheetData.Row[k].C = trimCell(v.C)
			}
xurime's avatar
xurime 已提交
108
			output, _ := xml.Marshal(sheet)
109
			f.saveFileList(path, replaceWorkSheetsRelationshipsNameSpaceBytes(output))
110 111 112 113
			ok := f.checked[path]
			if ok {
				f.checked[path] = false
			}
xurime's avatar
xurime 已提交
114 115
		}
	}
xurime's avatar
xurime 已提交
116 117
}

xurime's avatar
xurime 已提交
118
// trimCell provides a function to trim blank cells which created by completeCol.
119
func trimCell(column []xlsxC) []xlsxC {
120 121
	col := make([]xlsxC, len(column))
	i := 0
122
	for _, c := range column {
xurime's avatar
xurime 已提交
123
		if c.S != 0 || c.V != "" || c.F != nil || c.T != "" {
124 125
			col[i] = c
			i++
126 127
		}
	}
128
	return col[0:i]
129 130
}

131 132
// setContentTypes provides a function to read and update property of contents
// type of XLSX.
133
func (f *File) setContentTypes(index int) {
xurime's avatar
xurime 已提交
134
	content := f.contentTypesReader()
xurime's avatar
xurime 已提交
135
	content.Overrides = append(content.Overrides, xlsxOverride{
136 137
		PartName:    "/xl/worksheets/sheet" + strconv.Itoa(index) + ".xml",
		ContentType: "application/vnd.openxmlformats-officedocument.spreadsheetml.worksheet+xml",
xurime's avatar
xurime 已提交
138 139 140
	})
}

141
// setSheet provides a function to update sheet property by given index.
142
func (f *File) setSheet(index int, name string) {
xurime's avatar
xurime 已提交
143
	var xlsx xlsxWorksheet
144
	xlsx.Dimension.Ref = "A1"
xurime's avatar
xurime 已提交
145
	xlsx.SheetViews.SheetView = append(xlsx.SheetViews.SheetView, xlsxSheetView{
xurime's avatar
xurime 已提交
146
		WorkbookViewID: 0,
xurime's avatar
xurime 已提交
147
	})
148
	path := "xl/worksheets/sheet" + strconv.Itoa(index) + ".xml"
149
	f.sheetMap[trimSheetName(name)] = path
xurime's avatar
xurime 已提交
150
	f.Sheet[path] = &xlsx
xurime's avatar
xurime 已提交
151 152
}

153 154
// setWorkbook update workbook property of XLSX. Maximum 31 characters are
// allowed in sheet title.
155
func (f *File) setWorkbook(name string, sheetID, rid int) {
xurime's avatar
xurime 已提交
156
	content := f.workbookReader()
xurime's avatar
xurime 已提交
157
	content.Sheets.Sheet = append(content.Sheets.Sheet, xlsxSheet{
158
		Name:    trimSheetName(name),
159
		SheetID: sheetID,
160
		ID:      "rId" + strconv.Itoa(rid),
xurime's avatar
xurime 已提交
161
	})
xurime's avatar
xurime 已提交
162 163
}

xurime's avatar
xurime 已提交
164
// workbookRelsReader provides a function to read and unmarshal workbook
xurime's avatar
xurime 已提交
165 166 167 168
// relationships of XLSX file.
func (f *File) workbookRelsReader() *xlsxWorkbookRels {
	if f.WorkBookRels == nil {
		var content xlsxWorkbookRels
169
		_ = xml.Unmarshal(namespaceStrictToTransitional(f.readXML("xl/_rels/workbook.xml.rels")), &content)
xurime's avatar
xurime 已提交
170
		f.WorkBookRels = &content
xurime's avatar
xurime 已提交
171
	}
xurime's avatar
xurime 已提交
172
	return f.WorkBookRels
xurime's avatar
xurime 已提交
173 174
}

xurime's avatar
xurime 已提交
175
// workbookRelsWriter provides a function to save xl/_rels/workbook.xml.rels after
xurime's avatar
xurime 已提交
176 177 178 179
// serialize structure.
func (f *File) workbookRelsWriter() {
	if f.WorkBookRels != nil {
		output, _ := xml.Marshal(f.WorkBookRels)
180
		f.saveFileList("xl/_rels/workbook.xml.rels", output)
xurime's avatar
xurime 已提交
181
	}
xurime's avatar
xurime 已提交
182 183
}

184
// addXlsxWorkbookRels update workbook relationships property of XLSX.
185
func (f *File) addXlsxWorkbookRels(sheet int) int {
xurime's avatar
xurime 已提交
186
	content := f.workbookRelsReader()
187 188 189 190 191 192 193 194
	rID := 0
	for _, v := range content.Relationships {
		t, _ := strconv.Atoi(strings.TrimPrefix(v.ID, "rId"))
		if t > rID {
			rID = t
		}
	}
	rID++
195
	ID := bytes.Buffer{}
196
	ID.WriteString("rId")
197 198
	ID.WriteString(strconv.Itoa(rID))
	target := bytes.Buffer{}
199
	target.WriteString("worksheets/sheet")
200
	target.WriteString(strconv.Itoa(sheet))
201
	target.WriteString(".xml")
xurime's avatar
xurime 已提交
202
	content.Relationships = append(content.Relationships, xlsxWorkbookRelation{
203 204
		ID:     ID.String(),
		Target: target.String(),
205
		Type:   SourceRelationshipWorkSheet,
xurime's avatar
xurime 已提交
206
	})
207
	return rID
xurime's avatar
xurime 已提交
208 209
}

210
// setAppXML update docProps/app.xml file of XML.
211
func (f *File) setAppXML() {
212
	f.saveFileList("docProps/app.xml", []byte(templateDocpropsApp))
xurime's avatar
xurime 已提交
213 214
}

215 216 217 218 219 220 221
// replaceRelationshipsNameSpaceBytes; Some tools that read XLSX files have
// very strict requirements about the structure of the input XML. In
// particular both Numbers on the Mac and SAS dislike inline XML namespace
// declarations, or namespace prefixes that don't match the ones that Excel
// itself uses. This is a problem because the Go XML library doesn't multiple
// namespace declarations in a single element of a document. This function is
// a horrible hack to fix that after the XML marshalling is completed.
222 223 224 225 226 227
func replaceRelationshipsNameSpaceBytes(workbookMarshal []byte) []byte {
	oldXmlns := []byte(`<workbook xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main">`)
	newXmlns := []byte(`<workbook xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships" xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" mc:Ignorable="x15" xmlns:x15="http://schemas.microsoft.com/office/spreadsheetml/2010/11/main">`)
	return bytes.Replace(workbookMarshal, oldXmlns, newXmlns, -1)
}

228
// SetActiveSheet provides function to set default active worksheet of XLSX by
S
Sebastian Willing 已提交
229 230
// given index. Note that active index is different from the index returned by
// function GetSheetMap(). It should be greater than 0 and less than total
231
// worksheet numbers.
232
func (f *File) SetActiveSheet(index int) {
xurime's avatar
xurime 已提交
233 234 235
	if index < 1 {
		index = 1
	}
236 237 238 239 240 241 242 243 244 245 246
	wb := f.workbookReader()
	for activeTab, sheet := range wb.Sheets.Sheet {
		if sheet.SheetID == index {
			if len(wb.BookViews.WorkBookView) > 0 {
				wb.BookViews.WorkBookView[0].ActiveTab = activeTab
			} else {
				wb.BookViews.WorkBookView = append(wb.BookViews.WorkBookView, xlsxWorkBookView{
					ActiveTab: activeTab,
				})
			}
		}
xurime's avatar
xurime 已提交
247
	}
248 249
	for idx, name := range f.GetSheetMap() {
		xlsx := f.workSheetReader(name)
250 251 252
		if len(xlsx.SheetViews.SheetView) > 0 {
			xlsx.SheetViews.SheetView[0].TabSelected = false
		}
253
		if index == idx {
xurime's avatar
xurime 已提交
254 255 256 257 258 259 260 261 262 263 264
			if len(xlsx.SheetViews.SheetView) > 0 {
				xlsx.SheetViews.SheetView[0].TabSelected = true
			} else {
				xlsx.SheetViews.SheetView = append(xlsx.SheetViews.SheetView, xlsxSheetView{
					TabSelected: true,
				})
			}
		}
	}
}

xurime's avatar
xurime 已提交
265 266
// GetActiveSheetIndex provides a function to get active sheet index of the
// XLSX. If not found the active sheet will be return integer 0.
267
func (f *File) GetActiveSheetIndex() int {
268 269
	for idx, name := range f.GetSheetMap() {
		xlsx := f.workSheetReader(name)
270 271
		for _, sheetView := range xlsx.SheetViews.SheetView {
			if sheetView.TabSelected {
272
				return idx
273
			}
274 275
		}
	}
276
	return 0
277 278
}

xurime's avatar
xurime 已提交
279
// SetSheetName provides a function to set the worksheet name be given old and new
280
// worksheet name. Maximum 31 characters are allowed in sheet title and this
281 282 283
// function only changes the name of the sheet and will not update the sheet
// name in the formula or reference associated with the cell. So there may be
// problem formula error or reference missing.
284
func (f *File) SetSheetName(oldName, newName string) {
285 286
	oldName = trimSheetName(oldName)
	newName = trimSheetName(newName)
xurime's avatar
xurime 已提交
287
	content := f.workbookReader()
288 289 290
	for k, v := range content.Sheets.Sheet {
		if v.Name == oldName {
			content.Sheets.Sheet[k].Name = newName
291 292
			f.sheetMap[newName] = f.sheetMap[oldName]
			delete(f.sheetMap, oldName)
293 294 295 296
		}
	}
}

xurime's avatar
xurime 已提交
297
// GetSheetName provides a function to get worksheet name of XLSX by given
298 299
// worksheet index. If given sheet index is invalid, will return an empty
// string.
300
func (f *File) GetSheetName(index int) string {
xurime's avatar
xurime 已提交
301
	content := f.workbookReader()
302 303 304 305 306 307 308 309 310
	rels := f.workbookRelsReader()
	for _, rel := range rels.Relationships {
		rID, _ := strconv.Atoi(strings.TrimSuffix(strings.TrimPrefix(rel.Target, "worksheets/sheet"), ".xml"))
		if rID == index {
			for _, v := range content.Sheets.Sheet {
				if v.ID == rel.ID {
					return v.Name
				}
			}
311 312
		}
	}
313
	return ""
314 315
}

xurime's avatar
xurime 已提交
316
// GetSheetIndex provides a function to get worksheet index of XLSX by given sheet
317 318
// name. If given worksheet name is invalid, will return an integer type value
// 0.
319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334
func (f *File) GetSheetIndex(name string) int {
	content := f.workbookReader()
	rels := f.workbookRelsReader()
	for _, v := range content.Sheets.Sheet {
		if v.Name == name {
			for _, rel := range rels.Relationships {
				if v.ID == rel.ID {
					rID, _ := strconv.Atoi(strings.TrimSuffix(strings.TrimPrefix(rel.Target, "worksheets/sheet"), ".xml"))
					return rID
				}
			}
		}
	}
	return 0
}

xurime's avatar
xurime 已提交
335
// GetSheetMap provides a function to get worksheet name and index map of XLSX.
336
// For example:
337
//
xurime's avatar
xurime 已提交
338
//    xlsx, err := excelize.OpenFile("./Book1.xlsx")
339
//    if err != nil {
340
//        return
341
//    }
342 343
//    for index, name := range xlsx.GetSheetMap() {
//        fmt.Println(index, name)
344 345 346
//    }
//
func (f *File) GetSheetMap() map[int]string {
xurime's avatar
xurime 已提交
347
	content := f.workbookReader()
348
	rels := f.workbookRelsReader()
349 350
	sheetMap := map[int]string{}
	for _, v := range content.Sheets.Sheet {
351
		for _, rel := range rels.Relationships {
352 353 354
			relStr := strings.SplitN(rel.Target, "worksheets/sheet", 2)
			if rel.ID == v.ID && len(relStr) == 2 {
				rID, _ := strconv.Atoi(strings.TrimSuffix(relStr[1], ".xml"))
355 356 357
				sheetMap[rID] = v.Name
			}
		}
358 359
	}
	return sheetMap
xurime's avatar
xurime 已提交
360
}
361

xurime's avatar
xurime 已提交
362
// getSheetMap provides a function to get worksheet name and XML file path map of
363 364 365 366 367 368 369 370 371
// XLSX.
func (f *File) getSheetMap() map[string]string {
	maps := make(map[string]string)
	for idx, name := range f.GetSheetMap() {
		maps[name] = "xl/worksheets/sheet" + strconv.Itoa(idx) + ".xml"
	}
	return maps
}

xurime's avatar
xurime 已提交
372
// SetSheetBackground provides a function to set background picture by given
373
// worksheet name and file path.
374 375 376 377 378 379
func (f *File) SetSheetBackground(sheet, picture string) error {
	var err error
	// Check picture exists first.
	if _, err = os.Stat(picture); os.IsNotExist(err) {
		return err
	}
380
	ext, ok := supportImageTypes[path.Ext(picture)]
381
	if !ok {
xurime's avatar
xurime 已提交
382
		return errors.New("unsupported image extension")
383 384 385 386
	}
	pictureID := f.countMedia() + 1
	rID := f.addSheetRelationships(sheet, SourceRelationshipImage, "../media/image"+strconv.Itoa(pictureID)+ext, "")
	f.addSheetPicture(sheet, rID)
387 388
	file, _ := ioutil.ReadFile(picture)
	f.addMedia(file, ext)
389 390 391
	f.setContentTypePartImageExtensions()
	return err
}
392

xurime's avatar
xurime 已提交
393
// DeleteSheet provides a function to delete worksheet in a workbook by given
394
// worksheet name. Use this method with caution, which will affect changes in
395 396 397 398
// references such as formulas, charts, and so on. If there is any referenced
// value of the deleted worksheet, it will cause a file error when you open it.
// This function will be invalid when only the one worksheet is left.
func (f *File) DeleteSheet(name string) {
xurime's avatar
xurime 已提交
399
	content := f.workbookReader()
400
	for k, v := range content.Sheets.Sheet {
xurime's avatar
xurime 已提交
401 402
		if v.Name == trimSheetName(name) && len(content.Sheets.Sheet) > 1 {
			content.Sheets.Sheet = append(content.Sheets.Sheet[:k], content.Sheets.Sheet[k+1:]...)
403 404
			sheet := "xl/worksheets/sheet" + strconv.Itoa(v.SheetID) + ".xml"
			rels := "xl/worksheets/_rels/sheet" + strconv.Itoa(v.SheetID) + ".xml.rels"
xurime's avatar
xurime 已提交
405 406 407 408 409 410 411
			target := f.deleteSheetFromWorkbookRels(v.ID)
			f.deleteSheetFromContentTypes(target)
			delete(f.sheetMap, name)
			delete(f.XLSX, sheet)
			delete(f.XLSX, rels)
			delete(f.Sheet, sheet)
			f.SheetCount--
412 413
		}
	}
414
	f.SetActiveSheet(len(f.GetSheetMap()))
415 416
}

xurime's avatar
xurime 已提交
417
// deleteSheetFromWorkbookRels provides a function to remove worksheet
418 419
// relationships by given relationships ID in the file
// xl/_rels/workbook.xml.rels.
420
func (f *File) deleteSheetFromWorkbookRels(rID string) string {
xurime's avatar
xurime 已提交
421
	content := f.workbookRelsReader()
422
	for k, v := range content.Relationships {
xurime's avatar
xurime 已提交
423 424 425
		if v.ID == rID {
			content.Relationships = append(content.Relationships[:k], content.Relationships[k+1:]...)
			return v.Target
426 427 428 429 430
		}
	}
	return ""
}

xurime's avatar
xurime 已提交
431
// deleteSheetFromContentTypes provides a function to remove worksheet
432
// relationships by given target name in the file [Content_Types].xml.
433
func (f *File) deleteSheetFromContentTypes(target string) {
xurime's avatar
xurime 已提交
434
	content := f.contentTypesReader()
435
	for k, v := range content.Overrides {
xurime's avatar
xurime 已提交
436 437
		if v.PartName == "/xl/"+target {
			content.Overrides = append(content.Overrides[:k], content.Overrides[k+1:]...)
438 439 440
		}
	}
}
441

xurime's avatar
xurime 已提交
442
// CopySheet provides a function to duplicate a worksheet by gave source and
443 444 445 446
// target worksheet index. Note that currently doesn't support duplicate
// workbooks that contain tables, charts or pictures. For Example:
//
//    // Sheet1 already exists...
447 448
//    index := xlsx.NewSheet("Sheet2")
//    err := xlsx.CopySheet(1, index)
449
//    return err
450 451 452
//
func (f *File) CopySheet(from, to int) error {
	if from < 1 || to < 1 || from == to || f.GetSheetName(from) == "" || f.GetSheetName(to) == "" {
xurime's avatar
xurime 已提交
453
		return errors.New("invalid worksheet index")
454
	}
455 456
	f.copySheet(from, to)
	return nil
457 458
}

xurime's avatar
xurime 已提交
459
// copySheet provides a function to duplicate a worksheet by gave source and
460
// target worksheet name.
461
func (f *File) copySheet(from, to int) {
462
	sheet := f.workSheetReader("sheet" + strconv.Itoa(from))
463
	worksheet := deepcopy.Copy(sheet).(*xlsxWorksheet)
464 465 466 467 468 469 470
	path := "xl/worksheets/sheet" + strconv.Itoa(to) + ".xml"
	if len(worksheet.SheetViews.SheetView) > 0 {
		worksheet.SheetViews.SheetView[0].TabSelected = false
	}
	worksheet.Drawing = nil
	worksheet.TableParts = nil
	worksheet.PageSetUp = nil
471
	f.Sheet[path] = worksheet
472 473 474 475 476 477 478
	toRels := "xl/worksheets/_rels/sheet" + strconv.Itoa(to) + ".xml.rels"
	fromRels := "xl/worksheets/_rels/sheet" + strconv.Itoa(from) + ".xml.rels"
	_, ok := f.XLSX[fromRels]
	if ok {
		f.XLSX[toRels] = f.XLSX[fromRels]
	}
}
479

xurime's avatar
xurime 已提交
480
// SetSheetVisible provides a function to set worksheet visible by given worksheet
481 482
// name. A workbook must contain at least one visible worksheet. If the given
// worksheet has been activated, this setting will be invalidated. Sheet state
483
// values as defined by http://msdn.microsoft.com/en-us/library/office/documentformat.openxml.spreadsheet.sheetstatevalues.aspx
xurime's avatar
xurime 已提交
484 485 486 487 488
//
//    visible
//    hidden
//    veryHidden
//
489 490 491 492 493
// For example, hide Sheet1:
//
//    xlsx.SetSheetVisible("Sheet1", false)
//
func (f *File) SetSheetVisible(name string, visible bool) {
494 495
	name = trimSheetName(name)
	content := f.workbookReader()
496 497 498 499 500 501 502 503
	if visible {
		for k, v := range content.Sheets.Sheet {
			if v.Name == name {
				content.Sheets.Sheet[k].State = ""
			}
		}
		return
	}
504 505
	count := 0
	for _, v := range content.Sheets.Sheet {
506
		if v.State != "hidden" {
507 508 509 510
			count++
		}
	}
	for k, v := range content.Sheets.Sheet {
511
		xlsx := f.workSheetReader(f.GetSheetMap()[k])
512 513 514 515 516
		tabSelected := false
		if len(xlsx.SheetViews.SheetView) > 0 {
			tabSelected = xlsx.SheetViews.SheetView[0].TabSelected
		}
		if v.Name == name && count > 1 && !tabSelected {
517
			content.Sheets.Sheet[k].State = "hidden"
518 519 520 521
		}
	}
}

xurime's avatar
xurime 已提交
522
// parseFormatPanesSet provides a function to parse the panes settings.
523
func parseFormatPanesSet(formatSet string) (*formatPanes, error) {
xurime's avatar
xurime 已提交
524
	format := formatPanes{}
525 526
	err := json.Unmarshal([]byte(formatSet), &format)
	return &format, err
xurime's avatar
xurime 已提交
527 528
}

xurime's avatar
xurime 已提交
529
// SetPanes provides a function to create and remove freeze panes and split panes
530
// by given worksheet name and panes format set.
xurime's avatar
xurime 已提交
531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558
//
// activePane defines the pane that is active. The possible values for this
// attribute are defined in the following table:
//
//     Enumeration Value              | Description
//    --------------------------------+-------------------------------------------------------------
//     bottomLeft (Bottom Left Pane)  | Bottom left pane, when both vertical and horizontal
//                                    | splits are applied.
//                                    |
//                                    | This value is also used when only a horizontal split has
//                                    | been applied, dividing the pane into upper and lower
//                                    | regions. In that case, this value specifies the bottom
//                                    | pane.
//                                    |
//    bottomRight (Bottom Right Pane) | Bottom right pane, when both vertical and horizontal
//                                    | splits are applied.
//                                    |
//    topLeft (Top Left Pane)         | Top left pane, when both vertical and horizontal splits
//                                    | are applied.
//                                    |
//                                    | This value is also used when only a horizontal split has
//                                    | been applied, dividing the pane into upper and lower
//                                    | regions. In that case, this value specifies the top pane.
//                                    |
//                                    | This value is also used when only a vertical split has
//                                    | been applied, dividing the pane into right and left
//                                    | regions. In that case, this value specifies the left pane
//                                    |
xurime's avatar
xurime 已提交
559
//    topRight (Top Right Pane)       | Top right pane, when both vertical and horizontal
xurime's avatar
xurime 已提交
560 561 562 563 564 565 566
//                                    | splits are applied.
//                                    |
//                                    | This value is also used when only a vertical split has
//                                    | been applied, dividing the pane into right and left
//                                    | regions. In that case, this value specifies the right
//                                    | pane.
//
567
// Pane state type is restricted to the values supported currently listed in the following table:
xurime's avatar
xurime 已提交
568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595
//
//     Enumeration Value              | Description
//    --------------------------------+-------------------------------------------------------------
//     frozen (Frozen)                | Panes are frozen, but were not split being frozen. In
//                                    | this state, when the panes are unfrozen again, a single
//                                    | pane results, with no split.
//                                    |
//                                    | In this state, the split bars are not adjustable.
//                                    |
//     split (Split)                  | Panes are split, but not frozen. In this state, the split
//                                    | bars are adjustable by the user.
//
// x_split (Horizontal Split Position): Horizontal position of the split, in
// 1/20th of a point; 0 (zero) if none. If the pane is frozen, this value
// indicates the number of columns visible in the top pane.
//
// y_split (Vertical Split Position): Vertical position of the split, in 1/20th
// of a point; 0 (zero) if none. If the pane is frozen, this value indicates the
// number of rows visible in the left pane. The possible values for this
// attribute are defined by the W3C XML Schema double datatype.
//
// top_left_cell: Location of the top left visible cell in the bottom right pane
// (when in Left-To-Right mode).
//
// sqref (Sequence of References): Range of the selection. Can be non-contiguous
// set of ranges.
//
// An example of how to freeze column A in the Sheet1 and set the active cell on
xurime's avatar
xurime 已提交
596
// Sheet1!K16:
xurime's avatar
xurime 已提交
597
//
xurime's avatar
xurime 已提交
598
//    xlsx.SetPanes("Sheet1", `{"freeze":true,"split":false,"x_split":1,"y_split":0,"top_left_cell":"B1","active_pane":"topRight","panes":[{"sqref":"K16","active_cell":"K16","pane":"topRight"}]}`)
xurime's avatar
xurime 已提交
599 600
//
// An example of how to freeze rows 1 to 9 in the Sheet1 and set the active cell
xurime's avatar
xurime 已提交
601
// ranges on Sheet1!A11:XFD11:
xurime's avatar
xurime 已提交
602
//
xurime's avatar
xurime 已提交
603
//    xlsx.SetPanes("Sheet1", `{"freeze":true,"split":false,"x_split":0,"y_split":9,"top_left_cell":"A34","active_pane":"bottomLeft","panes":[{"sqref":"A11:XFD11","active_cell":"A11","pane":"bottomLeft"}]}`)
xurime's avatar
xurime 已提交
604 605 606 607
//
// An example of how to create split panes in the Sheet1 and set the active cell
// on Sheet1!J60:
//
xurime's avatar
xurime 已提交
608
//    xlsx.SetPanes("Sheet1", `{"freeze":false,"split":true,"x_split":3270,"y_split":1800,"top_left_cell":"N57","active_pane":"bottomLeft","panes":[{"sqref":"I36","active_cell":"I36"},{"sqref":"G33","active_cell":"G33","pane":"topRight"},{"sqref":"J60","active_cell":"J60","pane":"bottomLeft"},{"sqref":"O60","active_cell":"O60","pane":"bottomRight"}]}`)
xurime's avatar
xurime 已提交
609 610 611 612 613 614
//
// An example of how to unfreeze and remove all panes on Sheet1:
//
//    xlsx.SetPanes("Sheet1", `{"freeze":false,"split":false}`)
//
func (f *File) SetPanes(sheet, panes string) {
615
	fs, _ := parseFormatPanesSet(panes)
xurime's avatar
xurime 已提交
616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
	xlsx := f.workSheetReader(sheet)
	p := &xlsxPane{
		ActivePane:  fs.ActivePane,
		TopLeftCell: fs.TopLeftCell,
		XSplit:      float64(fs.XSplit),
		YSplit:      float64(fs.YSplit),
	}
	if fs.Freeze {
		p.State = "frozen"
	}
	xlsx.SheetViews.SheetView[len(xlsx.SheetViews.SheetView)-1].Pane = p
	if !(fs.Freeze) && !(fs.Split) {
		if len(xlsx.SheetViews.SheetView) > 0 {
			xlsx.SheetViews.SheetView[len(xlsx.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,
		})
	}
	xlsx.SheetViews.SheetView[len(xlsx.SheetViews.SheetView)-1].Selection = s
}

xurime's avatar
xurime 已提交
643
// GetSheetVisible provides a function to get worksheet visible by given worksheet
644 645 646 647 648
// name. For example, get visible state of Sheet1:
//
//    xlsx.GetSheetVisible("Sheet1")
//
func (f *File) GetSheetVisible(name string) bool {
649
	content := f.workbookReader()
650
	visible := false
651
	for k, v := range content.Sheets.Sheet {
xurime's avatar
xurime 已提交
652
		if v.Name == trimSheetName(name) {
653 654 655
			if content.Sheets.Sheet[k].State == "" || content.Sheets.Sheet[k].State == "visible" {
				visible = true
			}
656 657
		}
	}
658
	return visible
659 660
}

661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710
// SearchSheet provides a function to get coordinates by given worksheet name
// and cell value. This function only supports exact match of strings and
// numbers, doesn't support the calculated result, formatted numbers and
// conditional lookup currently. If it is a merged cell, it will return the
// coordinates of the upper left corner of the merged area. For example,
// search the coordinates of the value of "100" on Sheet1:
//
//    xlsx.SearchSheet("Sheet1", "100")
//
func (f *File) SearchSheet(sheet, value string) []string {
	xlsx := f.workSheetReader(sheet)
	result := []string{}
	name, ok := f.sheetMap[trimSheetName(sheet)]
	if !ok {
		return result
	}
	if xlsx != nil {
		output, _ := xml.Marshal(f.Sheet[name])
		f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpaceBytes(output))
	}
	xml.NewDecoder(bytes.NewReader(f.readXML(name)))
	d := f.sharedStringsReader()
	var inElement string
	var r xlsxRow
	decoder := xml.NewDecoder(bytes.NewReader(f.readXML(name)))
	for {
		token, _ := decoder.Token()
		if token == nil {
			break
		}
		switch startElement := token.(type) {
		case xml.StartElement:
			inElement = startElement.Name.Local
			if inElement == "row" {
				r = xlsxRow{}
				_ = decoder.DecodeElement(&r, &startElement)
				for _, colCell := range r.C {
					val, _ := colCell.getValueFrom(f, d)
					if val != value {
						continue
					}
					result = append(result, fmt.Sprintf("%s%d", strings.Map(letterOnlyMapF, colCell.R), r.R))
				}
			}
		default:
		}
	}
	return result
}

R
r-uchino 已提交
711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
// RegSearchSheet provides the ability to retrieve coordinates
// with the given worksheet name and regular expression
// For a merged cell, get the coordinates
// of the upper left corner of the merge area.
// :example)
// Search the coordinates where the numerical value in the range of "0-9" of Sheet 1 is described:
//
//    xlsx.RegSearchSheet("Sheet1", "[0-9]")
//
func (f *File) RegSearchSheet(sheet, value string) []string {
	xlsx := f.workSheetReader(sheet)
	result := []string{}
	name, ok := f.sheetMap[trimSheetName(sheet)]
	if !ok {
		return result
	}
	if xlsx != nil {
		output, _ := xml.Marshal(f.Sheet[name])
		f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpaceBytes(output))
	}
	xml.NewDecoder(bytes.NewReader(f.readXML(name)))
	d := f.sharedStringsReader()
	var inElement string
	var r xlsxRow
	decoder := xml.NewDecoder(bytes.NewReader(f.readXML(name)))
	for {
		token, _ := decoder.Token()
		if token == nil {
			break
		}
		switch startElement := token.(type) {
		case xml.StartElement:
			inElement = startElement.Name.Local
			if inElement == "row" {
				r = xlsxRow{}
				_ = decoder.DecodeElement(&r, &startElement)
				for _, colCell := range r.C {
					val, _ := colCell.getValueFrom(f, d)
					regex := regexp.MustCompile(value)
					if !regex.MatchString(val) {
						continue
					}
					result = append(result, fmt.Sprintf("%s%d", strings.Map(letterOnlyMapF, colCell.R), r.R))
				}
			}
		default:
		}
	}
	return result
}

762 763
// ProtectSheet provides a function to prevent other users from accidentally
// or deliberately changing, moving, or deleting data in a worksheet. For
764
// example, protect Sheet1 with protection settings:
765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802
//
//    xlsx.ProtectSheet("Sheet1", &excelize.FormatSheetProtection{
//        Password:      "password",
//        EditScenarios: false,
//    })
//
func (f *File) ProtectSheet(sheet string, settings *FormatSheetProtection) {
	xlsx := f.workSheetReader(sheet)
	if settings == nil {
		settings = &FormatSheetProtection{
			EditObjects:       true,
			EditScenarios:     true,
			SelectLockedCells: true,
		}
	}
	xlsx.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 != "" {
		xlsx.SheetProtection.Password = genSheetPasswd(settings.Password)
	}
}

803 804 805 806 807 808
// UnprotectSheet provides a function to unprotect an Excel worksheet.
func (f *File) UnprotectSheet(sheet string) {
	xlsx := f.workSheetReader(sheet)
	xlsx.SheetProtection = nil
}

xurime's avatar
xurime 已提交
809
// trimSheetName provides a function to trim invaild characters by given worksheet
810 811
// name.
func trimSheetName(name string) string {
812
	r := []rune{}
813
	for _, v := range name {
814 815 816 817 818 819 820 821
		switch v {
		case 58, 92, 47, 63, 42, 91, 93: // replace :\/?*[]
			continue
		default:
			r = append(r, v)
		}
	}
	name = string(r)
822 823
	if utf8.RuneCountInString(name) > 31 {
		name = string([]rune(name)[0:31])
824 825 826
	}
	return name
}