package controllers

import (
	"github.com/gin-gonic/gin"
	"github.com/xuri/excelize/v2"
	"silkserver/constvar"
	"silkserver/controllers/request"
	"silkserver/extend/code"
	"silkserver/extend/util"
	"silkserver/middleware"
	"silkserver/models"
	"silkserver/pkg/logx"
	"silkserver/pkg/timex"
	"time"
)

type AttendanceController struct {
}

// AttendanceInput
//
//	@Tags		考勤管理
//	@Summary	考勤导入
//	@Produce	application/xlsx
//	@Success	200		{object}	util.Response	"成功"
//	@Router		/api-jl/v1/attendance/attendanceInput [post]
func (slf AttendanceController) AttendanceInput(c *gin.Context) {
	file, _, err := c.Request.FormFile("file")
	if err != nil {
		logx.Error(err.Error())
		util.ResponseFormat(c, code.RequestParamError, "文件读取失败")
		return
	}
	defer file.Close()
	//读取excel文件
	f, err := excelize.OpenReader(file)
	if err != nil {
		logx.Error(err.Error())
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	defer f.Close()
	rows, err := f.GetRows("Sheet1")
	if err != nil {
		logx.Error(err.Error())
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	if len(rows) <= 1 {
		logx.Error(err.Error())
		util.ResponseFormat(c, code.RequestParamError, "该文件没有数据内容")
		return
	}

	attendances := make([]*models.AttendanceManage, 0)
	workerIds := make([]string, 0)
	idMap := make(map[string]interface{})
	info := middleware.GetUserInfo(c)
	for k, row := range rows {
		if k == 0 {
			continue
		}
		var am models.AttendanceManage
		am.Date = row[0]
		am.WorkerId = row[1]
		am.WorkerName = row[2]
		am.StartWorkTime = row[3]
		am.EndWorkTime = row[4]
		if am.WorkerId != "" {
			idMap[am.WorkerId] = ""
		}
		am.AddPeople = info.NickName
		am.CreateTime = timex.TimeToString2(time.Now())
		attendances = append(attendances, &am)
	}

	for K, _ := range idMap {
		workerIds = append(workerIds, K)
	}
	//查询班次信息
	schedules, err := models.NewShiftScheduleSearch().SetResourceIDs(workerIds).SetResourceType(constvar.ResourceTypeWorker).FindNotTotal()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	scheduleIds := make([]string, 0)
	for _, schedule := range schedules {
		scheduleIds = append(scheduleIds, schedule.ID)
	}
	shifts, _ := models.NewShiftScheduleShiftSearch().SetShiftScheduleIDs(scheduleIds).SetPreload(true).FindNotTotal()
	for _, schedule := range schedules {
		for _, shift := range shifts {
			if schedule.ID == shift.ShiftScheduleID {
				schedule.Shift = append(schedule.Shift, &shift.Shift)
			}
		}
	}
	for _, attendance := range attendances {
		for _, schedule := range schedules {
			if attendance.WorkerId == schedule.ResourceID {
				if len(schedule.Shift) > 0 {
					attendance.Classes = schedule.Shift[0].Name
					if len(schedule.Shift[0].ShiftTime) > 0 {
						t := schedule.Shift[0].ShiftTime[0]
						attendance.ClassesStartTime = t.StartTime
						attendance.ClassesEndTime = t.EndTime
					}
				}
				break
			}
		}
	}
	//插入考勤数据
	err = models.NewAttendanceManageSearch().CreateBatch(attendances)
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	util.ResponseFormat(c, code.Success, "导入成功")
}

// GetAttendanceList
//
//	@Tags		考勤管理
//	@Summary	获取考勤列表
//	@Produce	application/json
//	@Param		object	body		request.GetAttendanceList	true	"参数"
//	@Param     	Authorization	header string true "token"
//	@Success	200		{object}	util.ResponseList{data=[]models.AttendanceManage}		"成功"
//	@Router		/api-jl/v1/attendance/getAttendanceList [post]
func (slf AttendanceController) GetAttendanceList(c *gin.Context) {
	var params request.GetAttendanceList
	err := c.BindJSON(&params)
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	manages, total, err := models.NewAttendanceManageSearch().SetPage(params.Page, params.PageSize).Find()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	util.ResponseFormatList(c, code.Success, manages, total)
}

// DeleteAttendanceInfo
//
//	@Tags		考勤管理
//	@Summary	删除考勤信息
//	@Produce	application/json
//	@Param		object	body		request.DeleteAttendanceInfo	true	"参数"
//	@Param     	Authorization	header string true "token"
//	@Success	200		{object}	util.Response		"成功"
//	@Router		/api-jl/v1/attendance/deleteAttendanceInfo [delete]
func (slf AttendanceController) DeleteAttendanceInfo(c *gin.Context) {
	var params request.DeleteAttendanceInfo
	err := c.BindJSON(&params)
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	err = models.NewAttendanceManageSearch().SetIds(params.Ids).Delete()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err)
		return
	}
	util.ResponseFormat(c, code.Success, "删除成功")
}