mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 12:17:24 +00:00
Add RSS/Sitemap handling.
This commit is contained in:
49
server/utils/rss.go
Normal file
49
server/utils/rss.go
Normal file
@@ -0,0 +1,49 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"github.com/gofiber/fiber/v3/log"
|
||||
"net/url"
|
||||
"nysoure/server/model"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
RssFileName = "rss.xml"
|
||||
)
|
||||
|
||||
func GenerateRss(baseURL string, resources []model.Resource) {
|
||||
path := filepath.Join(GetStoragePath(), RssFileName)
|
||||
builder := strings.Builder{}
|
||||
builder.WriteString(`<?xml version="1.0" encoding="UTF-8"?>`)
|
||||
builder.WriteRune('\n')
|
||||
builder.WriteString(`<rss version="2.0">`)
|
||||
builder.WriteRune('\n')
|
||||
builder.WriteString(`<channel>`)
|
||||
builder.WriteRune('\n')
|
||||
for _, resource := range resources {
|
||||
builder.WriteString(" <item>\n")
|
||||
builder.WriteString(" <title>")
|
||||
builder.WriteString(url.PathEscape(resource.Title))
|
||||
builder.WriteString("</title>\n")
|
||||
builder.WriteString(" <link>")
|
||||
builder.WriteString(baseURL + "/resources/" + strconv.Itoa(int(resource.ID)))
|
||||
builder.WriteString("</link>\n")
|
||||
builder.WriteString(" <description>")
|
||||
builder.WriteString(url.PathEscape(ArticleToDescription(resource.Article, 255)))
|
||||
builder.WriteString("</description>\n")
|
||||
builder.WriteString(" <pubDate>")
|
||||
builder.WriteString(resource.UpdatedAt.Format("Mon, 02 Jan 2006 15:04:05 MST"))
|
||||
builder.WriteString("</pubDate>\n")
|
||||
builder.WriteString(" </item>\n")
|
||||
}
|
||||
builder.WriteString(`</channel>`)
|
||||
builder.WriteRune('\n')
|
||||
builder.WriteString(`</rss>`)
|
||||
data := builder.String()
|
||||
if err := os.WriteFile(path, []byte(data), 0644); err != nil {
|
||||
log.Error("failed to write RSS file", err)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user