// Package excel...
//
// Description : excel...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2021-11-19 3:59 下午
package excel

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

// NewRead 获取读取实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 4:00 下午 2021/11/19
func NewRead(fullFilePath string, filePassword string) (*Read, error) {
	var err error
	read := &Read{}
	if read.fileHandler, err = excelize.OpenFile(fullFilePath, excelize.Options{Password: filePassword}); nil != err {
		return nil, err
	}
	return read, nil
}

// Read ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 4:03 下午 2021/11/19
type Read struct {
	fileHandler *excelize.File
}

// GetAllSheetList ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 4:06 下午 2021/11/19
func (r *Read) GetAllSheetList() []string {
	return r.fileHandler.GetSheetList()
}

// GetAllData 读取全部数据
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 4:07 下午 2021/11/19
func (r *Read) GetAllData() (map[string][][]string, error) {
	var (
		sheetList []string
		result    map[string][][]string
		err       error
	)
	sheetList = r.GetAllSheetList()
	result = make(map[string][][]string)
	for _, sheetName := range sheetList {
		if result[sheetName], err = r.fileHandler.GetRows(sheetName); nil != err {
			return nil, err
		}
	}
	return result, nil
}

// GetAllDataToMap 读取全部数据,并返回map结构
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 4:20 下午 2021/11/19
func (r *Read) GetAllDataToMap(fieldList []string) (map[string][]map[string]string, error) {
	var (
		allData      map[string][][]string
		err          error
		formatResult map[string][]map[string]string
	)
	formatResult = make(map[string][]map[string]string)
	if allData, err = r.GetAllData(); nil != err {
		return nil, err
	}
	for sheetName, sheetData := range allData {
		formatResult[sheetName] = make([]map[string]string, 0)
		for _, lineData := range sheetData {
			tmpResult := make(map[string]string)
			for idx, colData := range lineData {
				if idx >= len(fieldList) {
					// 指定的字段列表较短, 自动剔除后面的字段
					break
				}
				tmpResult[fieldList[idx]] = colData
			}
			// 字段列表较长, 单元格不足, 自动补齐空字符串
			for i := len(lineData); i < len(fieldList); i++ {
				tmpResult[fieldList[i]] = ""
			}
			formatResult[sheetName] = append(formatResult[sheetName], tmpResult)
		}
	}
	// 格式化数据
	return formatResult, nil
}