package xlsx

import (
	"joybuywork/internal/plugins/export"
)

type Option func(e *Export)

// UseDefaultSheet 创建文件开始的默认工作簿
func UseDefaultSheet(name string) Option {
	return func(e *Export) {
		//旧的默认工作表
		index := e.CreateSheet(name)

		e.file.SetActiveSheet(index)

		//这个是组件自带的，所以删除掉
		e.file.DeleteSheet(DefaultSheetName)
		e.sheet = name
	}
}

type Sheet struct {
	Name        string             //当前sheet的名称
	Header      export.HeaderField //当前sheet的字段数
	SwitchField []SwitchField      //切换字段和标识值
	Total       int64              //当前sheet的写入的总条数，用于自动切sheet，当 SwitchField = nil 的时候才使用该值
	writeNum    int64              //写入当前sheet的条数
	sfIdx       int64              //当前sheet从第几行开始写入
	fields      map[string]string  //头部对应的表格列和字段名 map[string]string{"A":"字段key1", "B": "字段key2"}
	header      map[string]string  //表头数据map[string]string{"A1":"表头1", "B1": "表头2"}
}

type SwitchField struct {
	Key string      //当前sheet的字段标识
	Val interface{} //当前sheet的字段标识的值
}

// UseManySheet 多sheet导出
func UseManySheet(ss ...Sheet) Option {
	return func(e *Export) {
		var defaultIdx = -1
		for k, v := range ss {
			//工作表存在，代表已经设置过表头，不用再设置了
			_ = e.CreateSheet(v.Name)

			//初始化
			e.initSheet(&v, true)

			if defaultIdx == -1 && (v.Total > 0 || v.SwitchField != nil) {
				defaultIdx = k
			}

			//设置他的索引
			e.ss = append(e.ss, v)

			if v.SwitchField != nil {
				if e.sfv == nil {
					e.sfv = make(map[string]int)
				}
				for _, v1 := range v.SwitchField {
					e.sfv[e.getSfvKey(v1.Key, v1.Val)] = k
				}
			}
		}

		//从第一个有导出数据的的sheet开始
		e.initSheet(&ss[defaultIdx], false)
	}
}