package pick

import (
	"github.com/gogf/gf/v2/os/gtime"
	"zhiyuan/models/final"
	"zhiyuan/pkg/app"
	"zhiyuan/pkg/db"
	"zhiyuan/services/form"
)

var MatPickWork final.MatPickWork

func GetMatPickWork(where map[string]interface{}, fields []string, page app.Page, retVal interface{}) ([]*final.MatPickWork, error) {
	if page.PageNum > 0 && page.PageSize > 0 {
		where["_limit"] = db.GetOffset(uint(page.PageNum), uint(page.PageSize))
	}
	return MatPickWork.GetMulti(where, fields, retVal)
}

func getPickWorkMap(form form.WorkItemAdd) map[string]interface{} {
	pickMap := map[string]interface{}{
		"worker_id": form.WorkerId,
		"cycle":     form.Cycle,
		"pictures":  form.Pictures,
	}
	return pickMap
}

func AddPickWork(form form.WorkItemAdd) (int64, error) {

	return db.InsertOne(final.MatPickWork{}.TableName(), map[string]interface{}{
		"site_id":    form.SitId,
		"room_type":  form.RoomType,
		"photograph": form.Photograph,
		"cycle":      form.Cycle,
		"worker_id":  form.WorkerId,
	})
}

func EditPickWork(form form.WorkItemAdd, id int) (int64, error) {
	return db.Update(final.MatPickWork{}.TableName(), map[string]interface{}{"id": id}, getPickWorkMap(form))
}

func DelPickWork(Id, SiteId int) (int64, error) {
	return db.Update(final.MatPickWork{}.TableName(), map[string]interface{}{"id": Id, "site_id": SiteId}, map[string]interface{}{
		"deleted_at": gtime.Now().Unix(),
	})
}