package model

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

type (
	Contact struct {
		Id       int         `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
		Name     string      `json:"name" gorm:"column:name;type:varchar(255);comment:联系人姓名"`
		Number   string      `json:"number" gorm:"column:number;type:varchar(255);comment:联系人编号"`
		ClientId int         `json:"client_id" gorm:"column:client_id;type:int(11);comment:客户ID"`
		Client   Client      `json:"-" gorm:"foreignKey:ClientId"`
		Position string      `json:"position" gorm:"column:position;type:varchar(255);comment:职位"`
		Phone    string      `json:"phone" gorm:"column:phone;type:varchar(255);comment:电话"`
		MemberId int         `json:"member_id" gorm:"column:member_id;type:int(11);comment:负责人ID"`
		Member   User        `json:"member" gorm:"foreignKey:MemberId"`
		IsFirst  bool        `json:"is_first" gorm:"column:is_first;type:tinyint(1);comment:是否首要联系人"`
		Wechat   string      `json:"wechat" gorm:"column:wechat;type:varchar(255);comment:微信"`
		Birthday *CustomTime `json:"birthday" gorm:"column:birthday;type:datetime;comment:生日"`
		Email    string      `json:"email" gorm:"column:email;type:varchar(255);comment:邮箱"`
		Desc     string      `json:"desc" gorm:"column:desc;type:varchar(255);comment:备注"`
		Address
		gorm.Model `json:"-"`
	}

	ContactSearch struct {
		Contact

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

	ContactDetail struct {
		Contact
		Client       Client
		FollowRecord []FollowRecord `gorm:"foreignKey:ContactId"`
	}
)

func (Contact) TableName() string {
	return "contacts"
}

func NewContactSearch(db *gorm.DB) *ContactSearch {
	if db == nil {
		db = mysqlx.GetDB()
	}
	return &ContactSearch{
		Orm: db,
	}
}

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

	if slf.ClientId != 0 {
		db = db.Where("client_id = ?", slf.ClientId)
	}

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

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

				if key == "client_name" {
					//db = db.Joins("inner join clients on clients.id = contacts.client_id").Where("clients.name LIKE ?", "%"+v+"%")
					db = db.Joins("Client").Where("Client.name LIKE ?", "%"+v+"%")
				}

				if key == "is_first" {
					if v == "是" {
						db = db.Where("is_first = ?", true)
					} else if v == "否" {
						db = db.Where("is_first = ?", false)
					}
				}
			case int:
			case int64:
			case float64:
				if key == "client_id" {
					db = db.Where("client_id = ? and is_first = true", v)
				}
			}
		}
	}

	return db
}

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

func (slf *ContactSearch) Update(record *Contact) error {
	var db = slf.build()
	m := map[string]interface{}{
		"name":        record.Name,
		"number":      record.Number,
		"client_id":   record.ClientId,
		"position":    record.Position,
		"phone":       record.Phone,
		"member_id":   record.MemberId,
		"is_first":    record.IsFirst,
		"wechat":      record.Wechat,
		"birthday":    record.Birthday,
		"email":       record.Email,
		"desc":        record.Desc,
		"country_id":  record.CountryId,
		"province_id": record.ProvinceId,
		"city_id":     record.CityId,
		"region_id":   record.RegionId,
	}
	return db.Updates(m).Error
}

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

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

func (slf *ContactSearch) FindAll() ([]*ContactDetail, int64, error) {
	var db = slf.build()
	var records = make([]*ContactDetail, 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("Member").Preload("FollowRecord").Preload("Client").Preload("Country").Preload("Province").Preload("City").Preload("Region").Order("id desc").Find(&records).Error
	return records, total, err
}

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

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

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

func (slf *ContactSearch) SetClientId(clientId int) *ContactSearch {
	slf.ClientId = clientId
	return slf
}

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

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

func (slf *ContactSearch) SetOrder(order string) *ContactSearch {
	slf.OrderBy = order
	return slf
}
func (slf *ContactSearch) SetIds(ids []int) *ContactSearch {
	slf.Orm = slf.Orm.Where("id in (?)", ids)
	return slf
}

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