Trending page (#128)

Fixes #46 Reviewed-on: https://codeberg.org/rimgo/rimgo/pulls/128 Co-authored-by: orangix <orangix@noreply.codeberg.org> Co-committed-by: orangix <orangix@noreply.codeberg.org>
2025-12-13 19:55:21 +00:00 · 2023-08-14 22:47:59 +00:00
parent d69d8dba0e
commit a8abb43f3a
6 changed files with 255 additions and 8 deletions
--- a/api/trending.go
+++ b/api/trending.go
@@ -0,0 +1,107 @@
+package api
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+	"sync"
+
+	"codeberg.org/rimgo/rimgo/utils"
+	"github.com/patrickmn/go-cache"
+	"github.com/tidwall/gjson"
+)
+
+func (client *Client) FetchTrending(section, sort, page string) ([]Submission, error) {
+	cacheData, found := client.Cache.Get(fmt.Sprintf("trending-%s-%s-%s", section, sort, page))
+	if found {
+		return cacheData.([]Submission), nil
+	}
+
+	req, err := http.NewRequest("GET", "https://api.imgur.com/post/v1/posts", nil)
+	if err != nil {
+		return []Submission{}, err
+	}
+	utils.SetReqHeaders(req)
+
+	q := req.URL.Query()
+	q.Add("client_id", client.ClientID)
+	q.Add("include", "cover")
+	q.Add("page", page)
+
+	switch sort {
+	case "newest":
+		q.Add("filter[window]", "week")
+		q.Add("sort", "-time")
+	case "best":
+		q.Add("filter[window]", "all")
+		q.Add("sort", "-top")
+	case "popular":
+		fallthrough
+	default:
+		q.Add("filter[window]", "week")
+		q.Add("sort", "-viral")
+		sort = "popular"
+	}
+	switch section {
+	case "hot":
+		q.Add("filter[section]", "eq:hot")
+	case "new":
+		q.Add("filter[section]", "eq:new")
+	case "top":
+		q.Add("filter[section]", "eq:top")
+		q.Add("filter[window]", "day")
+	default:
+		q.Add("filter[section]", "eq:hot")
+		section = "hot"
+	}
+
+	req.URL.RawQuery = q.Encode()
+
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return []Submission{}, err
+	}
+
+	body, err := io.ReadAll(res.Body)
+	if err != nil {
+		return []Submission{}, err
+	}
+
+	data := gjson.Parse(string(body))
+
+	wg := sync.WaitGroup{}
+	posts := make([]Submission, 0)
+	data.ForEach(
+		func(key, value gjson.Result) bool {
+			wg.Add(1)
+
+			go func() {
+				defer wg.Done()
+				posts = append(posts, Submission{
+					Id:    value.Get("id").String(),
+					Title: value.Get("title").String(),
+					Link:  strings.ReplaceAll(value.Get("url").String(), "https://imgur.com", ""),
+					Cover: Media{
+						Id:   value.Get("cover_id").String(),
+						Type: value.Get("cover.type").String(),
+						Url:  strings.ReplaceAll(value.Get("cover.url").String(), "https://i.imgur.com", ""),
+					},
+					Points:    value.Get("point_count").Int(),
+					Upvotes:   value.Get("upvote_count").Int(),
+					Downvotes: value.Get("downvote_count").Int(),
+					Comments:  value.Get("comment_count").Int(),
+					Views:     value.Get("view_count").Int(),
+					IsAlbum:   value.Get("is_album").Bool(),
+				})
+			}()
+
+			return true
+		},
+	)
+
+	wg.Wait()
+
+	client.Cache.Set(fmt.Sprintf("trending-%s-%s-%s", section, sort, page), posts, cache.DefaultExpiration)
+	return posts, nil
+}