package model

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

type (
	SaleChance 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:销售线索编号"`
		ContactId             int                    `json:"contact_id" gorm:"column:contact_id;type:int(11);comment:联系人ID"`
		Contact               Contact                `json:"contact" gorm:"foreignKey:ContactId;references:Id"`
		ClientId              int                    `json:"client_id" gorm:"column:client_id;type:int(11);comment:客户ID"`
		SalesSourcesId        int                    `json:"sales_sources_id" gorm:"column:sales_sources_id;type:int(11);comment:商机来源ID"`
		SaleTypeId            int                    `json:"sale_type_id" gorm:"column:sale_type_id;type:int(11);comment:商机类型ID"`
		SaleType              SaleType               `json:"sale_type" `
		SaleStageId           int                    `json:"sale_stage_id" gorm:"column:sale_stage_id;type:int(11);comment:商机阶段ID"`
		SaleStage             SaleStage              `json:"sale_stage" gorm:"foreignKey:SaleStageId;references:Id"`
		MemberId              int                    `json:"member_id" gorm:"column:member_id;type:int(11);comment:销售负责人ID"`
		Member                User                   `json:"member" gorm:"foreignKey:MemberId"`
		RegularCustomersId    int                    `json:"regular_customers_id" gorm:"column:regular_customers_id;type:int(11);comment:常客ID"`
		RegularCustomers      RegularCustomers       `json:"regular_customers"`
		Competitors           string                 `json:"competitors" gorm:"column:competitors;type:varchar(255);comment:竞争对手"`
		PossibilitiesId       int                    `json:"possibilities_id" gorm:"column:possibilities_id;type:int(11);comment:可能性ID"`
		Possibility           Possibility            `json:"possibility" gorm:"foreignKey:PossibilitiesId;references:Id"`
		Budget                float64                `json:"budget" gorm:"column:budget;type:decimal(10,2);comment:预算"`
		ProjectedAmount       float64                `json:"projected_amount" gorm:"column:projected_amount;type:decimal(10,2);comment:预计金额"`
		Currency              int                    `json:"currency" gorm:"column:currency;type:int(11);comment:币种"`
		ExpectedTime          *CustomTime            `json:"expected_time" gorm:"column:expected_time;type:datetime;comment:预计成交时间"`
		StatusId              int                    `json:"status_id" gorm:"column:status_id;type:int(11);comment:状态ID"`
		Status                Status                 `json:"status"` //状态
		PainPoints            string                 `json:"pain_points" gorm:"column:pain_points;type:text;comment:痛点"`
		WhetherEstablished    string                 `json:"whether_established" gorm:"column:whether_established;type:text;comment:是否成立"`
		CapitalBudget         string                 `json:"capital_budget" gorm:"column:capital_budget;type:text;comment:资金预算"`
		KeyMaker              string                 `json:"key_maker" gorm:"column:key_maker;type:text;comment:关键人"`
		KeyFactors            string                 `json:"key_factors" gorm:"column:key_factors;type:text;comment:关键因素"`
		Process               string                 `json:"process" gorm:"column:process;type:text;comment:决策流程"`
		Solutions             string                 `json:"solutions" gorm:"column:solutions;type:text;comment:竞争对手解决方案"`
		Advantages            string                 `json:"advantages" gorm:"column:advantages;type:text;comment:竞争优势"`
		Disadvantages         string                 `json:"disadvantages" gorm:"column:disadvantages;type:text;comment:竞争劣势"`
		Opportunities         string                 `json:"opportunities" gorm:"column:opportunities;type:text;comment:竞争机会"`
		Threats               string                 `json:"threats" gorm:"column:threats;type:text;comment:竞争威胁"`
		DetailAddress         string                 `json:"detail_address" gorm:"column:detail_address;type:text;comment:详细地址"`
		Remark                string                 `json:"remark" gorm:"column:remark;type:text;comment:备注"`
		Client                Client                 `json:"client" gorm:"foreignKey:ClientId;references:Id"`
		CollectionProjections []CollectionProjection `json:"collection_projections" gorm:"foreignKey:SaleChanceId"`
		SalesSources          SalesSources           `json:"sales_sources"`
		CodeStandID           string                 `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
		Products              []*Product             `json:"products" gorm:"many2many:SaleChanceProduct;"`
		Address
		gorm.Model `json:"-"`
	}

	SaleChanceSearch struct {
		SaleChance

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

func (SaleChance) TableName() string {
	return "sale_chance"
}

func NewSaleChanceSearch() *SaleChanceSearch {
	return &SaleChanceSearch{Orm: mysqlx.GetDB()}
}

func (slf *SaleChanceSearch) build() *gorm.DB {
	var db = slf.Orm.Model(&SaleChance{})
	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 == "name" || key == "number" {
					db = db.Where(key+" LIKE ?", "%"+v+"%")
				}

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

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

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

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

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

				if key == "expected_time" {
					db = db.Where("expected_time LIKE ?", "%"+v+"%")
				}

			case int, float64:
				if key == "member_id" || key == "budget" || key == "projected_amount" || key == "client_id" {
					db = db.Where(key+" = ?", v)
				}
			case []int:
				if key == "member_ids" {
					db = db.Where("sale_chance.member_id in ?", v)
				}
			}
		}
	}

	return db
}

func (slf *SaleChanceSearch) Create(record *SaleChance) (err error) {
	var db = slf.build()
	err = db.Create(record).Error
	return
}

func (slf *SaleChanceSearch) Update(record *SaleChance) (err error) {
	var db = slf.build()
	err = db.Preload("Products").Updates(record).Error
	return
}

func (slf *SaleChanceSearch) Delete() (err error) {
	var db = slf.build()
	err = db.Delete(&SaleChance{}).Error
	return
}

func (slf *SaleChanceSearch) Find() (record SaleChance, err error) {
	var db = slf.build()
	err = db.Preload("Products").First(&record).Error
	return
}

func (slf *SaleChanceSearch) FindAll() ([]*SaleChance, int64, error) {
	var db = slf.build()
	var records = make([]*SaleChance, 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("SaleType").Preload("RegularCustomers").Preload("SalesSources").Preload("Products").
		Preload("Member").Preload("SaleStage").Preload("Possibility").
		Preload("CollectionProjections").Preload("Client").
		Preload("Province").Preload("City").Preload("Contact").Preload("Status").Order("id desc").Find(&records).Error
	return records, total, err
}

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

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

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

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

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

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

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

func (slf *SaleChanceSearch) 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
}