Initial commit

internal/widget/reddit.go

@@ -0,0 +1,66 @@
+package widget
+
+import (
+	"context"
+	"errors"
+	"html/template"
+	"time"
+
+	"github.com/glanceapp/glance/internal/assets"
+	"github.com/glanceapp/glance/internal/feed"
+)
+
+type Reddit struct {
+	widgetBase    `yaml:",inline"`
+	Posts         feed.ForumPosts `yaml:"-"`
+	Subreddit     string          `yaml:"subreddit"`
+	Style         string          `yaml:"style"`
+	Limit         int             `yaml:"limit"`
+	CollapseAfter int             `yaml:"collapse-after"`
+}
+
+func (widget *Reddit) Initialize() error {
+	if widget.Subreddit == "" {
+		return errors.New("no subreddit specified")
+	}
+
+	if widget.Limit <= 0 {
+		widget.Limit = 15
+	}
+
+	if widget.CollapseAfter == 0 || widget.CollapseAfter < -1 {
+		widget.CollapseAfter = 5
+	}
+
+	widget.withTitle("/r/" + widget.Subreddit).withCacheDuration(30 * time.Minute)
+
+	return nil
+}
+
+func (widget *Reddit) Update(ctx context.Context) {
+	posts, err := feed.FetchSubredditPosts(widget.Subreddit)
+
+	if !widget.canContinueUpdateAfterHandlingErr(err) {
+		return
+	}
+
+	if len(posts) > widget.Limit {
+		posts = posts[:widget.Limit]
+	}
+
+	posts.SortByEngagement()
+	widget.Posts = posts
+}
+
+func (widget *Reddit) Render() template.HTML {
+	if widget.Style == "horizontal-cards" {
+		return widget.render(widget, assets.RedditCardsHorizontalTemplate)
+	}
+
+	if widget.Style == "vertical-cards" {
+		return widget.render(widget, assets.RedditCardsVerticalTemplate)
+	}
+
+	return widget.render(widget, assets.ForumPostsTemplate)
+
+}