package model

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

type (
	// File 附件
	File struct {
		Name          string `gorm:"name" json:"name"`
		Size          int64  `gorm:"size" json:"size"`                    // 文件大小
		FilePath      string `gorm:"file_path" json:"-"`                  // 文件路径
		Key           string `gorm:"key" json:"key"`                      // 对象存储key
		Bucket        string `gorm:"bucket" json:"-"`                     // 对象存储bucket
		DownloadCount int    `gorm:"download_count" json:"downloadCount"` // 下次次数
		PreviewCount  int    `gorm:"preview_count" json:"previewCount"`   // 预览次数
		FileType      string `gorm:"file_type" json:"fileType"`           // 文件类型
		SourceType    string `gorm:"source_type" json:"sourceType"`       // 附件来源
		SourceId      int    `gorm:"source_id" json:"sourceId"`           // 来源id
		gorm.Model    `json:"-"`
	}

	// FileSearch 附件搜索条件
	FileSearch struct {
		File
		Orm         *gorm.DB
		QueryClass  constvar.FileQueryClass
		KeywordType constvar.FileKeywordType
		Keyword     string
		PageNum     int
		PageSize    int
	}
)

func (File) TableName() string {
	return "file"
}

func NewFileSearch() *FileSearch {
	return &FileSearch{
		Orm: mysqlx.GetDB(),
	}
}

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

	return db
}

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

func (slf *FileSearch) CreateBatch(records []*File) error {
	var db = slf.build()
	return db.Create(records).Error
}

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

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

func (slf *FileSearch) FindAll() ([]*File, error) {
	var db = slf.build()
	var record = make([]*File, 0)
	err := db.Find(&record).Error
	return record, err
}

func (slf *FileSearch) SetId(id uint) *FileSearch {
	slf.ID = id
	return slf
}

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

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

func (slf *FileSearch) Updates(values interface{}) error {
	var db = slf.build()
	return db.Updates(values).Error
}

func (slf *FileSearch) Save(record *File) error {
	if record.ID == 0 {
		return errors.New("id为空")
	}
	var db = slf.build()

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

	return nil
}

func (slf *FileSearch) Find() ([]*File, int64, error) {
	var db = slf.build()
	var records = make([]*File, 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.Find(&records).Error
	return records, total, err
}

// InitDefaultData 初始化数据
func (slf *FileSearch) InitDefaultData() error {
	var (
		db          = slf.Orm.Table(slf.TableName())
		total int64 = 0
	)
	if err := db.Count(&total).Error; err != nil {
		return err
	}
	if total != 0 {
		return nil
	}
	records := []*File{}
	return slf.CreateBatch(records)
}