refactor

html/feed.go

@@ -0,0 +1,109 @@
+package html
+
+import (
+	"fmt"
+	"log"
+	"net/url"
+	"strings"
+
+	"github.com/evan-buss/kobo-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)
+	}
+
+	resolved := baseUrl.ResolveReference(relativeUrl).String()
+	fmt.Println("Resolved URL: ", resolved)
+	return resolved
+}