rimgo/api/album.go

92 lines
2.4 KiB
Go
Raw Normal View History

2022-01-17 20:23:04 +00:00
package api
import (
"io"
"net/http"
"strings"
2022-01-17 22:34:05 +00:00
"codeberg.org/video-prize-ranch/rimgo/types"
2022-01-18 23:05:06 +00:00
"codeberg.org/video-prize-ranch/rimgo/utils"
2022-01-17 20:23:04 +00:00
"github.com/spf13/viper"
"github.com/tidwall/gjson"
)
func FetchAlbum(albumID string) (types.Album, error) {
// https://api.imgur.com/post/v1/albums/zk7mdKH?client_id=${CLIENT_ID}&include=media%2Caccount
res, err := http.Get("https://api.imgur.com/post/v1/albums/" + albumID + "?client_id=" + viper.GetString("RIMGU_IMGUR_CLIENT_ID") + "&include=media%2Caccount")
if err != nil {
return types.Album{}, err
}
body, err := io.ReadAll(res.Body)
if err != nil {
return types.Album{}, err
}
data := gjson.Parse(string(body))
2022-01-20 21:09:57 +00:00
album := types.Album{}
if data.Get("privacy").String() == "private" {
album, err = ParseAlbum(data)
} else {
album, err = FetchPosts(albumID)
}
return album, err
}
func FetchPosts(albumID string) (types.Album, error) {
res, err := http.Get("https://api.imgur.com/post/v1/posts/" + albumID + "?client_id=" + viper.GetString("RIMGU_IMGUR_CLIENT_ID") + "&include=media%2Caccount")
if err != nil {
return types.Album{}, err
}
body, err := io.ReadAll(res.Body)
if err != nil {
return types.Album{}, err
}
data := gjson.Parse(string(body))
return ParseAlbum(data)
}
func ParseAlbum(data gjson.Result) (types.Album, error) {
2022-01-17 22:11:33 +00:00
media := make([]types.Media, 0)
2022-01-17 20:23:04 +00:00
data.Get("media").ForEach(
func(key gjson.Result, value gjson.Result) bool {
url := value.Get("url").String()
2022-01-17 22:47:33 +00:00
url = strings.ReplaceAll(url, "https://i.imgur.com", "")
2022-01-17 22:11:33 +00:00
media = append(media, types.Media{
2022-01-17 22:34:05 +00:00
Id: value.Get("id").String(),
Name: value.Get("name").String(),
2022-01-18 23:05:06 +00:00
MimeType: value.Get("mime_type").String(),
Type: value.Get("type").String(),
2022-01-17 22:34:05 +00:00
Title: value.Get("metadata.title").String(),
2022-01-17 22:11:33 +00:00
Description: value.Get("metadata.description").String(),
2022-01-17 22:34:05 +00:00
Url: url,
2022-01-17 22:11:33 +00:00
})
2022-01-17 20:23:04 +00:00
return true
},
)
2022-01-18 23:05:06 +00:00
createdAt, err := utils.FormatDate(data.Get("created_at").String())
2022-01-17 20:23:04 +00:00
if err != nil {
return types.Album{}, err
}
return types.Album{
2022-01-17 22:34:05 +00:00
Id: data.Get("id").String(),
Title: data.Get("title").String(),
2022-01-20 21:09:57 +00:00
Privacy: data.Get("privacy").String(),
2022-01-17 22:34:05 +00:00
Views: data.Get("view_count").Int(),
2022-01-18 23:05:06 +00:00
Upvotes: data.Get("upvote_count").Int(),
Downvotes: data.Get("downvote_count").Int(),
Comments: data.Get("comment_count").Int(),
CreatedAt: createdAt,
2022-01-17 22:34:05 +00:00
Media: media,
2022-01-17 20:23:04 +00:00
}, nil
2022-01-20 21:09:57 +00:00
}