package db

import (
	"fmt"

	"gorm.io/gorm"
)

type (
	ModelRule struct {
		Id      string `gorm:"primary_key;column:id" json:"id"`
		ModelId string `gorm:"column:model_id" json:"modelId"`
		Scope   string `gorm:"column:scope" json:"scope"`
		RuleArg
	}

	RuleArg struct {
		Alias    string `gorm:"column:alias" json:"alias"`         // 参数的别名
		Name     string `gorm:"column:name" json:"name"`           // 参数名称
		Type     string `gorm:"column:type" json:"type"`           // 参数类型: input, option, range, image
		Must     bool   `gorm:"column:must" json:"must"`           // 是否必填
		Unit     string `gorm:"column:unit" json:"unit"`           // 单位
		Range    string `gorm:"column:range" json:"range"`         // 值的范围，eg：0,100表示从0到100
		Value    string `gorm:"column:value" json:"value"`         // 参数值
		ValType  string `gorm:"column:val_type" json:"valType"`    // 参数值类型 int, string, bool
		Operator string `gorm:"column:operator" json:"operator"`   // 运算符
		Sort     int    `gorm:"column:sort;default:0" json:"sort"` // 参数顺序
	}

	ModelRuleSet struct {
		Alias    string      `json:"alias"`    // 参数的别名
		Type     string      `json:"type"`     // 参数类型: input, option, range, image
		Name     string      `json:"name"`     // 参数名称
		Value    interface{} `json:"value"`    // 参数值
		ValType  string      `json:"valType"`  // 值类型
		Operator string      `json:"operator"` // 运算符
	}

	ModelRuleSearch struct {
		ModelRule
		Orm *gorm.DB
	}
)

func (e *ModelRule) TableName() string {
	return "model_rule"
}

func NewModelRuleSearch() *ModelRuleSearch {
	return &ModelRuleSearch{
		Orm: GetDB(),
	}
}

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

func (slf *ModelRuleSearch) SetModelID(id string) *ModelRuleSearch {
	slf.ModelId = id
	return slf
}

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

	if slf.ModelId != "" {
		db = db.Where("model_id = ?", slf.ModelId)
	}

	return db
}

func (slf *ModelRuleSearch) Find() ([]*ModelRule, error) {
	var (
		records = make([]*ModelRule, 0)
		db      = slf.build()
	)

	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find records err: %v", err)
	}

	return records, nil
}