mirror of https://github.com/jlelse/GoBlog
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/araddon/dateparse"
|
|
"github.com/gorilla/feeds"
|
|
)
|
|
|
|
type feedType string
|
|
|
|
const (
|
|
noFeed feedType = ""
|
|
rssFeed feedType = "rss"
|
|
atomFeed feedType = "atom"
|
|
jsonFeed feedType = "json"
|
|
|
|
feedAudioURL = "audio"
|
|
feedAudioType = "audiomime"
|
|
feedAudioLength = "audiolength"
|
|
)
|
|
|
|
func generateFeed(blog string, f feedType, w http.ResponseWriter, r *http.Request, posts []*post, title string, description string) {
|
|
now := time.Now()
|
|
if title == "" {
|
|
title = appConfig.Blogs[blog].Title
|
|
}
|
|
if description == "" {
|
|
description = appConfig.Blogs[blog].Description
|
|
}
|
|
feed := &feeds.Feed{
|
|
Title: title,
|
|
Description: description,
|
|
Link: &feeds.Link{Href: appConfig.Server.PublicAddress + strings.TrimSuffix(r.URL.Path, "."+string(f))},
|
|
Created: now,
|
|
Author: &feeds.Author{
|
|
Name: appConfig.User.Name,
|
|
Email: appConfig.User.Email,
|
|
},
|
|
Image: &feeds.Image{
|
|
Url: appConfig.User.Picture,
|
|
},
|
|
}
|
|
for _, p := range posts {
|
|
created, _ := dateparse.ParseLocal(p.Published)
|
|
updated, _ := dateparse.ParseLocal(p.Updated)
|
|
var enc *feeds.Enclosure
|
|
if p.firstParameter(feedAudioURL) != "" {
|
|
enc = &feeds.Enclosure{
|
|
Url: p.firstParameter(feedAudioURL),
|
|
Type: p.firstParameter(feedAudioType),
|
|
Length: p.firstParameter(feedAudioLength),
|
|
}
|
|
}
|
|
feed.Add(&feeds.Item{
|
|
Title: p.title(),
|
|
Link: &feeds.Link{Href: p.fullURL()},
|
|
Description: p.summary(),
|
|
Id: p.Path,
|
|
Content: string(p.absoluteHTML()),
|
|
Created: created,
|
|
Updated: updated,
|
|
Enclosure: enc,
|
|
})
|
|
}
|
|
var err error
|
|
var feedString, feedMediaType string
|
|
switch f {
|
|
case rssFeed:
|
|
feedMediaType = contentTypeRSS
|
|
feedString, err = feed.ToRss()
|
|
case atomFeed:
|
|
feedMediaType = contentTypeATOM
|
|
feedString, err = feed.ToAtom()
|
|
case jsonFeed:
|
|
feedMediaType = contentTypeJSONFeed
|
|
feedString, err = feed.ToJSON()
|
|
default:
|
|
return
|
|
}
|
|
if err != nil {
|
|
w.Header().Del(contentType)
|
|
serveError(w, r, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
w.Header().Set(contentType, feedMediaType+charsetUtf8Suffix)
|
|
_, _ = writeMinified(w, feedMediaType, []byte(feedString))
|
|
}
|