package originals

import (
	"ytdlp-site/database"
	"ytdlp-site/transcodes"

	"gorm.io/gorm"
)

type Status string

const (
	StatusNotStarted        Status = "not started"
	StatusMetadata          Status = "metadata"
	StatusDownloading       Status = "downloading"
	StatusDownloadCompleted Status = "download completed"
	StatusTranscoding       Status = "transcoding"
	StatusCompleted         Status = "completed"
	StatusFailed            Status = "failed"
)

type Original struct {
	gorm.Model
	UserID  uint
	URL     string
	Title   string
	Artist  string
	Status  Status
	Audio   bool // video download requested
	Video   bool // audio download requested
	Watched bool

	Playlist   bool // part of a playlist
	PlaylistID uint // Playlist.ID (if part of a playlist)
}

func SetStatus(id uint, status Status) error {
	db := database.Get()
	log.Debugln("original", id, "status -> ", status)
	return db.Model(&Original{}).Where("id = ?", id).Update("status", status).Error
}

// if there is an active transcode for this original,
// set the status to transcode. otherwise ,to completed
func SetStatusTranscodingOrCompleted(id uint) error {
	db := database.Get()

	var count int64
	err := db.Model(&transcodes.Transcode{}).Where("original_id = ?", id).Count(&count).Error
	if err != nil {
		return err
	}

	if count > 0 {
		log.Debugln("found transcodes for original", id)
		return SetStatus(id, StatusTranscoding)
	} else {
		log.Debugln("no transcodes for original", id)
		return SetStatus(id, StatusCompleted)
	}

}