package models

import (
	"fmt"
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
	"math/rand"
	"silkserver/constvar"
	"silkserver/pkg/mysqlx"
	"silkserver/utils"
)

// 工时统计
type (
	// PayrollWorkingHours 工资计算-员工的工时统计
	PayrollWorkingHours struct {
		BaseModelInt
		Cycle             string                     `json:"cycle" gorm:"index;size:20;not null;comment:统计周期（年月日）"`   //统计周期（年月日）
		WorkTypeID        uint                       `json:"workTypeID" gorm:"type:bigint(20);not null;comment:工种ID"` //工种ID
		WorkTypeCode      constvar.JobType           `json:"workTypeCode" gorm:"size:255;not null;comment:工种编码"`      //工种编码
		WorkType          WorkTypeManage             `json:"workType" gorm:"foreignKey:WorkTypeID;references:ID"`
		WorkerID          string                     `json:"workerId" gorm:"size:200;not null;comment:员工ID"` //员工ID
		Worker            Worker                     `json:"worker" gorm:"foreignKey:WorkerID;references:ID"`
		WorkshopId        uint                       `json:"workshopId" gorm:"type:int(11);comment:车间Id"`
		WorkshopNumber    string                     `json:"workshopNumber" gorm:"size:255;not null;comment:车间编号"`     // 车间编号
		GroupNumber       int                        `json:"groupNumber" gorm:"size:11;not null;comment:组别"`           // 组别
		StartCarNumber    int                        `json:"startCarNumbers"  gorm:"size:11;comment:车台号开始"`            // 车台号列表
		EndCarNumber      int                        `json:"endCarNumbers"  gorm:"size:11;comment:车台号结束"`              // 车台号列表
		ShiftTime         string                     `json:"shiftStartTime" gorm:"size:100;comment:班次时间段" `            // 班次时间段
		ShiftCrossDay     bool                       `json:"shiftCrossDay" gorm:"size:1;comment:是否为跨天班次" `             // 是否为跨天班次
		ShiftClockInTime  string                     `json:"shiftClockInTime" gorm:"size:50;comment:上班打卡时间" `          // 上班打卡时间
		ShiftClockOutTime string                     `json:"shiftClockOutTime" gorm:"size:50;comment:下班打卡时间" `         // 下班打卡时间
		OvertimeType      constvar.ShiftOvertimeType `json:"overtimeType" gorm:"size:50;comment:加班类型" `                // 加班类型
		OvertimeDuration  decimal.Decimal            `json:"overtimeDuration" gorm:"type:decimal(12,4);comment:加班时长" ` // 加班时长
		OvertimePay       decimal.Decimal            `json:"overtimePay" gorm:"type:decimal(12,4);comment:加班工资"`       // 加班工资
	}

	PayrollWorkingHoursSearch struct {
		PayrollWorkingHours
		Monthly  string
		Order    string
		PageNum  int
		PageSize int
		Preload  bool
		Orm      *gorm.DB
	}
)

func (slf PayrollWorkingHours) TableName() string {
	return "silk_payroll_working_hours"
}

// NewPayrollWorkingHoursSearch 员工的工时统计
func NewPayrollWorkingHoursSearch() *PayrollWorkingHoursSearch {
	return &PayrollWorkingHoursSearch{Orm: mysqlx.GetDB()}
}

func (slf *PayrollWorkingHoursSearch) SetOrm(tx *gorm.DB) *PayrollWorkingHoursSearch {
	slf.Orm = tx
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetPage(page, size int) *PayrollWorkingHoursSearch {
	slf.PageNum, slf.PageSize = page, size
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetOrder(order string) *PayrollWorkingHoursSearch {
	slf.Order = order
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetID(id uint) *PayrollWorkingHoursSearch {
	slf.ID = id
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetCycle(cycle string) *PayrollWorkingHoursSearch {
	slf.Cycle = cycle
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetMonthly(monthly string) *PayrollWorkingHoursSearch {
	slf.Monthly = monthly
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetWorkTypeID(workTypeID uint) *PayrollWorkingHoursSearch {
	slf.WorkTypeID = workTypeID
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetWorkTypeCode(workTypeCode constvar.JobType) *PayrollWorkingHoursSearch {
	slf.WorkTypeCode = workTypeCode
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetWorkerID(workerID string) *PayrollWorkingHoursSearch {
	slf.WorkerID = workerID
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetWorkshopNumber(workshopNumber string) *PayrollWorkingHoursSearch {
	slf.WorkshopNumber = workshopNumber
	return slf
}

func (slf *PayrollWorkingHoursSearch) SetGroupNumber(groupNumber int) *PayrollWorkingHoursSearch {
	slf.GroupNumber = groupNumber
	return slf
}

func (slf *PayrollWorkingHoursSearch) build() *gorm.DB {
	var db = slf.Orm.Table(slf.TableName())

	if slf.Preload {
		db = db.Model(&PayrollWorkingHours{}).Preload("Worker").Preload("WorkTypeManage")
	}

	if slf.ID > 0 {
		db = db.Where("id = ?", slf.ID)
	}

	if slf.Cycle != "" {
		db = db.Where("cycle = ?", slf.Cycle)
	}

	if slf.Monthly != "" {
		db = db.Where("cycle like ?", slf.Monthly+"%")
	}

	if slf.WorkTypeID > 0 {
		db = db.Where("work_type_id = ?", slf.WorkTypeID)
	}

	if slf.WorkTypeCode != "" {
		db = db.Where("work_type_code = ?", slf.WorkTypeCode)
	}

	if slf.WorkerID != "" {
		db = db.Where("worker_id = ?", slf.WorkerID)
	}

	if slf.WorkshopNumber != "" {
		db = db.Where("workshop_number = ?", slf.WorkshopNumber)
	}

	if slf.GroupNumber > 0 {
		db = db.Where("group_number = ?", slf.GroupNumber)
	}

	db.Where("1 = 1")
	if slf.Order != "" {
		db = db.Order(slf.Order)
	}

	return db
}

// Create 单条插入
func (slf *PayrollWorkingHoursSearch) Create(record *PayrollWorkingHours) error {
	var db = slf.build()

	if err := db.Create(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}

	return nil
}

// CreateBatch 批量插入
func (slf *PayrollWorkingHoursSearch) CreateBatch(records []*PayrollWorkingHours) error {
	var db = slf.build()

	if err := db.Create(&records).Error; err != nil {
		return fmt.Errorf("create batch err: %v, records: %+v", err, records)
	}

	return nil
}

// Save 单条更新
func (slf *PayrollWorkingHoursSearch) Save(record *PayrollWorkingHours) error {
	var db = slf.build()

	if err := db.Omit("CreatedAt").Save(record).Error; err != nil {
		return fmt.Errorf("save err: %v, record: %+v", err, record)
	}

	return nil
}

// SaveBatch 批量更新
func (slf *PayrollWorkingHoursSearch) SaveBatch(record []*PayrollWorkingHours) error {
	var db = slf.build()

	if err := db.Omit("CreatedAt").Save(record).Error; err != nil {
		return fmt.Errorf("save err: %v, record: %+v", err, record)
	}

	return nil
}

// UpdateByMap 单条更新
func (slf *PayrollWorkingHoursSearch) UpdateByMap(upMap map[string]interface{}) error {
	var (
		db = slf.build()
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
	}

	return nil
}

// UpdateByQuery 批量更新
func (slf *PayrollWorkingHoursSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
	var (
		db = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
	}

	return nil
}

// Delete 删除
func (slf *PayrollWorkingHoursSearch) Delete() error {
	var db = slf.build()

	if err := db.Unscoped().Delete(&PayrollWorkingHours{}).Error; err != nil {
		return err
	}

	return nil
}

// First 根据条件查询一条记录
func (slf *PayrollWorkingHoursSearch) First() (*PayrollWorkingHours, error) {
	var (
		record = new(PayrollWorkingHours)
		db     = slf.build()
	)

	if err := db.First(record).Error; err != nil {
		return record, err
	}

	return record, nil
}

// Find 指定条件查询（包含总条数）
func (slf *PayrollWorkingHoursSearch) Find() ([]*PayrollWorkingHours, int64, error) {
	var (
		records = make([]*PayrollWorkingHours, 0)
		total   int64
		db      = slf.build()
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find count err: %v", err)
	}
	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find records err: %v", err)
	}

	return records, total, nil
}

// FindNotTotal 指定条件查询
func (slf *PayrollWorkingHoursSearch) FindNotTotal() ([]*PayrollWorkingHours, error) {
	var (
		records = make([]*PayrollWorkingHours, 0)
		db      = slf.build()
	)

	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find records err: %v", err)
	}

	return records, nil
}

// FindByQuery 指定条件查询（包含总条数）
func (slf *PayrollWorkingHoursSearch) FindByQuery(query string, args []interface{}) ([]*PayrollWorkingHours, int64, error) {
	var (
		records = make([]*PayrollWorkingHours, 0)
		total   int64
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find by query count err: %v", err)
	}
	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, total, nil
}

// FindByQueryNotTotal 指定条件查询&不查询总条数.
func (slf *PayrollWorkingHoursSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*PayrollWorkingHours, error) {
	var (
		records = make([]*PayrollWorkingHours, 0)
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, nil
}

type GroupWorker struct {
	Cycle          string           `json:"cycle"`          // 统计周期（年月日）
	WorkTypeCode   constvar.JobType `json:"workTypeCode"`   // 工种编码
	WorkshopNumber string           `json:"workshopNumber"` // 车间编号
	GroupNumber    int              `json:"groupNumber"`    // 组别
	WorkerIds      string           `json:"workerIds"`      // 员工ID
	WorkerCount    int              `json:"workerCount"`    // 挡车工数量
}

// GroupWorker 每天小组人员
func (slf *PayrollWorkingHoursSearch) GroupWorker(monthly string, workTypeCode constvar.JobType) ([]*GroupWorker, error) {
	var (
		records = make([]*GroupWorker, 0)
		db      = slf.Orm.Table(slf.TableName())
	)
	db.Select("cycle, work_type_code, workshop_number, group_number, group_concat(worker_id) as worker_ids, count(id) as worker_count")
	db.Where("cycle like ?", monthly+"%")
	if workTypeCode != "" {
		db.Where("work_type_code = ?", workTypeCode)
	}
	db.Group("cycle, work_type_code, workshop_number, group_number")

	return records, db.Find(&records).Error
}

// InitDefaultData 初始化数据
func (slf *PayrollWorkingHoursSearch) InitDefaultData() error {
	var (
		db          = slf.Orm.Table(slf.TableName())
		total int64 = 0
	)

	firstDay, lastDay := utils.GetLastMonthPeriod(utils.GetMonthByOffset(-1))

	for i := 0; i < (lastDay.Day()); i++ {
		date := firstDay.AddDate(0, 0, i)
		if err := db.Where("cycle = ?", date.Format("2006-01-02")).Count(&total).Error; err != nil {
			return err
		}
		if total != 0 {
			return nil
		}

		data := make([]*PayrollWorkingHours, 0)
		workers, _ := NewWorkerSearch().FindNotTotal()
		workshop := rand.Intn(10)
		for _, record := range workers {
			round := rand.Intn(10)
			info := PayrollWorkingHours{
				Cycle:             date.Format("2006-01-02"),
				WorkTypeID:        uint(round + 1),
				WorkTypeCode:      constvar.JobTypeArr[round],
				WorkerID:          record.ID,
				WorkshopId:        uint(workshop + 1),
				WorkshopNumber:    fmt.Sprintf("100%v", workshop),
				GroupNumber:       round,
				StartCarNumber:    round*10 + 1,
				EndCarNumber:      (round + 1) * 10,
				ShiftTime:         "08:00-18:00",
				ShiftCrossDay:     false,
				ShiftClockInTime:  fmt.Sprintf("07:5%v", round),
				ShiftClockOutTime: fmt.Sprintf("20:0%v", round),
			}
			//if info.WorkTypeCode == constvar.JobTypeWeavers {
			//	info.StartCarNumber = i*10 + 1
			//	info.EndCarNumber = (i + 1) * 10
			//}

			if date.Weekday() == 0 {
				info.OvertimeType = constvar.ShiftOvertimeTypeOvertime
				info.OvertimeDuration = decimal.NewFromInt32(1)
				info.OvertimePay = decimal.NewFromInt32(1 * 90)
			} else {
				info.OvertimeType = constvar.ShiftOvertimeTypeTimeout
				info.OvertimeDuration = decimal.NewFromInt32(int32(workshop))
				info.OvertimePay = decimal.NewFromInt32(int32(workshop * 12))
			}
			data = append(data, &info)
		}

		if err := slf.CreateBatch(data); err != nil {
			return err
		}
	}

	return nil
}