package innerserverser

import (
	"errors"
	"fmt"
	"github.com/360EntSecGroup-Skylar/excelize"
	"stat/api/proto"
	"stat/common/log"
	"stat/common/utils"
	"strconv"
	"time"
)

var exportTransWaterTitle = []string{"日活跃", "日注册", "视频日充值", "棋牌日充值", "交易笔数"}

const sheetName = "Sheet1"

// 按字母
func NewXlslFile(f *excelize.File, list []*proto.InnerDailyRecharge) {
	// 设置表头样式
	headStyleID, _ := f.NewStyle(titlestyle)
	filePath := "/usr/local" + "/bakexel/" + "vvvv.xlsx"
	// 设置列宽
	f.SetColWidth("Sheet1", "A", "L", 23)

	// 设置表头行高
	//f.SetRowHeight("Sheet1", 1, 25)
	f.SetCellStyle("Sheet1", "A1", "L1", headStyleID)
	idx := f.NewSheet(sheetName)
	f.MergeCell("Sheet1", "A1", "A2")
	f.SetCellValue(sheetName, "A1", "日期")
	f.SetCellStyle(sheetName, "A4", "A50", headStyleID)
	f.SetActiveSheet(idx)
	now := time.Now()
	day := now.Format("2006-01")
	begin, end := utils.MonthDateRange(now)
	beginday := utils.BeginOfMonth(now).Day()
	endday := utils.EndOfMonth(now).Day()
	index := "A"
	for i := beginday; i <= endday; i++ {
		tmp := fmt.Sprintf("%s%d", index, i+2)
		vvvvv := fmt.Sprintf("%s-%d", day, i)
		f.SetCellValue(sheetName, tmp, vvvvv)
	}
	one := 'A'
	begin0 := 'B'
	temp0 := begin0
	pre := ""
	for _, info := range list {
		t0 := temp0
		countt0 := 1
		for i := 1; i < 5; i++ {
			temp0 = t0 + int32(countt0)
			countt0++
			if temp0 > 'Z' {
				one++
				temp0 = 'A'
				t0 = temp0
				countt0 = 1
			}
		}
		//merge := setcellindex + 5 + int32(index*5)
		beginmerge := fmt.Sprintf("%s%s1", pre, string(begin0))
		if one > 'A' {
			pre = string(one - 1)
		}
		endmerge := fmt.Sprintf("%s%s1", pre, string(temp0))
		temp0 += 2
		if temp0 > 'Z' {
			one++
			temp0 = 'A' + temp0 - 'Z' - 1
			if one > 'A' {
				pre = string(one - 1)
			}
		}
		begin0 = temp0
		f.MergeCell("Sheet1", beginmerge, endmerge)
		//tmp := fmt.Sprintf("%s1", string(index0))
		f.SetCellValue(sheetName, beginmerge, info.Name)
		fmt.Println(beginmerge, endmerge)
	}
	err := f.SaveAs(filePath)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Println(begin, end)
}

// 按行列索引
func NewXlslFile0(f *excelize.File, list []*proto.InnerDailyRecharge, titlestyle, cellstyle int) error {
	// 设置列宽
	f.SetColWidth(sheetName, "A", "AAA", 19)

	// 设置表头行高
	f.SetRowHeight(sheetName, 1, 25)
	f.SetCellStyle(sheetName, "A1", "ZZ1", titlestyle)
	f.SetCellStyle(sheetName, "A2", "ZZ2", titlestyle)
	f.SetCellStyle(sheetName, "A3", "ZZ3", titlestyle)

	idx := f.NewSheet(sheetName)
	f.SetCellValue(sheetName, "A1", "日期")
	f.SetActiveSheet(idx)

	now := time.Now()
	dayformat := now.Format("2006-01")

	//begin, end := utils.MonthDateRange(now)
	beginday := utils.BeginOfMonth(now).Day()
	endday := utils.EndOfMonth(now).Day()

	//设置日期
	index := "A"
	for i := beginday; i <= endday; i++ {
		tmp := fmt.Sprintf("%s%d", index, i+3)
		rowday := fmt.Sprintf("%s-%02d", dayformat, i)

		f.SetCellValue(sheetName, tmp, rowday)
	}

	//新文件设置固定的主题和style
	row := 1
	col := 2
	count := col
	for _, info := range list {
		begincol := count
		count = count + 4
		titleindex := 0
		titlemap := make(map[string]string, 0)
		for i := begincol; i <= count; i++ {
			if titleindex < len(exportTransWaterTitle) {
				titlecell, serr := utils.CoordinatesToCellName(begincol+titleindex, row+2)
				if serr != nil {
					log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
						log.Any("title", exportTransWaterTitle[titleindex]))
					return serr
				}
				titlemap[titlecell] = exportTransWaterTitle[titleindex]
			}
			titleindex++
		}
		info.BeginCol = begincol
		// 第一行产品
		begincell, serr := utils.CoordinatesToCellName(begincol, row)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		endcell, serr := utils.CoordinatesToCellName(count, row)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		// 第二行appid
		appidbegincell, serr := utils.CoordinatesToCellName(begincol, row+1)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoappid", info.AppId))
			return serr
		}
		appidendcell, serr := utils.CoordinatesToCellName(count, row+1)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoappid", info.AppId))
			return serr
		}
		f.MergeCell(sheetName, begincell, endcell)

		f.MergeCell(sheetName, appidbegincell, appidendcell)
		//tmp := fmt.Sprintf("%s1", string(index0))
		/*if info.AppId != 0 {
			if info.AppId == 888 {
				//tmp := fmt.Sprintf("%s1", string(index0))
				f.SetCellValue(sheetName, begincell, info.Name+"(集团)")
			} else {
				f.SetCellValue(sheetName, begincell, info.Name+"(内部)")
			}
		} else {
		*/
		f.SetCellValue(sheetName, begincell, info.Name)
		//}
		f.SetCellValue(sheetName, appidbegincell, info.AppId)

		//第三行app统计的信息
		for key, value := range titlemap {
			//f.SetCellStyle(sheetName, key, key, cellstyle)
			f.SetCellValue(sheetName, key, value)
		}
		count += 2
	}
	return nil
}

// 合并单元格
func MergeCell(f *excelize.File, sheetname string, col int, list []*proto.InnerDailyRecharge) error {
	row := 1
	count := col + 1
	for _, info := range list {
		begincol := count
		count = count + 4
		titleindex := 0
		titlemap := make(map[string]string, 0)
		for i := begincol; i <= count; i++ {
			if titleindex < len(exportTransWaterTitle) {
				titlecell, serr := utils.CoordinatesToCellName(begincol+titleindex, row+2)
				if serr != nil {
					log.Error("MergeCell CoordinatesToCellName ", log.Any("err", serr),
						log.Any("title", exportTransWaterTitle[titleindex]))
					return serr
				}
				titlemap[titlecell] = exportTransWaterTitle[titleindex]
			}
			titleindex++
		}
		begincell, serr := utils.CoordinatesToCellName(begincol, row)
		if serr != nil {
			log.Error("MergeCell CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		endcell, serr := utils.CoordinatesToCellName(count, row)
		if serr != nil {
			log.Error("MergeCell CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		// 第一行appid
		appidbegincell, serr := utils.CoordinatesToCellName(begincol, row+1)
		if serr != nil {
			log.Error("MergeCell CoordinatesToCellName ", log.Any("err", serr),
				log.Any("appid", info.AppId))
			return serr
		}
		appidendcell, serr := utils.CoordinatesToCellName(count, row+1)
		if serr != nil {
			log.Error("MergeCell CoordinatesToCellName ", log.Any("err", serr),
				log.Any("appid", info.AppId))
			return serr
		}
		info.BeginCol = begincol
		f.MergeCell(sheetname, begincell, endcell)
		/*if info.AppId != 0 {
			if info.AppId == 888 {
				f.SetCellValue(sheetname, begincell, info.Name+"(集团)")
			} else {
				f.SetCellValue(sheetname, begincell, info.Name+"(内部)")
			}
		} else {
		*/
		f.SetCellValue(sheetname, begincell, info.Name)
		//}

		f.MergeCell(sheetname, appidbegincell, appidendcell)
		//tmp := fmt.Sprintf("%s1", string(index0))
		f.SetCellValue(sheetname, appidbegincell, info.AppId)
		for key, value := range titlemap {
			f.SetCellValue(sheetname, key, value)
		}
		count += 2
	}

	return nil
}

func OpenOldXlsl(f *excelize.File, list []*proto.InnerDailyRecharge) error {
	idx := f.GetSheetIndex(sheetName)
	f.SetActiveSheet(idx)

	cols := f.GetRows(sheetName)
	begincol := 0
	//查找数据开始单元格，新数据先合并单元格
	/*for rowindex, col := range cols {
		for colindex, rowCell := range col {
			if rowCell == "" {
				continue
			}
			if rowindex == 1 {
				for _, info := range list {
					if rowCell == string(info.AppId) {
						info.BeginRow = rowindex + 1
						info.BeginCol = colindex + 1
						info.IsOld = true
					}
				}
			}
		}
	}
	*/

	if len(cols) < 2 || len(cols[1]) < len(list) {
		return errors.New("OpenOldXlsl cols < 0")
	}

	/*for rowindex := 0; rowindex < len(cols); rowindex++ {
		if cols[rowindex][0] != "" {
			for _, info := range list {
				if cols[rowindex][0] == info.DateTime {
					info.BeginRow = rowindex + 1
					info.IsOld = true
				}
			}
		}
	}
	*/
	//旧的数据单元格开始信息
	for colindex := 0; colindex < len(cols[1]); colindex++ {
		for _, info := range list {
			appid := strconv.Itoa(int(info.AppId))
			if cols[1][colindex] != "" && cols[1][colindex] == appid {
				info.BeginCol = colindex + 1
				info.IsOld = true
			}
		}
	}

	//新的单元格从第几列信息
	newlist := make([]*proto.InnerDailyRecharge, 0)
	for _, info := range list {
		if info.IsOld {
			if info.BeginCol > begincol {
				begincol = info.BeginCol
			}
		} else {
			newlist = append(newlist, info)
		}
	}
	if len(newlist) > 0 {
		serr := MergeCell(f, sheetName, begincol+5, newlist)
		if serr != nil {
			log.Error("OpenOldXlsl MergeCell ", log.Any("err", serr))
			return serr
		}
	}

	return nil
}

func NewXlslFile2(f *excelize.File, list []*AppInformation, titlestyle, cellstyle int, now time.Time) error {
	// 设置列宽
	f.SetColWidth(sheetName, "A", "AAA", 19)

	// 设置表头行高
	f.SetRowHeight(sheetName, 1, 25)
	f.SetCellStyle(sheetName, "A1", "ZZ1", titlestyle)
	f.SetCellStyle(sheetName, "A2", "ZZ2", titlestyle)
	f.SetCellStyle(sheetName, "A3", "ZZ3", titlestyle)

	idx := f.NewSheet(sheetName)
	f.SetCellValue(sheetName, "A1", "日期")
	f.SetActiveSheet(idx)

	dayformat := now.Format("2006-01")

	//begin, end := utils.MonthDateRange(now)
	beginday := utils.BeginOfMonth(now).Day()
	endday := utils.EndOfMonth(now).Day()

	//设置日期
	index := "A"
	for i := beginday; i <= endday; i++ {
		tmp := fmt.Sprintf("%s%d", index, i+3)
		rowday := fmt.Sprintf("%s-%02d", dayformat, i)
		f.SetCellValue(sheetName, tmp, rowday)
	}
	// 设置汇总
	tmp := fmt.Sprintf("%s%d", index, endday+1+3)
	f.SetCellValue(sheetName, tmp, "汇总")

	//新文件设置固定的主题和style
	row := 1
	col := 2
	count := col
	for _, info := range list {
		begincol := count
		count = count + 4
		titleindex := 0
		titlemap := make(map[string]string, 0)
		for i := begincol; i <= count; i++ {
			if titleindex < len(exportTransWaterTitle) {
				titlecell, serr := utils.CoordinatesToCellName(begincol+titleindex, row+2)
				if serr != nil {
					log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
						log.Any("title", exportTransWaterTitle[titleindex]))
					return serr
				}
				titlemap[titlecell] = exportTransWaterTitle[titleindex]
			}
			titleindex++
		}
		info.BeginCol = begincol
		// 第一行产品
		begincell, serr := utils.CoordinatesToCellName(begincol, row)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		endcell, serr := utils.CoordinatesToCellName(count, row)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoname", info.Name))
			return serr
		}
		// 第二行appid
		appidbegincell, serr := utils.CoordinatesToCellName(begincol, row+1)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoappid", info.Id))
			return serr
		}
		appidendcell, serr := utils.CoordinatesToCellName(count, row+1)
		if serr != nil {
			log.Error("NewXlslFile0 CoordinatesToCellName ", log.Any("err", serr),
				log.Any("infoappid", info.Id))
			return serr
		}
		f.MergeCell(sheetName, begincell, endcell)
		f.MergeCell(sheetName, appidbegincell, appidendcell)
		f.SetCellValue(sheetName, begincell, info.Name)
		f.SetCellValue(sheetName, appidbegincell, info.Id)
		//第三行app统计的信息
		for key, value := range titlemap {
			f.SetCellValue(sheetName, key, value)
		}
		count += 2
	}
	return nil
}