// Package excel -----------------------------
// @file      : templateInter_test.go
// @author    : JJXu
// @contact   : wavingbear@163.com
// @time      : 2022/7/23 15:59
// -------------------------------------------
package example

import (
	"fmt"
	"gingogo2/utils/excel"
	"path/filepath"
	"strings"
	"testing"
	"time"
)

//定义sheet表结构
type sheet1Define struct {
	Xid  int    `json:"xid" form:"xid" db:"column:xid;comment: "`
	Name string `json:"name" form:"name" db:"column:name;comment: "`
	Age  int    `json:"age" form:"age" db:"column:age;comment: "`
}

func TestExcelTpl_WriteToExcel(t *testing.T) {
	var sheet1Data = []sheet1Define{
		{1, "张三", 16},
		{2, "黑猫警长", 18},
	}

	var sheet1 = excel.NewSheet("Sheet1", sheet1Data)
	//var suffixFunc = func() string { return fmt.Sprintf("%v", time.Now().Unix()) }
	exCreator, err := excel.NewExcelTemplate("demo.xlsx", "./", "./demo.xlsx", sheet1)
	if err != nil {
		t.Log(err)
	}
	exCreator.UseOption(func(excel *excel.Excel) {
		ext := filepath.Ext(excel.SaveName)
		name := strings.Split(excel.SaveName, ext)[0]
		excel.SaveName = fmt.Sprintf("%s_%v%s", name, time.Now().Unix(), ext)
	})
	path, name, err := exCreator.WriteToFile()
	if err != nil {
		t.Log(err)
	} else {
		t.Log(path, name)
	}
}

func TestReadExcel(t *testing.T) {
	var file = "demo_1671435052.xlsx"
	ex := excel.Excel{OriginFilePath: file}
	var datas []sheet1Define
	err := ex.ReadSheetData("Sheet1", func(rowIndex int, rows []string) {
		if rowIndex == 0 {
			//跳过首行
			return
		}
		datas = append(datas, sheet1Define{
			Xid:  excel.Int[int](rows[0]),
			Name: rows[1],
			Age:  excel.Int[int](rows[2]),
		})
	})
	if err != nil {
		t.Error(err.Error())
	}
	fmt.Println(datas)

}