package dav

import (
	"fmt"
	"path/filepath"

	"github.com/debridmediamanager.com/zurg/internal/torrent"
	"github.com/debridmediamanager.com/zurg/pkg/dav"
	"github.com/debridmediamanager.com/zurg/pkg/realdebrid"
)

// createMultiTorrentResponse creates a WebDAV response for a list of torrents
func createMultiTorrentResponse(torrents []realdebrid.Torrent) (*dav.MultiStatus, error) {
	var responses []dav.Response
	responses = append(responses, dav.Directory("/torrents"))

	seen := make(map[string]bool)

	for _, item := range torrents {
		if item.Progress != 100 {
			continue
		}
		if _, exists := seen[item.Filename]; exists {
			continue
		}
		seen[item.Filename] = true

		path := filepath.Join("/torrents", item.Filename)
		responses = append(responses, dav.Directory(path))
	}

	return &dav.MultiStatus{
		XMLNS:    "DAV:",
		Response: responses,
	}, nil
}

// createTorrentResponse creates a WebDAV response for torrents with the same name
func createCombinedTorrentResponse(torrents []realdebrid.Torrent, t *torrent.TorrentManager) (*dav.MultiStatus, error) {
	var responses []dav.Response
	// initial response is the directory itself
	currentPath := filepath.Join("/torrents", torrents[0].Filename)
	responses = append(responses, dav.Directory(currentPath))

	seen := make(map[string]bool)
	idx := 0

	var torrentResponses []dav.Response
	for _, torrent := range torrents {
		info := t.GetInfo(torrent.ID)
		if info == nil {
			continue
		}

		var selectedFiles []realdebrid.File
		for _, file := range info.Files {
			if file.Selected == 0 {
				continue
			}
			selectedFiles = append(selectedFiles, file)
		}

		if len(selectedFiles) != len(info.Links) {
			fmt.Println("Links and files do not match", info.Filename)
			// TODO: Add auto-healing for this
			// for _, link := range info.Links {
			// 	unrestrictFn := func() (*realdebrid.UnrestrictResponse, error) {
			// 		return realdebrid.UnrestrictCheck(os.Getenv("RD_TOKEN"), link)
			// 	}
			// 	resp := realdebrid.RetryUntilOk(unrestrictFn)
			// 	if resp == nil {
			// 		continue
			// 	} else {
			// 		if _, exists := seen[resp.Filename]; exists {
			// 			continue
			// 		}
			// 		seen[resp.Filename] = true
			// 		filePath := filepath.Join(currentPath, resp.Filename)
			// 		torrentResponses = append(torrentResponses,
			// 			dav.File(
			// 				filePath,
			// 				resp.Filesize,
			// 				info.Added,
			// 				resp.Link,
			// 			),
			// 		)
			// 	}
			// }
		} else {
			for _, file := range selectedFiles {
				filename := filepath.Base(file.Path)
				if _, exists := seen[filename]; exists {
					continue
				}
				seen[filename] = true
				filePath := filepath.Join(currentPath, filename)
				torrentResponses = append(torrentResponses, dav.File(
					filePath,
					file.Bytes,
					convertDate(info.Added),
					info.Links[idx],
				))
				idx++
			}
		}
	}
	responses = append(responses, torrentResponses...)

	return &dav.MultiStatus{
		XMLNS:    "DAV:",
		Response: responses,
	}, nil
}