mirror of
https://github.com/jlelse/GoBlog
synced 2024-06-26 17:47:35 +00:00
90 lines
2.2 KiB
Go
90 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,
|
|
},
|
|
Image: &feeds.Image{
|
|
Url: appConfig.User.Picture,
|
|
},
|
|
}
|
|
for _, p := range posts {
|
|
created, _ := dateparse.ParseIn(p.Published, time.Local)
|
|
updated, _ := dateparse.ParseIn(p.Updated, time.Local)
|
|
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.html()),
|
|
Created: created,
|
|
Updated: updated,
|
|
Enclosure: enc,
|
|
})
|
|
}
|
|
var feedStr string
|
|
var err error
|
|
switch f {
|
|
case rssFeed:
|
|
w.Header().Add(contentType, "application/rss+xml; charset=utf-8")
|
|
feedStr, err = feed.ToRss()
|
|
case atomFeed:
|
|
w.Header().Add(contentType, "application/atom+xml; charset=utf-8")
|
|
feedStr, err = feed.ToAtom()
|
|
case jsonFeed:
|
|
w.Header().Add(contentType, "application/feed+json; charset=utf-8")
|
|
feedStr, err = feed.ToJSON()
|
|
default:
|
|
return
|
|
}
|
|
if err != nil {
|
|
w.Header().Del(contentType)
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(feedStr))
|
|
}
|