sheet.go 35.2 KB
Newer Older
xurime's avatar
xurime 已提交
1
// Copyright 2016 - 2019 The excelize Authors. All rights reserved. Use of
xurime's avatar
xurime 已提交
2 3 4 5 6 7 8
// 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
// SearchSheet provides a function to get coordinates by given worksheet name,
// cell value, and regular expression. The function doesn't support searching
// on 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.
//
// An example of search the coordinates of the value of "100" on Sheet1:
668 669 670
//
//    xlsx.SearchSheet("Sheet1", "100")
//
671 672
// An example of search the coordinates where the numerical value in the range
// of "0-9" of Sheet1 is described:
R
r-uchino 已提交
673
//
674
//    xlsx.SearchSheet("Sheet1", "[0-9]", true)
R
r-uchino 已提交
675
//
676 677 678 679 680
func (f *File) SearchSheet(sheet, value string, reg ...bool) []string {
	var regSearch bool
	for _, r := range reg {
		regSearch = r
	}
R
r-uchino 已提交
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
	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)
709 710 711 712 713 714 715 716 717
					if regSearch {
						regex := regexp.MustCompile(value)
						if !regex.MatchString(val) {
							continue
						}
					} else {
						if val != value {
							continue
						}
R
r-uchino 已提交
718 719 720 721 722 723 724 725 726 727
					}
					result = append(result, fmt.Sprintf("%s%d", strings.Map(letterOnlyMapF, colCell.R), r.R))
				}
			}
		default:
		}
	}
	return result
}

728 729
// ProtectSheet provides a function to prevent other users from accidentally
// or deliberately changing, moving, or deleting data in a worksheet. For
730
// example, protect Sheet1 with protection settings:
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 762 763 764 765 766 767 768
//
//    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)
	}
}

769 770 771 772 773 774
// 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 已提交
775
// trimSheetName provides a function to trim invaild characters by given worksheet
776 777
// name.
func trimSheetName(name string) string {
778
	r := []rune{}
779
	for _, v := range name {
780 781 782 783 784 785 786 787
		switch v {
		case 58, 92, 47, 63, 42, 91, 93: // replace :\/?*[]
			continue
		default:
			r = append(r, v)
		}
	}
	name = string(r)
788 789
	if utf8.RuneCountInString(name) > 31 {
		name = string([]rune(name)[0:31])
790 791 792
	}
	return name
}
793 794 795 796 797 798 799 800 801 802 803 804 805

// PageLayoutOption is an option of a page layout of a worksheet. See
// SetPageLayout().
type PageLayoutOption interface {
	setPageLayout(layout *xlsxPageSetUp)
}

// PageLayoutOptionPtr is a writable PageLayoutOption. See GetPageLayout().
type PageLayoutOptionPtr interface {
	PageLayoutOption
	getPageLayout(layout *xlsxPageSetUp)
}

806 807 808 809 810 811 812
type (
	// PageLayoutOrientation defines the orientation of page layout for a
	// worksheet.
	PageLayoutOrientation string
	// PageLayoutPaperSize defines the paper size of the worksheet
	PageLayoutPaperSize int
)
813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835

const (
	// OrientationPortrait indicates page layout orientation id portrait.
	OrientationPortrait = "portrait"
	// OrientationLandscape indicates page layout orientation id landscape.
	OrientationLandscape = "landscape"
)

// setPageLayout provides a method to set the orientation for the worksheet.
func (o PageLayoutOrientation) setPageLayout(ps *xlsxPageSetUp) {
	ps.Orientation = string(o)
}

// getPageLayout provides a method to get the orientation for the worksheet.
func (o *PageLayoutOrientation) getPageLayout(ps *xlsxPageSetUp) {
	// Excel default: portrait
	if ps == nil || ps.Orientation == "" {
		*o = OrientationPortrait
		return
	}
	*o = PageLayoutOrientation(ps.Orientation)
}

836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
// setPageLayout provides a method to set the paper size for the worksheet.
func (p PageLayoutPaperSize) setPageLayout(ps *xlsxPageSetUp) {
	ps.PaperSize = int(p)
}

// getPageLayout provides a method to get the paper size for the worksheet.
func (p *PageLayoutPaperSize) getPageLayout(ps *xlsxPageSetUp) {
	// Excel default: 1
	if ps == nil || ps.PaperSize == 0 {
		*p = 1
		return
	}
	*p = PageLayoutPaperSize(ps.PaperSize)
}

851 852 853 854
// SetPageLayout provides a function to sets worksheet page layout.
//
// Available options:
//   PageLayoutOrientation(string)
855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977
//	 PageLayoutPaperSize(int)
//
// The following shows the paper size sorted by excelize index number:
//
//     Index | Paper Size
//    -------+-----------------------------------------------
//       1   | Letter paper (8.5 in. by 11 in.)
//       2   | Letter small paper (8.5 in. by 11 in.)
//       3   | Tabloid paper (11 in. by 17 in.)
//       4   | Ledger paper (17 in. by 11 in.)
//       5   | Legal paper (8.5 in. by 14 in.)
//       6   | Statement paper (5.5 in. by 8.5 in.)
//       7   | Executive paper (7.25 in. by 10.5 in.)
//       8   | A3 paper (297 mm by 420 mm)
//       9   | A4 paper (210 mm by 297 mm)
//       10  | A4 small paper (210 mm by 297 mm)
//       11  | A5 paper (148 mm by 210 mm)
//       12  | B4 paper (250 mm by 353 mm)
//       13  | B5 paper (176 mm by 250 mm)
//       14  | Folio paper (8.5 in. by 13 in.)
//       15  | Quarto paper (215 mm by 275 mm)
//       16  | Standard paper (10 in. by 14 in.)
//       17  | Standard paper (11 in. by 17 in.)
//       18  | Note paper (8.5 in. by 11 in.)
//       19  | #9 envelope (3.875 in. by 8.875 in.)
//       20  | #10 envelope (4.125 in. by 9.5 in.)
//       21  | #11 envelope (4.5 in. by 10.375 in.)
//       22  | #12 envelope (4.75 in. by 11 in.)
//       23  | #14 envelope (5 in. by 11.5 in.)
//       24  | C paper (17 in. by 22 in.)
//       25  | D paper (22 in. by 34 in.)
//       26  | E paper (34 in. by 44 in.)
//       27  | DL envelope (110 mm by 220 mm)
//       28  | C5 envelope (162 mm by 229 mm)
//       29  | C3 envelope (324 mm by 458 mm)
//       30  | C4 envelope (229 mm by 324 mm)
//       31  | C6 envelope (114 mm by 162 mm)
//       32  | C65 envelope (114 mm by 229 mm)
//       33  | B4 envelope (250 mm by 353 mm)
//       34  | B5 envelope (176 mm by 250 mm)
//       35  | B6 envelope (176 mm by 125 mm)
//       36  | Italy envelope (110 mm by 230 mm)
//       37  | Monarch envelope (3.875 in. by 7.5 in.).
//       38  | 6 3/4 envelope (3.625 in. by 6.5 in.)
//       39  | US standard fanfold (14.875 in. by 11 in.)
//       40  | German standard fanfold (8.5 in. by 12 in.)
//       41  | German legal fanfold (8.5 in. by 13 in.)
//       42  | ISO B4 (250 mm by 353 mm)
//       43  | Japanese double postcard (200 mm by 148 mm)
//       44  | Standard paper (9 in. by 11 in.)
//       45  | Standard paper (10 in. by 11 in.)
//       46  | Standard paper (15 in. by 11 in.)
//       47  | Invite envelope (220 mm by 220 mm)
//       50  | Letter extra paper (9.275 in. by 12 in.)
//       51  | Legal extra paper (9.275 in. by 15 in.)
//       52  | Tabloid extra paper (11.69 in. by 18 in.)
//       53  | A4 extra paper (236 mm by 322 mm)
//       54  | Letter transverse paper (8.275 in. by 11 in.)
//       55  | A4 transverse paper (210 mm by 297 mm)
//       56  | Letter extra transverse paper (9.275 in. by 12 in.)
//       57  | SuperA/SuperA/A4 paper (227 mm by 356 mm)
//       58  | SuperB/SuperB/A3 paper (305 mm by 487 mm)
//       59  | Letter plus paper (8.5 in. by 12.69 in.)
//       60  | A4 plus paper (210 mm by 330 mm)
//       61  | A5 transverse paper (148 mm by 210 mm)
//       62  | JIS B5 transverse paper (182 mm by 257 mm)
//       63  | A3 extra paper (322 mm by 445 mm)
//       64  | A5 extra paper (174 mm by 235 mm)
//       65  | ISO B5 extra paper (201 mm by 276 mm)
//       66  | A2 paper (420 mm by 594 mm)
//       67  | A3 transverse paper (297 mm by 420 mm)
//       68  | A3 extra transverse paper (322 mm by 445 mm)
//       69  | Japanese Double Postcard (200 mm x 148 mm)
//       70  | A6 (105 mm x 148 mm)
//       71  | Japanese Envelope Kaku #2
//       72  | Japanese Envelope Kaku #3
//       73  | Japanese Envelope Chou #3
//       74  | Japanese Envelope Chou #4
//       75  | Letter Rotated (11in x 8 1/2 11 in)
//       76  | A3 Rotated (420 mm x 297 mm)
//       77  | A4 Rotated (297 mm x 210 mm)
//       78  | A5 Rotated (210 mm x 148 mm)
//       79  | B4 (JIS) Rotated (364 mm x 257 mm)
//       80  | B5 (JIS) Rotated (257 mm x 182 mm)
//       81  | Japanese Postcard Rotated (148 mm x 100 mm)
//       82  | Double Japanese Postcard Rotated (148 mm x 200 mm)
//       83  | A6 Rotated (148 mm x 105 mm)
//       84  | Japanese Envelope Kaku #2 Rotated
//       85  | Japanese Envelope Kaku #3 Rotated
//       86  | Japanese Envelope Chou #3 Rotated
//       87  | Japanese Envelope Chou #4 Rotated
//       88  | B6 (JIS) (128 mm x 182 mm)
//       89  | B6 (JIS) Rotated (182 mm x 128 mm)
//       90  | (12 in x 11 in)
//       91  | Japanese Envelope You #4
//       92  | Japanese Envelope You #4 Rotated
//       93  | PRC 16K (146 mm x 215 mm)
//       94  | PRC 32K (97 mm x 151 mm)
//       95  | PRC 32K(Big) (97 mm x 151 mm)
//       96  | PRC Envelope #1 (102 mm x 165 mm)
//       97  | PRC Envelope #2 (102 mm x 176 mm)
//       98  | PRC Envelope #3 (125 mm x 176 mm)
//       99  | PRC Envelope #4 (110 mm x 208 mm)
//       100 | PRC Envelope #5 (110 mm x 220 mm)
//       101 | PRC Envelope #6 (120 mm x 230 mm)
//       102 | PRC Envelope #7 (160 mm x 230 mm)
//       103 | PRC Envelope #8 (120 mm x 309 mm)
//       104 | PRC Envelope #9 (229 mm x 324 mm)
//       105 | PRC Envelope #10 (324 mm x 458 mm)
//       106 | PRC 16K Rotated
//       107 | PRC 32K Rotated
//       108 | PRC 32K(Big) Rotated
//       109 | PRC Envelope #1 Rotated (165 mm x 102 mm)
//       110 | PRC Envelope #2 Rotated (176 mm x 102 mm)
//       111 | PRC Envelope #3 Rotated (176 mm x 125 mm)
//       112 | PRC Envelope #4 Rotated (208 mm x 110 mm)
//       113 | PRC Envelope #5 Rotated (220 mm x 110 mm)
//       114 | PRC Envelope #6 Rotated (230 mm x 120 mm)
//       115 | PRC Envelope #7 Rotated (230 mm x 160 mm)
//       116 | PRC Envelope #8 Rotated (309 mm x 120 mm)
//       117 | PRC Envelope #9 Rotated (324 mm x 229 mm)
//       118 | PRC Envelope #10 Rotated (458 mm x 324 mm)
//
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995
func (f *File) SetPageLayout(sheet string, opts ...PageLayoutOption) error {
	s := f.workSheetReader(sheet)
	ps := s.PageSetUp
	if ps == nil {
		ps = new(xlsxPageSetUp)
		s.PageSetUp = ps
	}

	for _, opt := range opts {
		opt.setPageLayout(ps)
	}
	return nil
}

// GetPageLayout provides a function to gets worksheet page layout.
//
// Available options:
//   PageLayoutOrientation(string)
996
//   PageLayoutPaperSize(int)
997 998 999 1000 1001 1002 1003 1004 1005
func (f *File) GetPageLayout(sheet string, opts ...PageLayoutOptionPtr) error {
	s := f.workSheetReader(sheet)
	ps := s.PageSetUp

	for _, opt := range opts {
		opt.getPageLayout(ps)
	}
	return nil
}