package model

import (
	"aps_crm/constvar"
	"aps_crm/pkg/mysqlx"
	"fmt"
	"gorm.io/gorm"
)

type (
	SalesLeads struct {
		Id              int                  `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
		Name            string               `json:"name" gorm:"column:name;uniqueIndex:name_isDeleted_idx;;type:varchar(255);comment:客户名称"`
		IsDeleted       bool                 `json:"-" gorm:"column:is_deleted;uniqueIndex:name_isDeleted_idx;type:tinyint(1);comment:是否删除"`
		Number          string               `json:"number" gorm:"column:number;type:varchar(255);comment:销售线索编号"`
		ContactName     string               `json:"contact_name" gorm:"column:contact_name;type:varchar(255);comment:联系人姓名"`
		ContactPhone    string               `json:"contact_phone" gorm:"column:contact_phone;type:varchar(255);comment:联系人电话"`
		ContactPosition string               `json:"contact_position" gorm:"column:contact_position;type:varchar(255);comment:联系人职位"`
		SalesSourcesId  int                  `json:"sales_sources_id" gorm:"column:sales_sources_id;type:int(11);comment:商机来源ID"`
		SalesSources    SalesSources         `json:"sales_sources" gorm:"foreignKey:SalesSourcesId"`
		MemberId        int                  `json:"member_id" gorm:"column:member_id;type:int(11);comment:销售负责人ID"`
		Member          User                 `json:"member" gorm:"foreignKey:MemberId"`
		SalesStatus     constvar.SalesStatus `json:"sales_status" gorm:"column:sales_status;type:int(11);comment:销售状态"`
		Desc            string               `json:"desc" gorm:"column:desc;type:varchar(255);comment:备注"`
		Reason          string               `json:"reason" gorm:"column:reason;type:text;comment:失败原因"`
		FollowRecord    []FollowRecord       `gorm:"foreignKey:SalesLeadsId"`
		DetailAddress   string               `json:"detail_address" gorm:"column:detail_address;type:varchar(255);comment:详细地址"`
		CodeStandID     string               `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
		Address
		gorm.Model `json:"-"`
	}

	SalesLeadsSearch struct {
		SalesLeads

		Orm       *gorm.DB
		SearchMap map[string]interface{}
		OrderBy   string
		PageNum   int
		PageSize  int
	}
)

func (SalesLeads) TableName() string {
	return "sales_leads"
}

func NewSalesLeadsSearch(db *gorm.DB) *SalesLeadsSearch {
	if db == nil {
		db = mysqlx.GetDB()
	}

	return &SalesLeadsSearch{
		Orm: db,
	}
}

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

	if len(slf.SearchMap) > 0 {
		for key, value := range slf.SearchMap {
			switch v := value.(type) {
			case string:
				if key == "number" || key == "contact_name" || key == "contact_phone" || key == "name" {
					db = db.Where(key+" LIKE ?", "%"+v+"%")
				}

				if key == "city" {
					db = db.Joins("City").Where("City.name LIKE ?", "%"+v+"%")
				}

				if key == "province" {
					db = db.Joins("Province").Where("Province.name LIKE ?", "%"+v+"%")
				}

				if key == "sales_sources" {
					db = db.Joins("SalesSources").Where("SalesSources.name LIKE ?", "%"+v+"%")
				}

				if key == "member_name" {
					db = db.Joins("Member").Where("Member.username LIKE ?", "%"+v+"%")
				}

			case int:
			case []int:
				if key == "member_ids" {
					db = db.Where("sales_leads.member_id in ?", v)
				}
			}
		}
	}

	return db
}

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

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

func (slf *SalesLeadsSearch) Delete() error {
	var db = slf.build()
	db.Update("is_deleted", 1)
	return db.Delete(&SalesLeads{}).Error
}

func (slf *SalesLeadsSearch) Find() (result []SalesLeads, err error) {
	var db = slf.build()
	err = db.Find(&result).Error
	return
}

func (slf *SalesLeadsSearch) FindOne() (result SalesLeads, err error) {
	var db = slf.build()
	err = db.First(&result).Error
	return
}

func (slf *SalesLeadsSearch) FindAll() ([]*SalesLeads, int64, error) {
	var db = slf.build()
	var records = make([]*SalesLeads, 0)
	var total int64
	if err := db.Count(&total).Error; err != nil {
		return records, total, err
	}
	if slf.PageNum > 0 && slf.PageSize > 0 {
		db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize)
	}

	err := db.Preload("SalesSources").Preload("Member").Preload("Province").Preload("City").Preload("FollowRecord").Order("id desc").Find(&records).Error
	return records, total, err
}

func (slf *SalesLeadsSearch) Count() (int64, error) {
	var db = slf.build()
	var total int64
	err := db.Count(&total).Error
	return total, err
}

func (slf *SalesLeadsSearch) MaxAutoIncr() (int, error) {
	type Result struct {
		Max int
	}

	var (
		result Result
		db     = slf.build()
	)

	err := db.Select("MAX(id) as max").Scan(&result).Error
	if err != nil {
		return result.Max, fmt.Errorf("max err: %v", err)
	}
	return result.Max, nil
}

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

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

func (slf *SalesLeadsSearch) First() (result SalesLeads, err error) {
	var db = slf.build()
	err = db.First(&result).Error
	return
}

func (slf *SalesLeadsSearch) SetPage(page, size int) *SalesLeadsSearch {
	slf.PageNum, slf.PageSize = page, size
	return slf
}

func (slf *SalesLeadsSearch) SetOrder(order string) *SalesLeadsSearch {
	slf.OrderBy = order
	return slf
}

func (slf *SalesLeadsSearch) UpdateMap(data map[string]interface{}) error {
	var db = slf.build()
	return db.Updates(data).Error
}
func (slf *SalesLeadsSearch) SetIds(ids []int) *SalesLeadsSearch {
	slf.Orm = slf.Orm.Where("id in (?)", ids)
	return slf
}

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

func (slf *SalesLeadsSearch) SetSearchMap(data map[string]interface{}) *SalesLeadsSearch {
	slf.SearchMap = data
	return slf
}