package pick

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

var MatPickBuild models.MatPickBuild

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

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

func AddBuild(form form.MaterialPickBuildAdd) (int64, error) {
	subMap := map[string]interface{}{
		"build_name": form.BuildName,
		"pkg_id":     form.PkgID,
		"build_type": form.BuildType,
		"unit":       form.Unit,
		"price":      form.Price,
	}
	subID, err := db.InsertOne(MatPickBuild.TableName(), subMap)
	if err != nil {
		return 0, nil
	}
	return subID, nil
}

func EditBuild(form form.MaterialPickBuildAdd, id int) (int64, error) {
	subMap := map[string]interface{}{
		"build_name": form.BuildName,
		"pkg_id":     form.PkgID,
		"build_type": form.BuildType,
		"unit":       form.Unit,
		"price":      form.Price,
	}
	return db.Update(MatPickBuild.TableName(), map[string]interface{}{"id": id}, subMap)
}

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