package service

import (
	model2 "github.com/IceWhaleTech/CasaOS/service/model"
	"gorm.io/gorm"
)

type DownloadService interface {
	AddDownloadTask(m model2.PersonDownloadDBModel)   //添加下载任务
	EditDownloadState(m model2.PersonDownloadDBModel) //只修改状态
	SaveDownload(m model2.PersonDownloadDBModel)
	DelDownload(uuid string)
	GetDownloadById(uuid string) model2.PersonDownloadDBModel
	GetDownloadListByState(state string, t int) []model2.PersonDownloadDBModel
	SetDownloadError(m model2.PersonDownloadDBModel)
	GetDownloadListByPath(m model2.PersonDownloadDBModel) int
}
type downloadService struct {
	db *gorm.DB
}

func (d *downloadService) GetDownloadListByPath(m model2.PersonDownloadDBModel) int {
	var list []model2.PersonDownloadDBModel
	d.db.Select("path").Where("path = ? AND `from` = ? AND state = 0", m.Path, m.From).Find(&list)
	return len(list)
}

func (d *downloadService) AddDownloadTask(m model2.PersonDownloadDBModel) {

	d.db.Create(&m)
}
func (d *downloadService) EditDownloadState(m model2.PersonDownloadDBModel) {

	d.db.Model(&m).Where("uuid = ?", m.UUID).Update("state", m.State)
}

//failed during download
func (d *downloadService) SetDownloadError(m model2.PersonDownloadDBModel) {
	d.db.Model(&m).Updates(m)
}

func (d *downloadService) DelDownload(uuid string) {
	var m model2.PersonDownloadDBModel
	d.db.Where("uuid = ?", uuid).Delete(&m)
}
func (d *downloadService) GetDownloadById(uuid string) model2.PersonDownloadDBModel {
	var m model2.PersonDownloadDBModel
	d.db.Model(m).Where("uuid = ?", uuid).First(&m)
	return m
}
func (d *downloadService) GetDownloadListByState(state string, t int) (list []model2.PersonDownloadDBModel) {
	if len(state) == 0 {
		d.db.Where("type = ?", t).Find(&list)
	} else {
		d.db.Where("state = ? AND type= ?", state, t).Find(&list)
	}

	return
}

func (d *downloadService) SaveDownload(m model2.PersonDownloadDBModel) {
	d.db.Save(&m)
}
func NewDownloadService(db *gorm.DB) DownloadService {
	return &downloadService{db: db}
}