package db

import (
	"fmt"
	"gorm.io/gorm"
	"time"
)

type ModelTaskResults struct {
	BaseModel
	Title       string `json:"name" gorm:"type:varchar(255)"`                                            //预警名称,暂时用任务名称
	Event       string `json:"event" gorm:"type:varchar(255)"`                                           //预警事件
	ModelID     string `json:"modelID" gorm:"type:varchar(255)"`                                         //模型ID
	ModelTaskID string `json:"modelTaskID" gorm:"type:varchar(255)"`                                     //模型任务ID
	OrgID       string `json:"orgID" gorm:"index;column:org_id;type:varchar(299);not null;"`             //派出所 domain unit ID
	CommunityId string `json:"communityID" gorm:"index;column:community_id;type:varchar(299);not null;"` //小区ID
	ObjectIds   string `json:"objectIds" gorm:"type:text"`                                               //事件对象，可以是人，多个用逗号分隔
	Location    string `json:"location" gorm:"type:varchar(255)"`                                        //发生地点
	Building    string `json:"building" gorm:"type:varchar(255);"`                                       //楼栋
	Floor       string `json:"floor"       gorm:"type:varchar(255);"`                                    //楼层
	PicDate     string `json:"picDate"       gorm:"type:varchar(255);"`                                  //抓拍时间
}

func (m *ModelTaskResults) TableName() string {
	return "model_task_results"
}

type ModelTaskResultsSearch struct {
	ModelTaskResults
	Orm      *gorm.DB
	PageNum  int
	PageSize int
	Order    string
	Keyword  string
}

func NewModelTaskResultsSearch() *ModelTaskResultsSearch {
	return &ModelTaskResultsSearch{
		Orm:      GetDB(),
		PageNum:  1,
		PageSize: 10,
	}
}

func (slf *ModelTaskResultsSearch) SetOrm(tx *gorm.DB) *ModelTaskResultsSearch {
	slf.Orm = tx
	return slf
}

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

func (slf *ModelTaskResultsSearch) SetOrder(order string) *ModelTaskResultsSearch {
	slf.Order = order
	return slf
}

func (slf *ModelTaskResultsSearch) SetID(id string) *ModelTaskResultsSearch {
	slf.ID = id
	return slf
}

func (slf *ModelTaskResultsSearch) SetKeyword(kw string) *ModelTaskResultsSearch {
	slf.Keyword = kw
	return slf
}

func (slf *ModelTaskResultsSearch) build() *gorm.DB {
	var db = slf.Orm.Table(slf.TableName())
	if slf.Order != "" {
		db = db.Order(slf.Order)
	}

	if slf.ID != "" {
		db = db.Where("id = ?", slf.ID)
	}

	if slf.Keyword != "" {
		kw := "%" + slf.Keyword + "%"
		db = db.Where("name like ?", kw)
	}

	return db
}

func (slf *ModelTaskResultsSearch) First() (*ModelTaskResults, error) {
	var (
		record = new(ModelTaskResults)
		db     = slf.build()
	)

	if err := db.First(record).Error; err != nil {
		return record, err
	}

	return record, nil
}

func (slf *ModelTaskResultsSearch) Find() ([]*ModelTaskResults, int64, error) {
	var (
		records = make([]*ModelTaskResults, 0)
		total   int64
		db      = slf.build()
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find count err: %v", err)
	}
	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find records err: %v", err)
	}

	return records, total, nil
}

func (slf *ModelTaskResultsSearch) FindAll() ([]*ModelTaskResults, error) {
	var (
		records = make([]*ModelTaskResults, 0)
		db      = slf.build()
	)
	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find records err: %v", err)
	}

	return records, nil
}

func (slf *ModelTaskResultsSearch) Count() int64 {
	var (
		count int64
		db    = slf.build()
	)

	if err := db.Count(&count).Error; err != nil {
		return count
	}

	return count
}

func (slf *ModelTaskResultsSearch) Create(record *ModelTaskResults) error {
	var db = slf.build()

	if err := db.Create(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ModelTaskResultsSearch) BatchCreate(record []*ModelTaskResults) error {
	var db = slf.build()

	if err := db.Create(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ModelTaskResultsSearch) Save(record *ModelTaskResults) error {
	var db = slf.build()
	record.UpdatedAt = time.Now()
	if err := db.Omit("CreatedAt").Save(record).Error; err != nil {
		return fmt.Errorf("save err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ModelTaskResultsSearch) Update(record *ModelTaskResults) error {
	var db = slf.build()

	if err := db.Updates(record).Error; err != nil {
		return fmt.Errorf("update err: %v, record: %+v", err, record)
	}

	return nil
}

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