package service

import (
	butil "enterprise/base/util"
	"enterprise/common/config"
	"enterprise/common/dao"
	"enterprise/common/model"
	"fmt"
	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cast"
	"os"
	"time"

	"github.com/xuri/excelize/v2"
	"net/http"
)

var (
	StaffSalaryTypeBank    = "bank"
	StaffSalaryTypeSummary = "summary"
	StaffSalaryTypeAgent   = "agent"
)

type StaffSalary struct {
}

func (s *StaffSalary) Agent(month string, ctx *gin.Context) {
	xls := ctx.Query("xls")
	staffSalarys, err := dao.NewStaffSalaryDao().Query(month)
	if err != nil {
		panic(config.ErrDb.New().Append(err))
	}

	header := []string{"姓名", "身份证号", "电话", "基本工资", "出勤工资", "社保扣除", "个税扣除", "应发工资"}
	datas := make([][]string, 0)
	for _, staff := range staffSalarys {
		baseInfo, err := dao.NewStaffInfoDao().GetByUsername(staff.Username)
		if err != nil {
			log.Errorf("db error :%s", err.Error())
			continue
		}
		if baseInfo == nil {
			continue
		}
		if staff.BaseSalary == 0 {
			continue
		}
		item := make([]string, 0)
		item = append(item, baseInfo.Realname)
		item = append(item, cast.ToString(baseInfo.Idno))
		item = append(item, cast.ToString(baseInfo.Phone))
		item = append(item, cast.ToString(staff.BaseSalary))
		item = append(item, cast.ToString(staff.RealSalary))
		item = append(item, cast.ToString(staff.SocialInsurence))
		item = append(item, cast.ToString(staff.PersonalTax))
		item = append(item, cast.ToString(staff.RealSalary-staff.SocialInsurence-staff.PersonalTax))

		datas = append(datas, item)
	}

	if xls != "" {
		filename := fmt.Sprintf("agent_%s.xlsx", time.Now().Format("20060102_150405"))
		s.toExcel(filename, header, datas, ctx)
		os.Remove(filename)
	} else {
		links := make([]map[string]string, 0)
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=agent&xls=1",
			"name": "下载",
		})
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=bank",
			"name": "银行",
		})
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=summary",
			"name": "汇总",
		})
		ctx.HTML(http.StatusOK, "salary.html", gin.H{
			"title":  month + "工资汇总",
			"header": header,
			"data":   datas,
			"link":   links,
		})
	}
}

func (s *StaffSalary) Bank(month string, ctx *gin.Context) {
	xls := ctx.Query("xls")
	staffSalarys, err := dao.NewStaffSalaryDao().Query(month)
	if err != nil {
		panic(config.ErrDb.New().Append(err))
	}

	header := []string{"账号", "户名", "金额", "开户行", "开户地", "收款备注"}
	datas := make([][]string, 0)
	for _, staff := range staffSalarys {
		baseInfo, err := dao.NewStaffInfoDao().GetByUsername(staff.Username)
		if err != nil {
			log.Errorf("db error :%s", err.Error())
			continue
		}
		if baseInfo == nil {
			continue
		}
		if staff.BaseSalary == 0 {
			continue
		}
		item := make([]string, 0)
		item = append(item, baseInfo.BankCard)
		item = append(item, baseInfo.Realname)
		item = append(item, cast.ToString(staff.RealSalary-staff.SocialInsurence-staff.PersonalTax))
		item = append(item, cast.ToString(baseInfo.BankName))
		item = append(item, "重庆市")
		item = append(item, fmt.Sprintf("%s工资", month))
		datas = append(datas, item)
	}

	if xls != "" {
		filename := fmt.Sprintf("bank_%s.xlsx", time.Now().Format("20060102_150405"))
		s.toExcel(filename, header, datas, ctx)
		os.Remove(filename)
	} else {
		links := make([]map[string]string, 0)
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=bank&xls=1",
			"name": "下载",
		})
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=summary",
			"name": "汇总",
		})
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=agent",
			"name": "代理",
		})
		ctx.HTML(http.StatusOK, "salary.html", gin.H{
			"title":  month + "工资汇总",
			"header": header,
			"data":   datas,
			"link":   links,
		})
	}
}

func (s *StaffSalary) Summary(month string, ctx *gin.Context) {
	xls := ctx.Query("xls")
	staffSalarys, err := dao.NewStaffSalaryDao().Query(month)
	if err != nil {
		panic(config.ErrDb.New().Append(err))
	}

	header := []string{"姓名", "身份证号", "入职日期", "转正日期", "基本工资", "出勤工资", "社保扣除", "个税扣除", "请假天数", "实发工资"}
	datas := make([][]string, 0)
	summary := new(model.StaffSalary)
	for _, staff := range staffSalarys {
		baseInfo, err := dao.NewStaffInfoDao().GetByUsername(staff.Username)
		if err != nil {
			log.Errorf("db error :%s", err.Error())
			continue
		}
		if baseInfo == nil {
			continue
		}

		item := make([]string, 0)
		item = append(item, baseInfo.Realname)
		item = append(item, cast.ToString(baseInfo.Idno))
		item = append(item, cast.ToString(baseInfo.EntryDate))
		item = append(item, cast.ToString(baseInfo.OfficialDate))
		item = append(item, cast.ToString(staff.BaseSalary))
		item = append(item, cast.ToString(staff.RealSalary))
		item = append(item, cast.ToString(staff.SocialInsurence))
		item = append(item, cast.ToString(staff.PersonalTax))
		item = append(item, cast.ToString(staff.Holiday))
		item = append(item, cast.ToString(staff.RealSalary-staff.SocialInsurence-staff.PersonalTax))

		datas = append(datas, item)
		summary.BaseSalary += staff.BaseSalary
		summary.RealSalary += staff.RealSalary
		summary.SocialInsurence += staff.SocialInsurence
		summary.PersonalTax += staff.PersonalTax
		summary.Holiday += staff.Holiday
	}
	summary.Holiday = butil.FloatCut(summary.Holiday)
	summary.BaseSalary = butil.FloatCut(summary.BaseSalary)
	summary.RealSalary = butil.FloatCut(summary.RealSalary)
	summary.SocialInsurence = butil.FloatCut(summary.SocialInsurence)
	summary.PersonalTax = butil.FloatCut(summary.PersonalTax)
	datas = append(datas, []string{"合计", "-", "-", "-",
		cast.ToString(summary.BaseSalary), cast.ToString(summary.RealSalary), cast.ToString(summary.SocialInsurence),
		cast.ToString(summary.PersonalTax), cast.ToString(summary.Holiday),
		cast.ToString(summary.RealSalary - summary.SocialInsurence - summary.PersonalTax)})

	if xls != "" {
		filename := fmt.Sprintf("summary_%s.xlsx", time.Now().Format("20060102_150405"))
		s.toExcel(filename, header, datas, ctx)
		os.Remove(filename)
	} else {
		links := make([]map[string]string, 0)
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=bank",
			"name": "银行",
		})
		links = append(links, map[string]string{
			"url":  "/api/staff/salary?type=agent",
			"name": "代理",
		})
		ctx.HTML(http.StatusOK, "salary.html", gin.H{
			"title":  month + "工资汇总",
			"header": header,
			"data":   datas,
			"link":   links,
		})
	}
}

func (s *StaffSalary) toExcel(filePath string, header []string, datas [][]string, ctx *gin.Context) {
	f := excelize.NewFile()
	//声明工作表的名称
	sheetName := "工资单"
	f.SetSheetName("Sheet1", sheetName)

	f.SetSheetRow(sheetName, "A1", &header)
	for i, data := range datas {
		f.SetSheetRow(sheetName, fmt.Sprintf("A%d", i+2), &data)
	}
	if err := f.SaveAs(filePath); err != nil {
		panic(config.ErrInternal.New().Append(err))
	}

	ctx.Writer.Header().Add("Content-Disposition", fmt.Sprintf("attachment; filename=%s", filePath))
	ctx.Writer.Header().Add("Content-Type", "application/msexcel")
	ctx.File(filePath)
}