package pick

import (
	"zhiyuan/models"
	"zhiyuan/pkg/app"
	"zhiyuan/pkg/db"
	"zhiyuan/services/form"
)

var MatPick models.MatPick

func CountPick(where map[string]interface{}) (int64, error) {
	return db.Count(MatPick.TableName(), where)
}

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

func GetPick(where map[string]interface{}, fields []string, dest interface{}) (*models.MatPick, error) {
	return MatPick.GetOne(where, fields, dest)
}

func getPickMap(form form.MaterialPickAdd) map[string]interface{} {
	pickMap := map[string]interface{}{
		"pick_name": form.PickName,
		//"pkg_id":       form.PkgID,
		"room_type":    form.RoomType,
		"standard":     form.Standard,
		"confirm_type": form.ConfirmType,
		"cycle":        form.Cycle,
		"level":        form.Level,
	}
	return pickMap
}

func AddPick(form form.MaterialPickAdd) (int64, error) {
	return db.InsertOne(MatPick.TableName(), getPickMap(form))
}

func EditPick(form form.MaterialPickAdd, id int) (int64, error) {
	return db.Update(MatPick.TableName(), map[string]interface{}{"id": id}, getPickMap(form))
}

func DelPick(id int) (int64, error) {
	return db.Delete(MatPick.TableName(), map[string]interface{}{"id": id})
}