Use xml marshaling to create the rss feed

lib/rss.go

@@ -0,0 +1,158 @@
+package trantor
+
+import (
+	"encoding/xml"
+	"errors"
+	"net/http"
+	"strings"
+
+	"gitlab.com/trantor/trantor/lib/database"
+)
+
+type rss struct {
+	XMLName xml.Name `xml:"rss"`
+	Version string   `xml:"version,attr"`
+	Channel channel  `xml:"channel"`
+}
+
+type channel struct {
+	Title          string `xml:"title"`
+	Description    string `xml:"description"`
+	Link           string `xml:"link"`
+	ManagingEditor string `xml:"managingEditor,omitempty"`
+	Item           []item `xml:"item"`
+}
+
+type item struct {
+	Title       string     `xml:"title"`
+	Description string     `xml:"description"`
+	Link        string     `xml:"link"`
+	GUID        *guid      `xml:"guid,omitempty"`
+	Enclosure   *enclosure `xml:"enclosure,omitempty"`
+	Category    []string   `xml:"category"`
+}
+
+type guid struct {
+	IsPermaLink bool   `xml:"isPermaLink,attr"`
+	Chardata    string `xml:",chardata"`
+}
+
+type enclosure struct {
+	URL    string `xml:"url,attr"`
+	Length int    `xml:"length,attr"`
+	Type   string `xml:"type,attr"`
+}
+
+func loadRss(w http.ResponseWriter, tmpl string, data interface{}) error {
+	var err error
+	var res rss
+	switch tmpl {
+	case "news":
+		res, err = newsRss(data)
+	case "search":
+		res, err = searchRss(data)
+	case "list":
+		res, err = listRss(data)
+	}
+	if err != nil {
+		return err
+	}
+	w.Write([]byte(xml.Header))
+	enc := xml.NewEncoder(w)
+	enc.Indent("", "    ")
+	return enc.Encode(res)
+}
+
+func newsRss(data interface{}) (rss, error) {
+	news, ok := data.(newsData)
+	if !ok {
+		return rss{}, errors.New("Data is not valid")
+	}
+
+	feed := rss{
+		Version: "2.0",
+		Channel: channel{
+			Title:       news.S.Title,
+			Description: "News of the library",
+			Link:        news.S.BaseURL + "/news/",
+			Item:        make([]item, len(news.News)),
+		},
+	}
+	for i, n := range news.News {
+		feed.Channel.Item[i] = item{
+			Title:       n.Date,
+			Description: n.Text,
+			Link:        news.S.BaseURL + "/news/",
+		}
+	}
+
+	return feed, nil
+}
+
+func searchRss(data interface{}) (rss, error) {
+	search, ok := data.(searchData)
+	if !ok {
+		return rss{}, errors.New("Data is not valid")
+	}
+
+	description := "Last books added"
+	link := search.S.BaseURL
+	if search.S.Search != "" {
+		description = "Book search: " + search.S.Search
+		link += "/search/?q=" + search.S.Search
+	}
+
+	feed := rss{
+		Version: "2.0",
+		Channel: channel{
+			Title:       search.S.Title,
+			Description: description,
+			Link:        link,
+			Item:        bookListRss(search.Books, search.S.BaseURL),
+		},
+	}
+	return feed, nil
+}
+
+func listRss(data interface{}) (rss, error) {
+	lb, ok := data.(listData)
+	if !ok {
+		return rss{}, errors.New("Data is not valid")
+	}
+
+	feed := rss{
+		Version: "2.0",
+		Channel: channel{
+			Title:          lb.S.Title,
+			Description:    strings.Join(lb.List.Description, "\n"),
+			Link:           lb.S.BaseURL + "/list/" + lb.List.ListID,
+			ManagingEditor: lb.List.User.Username,
+			Item:           bookListRss(lb.List.Books, lb.S.BaseURL),
+		},
+	}
+	return feed, nil
+}
+
+func bookListRss(books []database.Book, baseURL string) []item {
+	items := make([]item, len(books))
+	for i, b := range books {
+		items[i] = item{
+			Title:       b.Title,
+			Description: b.Description,
+			Link:        baseURL + "/download/" + b.ID,
+			Category:    b.Tags,
+			Enclosure: &enclosure{
+				URL:    baseURL + "/download/" + b.ID + "/" + b.Title + ".epub",
+				Length: b.FileSize,
+				Type:   "application/epub+zip",
+			},
+		}
+		if b.Isbn != "" {
+			items[i].GUID = &guid{
+				IsPermaLink: false,
+				Chardata:    "ISBN: " + b.Isbn,
+			}
+		}
+	}
+	return items
+}

lib/stats.go

@@ -35,7 +35,7 @@ func (h handler) load(tmpl string, data interface{}) {
 	fmt := h.r.FormValue("fmt")
 	switch fmt {
 	case "rss":
-		err = h.template.rss.ExecuteTemplate(h.w, tmpl+".rss", data)
+		err = loadRss(h.w, tmpl, data)
 	case "opds":
 		err = h.template.opds.ExecuteTemplate(h.w, tmpl+".opds", data)
 	case "json":

lib/template.go

@@ -55,7 +55,6 @@ func GetStatus(h handler) Status {
 
 type Template struct {
 	html *html_tmpl.Template
-	rss  *txt_tmpl.Template
 	opds *txt_tmpl.Template
 }
 
@@ -71,18 +70,6 @@ func InitTemplate(assetsPath string) *Template {
 		log.Critical("Error loading html templates: ", err)
 	}
 
-	t.rss, err = txt_tmpl.New("rss").Funcs(txt_tmpl.FuncMap{
-		"book_list": func(books, baseURL interface{}) (map[string]interface{}, error) {
-			data := make(map[string]interface{}, 2)
-			data["Books"] = books
-			data["BaseURL"] = baseURL
-			return data, nil
-		},
-	}).ParseGlob(path.Join(templatePath, "*.rss"))
-	if err != nil {
-		log.Critical("Error loading rss templates: ", err)
-	}
-
 	t.opds, err = txt_tmpl.ParseGlob(path.Join(templatePath, "*.opds"))
 	if err != nil {
 		log.Critical("Error loading opds templates: ", err)

templates/book_list.rss

@@ -1,17 +0,0 @@
-{{$baseURL := .BaseURL}}
-{{range .Books}}
- <item>
-  <title>{{.Title}} - {{index .Authors 0}}</title>
-  <description>{{.Description}}</description>
-  <link>{{$baseURL}}/book/{{.ID}}</link>
-{{if .Isbn}}
-  <guid isPermaLink="false">ISBN: {{.Isbn}}</guid>
-{{end}}
-  <enclosure url="{{$baseURL}}/download/{{.ID}}/{{.Title}}.epub" length="{{.FileSize}}" type="application/epub+zip" />
-{{range .Authors}}
-  {{if .}}
-  <category>{{.}}</category>
-  {{end}}
-{{end}}
- </item>
-{{end}}

templates/list.rss

@@ -1,14 +0,0 @@
-<?xml version="1.0" encoding="UTF-8" ?>
-<rss version="2.0">
-<channel>
- <title>{{.S.Title}}</title>
- <description>{{range .List.Description}}
-	 {{.}}
-{{end}}</description>
- <link>{{.S.BaseURL}}/list/{{.List.ListID}}</link>
- <managingEditor>{{.List.User.Username}}</managingEditor>
- 
-{{template "book_list.rss" book_list .List.Books .S.BaseURL}}
- 
-</channel>
-</rss>

templates/news.rss

@@ -1,20 +0,0 @@
-<?xml version="1.0" encoding="UTF-8" ?>
-<rss version="2.0">
-<channel>
-{{with .S}}
- <title>{{.Title}}</title>
- <description>News of the library</description>
- <link>{{.BaseURL}}/news/</link>
-{{end}}
- 
-{{$baseURL := .S.BaseURL}}
-{{range .News}}
- <item>
-  <title>{{.Date}}</title>
-  <description>{{.Text}}</description>
-  <link>{{$baseURL}}/news/</link>
- </item>
-{{end}}
- 
-</channel>
-</rss>

templates/search.rss

@@ -1,17 +0,0 @@
-<?xml version="1.0" encoding="UTF-8" ?>
-<rss version="2.0">
-<channel>
-{{with .S}}
- <title>{{.Title}}</title>
-{{if .Search}}
- <description>Book search: {{.Search}}</description>
-{{else}}
- <description>Last books added</description>
-{{end}}
- <link>{{.BaseURL}}</link>
-{{end}}
- 
-{{template "book_list.rss" book_list .Books .S.BaseURL}}
- 
-</channel>
-</rss>