package model

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

type (
	Client struct {
		Id                int            `json:"id" gorm:"column:id;primaryKey;autoIncrement;not null"`
		Name              string         `json:"name" gorm:"column:name;unique;type:varchar(255);comment:客户名称"`
		Number            string         `json:"number" gorm:"column:number;type:varchar(255);comment:客户编号"`
		ClientStatusId    int            `json:"client_status_id" gorm:"column:client_status_id;type:int(11);comment:客户状态ID"`
		ClientTypeId      int            `json:"client_type_id" gorm:"column:client_type_id;type:int(11);comment:客户类型ID"`
		ClientOriginId    int            `json:"client_origin_id" gorm:"column:client_origin_id;type:int(11);comment:客户来源ID"`
		ClientLevelId     int            `json:"client_level_id" gorm:"column:client_level_id;type:int(11);comment:客户等级ID"`
		Contacts          []Contact      `json:"contacts" gorm:"foreignKey:ClientId"`
		MemberId          int            `json:"member_id" gorm:"column:member_id;type:int(11);comment:销售负责人ID"`
		ServiceMemberId   int            `json:"service_member_id" gorm:"column:service_member_id;type:int(11);comment:服务负责人ID"`
		DetailAddress     string         `json:"detail_address" gorm:"column:detail_address;type:varchar(255);comment:详细地址"`
		Remark            string         `json:"remark" gorm:"column:remark;type:varchar(255);comment:备注"`
		NextVisitTime     time.Time      `json:"next_visit_time" gorm:"column:next_visit_time;type:datetime;comment:下次回访时间"`
		LatestServiceTime time.Time      `json:"latest_service_time" gorm:"column:latest_service_time;type:datetime;comment:最晚服务时间"`
		FollowRecord      []FollowRecord `json:"follow_record" gorm:"foreignKey:ClientId"`
		Address
		Business
		gorm.Model `json:"-"`
	}

	ClientSearch struct {
		Client
		Orm *gorm.DB
	}
)

func (Client) TableName() string {
	return "clients"
}

func NewClientSearch() *ClientSearch {
	return &ClientSearch{
		Orm: mysqlx.GetDB(),
	}
}

func (slf *ClientSearch) build() *gorm.DB {
	var db = slf.Orm.Model(&Client{})
	if slf.Id != 0 {
		db.Where("id = ?", slf.Id)
	}
	if slf.Name != "" {
		db.Where("name = ?", slf.Name)
	}

	return db
}

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

func (slf *ClientSearch) Update(record *Client) error {
	var db = slf.build()
	return db.Save(record).Error
}

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

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

func (slf *ClientSearch) SetName(name string) *ClientSearch {
	slf.Name = name
	return slf
}

func (slf *ClientSearch) SetNumber(number string) *ClientSearch {
	slf.Number = number
	return slf
}

func (slf *ClientSearch) First() (*Client, error) {
	var db = slf.build()
	var record = &Client{}
	err := db.First(record).Error
	return record, err
}

func (slf *ClientSearch) Find() ([]*Client, error) {
	var db = slf.build()
	var records = make([]*Client, 0)
	err := db.Preload("FollowRecord").Preload("EnterpriseNature").Preload("RegisteredCapital").Preload("Industry").Preload("EnterpriseScale").Preload("Contacts").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error
	return records, err
}

func (slf *ClientSearch) UpdateByMap(data map[string]interface{}) error {
	var db = slf.build()
	return db.Updates(data).Error
}