package model

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

type (
	ServiceFollowup struct {
		Id             int          `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
		ClientId       int          `json:"clientId" gorm:"column:client_id;type:int;comment:客户id"`
		Client         Client       `json:"client" gorm:"foreignKey:ClientId"`
		Number         string       `json:"number" gorm:"column:number;type:varchar(255);comment:合同编号"`
		ContactId      int          `json:"contactId" gorm:"column:contact_id;type:int;comment:联系人id"`
		Contact        Contact      `json:"contact"  gorm:"foreignKey:ContactId"`
		ServiceOrderId int          `json:"serviceOrderId" gorm:"column:service_order_id;type:int;comment:客户服务单id"`
		ServiceOrder   ServiceOrder `gorm:"foreignKey:ServiceOrderId"`
		MemberId       int          `json:"memberId" gorm:"column:member_id;type:int;comment:服务人员id"`
		Member         User         `json:"Member" gorm:"foreignKey:MemberId"`
		PlanId         int          `json:"planId" gorm:"column:plan_id;type:int;comment:服务计划id"`
		SatisfactionId int          `json:"satisfactionId" gorm:"column:satisfaction_id;type:int;comment:满意度id"`
		Satisfaction   Satisfaction `json:"satisfaction" gorm:"foreignKey:SatisfactionId"`
		TimelyRateId   int          `json:"timelyRateId" gorm:"column:timely_rate_id;type:int;comment:及时率id"`
		TimelyRate     TimelyRate   `json:"timelyRate" gorm:"foreignKey:TimelyRateId"`
		SolveRateId    int          `json:"solveRateId" gorm:"column:solve_rate_id;type:int;comment:解决率id"`
		SolveRate      SolveRate    `json:"solveRate" gorm:"foreignKey:SolveRateId"`
		IsVisitId      int          `json:"isVisitId" gorm:"column:is_visit_id;type:int;comment:服务人员是否来过id"`
		IsVisit        IsVisit      `json:"isVisit"  gorm:"foreignKey:IsVisitId"`
		OldMemberId    int          `json:"oldMemberId" gorm:"column:old_member_id;type:int;comment:原服务人员"`
		OldMember      User         `json:"oldMember" gorm:"foreignKey:OldMemberId"`
		Remark         string       `json:"remark" gorm:"column:remark;type:text;comment:备注"`
		File           string       `json:"file" gorm:"column:file;type:varchar(255);comment:附件"`
		CodeStandID    string       `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
		CrmModel
	}

	ServiceFollowupSearch struct {
		ServiceFollowup
		Orm         *gorm.DB
		KeywordType constvar.ServiceFollowupKeywordType
		Keyword     string
		OrderBy     string
		PageNum     int
		PageSize    int
		Preload     bool
		MemberIds   []int
	}
)

func (*ServiceFollowup) TableName() string {
	return "service_followup"
}

func NewServiceFollowupSearch() *ServiceFollowupSearch {
	return &ServiceFollowupSearch{
		Orm: mysqlx.GetDB(),
	}
}

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

	switch slf.KeywordType {
	case constvar.ServiceFollowupKeywordFollowupNo:
		db = db.Where("number like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
	case constvar.ServiceFollowupKeywordCustomerName:
		db = db.Joins("Client", clause.LeftJoin).Where("Client.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
	case constvar.ServiceFollowupKeywordContactName:
		db = db.Joins("Contact").Where("Contact.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
	case constvar.ServiceFollowupKeywordCustomerServiceNo:
		db = db.Joins("ServiceOrder").Where("ServiceOrder.service_number like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
	case constvar.ServiceFollowupKeywordVisitor:
		db = db.Joins("left join user on user.id = service_followup.member_id").Where("user.username like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
	case constvar.ServiceFollowupKeywordSatisfactionDegree:
		db = db.Joins("Satisfaction").Where("Satisfaction.name like ?", fmt.Sprintf("%%%s%%", slf.Keyword))

	}

	if slf.ServiceOrderId > 0 {
		db = db.Where("service_order_id = ?", slf.ServiceOrderId)
	}

	if slf.Preload {
		db = db.Preload("Client").
			Preload("Contact").
			Preload("ServiceOrder").
			Preload("Member").
			Preload("Satisfaction").
			Preload("TimelyRate").
			Preload("SolveRate").
			Preload("IsVisit").
			Preload("OldMember")
	}

	if len(slf.MemberIds) > 0 {
		db = db.Where("service_followup.member_id in ?", slf.MemberIds)
	}
	if slf.Number != "" {
		db = db.Where("number = ?", slf.Number)
	}

	return db
}

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

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

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

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

func (slf *ServiceFollowupSearch) FindAll() ([]*ServiceFollowup, int64, error) {
	var db = slf.build()
	var records = make([]*ServiceFollowup, 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.Order("id desc").Find(&records).Error
	return records, total, err
}

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

func (slf *ServiceFollowupSearch) 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 *ServiceFollowupSearch) SetNumber(number string) *ServiceFollowupSearch {
	slf.Number = number
	return slf
}

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

func (slf *ServiceFollowupSearch) SetServiceOrderId(id int) *ServiceFollowupSearch {
	slf.ServiceOrderId = id
	return slf
}

func (slf *ServiceFollowupSearch) SetPreload(preload bool) *ServiceFollowupSearch {
	slf.Preload = preload
	return slf
}

func (slf *ServiceFollowupSearch) SetKeywordType(keyword constvar.ServiceFollowupKeywordType) *ServiceFollowupSearch {
	slf.KeywordType = keyword
	return slf
}

func (slf *ServiceFollowupSearch) SetKeyword(keyword string) *ServiceFollowupSearch {
	slf.Keyword = keyword
	return slf
}

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

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

func (slf *ServiceFollowupSearch) SetMemberIds(memberIds []int) *ServiceFollowupSearch {
	slf.MemberIds = memberIds
	return slf
}

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