package domain

import (
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/infrastructure/utils"
	"time"
)

// 工厂日历
type ProductCalendar struct {
	// 工厂日历ID
	ProductCalendarId int `json:"productCalendarId,omitempty"`
	// 企业id
	CompanyId int `json:"companyId,omitempty"`
	// 组织ID
	OrgId int `json:"orgId,omitempty"`
	// 工作位置
	WorkStation *WorkStation `json:"workStation,omitempty"`
	// 上班班次 1:全天  2：白班 4：中班  8：夜班
	WorkOn int `json:"workOn,omitempty"`
	// 日历选择
	CalendarSelected []string `json:"calendarSelected,omitempty"`
	// 上岗时间
	InWorkAt string `json:"inWorkAt,omitempty"`
	// 下岗时间
	OutWorkAt string `json:"outWorkAt,omitempty"`
	// 休息时间 （单位 h）
	BreakTime float64 `json:"breakTime,omitempty"`
	// 工时 （单位 h）
	WorkTime float64 `json:"workTime,omitempty"`
	// 创建时间
	CreatedAt time.Time `json:"createdAt,omitempty"`
	// 更新时间
	UpdatedAt time.Time `json:"updatedAt,omitempty"`
	// 删除时间
	DeletedAt time.Time `json:"deletedAt,omitempty"`
	// 扩展数据
	Ext *Ext `json:"ext,omitempty"`
}

type ProductCalendarRepository interface {
	Save(productCalendar *ProductCalendar) (*ProductCalendar, error)
	Remove(productCalendar *ProductCalendar) (*ProductCalendar, error)
	FindOne(queryOptions map[string]interface{}) (*ProductCalendar, error)
	Find(queryOptions map[string]interface{}) (int64, []*ProductCalendar, error)
}

func (productCalendar *ProductCalendar) Identify() interface{} {
	if productCalendar.ProductCalendarId == 0 {
		return nil
	}
	return productCalendar.ProductCalendarId
}

func (productCalendar *ProductCalendar) Update(data map[string]interface{}) error {
	productCalendar.UpdatedAt = time.Now()
	return nil
}

func (productCalendar *ProductCalendar) ResetWorkTime(v float64) error {
	if v != 0 {
		productCalendar.WorkTime = v
		return nil
	}
	td, err := utils.ComputeTimeDuration(productCalendar.InWorkAt, productCalendar.OutWorkAt)
	if err != nil {
		return err
	}
	if td.Hours() <= productCalendar.BreakTime {
		productCalendar.WorkTime = 0
		return nil
	}
	// 工作时长 = （上班时间-下班时间）- 休息时间
	productCalendar.WorkTime = td.Hours() - productCalendar.BreakTime
	return nil
}