370 lines
10 KiB
Go
370 lines
10 KiB
Go
// Copyright 2016 - 2022 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 XLAM / XLSM / XLSX / XLTM / XLTX files. Supports reading and
|
|
// writing spreadsheet documents generated by Microsoft Excel™ 2007 and later.
|
|
// Supports complex components by high compatibility, and provided streaming
|
|
// API for generating or reading data from a worksheet with huge amounts of
|
|
// data. This library needs Go version 1.15 or later.
|
|
|
|
package excelize
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/xml"
|
|
"io"
|
|
"strings"
|
|
)
|
|
|
|
type adjustDirection bool
|
|
|
|
const (
|
|
columns adjustDirection = false
|
|
rows adjustDirection = true
|
|
)
|
|
|
|
// adjustHelper provides a function to adjust rows and columns dimensions,
|
|
// hyperlinks, merged cells and auto filter when inserting or deleting rows or
|
|
// columns.
|
|
//
|
|
// sheet: Worksheet name that we're editing
|
|
// column: Index number of the column we're inserting/deleting before
|
|
// row: Index number of the row we're inserting/deleting before
|
|
// offset: Number of rows/column to insert/delete negative values indicate deletion
|
|
//
|
|
// TODO: adjustPageBreaks, adjustComments, adjustDataValidations, adjustProtectedCells
|
|
func (f *File) adjustHelper(sheet string, dir adjustDirection, num, offset int) error {
|
|
ws, err := f.workSheetReader(sheet)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sheetID := f.getSheetID(sheet)
|
|
if dir == rows {
|
|
err = f.adjustRowDimensions(ws, num, offset)
|
|
} else {
|
|
err = f.adjustColDimensions(ws, num, offset)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f.adjustHyperlinks(ws, sheet, dir, num, offset)
|
|
f.adjustTable(ws, sheet, dir, num, offset)
|
|
if err = f.adjustMergeCells(ws, dir, num, offset); err != nil {
|
|
return err
|
|
}
|
|
if err = f.adjustAutoFilter(ws, dir, num, offset); err != nil {
|
|
return err
|
|
}
|
|
if err = f.adjustCalcChain(dir, num, offset, sheetID); err != nil {
|
|
return err
|
|
}
|
|
checkSheet(ws)
|
|
_ = checkRow(ws)
|
|
|
|
if ws.MergeCells != nil && len(ws.MergeCells.Cells) == 0 {
|
|
ws.MergeCells = nil
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// adjustColDimensions provides a function to update column dimensions when
|
|
// inserting or deleting rows or columns.
|
|
func (f *File) adjustColDimensions(ws *xlsxWorksheet, col, offset int) error {
|
|
for rowIdx := range ws.SheetData.Row {
|
|
for colIdx, v := range ws.SheetData.Row[rowIdx].C {
|
|
cellCol, cellRow, _ := CellNameToCoordinates(v.R)
|
|
if col <= cellCol {
|
|
if newCol := cellCol + offset; newCol > 0 {
|
|
if newCol > MaxColumns {
|
|
return ErrColumnNumber
|
|
}
|
|
ws.SheetData.Row[rowIdx].C[colIdx].R, _ = CoordinatesToCellName(newCol, cellRow)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// adjustRowDimensions provides a function to update row dimensions when
|
|
// inserting or deleting rows or columns.
|
|
func (f *File) adjustRowDimensions(ws *xlsxWorksheet, row, offset int) error {
|
|
for i := range ws.SheetData.Row {
|
|
r := &ws.SheetData.Row[i]
|
|
if newRow := r.R + offset; r.R >= row && newRow > 0 {
|
|
if newRow >= TotalRows {
|
|
return ErrMaxRows
|
|
}
|
|
f.adjustSingleRowDimensions(r, newRow)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// adjustSingleRowDimensions provides a function to adjust single row dimensions.
|
|
func (f *File) adjustSingleRowDimensions(r *xlsxRow, num int) {
|
|
r.R = num
|
|
for i, col := range r.C {
|
|
colName, _, _ := SplitCellName(col.R)
|
|
r.C[i].R, _ = JoinCellName(colName, num)
|
|
}
|
|
}
|
|
|
|
// adjustHyperlinks provides a function to update hyperlinks when inserting or
|
|
// deleting rows or columns.
|
|
func (f *File) adjustHyperlinks(ws *xlsxWorksheet, sheet string, dir adjustDirection, num, offset int) {
|
|
// short path
|
|
if ws.Hyperlinks == nil || len(ws.Hyperlinks.Hyperlink) == 0 {
|
|
return
|
|
}
|
|
|
|
// order is important
|
|
if offset < 0 {
|
|
for i := len(ws.Hyperlinks.Hyperlink) - 1; i >= 0; i-- {
|
|
linkData := ws.Hyperlinks.Hyperlink[i]
|
|
colNum, rowNum, _ := CellNameToCoordinates(linkData.Ref)
|
|
|
|
if (dir == rows && num == rowNum) || (dir == columns && num == colNum) {
|
|
f.deleteSheetRelationships(sheet, linkData.RID)
|
|
if len(ws.Hyperlinks.Hyperlink) > 1 {
|
|
ws.Hyperlinks.Hyperlink = append(ws.Hyperlinks.Hyperlink[:i],
|
|
ws.Hyperlinks.Hyperlink[i+1:]...)
|
|
} else {
|
|
ws.Hyperlinks = nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if ws.Hyperlinks == nil {
|
|
return
|
|
}
|
|
for i := range ws.Hyperlinks.Hyperlink {
|
|
link := &ws.Hyperlinks.Hyperlink[i] // get reference
|
|
colNum, rowNum, _ := CellNameToCoordinates(link.Ref)
|
|
if dir == rows {
|
|
if rowNum >= num {
|
|
link.Ref, _ = CoordinatesToCellName(colNum, rowNum+offset)
|
|
}
|
|
} else {
|
|
if colNum >= num {
|
|
link.Ref, _ = CoordinatesToCellName(colNum+offset, rowNum)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// adjustTable provides a function to update the table when inserting or
|
|
// deleting rows or columns.
|
|
func (f *File) adjustTable(ws *xlsxWorksheet, sheet string, dir adjustDirection, num, offset int) {
|
|
if ws.TableParts == nil || len(ws.TableParts.TableParts) == 0 {
|
|
return
|
|
}
|
|
for idx := 0; idx < len(ws.TableParts.TableParts); idx++ {
|
|
tbl := ws.TableParts.TableParts[idx]
|
|
target := f.getSheetRelationshipsTargetByID(sheet, tbl.RID)
|
|
tableXML := strings.ReplaceAll(target, "..", "xl")
|
|
content, ok := f.Pkg.Load(tableXML)
|
|
if !ok {
|
|
continue
|
|
}
|
|
t := xlsxTable{}
|
|
if err := f.xmlNewDecoder(bytes.NewReader(namespaceStrictToTransitional(content.([]byte)))).
|
|
Decode(&t); err != nil && err != io.EOF {
|
|
return
|
|
}
|
|
coordinates, err := areaRefToCoordinates(t.Ref)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// Remove the table when deleting the header row of the table
|
|
if dir == rows && num == coordinates[0] {
|
|
ws.TableParts.TableParts = append(ws.TableParts.TableParts[:idx], ws.TableParts.TableParts[idx+1:]...)
|
|
ws.TableParts.Count = len(ws.TableParts.TableParts)
|
|
idx--
|
|
continue
|
|
}
|
|
coordinates = f.adjustAutoFilterHelper(dir, coordinates, num, offset)
|
|
x1, y1, x2, y2 := coordinates[0], coordinates[1], coordinates[2], coordinates[3]
|
|
if y2-y1 < 2 || x2-x1 < 1 {
|
|
ws.TableParts.TableParts = append(ws.TableParts.TableParts[:idx], ws.TableParts.TableParts[idx+1:]...)
|
|
ws.TableParts.Count = len(ws.TableParts.TableParts)
|
|
idx--
|
|
continue
|
|
}
|
|
t.Ref, _ = f.coordinatesToAreaRef([]int{x1, y1, x2, y2})
|
|
if t.AutoFilter != nil {
|
|
t.AutoFilter.Ref = t.Ref
|
|
}
|
|
_, _ = f.setTableHeader(sheet, x1, y1, x2)
|
|
table, _ := xml.Marshal(t)
|
|
f.saveFileList(tableXML, table)
|
|
}
|
|
}
|
|
|
|
// adjustAutoFilter provides a function to update the auto filter when
|
|
// inserting or deleting rows or columns.
|
|
func (f *File) adjustAutoFilter(ws *xlsxWorksheet, dir adjustDirection, num, offset int) error {
|
|
if ws.AutoFilter == nil {
|
|
return nil
|
|
}
|
|
|
|
coordinates, err := areaRefToCoordinates(ws.AutoFilter.Ref)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
x1, y1, x2, y2 := coordinates[0], coordinates[1], coordinates[2], coordinates[3]
|
|
|
|
if (dir == rows && y1 == num && offset < 0) || (dir == columns && x1 == num && x2 == num) {
|
|
ws.AutoFilter = nil
|
|
for rowIdx := range ws.SheetData.Row {
|
|
rowData := &ws.SheetData.Row[rowIdx]
|
|
if rowData.R > y1 && rowData.R <= y2 {
|
|
rowData.Hidden = false
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
coordinates = f.adjustAutoFilterHelper(dir, coordinates, num, offset)
|
|
x1, y1, x2, y2 = coordinates[0], coordinates[1], coordinates[2], coordinates[3]
|
|
|
|
if ws.AutoFilter.Ref, err = f.coordinatesToAreaRef([]int{x1, y1, x2, y2}); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// adjustAutoFilterHelper provides a function for adjusting auto filter to
|
|
// compare and calculate cell axis by the given adjust direction, operation
|
|
// axis and offset.
|
|
func (f *File) adjustAutoFilterHelper(dir adjustDirection, coordinates []int, num, offset int) []int {
|
|
if dir == rows {
|
|
if coordinates[1] >= num {
|
|
coordinates[1] += offset
|
|
}
|
|
if coordinates[3] >= num {
|
|
coordinates[3] += offset
|
|
}
|
|
return coordinates
|
|
}
|
|
if coordinates[0] >= num {
|
|
coordinates[0] += offset
|
|
}
|
|
if coordinates[2] >= num {
|
|
coordinates[2] += offset
|
|
}
|
|
return coordinates
|
|
}
|
|
|
|
// adjustMergeCells provides a function to update merged cells when inserting
|
|
// or deleting rows or columns.
|
|
func (f *File) adjustMergeCells(ws *xlsxWorksheet, dir adjustDirection, num, offset int) error {
|
|
if ws.MergeCells == nil {
|
|
return nil
|
|
}
|
|
|
|
for i := 0; i < len(ws.MergeCells.Cells); i++ {
|
|
areaData := ws.MergeCells.Cells[i]
|
|
coordinates, err := areaRefToCoordinates(areaData.Ref)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
x1, y1, x2, y2 := coordinates[0], coordinates[1], coordinates[2], coordinates[3]
|
|
if dir == rows {
|
|
if y1 == num && y2 == num && offset < 0 {
|
|
f.deleteMergeCell(ws, i)
|
|
i--
|
|
continue
|
|
}
|
|
|
|
y1, y2 = f.adjustMergeCellsHelper(y1, y2, num, offset)
|
|
} else {
|
|
if x1 == num && x2 == num && offset < 0 {
|
|
f.deleteMergeCell(ws, i)
|
|
i--
|
|
continue
|
|
}
|
|
|
|
x1, x2 = f.adjustMergeCellsHelper(x1, x2, num, offset)
|
|
}
|
|
if x1 == x2 && y1 == y2 {
|
|
f.deleteMergeCell(ws, i)
|
|
i--
|
|
continue
|
|
}
|
|
areaData.rect = []int{x1, y1, x2, y2}
|
|
if areaData.Ref, err = f.coordinatesToAreaRef([]int{x1, y1, x2, y2}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// adjustMergeCellsHelper provides a function for adjusting merge cells to
|
|
// compare and calculate cell axis by the given pivot, operation axis and
|
|
// offset.
|
|
func (f *File) adjustMergeCellsHelper(p1, p2, num, offset int) (int, int) {
|
|
if p2 < p1 {
|
|
p1, p2 = p2, p1
|
|
}
|
|
|
|
if offset >= 0 {
|
|
if num <= p1 {
|
|
p1 += offset
|
|
p2 += offset
|
|
} else if num <= p2 {
|
|
p2 += offset
|
|
}
|
|
return p1, p2
|
|
}
|
|
if num < p1 || (num == p1 && num == p2) {
|
|
p1 += offset
|
|
p2 += offset
|
|
} else if num <= p2 {
|
|
p2 += offset
|
|
}
|
|
return p1, p2
|
|
}
|
|
|
|
// deleteMergeCell provides a function to delete merged cell by given index.
|
|
func (f *File) deleteMergeCell(ws *xlsxWorksheet, idx int) {
|
|
if idx < 0 {
|
|
return
|
|
}
|
|
if len(ws.MergeCells.Cells) > idx {
|
|
ws.MergeCells.Cells = append(ws.MergeCells.Cells[:idx], ws.MergeCells.Cells[idx+1:]...)
|
|
ws.MergeCells.Count = len(ws.MergeCells.Cells)
|
|
}
|
|
}
|
|
|
|
// adjustCalcChain provides a function to update the calculation chain when
|
|
// inserting or deleting rows or columns.
|
|
func (f *File) adjustCalcChain(dir adjustDirection, num, offset, sheetID int) error {
|
|
if f.CalcChain == nil {
|
|
return nil
|
|
}
|
|
for index, c := range f.CalcChain.C {
|
|
if c.I != sheetID {
|
|
continue
|
|
}
|
|
colNum, rowNum, err := CellNameToCoordinates(c.R)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if dir == rows && num <= rowNum {
|
|
if newRow := rowNum + offset; newRow > 0 {
|
|
f.CalcChain.C[index].R, _ = CoordinatesToCellName(colNum, newRow)
|
|
}
|
|
}
|
|
if dir == columns && num <= colNum {
|
|
if newCol := colNum + offset; newCol > 0 {
|
|
f.CalcChain.C[index].R, _ = CoordinatesToCellName(newCol, rowNum)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|