package model

import (
	"aps_crm/pkg/mysqlx"
	"gorm.io/gorm"
	"time"
)

type (
	ServiceFeeManage struct {
		Id         int       `json:"id" gorm:"column:id;primaryKey;autoIncrement;not null"`
		ClientId   int       `json:"client_id" gorm:"column:client_id;type:int(11);comment:客户ID"`
		Client     *Client   `json:"client" gorm:"foreignKey:ClientId"`
		MemberId   int       `json:"member_id" gorm:"column:member_id;type:int(11);comment:员工ID"`
		LatestDate time.Time `json:"latest_date" gorm:"column:latest_date;type:datetime;comment:最晚服务时间"`
		Remark     string    `json:"remark" gorm:"column:remark;type:varchar(255);comment:备注"`
		File       string    `json:"file" gorm:"column:file;type:varchar(255);comment:文件"`
		gorm.Model `json:"-"`
	}

	ServiceFeeManageSearch struct {
		ServiceFeeManage
		Orm *gorm.DB
	}
)

func (ServiceFeeManage) TableName() string {
	return "service_fee_manage"
}

func NewServiceFeeManageSearch(db *gorm.DB) *ServiceFeeManageSearch {
	if db == nil {
		db = mysqlx.GetDB()
	}

	return &ServiceFeeManageSearch{
		Orm: db,
	}
}

func (slf *ServiceFeeManageSearch) build() *gorm.DB {
	var db = slf.Orm.Model(&ServiceFeeManage{})
	if slf.Id != 0 {
		db.Where("id = ?", slf.Id)
	}
	if slf.ClientId != 0 {
		db.Where("client_id = ?", slf.ClientId)
	}

	return db
}

func (slf *ServiceFeeManageSearch) Create(record *ServiceFeeManage) error {
	var db = slf.build()
	return db.Create(record).Error
}

func (slf *ServiceFeeManageSearch) Update(record *ServiceFeeManage) error {
	var db = slf.build()
	return db.Updates(record).Error
}

func (slf *ServiceFeeManageSearch) Delete() error {
	var db = slf.build()
	return db.Delete(&ServiceFeeManage{}).Error
}

func (slf *ServiceFeeManageSearch) SetId(id int) *ServiceFeeManageSearch {
	slf.Id = id
	return slf
}

func (slf *ServiceFeeManageSearch) Find() (*ServiceFeeManage, error) {
	var db = slf.build()
	var record = new(ServiceFeeManage)
	err := db.First(record).Error
	return record, err
}

func (slf *ServiceFeeManageSearch) FindAll() ([]*ServiceFeeManage, error) {
	var db = slf.build()
	var records = make([]*ServiceFeeManage, 0)
	err := db.Preload("Client").Find(&records).Error
	return records, err
}