title.go

package exportToExcel

import (
	"github.com/xuri/excelize/v2"
)

func NewTitle(sheet *Sheet) Title {
	return Title{sheet: sheet, FontScaleFactor: 2, BaseFontSize: 12, MaxFontSize: 20, MaxLevel: 5}
}

type Title struct {
	sheet           *Sheet
	titles          []*TitleItem
	MaxLevel        int
	FontScaleFactor float64
	BaseFontSize    float64
	MaxFontSize     float64
	occupiedRow     []int
	colNum          int
}

// Gen
// generate a TitleItem
func (t *Title) Gen(titles ...*TitleItem) (err error) {
	for _, title := range titles {
		//insert empty row
		if len(t.occupiedRow) > 0 {
			for thisRow := title.InitRow; thisRow < title.MergeRowTo; thisRow++ {
				var sameRowExisted bool
				for _, occRow := range t.occupiedRow {
					if occRow == thisRow {
						sameRowExisted = true
						break
					}
				}
				if !sameRowExisted {
					t.sheet.rowNum++
					_ = t.sheet.file.InsertRows(t.sheet.SheetName(), title.InitRow, 1)
					t.occupiedRow = append(t.occupiedRow, thisRow)
				}
			}
		} else if t.sheet.rowNum > title.InitRow {
			_ = t.sheet.file.InsertRows(t.sheet.SheetName(), title.InitRow, title.MergeRowTo-title.InitRow+1)
		}

		title.dataCheck()

		var (
			cellL = GetCellCoord(title.InitRow, title.InitCol)
			cellR = GetCellCoord(title.MergeRowTo, title.MergeColTo)
		)
		//set text
		if err = t.sheet.file.SetCellStr(t.sheet.SheetName(), cellL, title.Text); err != nil {
			return err
		}
		//Merge cells
		if title.MergeColTo != 0 && title.MergeRowTo != 0 && (title.MergeRowTo != title.InitRow || title.MergeColTo != title.InitCol) {
			if err = t.sheet.file.MergeCell(t.sheet.SheetName(), cellL, cellR); err != nil {
				return err
			}
		}
		//set Style
		if title.style == nil {
			title.style = t.sheet.titleStyle()
			title.style.Font.Size = t.setFontSize(title.Level)
		}
		headerStyleID, errs := t.sheet.file.NewStyle(title.style)
		if errs != nil {
			return
		}
		if err = t.sheet.file.SetCellStyle(t.sheet.SheetName(), cellL, cellR, headerStyleID); err != nil {
			return
		}
		colNum := title.MergeColTo - title.InitCol + 1
		if t.colNum < colNum {
			t.colNum = colNum
		}
		t.titles = append(t.titles, title)
	}
	return nil
}

// setFontSize
func (t *Title) setFontSize(titleLevel int) float64 {
	if titleLevel > t.MaxLevel || titleLevel < 0 {
		return t.BaseFontSize
	}
	fontSize := t.MaxFontSize - float64(titleLevel-1)*t.FontScaleFactor
	if fontSize < t.BaseFontSize {
		return t.BaseFontSize
	}
	return fontSize
}

// NewTitleItem
func (t *Title) NewTitleItem(level int, text string, initRow, initCol int, style ...*excelize.Style) *TitleItem {
	tm := &TitleItem{
		Level:   level,
		Text:    text,
		InitRow: initRow,
		InitCol: initCol,
		sheet:   t.sheet,
	}
	tm.dataCheck()
	if style != nil {
		tm.style = style[0]
	}
	t.titles = append(t.titles, tm)
	tm.id = len(t.titles) - 1
	return tm
}

type TitleItem struct {
	id         int
	Level      int
	Text       string
	InitRow    int //所在行
	InitCol    int //default is 1
	MergeColTo int
	MergeRowTo int
	style      *excelize.Style
	sheet      *Sheet
}

// GetId
func (t *TitleItem) GetId() int {
	return t.id
}

// SetStyle
// By default, the style is set automatically based on the title level.
func (t *TitleItem) SetStyle(style *excelize.Style) *TitleItem {
	t.style = style
	return t
}

// SetMergeColNum
func (t *TitleItem) SetMergeColNum(num int) *TitleItem {
	t.MergeColTo = t.InitCol + num - 1
	return t
}

// SetMergeRowNum
func (t *TitleItem) SetMergeRowNum(num int) *TitleItem {
	t.MergeRowTo = t.InitRow + num - 1
	return t
}

// SetMergeColTo
func (t *TitleItem) SetMergeColTo(colTo int) *TitleItem {
	t.MergeColTo = colTo
	return t
}

// SetMergeRowTo
func (t *TitleItem) SetMergeRowTo(colTo int) *TitleItem {
	t.MergeColTo = colTo
	return t
}

// SetMergeRowTo
func (t *TitleItem) SetMergeTo(row, col int) *TitleItem {
	t.MergeRowTo = row
	t.MergeColTo = col
	return t
}

// SetFullHorizontalMerge
func (t *TitleItem) SetFullHorizontalMerge(defaultColNum ...int) *TitleItem {
	t.InitCol = 1
	if defaultColNum != nil {
		t.MergeColTo = defaultColNum[0]
	} else {
		t.MergeColTo = len(t.sheet.Fields())
		t.dataCheck()
	}
	return t
}

// dataCheck
func (t *TitleItem) dataCheck() {
	if t.InitCol == 0 {
		t.InitCol = 1
	}
	if t.InitRow == 0 {
		t.InitRow = 1
	}
	if t.MergeRowTo == 0 {
		t.MergeRowTo = t.InitRow
	}
	if t.MergeColTo == 0 {
		t.MergeColTo = t.InitCol
	}
	if t.Level == 0 {
		t.Level = 5
	}
}