From e849e705c30cceec3cf7336a21bed96c8a911e90 Mon Sep 17 00:00:00 2001
From: Alexander Neonxp Kiryukhin <i@neonxp.ru>
Date: Tue, 8 Oct 2024 03:43:08 +0300
Subject: Добавил рейтинг Добавил страницу топа Добавил rss/xml/json feed
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/handler/feed/feed.go    | 73 +++++++++++++++++++++++++++++++++++++++++++++
 pkg/handler/feed/handler.go | 21 +++++++++++++
 2 files changed, 94 insertions(+)
 create mode 100644 pkg/handler/feed/feed.go
 create mode 100644 pkg/handler/feed/handler.go

(limited to 'pkg/handler/feed')

diff --git a/pkg/handler/feed/feed.go b/pkg/handler/feed/feed.go
new file mode 100644
index 0000000..05921e9
--- /dev/null
+++ b/pkg/handler/feed/feed.go
@@ -0,0 +1,73 @@
+package feed
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gorilla/feeds"
+	"github.com/labstack/echo/v4"
+	"sh.org.ru/pkg/model"
+)
+
+func (h *Handler) Feed(c echo.Context) error {
+	feedType := c.Param("type")
+
+	quotes := make([]model.Quote, 0, 20)
+	err := h.db.NewSelect().
+		Model((*model.Quote)(nil)).
+		Order("id DESC").
+		Where("approved = ?", true).
+		Limit(20).
+		Scan(c.Request().Context(), &quotes)
+	if err != nil {
+		return err
+	}
+
+	feed := &feeds.Feed{
+		Title:       "sh.org.ru - Новый цитатник Рунета",
+		Link:        &feeds.Link{Href: h.cfg.Host},
+		Description: "",
+		Author:      &feeds.Author{Name: "NeonXP", Email: "i@neonxp.ru"},
+		Created:     time.Now(),
+	}
+
+	for _, q := range quotes {
+		uid := fmt.Sprintf("%s/quote/%d", h.cfg.Host, q.ID)
+		feed.Items = append(feed.Items, &feeds.Item{
+			Id:          uid,
+			Title:       fmt.Sprintf("Цитата #%d", q.ID),
+			Link:        &feeds.Link{Href: uid},
+			Created:     q.CreatedAt,
+			Description: q.Text(),
+		})
+	}
+	switch feedType {
+	case "rss":
+		result, err := feed.ToRss()
+		if err != nil {
+			return err
+		}
+
+		c.Response().Header().Set("Content-Type", "application/rss+xml")
+		return c.String(http.StatusOK, result)
+	case "atom":
+		result, err := feed.ToAtom()
+		if err != nil {
+			return err
+		}
+
+		c.Response().Header().Set("Content-Type", "application/atom+xml")
+		return c.String(http.StatusOK, result)
+	case "json":
+		result, err := feed.ToJSON()
+		if err != nil {
+			return err
+		}
+
+		c.Response().Header().Set("Content-Type", "application/json")
+		return c.String(http.StatusOK, result)
+	default:
+		return echo.ErrNotFound
+	}
+}
diff --git a/pkg/handler/feed/handler.go b/pkg/handler/feed/handler.go
new file mode 100644
index 0000000..868e477
--- /dev/null
+++ b/pkg/handler/feed/handler.go
@@ -0,0 +1,21 @@
+package feed
+
+import (
+	"github.com/labstack/echo/v4"
+	"github.com/uptrace/bun"
+	"sh.org.ru/pkg/config"
+)
+
+type Handler struct {
+	db  *bun.DB
+	cfg *config.Config
+}
+
+// NewHandler returns new Handler.
+func NewHandler(db *bun.DB, cfg *config.Config) *Handler {
+	return &Handler{db: db, cfg: cfg}
+}
+
+func (h *Handler) Register(g *echo.Group) {
+	g.GET("/:type", h.Feed)
+}
-- 
cgit v1.2.3