package excel

import "github.com/tealeg/xlsx"

type LuckySheet struct {
	SheetList []LuckySheetData `description:"sheet数据"`
}

func (item *LuckySheet) GetExcelData(isRemoveBlankCell bool) (xlsxFile *xlsx.File, err error) {
	xlsxFile = xlsx.NewFile()
	if err != nil {
		return
	}

	for _, sheet := range item.SheetList {
		tableData, tmpErr := sheet.GetTableDataByLuckySheetDataStr(isRemoveBlankCell)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		err = tableData.WriteExcelSheetData(xlsxFile, sheet.Name)
		if err != nil {
			return
		}
	}

	return
}

func (item *LuckySheet) ToExcel(isRemoveBlankCell bool) (downloadFilePath string, err error) {
	downloadFilePath, err = getDownloadPath()
	if err != nil {
		return
	}

	xlsxFile, err := item.GetExcelData(isRemoveBlankCell)
	if err != nil {
		return
	}

	err = xlsxFile.Save(downloadFilePath)

	return
}