package html

import (
	"log"
	"net/url"
	"strings"

	"github.com/evan-buss/opds-proxy/opds"
)

type FeedViewModel struct {
	Title      string
	Search     string
	Navigation []NavigationViewModel
	Links      []LinkViewModel
}
type NavigationViewModel struct {
	Href  string
	Label string
}

type LinkViewModel struct {
	Title      string
	Author     string
	ImageURL   string
	Content    string
	Href       string
	IsDownload bool
}

func convertFeed(p *FeedParams) FeedViewModel {
	vm := FeedViewModel{
		Title:      p.Feed.Title,
		Search:     "",
		Links:      make([]LinkViewModel, 0),
		Navigation: make([]NavigationViewModel, 0),
	}

	for _, link := range p.Feed.Links {
		if link.Rel == "search" {
			vm.Search = resolveHref(p.URL, link.Href)
		}

		if link.TypeLink == "application/atom+xml;type=feed;profile=opds-catalog" {
			vm.Navigation = append(vm.Navigation, NavigationViewModel{
				Href:  resolveHref(p.URL, link.Href),
				Label: strings.ToUpper(link.Rel[:1]) + link.Rel[1:],
			})
		}
	}

	for _, entry := range p.Feed.Entries {
		vm.Links = append(vm.Links, constructLink(p.URL, entry))
	}

	return vm
}

func constructLink(url string, entry opds.Entry) LinkViewModel {
	vm := LinkViewModel{
		Title:   entry.Title,
		Content: entry.Content.Content,
	}

	authors := make([]string, 0)
	for _, author := range entry.Author {
		authors = append(authors, author.Name)
	}
	vm.Author = strings.Join(authors, " & ")

	for _, link := range entry.Links {
		vm.IsDownload = link.IsDownload()
		if link.IsNavigation() || link.IsDownload() {
			vm.Href = resolveHref(url, link.Href)
		}

		// Prefer the first "thumbnail" image we find
		if vm.ImageURL == "" && link.IsImage("thumbnail") {
			vm.ImageURL = resolveHref(url, link.Href)
		}
	}

	// If we didn't find a thumbnail, use the first image we find
	if vm.ImageURL == "" {
		for _, link := range entry.Links {
			if link.IsImage("") {
				vm.ImageURL = resolveHref(url, link.Href)
				break
			}
		}
	}

	return vm
}

func resolveHref(feedUrl string, relativePath string) string {
	baseUrl, err := url.Parse(feedUrl)
	if err != nil {
		log.Fatal(err)
	}
	relativeUrl, err := url.Parse(relativePath)
	if err != nil {
		log.Fatal(err)
	}

	return baseUrl.ResolveReference(relativeUrl).String()
}