GoBlog/feeds.go

80 lines
2.0 KiB
Go
Raw Normal View History

2020-09-02 15:48:37 +00:00
package main
import (
"io"
2020-09-02 15:48:37 +00:00
"net/http"
"strings"
2020-09-02 15:48:37 +00:00
"time"
2020-10-06 17:07:48 +00:00
2020-11-01 17:37:21 +00:00
"github.com/araddon/dateparse"
"github.com/jlelse/feeds"
"go.goblog.app/app/pkgs/bufferpool"
"go.goblog.app/app/pkgs/contenttype"
2020-09-02 15:48:37 +00:00
)
type feedType string
const (
2020-10-06 17:07:48 +00:00
noFeed feedType = ""
rssFeed feedType = "rss"
atomFeed feedType = "atom"
jsonFeed feedType = "json"
2020-09-02 15:48:37 +00:00
)
2022-03-23 20:48:28 +00:00
func (a *goBlog) generateFeed(blog string, f feedType, w http.ResponseWriter, r *http.Request, posts []*post, title, description string) {
2020-09-02 15:48:37 +00:00
now := time.Now()
title = a.renderMdTitle(defaultIfEmpty(title, a.cfg.Blogs[blog].Title))
description = defaultIfEmpty(description, a.cfg.Blogs[blog].Description)
2020-09-02 15:48:37 +00:00
feed := &feeds.Feed{
Title: title,
Description: description,
Link: &feeds.Link{Href: a.getFullAddress(strings.TrimSuffix(r.URL.Path, "."+string(f)))},
2020-09-02 15:48:37 +00:00
Created: now,
2020-11-01 17:37:21 +00:00
Author: &feeds.Author{
Name: a.cfg.User.Name,
Email: a.cfg.User.Email,
2020-11-01 17:37:21 +00:00
},
Image: &feeds.Image{
Url: a.cfg.User.Picture,
2020-11-01 17:37:21 +00:00
},
2020-09-02 15:48:37 +00:00
}
2020-10-15 15:32:46 +00:00
for _, p := range posts {
buf := bufferpool.Get()
a.feedHtml(buf, p)
2020-09-02 15:48:37 +00:00
feed.Add(&feeds.Item{
2021-08-05 06:09:34 +00:00
Title: p.RenderedTitle,
Link: &feeds.Link{Href: a.fullPostURL(p)},
Description: a.postSummary(p),
2020-10-15 15:32:46 +00:00
Id: p.Path,
Content: buf.String(),
Created: timeNoErr(dateparse.ParseLocal(p.Published)),
Updated: timeNoErr(dateparse.ParseLocal(p.Updated)),
2020-09-02 15:48:37 +00:00
})
bufferpool.Put(buf)
2020-09-02 15:48:37 +00:00
}
var feedWriteFunc func(w io.Writer) error
2021-12-30 11:40:21 +00:00
var feedMediaType string
2020-09-02 15:48:37 +00:00
switch f {
2020-10-06 17:07:48 +00:00
case rssFeed:
feedMediaType = contenttype.RSS
feedWriteFunc = feed.WriteRss
2020-10-06 17:07:48 +00:00
case atomFeed:
feedMediaType = contenttype.ATOM
feedWriteFunc = feed.WriteAtom
2020-10-06 17:07:48 +00:00
case jsonFeed:
feedMediaType = contenttype.JSONFeed
feedWriteFunc = feed.WriteJSON
default:
2021-12-30 11:40:21 +00:00
a.serve404(w, r)
return
2020-09-02 15:48:37 +00:00
}
pipeReader, pipeWriter := io.Pipe()
go func() {
writeErr := feedWriteFunc(pipeWriter)
_ = pipeWriter.CloseWithError(writeErr)
}()
w.Header().Set(contentType, feedMediaType+contenttype.CharsetUtf8Suffix)
minifyErr := a.min.Minify(feedMediaType, w, pipeReader)
_ = pipeReader.CloseWithError(minifyErr)
2020-09-02 15:48:37 +00:00
}