package ExcelUtil

import (
	"dsBaseRpc/Const"
	"dsBaseRpc/Utils/CommonUtil"
	"dsBaseRpc/Utils/DbUtil"
	"dsBaseRpc/Utils/FileUtil"
	"encoding/json"
	"fmt"
	"github.com/xuri/excelize/v2"
	"strconv"
	"strings"
)

//中文文档: xuri.me/excelize/zh-hans

type ColStruct struct {
	ColName      string        `json:"col_name"`
	ColField     string        `json:"col_field"`
	Width        float64       `json:"width"`
	Require      bool          `json:"require"`
	Level1Sql    bool          `json:"level_1_sql"`
	Level2Sql    bool          `json:"level_2_sql"`
	SqlNum       int64         `json:"sql_num"`
	SqlParameter []interface{} `json:"sql_parameter"`
}
type TemplateStruct struct {
	Title     string      `json:"title"`
	Level1Sql []string    `json:"level_1_sql"`
	Level2Sql [][]string  `json:"level_2_sql"`
	Cols      []ColStruct `json:"cols"`
}

var db = DbUtil.Engine

//样式文件的前缀路径
var stylePathPrefix = "./Config/ExcelStyle/"

/**
功能：设置指定EXCEL SHEET的某个单元格的背景色为红色
作者：黄海
时间：2020-06-08
*/
func GetBackgroundColorStyle(hexColor string) string {
	tableStyleWarningPath := stylePathPrefix + "tableStyleTemplate.json"
	if !FileUtil.PathExists(tableStyleWarningPath) {
		tableStyleWarningPath = "." + tableStyleWarningPath
	}
	content := FileUtil.ReadFileContent(tableStyleWarningPath)
	content = strings.Replace(content, "\r\n", "", -1)
	styleTemplate := strings.Replace(content, "#color#", hexColor, -1)
	return styleTemplate
}

//导出标识
const ExportFlag = 1

//导入标识
const ImportFlag = 2

/**
功能：判断是不是合法的导入模板
作者：黄海
时间：2020-06-17
*/
func IsValidTemplate(excelPath string, s1 TemplateStruct) bool {
	f, _ := excelize.OpenFile(excelPath)
	//如果 sheet名称不正确
	if f.GetSheetName(0) != s1.Title {
		return false
	}
	var isValidTemplate = true
	for i := range s1.Cols {
		colName, _ := excelize.ColumnNumberToName(i + 1)
		v, _ := f.GetCellValue(s1.Title, colName+CommonUtil.ConvertIntToString(HiddenRows+1))
		if s1.Cols[i].Require {
			if v != s1.Cols[i].ColName+"[*]" {
				isValidTemplate = false
				break
			}
		} else {
			if v != s1.Cols[i].ColName {
				isValidTemplate = false
				break
			}
		}
	}
	return isValidTemplate
}

/**
功能：读取配置JSON
作者：黄海
时间：2020-06-17
*/
func ReadJson(jsonTemplate string, ExportOrImport int) TemplateStruct {
	if ExportOrImport == 1 {
		jsonTemplate = "./Config/ExportExcelTemplate/" + jsonTemplate
	} else {
		jsonTemplate = "./Config/ImportExcelTemplate/" + jsonTemplate
	}
	if !FileUtil.PathExists(jsonTemplate) {
		jsonTemplate = "." + jsonTemplate
	}
	var jsonContent = FileUtil.ReadFileContent(jsonTemplate)
	jsonContent = strings.Replace(jsonContent, "\r\n", "", -1)
	var s1 TemplateStruct
	json.Unmarshal([]byte(jsonContent), &s1)
	return s1
}

/**
功能：通过JSON模板定义导出EXCEL
作者：黄海
时间：2020-06-05
*/
func ExportExcel(jsonTemplate string, listData []map[string]interface{}, targetPath string) {
	var s1 = ReadJson(jsonTemplate, ExportFlag)
	rows := len(listData) //行数
	cols := len(s1.Cols)  //列数
	//初始化excel
	f := excelize.NewFile()
	titleStylePath := stylePathPrefix + "titleStyle.json"
	tableStylePath := stylePathPrefix + "tableStyleNormal.json"

	if !FileUtil.PathExists(titleStylePath) {
		titleStylePath = "." + titleStylePath
	}
	if !FileUtil.PathExists(tableStylePath) {
		tableStylePath = "." + tableStylePath
	}

	titleStyle, _ := f.NewStyle(FileUtil.ReadFileContent(titleStylePath))
	tableStyle, _ := f.NewStyle(FileUtil.ReadFileContent(tableStylePath))

	var oldName = "Sheet1"
	var SheetName = s1.Title
	var RowHeight = 28.0
	f.SetSheetName(oldName, SheetName)
	//设置标题行行高
	f.SetRowHeight(SheetName, 1, RowHeight)
	//合并单元格(标题行)
	lastCol, _ := excelize.ColumnNumberToName(cols)
	f.MergeCell(SheetName, "A1", lastCol+"1")
	// 设置单元格的值(标题)
	f.SetCellValue(SheetName, "A1", s1.Title)
	//测试设置单元格背景颜色
	//设置字体(标题)
	f.SetCellStyle(SheetName, "A1", "A1", titleStyle)
	//设置列宽
	for i := 0; i < cols; i++ {
		m, _ := excelize.ColumnNumberToName(i + 1)
		f.SetColWidth(SheetName, m, m, s1.Cols[i].Width)
	}
	//设置边框
	for i := 2; i <= rows+2; i++ {
		for j := 0; j < cols; j++ {
			cName, _ := excelize.ColumnNumberToName(j + 1)
			cName = cName + strconv.Itoa(i)
			f.SetCellStyle(SheetName, cName, cName, tableStyle)
			f.SetRowHeight(SheetName, i, RowHeight)
		}
	}
	//写入标题头
	for i := 0; i < cols; i++ {
		cName, _ := excelize.ColumnNumberToName(i + 1)
		cName = cName + strconv.Itoa(2)
		f.SetCellValue(SheetName, cName, s1.Cols[i].ColName)
	}
	//写入内容
	for i := 0; i < rows; i++ {
		for j := 0; j < cols; j++ {
			cName, _ := excelize.ColumnNumberToName(j + 1)
			cName = cName + strconv.Itoa(i+1+2)
			fieldName := s1.Cols[j].ColField
			if fieldName == "number" {
				f.SetCellValue(SheetName, cName, CommonUtil.ConvertIntToString(i+1))
			} else {
				f.SetCellValue(SheetName, cName, listData[i][fieldName])
			}
		}
	}
	// 根据指定路径保存文件
	if err := f.SaveAs(targetPath); err != nil {
		println(err.Error())
	}
}

//5000个
const RowsCount = 5000

//隐藏用来装字典的行数
const HiddenRows = 200

/**
功能：通用的导入模板函数
作者：黄海
时间：2020-06-06
*/
func GenerateTemplate(jsonTemplate string, excelPath string) {
	var f *excelize.File
	var s1 TemplateStruct
	var SheetName string
	jsonTemplate = "./Config/ImportExcelTemplate/" + jsonTemplate
	if !FileUtil.PathExists(jsonTemplate) {
		jsonTemplate = "." + jsonTemplate
	}
	var jsonContent = FileUtil.ReadFileContent(jsonTemplate)
	//2、还原成结构体
	json.Unmarshal([]byte(jsonContent), &s1)
	//3、准备根据结构体进行创建导入模板
	var oldName = "Sheet1"
	SheetName = s1.Title
	f = excelize.NewFile()
	f.SetSheetName(oldName, SheetName)
	cols := len(s1.Cols) //列数
	//隐藏前hiddenRows行
	for i := 0; i < HiddenRows; i++ {
		f.SetRowVisible(SheetName, i+1, false)
	}
	//4、声明样式描述文件
	titleStylePath := stylePathPrefix + "titleStyle.json"
	tableStylePath := stylePathPrefix + "tableStyleNormal.json"
	if !FileUtil.PathExists(titleStylePath) {
		titleStylePath = "." + titleStylePath
	}
	if !FileUtil.PathExists(tableStylePath) {
		tableStylePath = "." + tableStylePath
	}
	tableStyle, _ := f.NewStyle(FileUtil.ReadFileContent(tableStylePath))

	//5、表头
	for i := 0; i < cols; i++ {
		//设置列宽
		cName, _ := excelize.ColumnNumberToName(i + 1)
		f.SetColWidth(SheetName, cName, cName, s1.Cols[i].Width)
		//写入文字
		cName = cName + strconv.Itoa(HiddenRows+1) //第hiddenRows+1行
		f.SetCellStyle(SheetName, cName, cName, tableStyle)
		//加上红星提醒
		if s1.Cols[i].Require {
			f.SetCellValue(SheetName, cName, s1.Cols[i].ColName)
			f.SetCellRichText(SheetName, cName, []excelize.RichTextRun{
				{
					Text: s1.Cols[i].ColName,
					Font: &excelize.Font{
						Family: "宋体",
						Size:   14,
					},
				},
				{
					Text: "[*]",
					Font: &excelize.Font{
						Bold:   true,
						Color:  "e83723",
						Family: "宋体",
						Size:   14,
					},
				},
			})
		} else {
			f.SetCellValue(SheetName, cName, s1.Cols[i].ColName)
		}
	}
	//6、设置足够多的行是28这个高度，RowsCount行应该够用了
	var RowHeight = 28.0
	//表头设置行高
	f.SetRowHeight(SheetName, 1, RowHeight)
	//设置边框
	for i := HiddenRows + 1; i <= HiddenRows+RowsCount; i++ {
		for j := 0; j < cols; j++ {
			cName, _ := excelize.ColumnNumberToName(j + 1)
			cName = cName + strconv.Itoa(i)
			f.SetCellStyle(SheetName, cName, cName, tableStyle)
		}
		f.SetRowHeight(SheetName, i, RowHeight)
	}
	//白色
	WhiteJson := GetBackgroundColorStyle(Const.WhiteStyle)
	WhiteStyle, _ := f.NewStyle(WhiteJson)
	f.SetCellStyle(SheetName, Const.WhiteCell, Const.WhiteCell, WhiteStyle)
	//红色
	RedJson := GetBackgroundColorStyle(Const.RedStyle)
	RedStyle, _ := f.NewStyle(RedJson)
	f.SetCellStyle(SheetName, Const.RedCell, Const.RedCell, RedStyle)
	//蓝色
	BlueJson := GetBackgroundColorStyle(Const.BlueStyle)
	BlueStyle, _ := f.NewStyle(BlueJson)
	f.SetCellStyle(SheetName, Const.BlueCell, Const.BlueCell, BlueStyle)
	//黄色
	YellowJson := GetBackgroundColorStyle(Const.YellowStyle)
	YellowStyle, _ := f.NewStyle(YellowJson)
	f.SetCellStyle(SheetName, Const.YellowCell, Const.YellowCell, YellowStyle)
	//保存
	// 根据指定路径保存文件(不管是不是通过，都保存一次)
	if err := f.SaveAs(excelPath); err != nil {
		println(err.Error())
	}
	return
}

/**
功能：按SQL定义添加字典
作者：黄海
时间：2020-06-18
*/
func TemplateAddDict(excelPath string, jsonTemplate string) (*excelize.File, TemplateStruct, int, string) {
	excelPath = "./Config/ImportExcelTemplate/" + excelPath
	if !FileUtil.PathExists(excelPath) {
		excelPath = "." + excelPath
	}
	f, _ := excelize.OpenFile(excelPath)
	var s1 TemplateStruct

	jsonTemplate = "./Config/ImportExcelTemplate/" + jsonTemplate
	if !FileUtil.PathExists(jsonTemplate) {
		jsonTemplate = "." + jsonTemplate
	}
	var jsonContent = FileUtil.ReadFileContent(jsonTemplate)
	//2、还原成结构体
	json.Unmarshal([]byte(jsonContent), &s1)
	cols := len(s1.Cols) //列数
	var SheetName = s1.Title
	//字典的列开始值
	dictCols := cols + 1
	//设置下拉框
	for i := 0; i < cols; i++ {
		//如果需要设置一级下接框
		if s1.Cols[i].Level1Sql {
			//查询值域
			args := s1.Cols[i].SqlParameter
			list, err := db.SQL(s1.Level1Sql[s1.Cols[i].SqlNum-1], args...).Query().List()
			if err != nil {
				fmt.Println(err.Error())
			}
			//字典列从cols+1开始
			dictCName, _ := excelize.ColumnNumberToName(dictCols)
			dictCols++
			for j := range list {
				f.SetCellValue(SheetName, dictCName+CommonUtil.ConvertIntToString(j+1), list[j]["dict_value"])
			}
			//隐藏字典列
			f.SetColVisible(SheetName, dictCName, false)
			//要设置下拉框的列
			cName, _ := excelize.ColumnNumberToName(i + 1)
			dvRange := excelize.NewDataValidation(true)
			dvRange.Sqref = cName + CommonUtil.ConvertIntToString(2) + ":" + cName + CommonUtil.ConvertIntToString(RowsCount)
			c := "$" + dictCName + "$1:$" + dictCName + "$" + CommonUtil.ConvertIntToString(len(list))
			dvRange.SetSqrefDropList(c)
			f.AddDataValidation(SheetName, dvRange)
		}
	}
	return f, s1, dictCols, SheetName
}