forked from Ivasoft/opds-proxy
108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
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()
|
|
}
|