package qbit

import (
	"cmp"
	"goBlack/pkg/debrid"
	"os"
	"path/filepath"
	"time"
)

// All torrent related helpers goes here

func (q *QBit) MarkAsFailed(t *Torrent) *Torrent {
	t.State = "error"
	q.storage.AddOrUpdate(t)
	return t
}

func (q *QBit) UpdateTorrent(t *Torrent, debridTorrent *debrid.Torrent) *Torrent {
	rcLoneMount := q.debrid.GetMountPath()
	if debridTorrent == nil && t.ID != "" {
		debridTorrent, _ = q.debrid.GetTorrent(t.ID)
	}
	if debridTorrent == nil {
		q.logger.Printf("Torrent with ID %s not found in %s", t.ID, q.debrid.GetName())
		return t
	}
	if debridTorrent.Status != "downloaded" {
		debridTorrent, _ = q.debrid.GetTorrent(t.ID)
	}

	if t.TorrentPath == "" {
		t.TorrentPath = filepath.Base(debridTorrent.GetMountFolder(rcLoneMount))
	}

	totalSize := float64(cmp.Or(debridTorrent.Bytes, 1.0))
	progress := cmp.Or(debridTorrent.Progress, 100.0)
	progress = progress / 100.0
	var sizeCompleted int64

	sizeCompleted = int64(totalSize * progress)
	savePath := filepath.Join(q.DownloadFolder, t.Category) + string(os.PathSeparator)
	torrentPath := filepath.Join(savePath, t.TorrentPath) + string(os.PathSeparator)

	var speed int64
	if debridTorrent.Speed != 0 {
		speed = debridTorrent.Speed
	}
	var eta int64
	if speed != 0 {
		eta = int64((totalSize - float64(sizeCompleted)) / float64(speed))
	}
	
	t.Size = debridTorrent.Bytes
	t.DebridTorrent = debridTorrent
	t.Completed = sizeCompleted
	t.Downloaded = sizeCompleted
	t.DownloadedSession = sizeCompleted
	t.Uploaded = sizeCompleted
	t.UploadedSession = sizeCompleted
	t.AmountLeft = int64(totalSize) - sizeCompleted
	t.Progress = float32(progress)
	t.SavePath = savePath
	t.ContentPath = torrentPath
	t.Eta = eta
	t.Dlspeed = speed
	t.Upspeed = speed

	if t.IsReady() {
		t.State = "pausedUP"
		q.storage.AddOrUpdate(t)
		return t
	}
	ticker := time.NewTicker(3 * time.Second)
	for {
		select {
		case <-ticker.C:
			if t.IsReady() {
				t.State = "pausedUP"
				q.storage.AddOrUpdate(t)
				ticker.Stop()
				return t
			} else {
				return q.UpdateTorrent(t, debridTorrent)
			}
		}
	}
}

func (q *QBit) ResumeTorrent(t *Torrent) bool {
	return true
}

func (q *QBit) PauseTorrent(t *Torrent) bool {
	return true
}

func (q *QBit) RefreshTorrent(t *Torrent) bool {
	return true
}

func (q *QBit) GetTorrentProperties(t *Torrent) *TorrentProperties {
	return &TorrentProperties{
		AdditionDate:           t.AddedOn,
		Comment:                "Debrid Blackhole <https://github.com/sirrobot01/debrid-blackhole>",
		CreatedBy:              "Debrid Blackhole <https://github.com/sirrobot01/debrid-blackhole>",
		CreationDate:           t.AddedOn,
		DlLimit:                -1,
		UpLimit:                -1,
		DlSpeed:                t.Dlspeed,
		UpSpeed:                t.Upspeed,
		TotalSize:              t.Size,
		TotalUploaded:          t.Uploaded,
		TotalDownloaded:        t.Downloaded,
		TotalUploadedSession:   t.UploadedSession,
		TotalDownloadedSession: t.DownloadedSession,
		LastSeen:               time.Now().Unix(),
		NbConnectionsLimit:     100,
		Peers:                  0,
		PeersTotal:             2,
		SeedingTime:            1,
		Seeds:                  100,
		ShareRatio:             100,
	}
}