package api import ( "io" "net/http" "strings" "sync" "time" "codeberg.org/rimgo/rimgo/utils" "github.com/patrickmn/go-cache" "github.com/tidwall/gjson" ) type User struct { Id int64 Bio string Username string Points int64 Cover string Avatar string CreatedAt string } type Submission struct { Id string Title string Link string Cover Media Points int64 Upvotes int64 Downvotes int64 Comments int64 Views int64 IsAlbum bool } func (client *Client) FetchUser(username string) (User, error) { cacheData, found := client.Cache.Get(username + "-user") if found { return cacheData.(User), nil } res, err := http.Get("https://api.imgur.com/account/v1/accounts/" + username + "?client_id=" + utils.Config.ImgurId) if err != nil { return User{}, err } body, err := io.ReadAll(res.Body) if err != nil { return User{}, err } data := gjson.Parse(string(body)) createdTime, _ := time.Parse(time.RFC3339, data.Get("created_at").String()) user := User{ Id: data.Get("id").Int(), Bio: data.Get("bio").String(), Username: data.Get("username").String(), Points: data.Get("reputation_count").Int(), Cover: strings.ReplaceAll(data.Get("cover_url").String(), "https://imgur.com", ""), Avatar: strings.ReplaceAll(data.Get("avatar_url").String(), "https://i.imgur.com", ""), CreatedAt: createdTime.Format("January 2, 2006"), } client.Cache.Set(username+"-user", user, 1*time.Hour) return user, nil } func (client *Client) FetchSubmissions(username string, sort string, page string) ([]Submission, error) { cacheData, found := client.Cache.Get(username + "-submissions") if found { return cacheData.([]Submission), nil } data, err := utils.GetJSON("https://api.imgur.com/3/account/" + username + "/submissions/" + page + "/" + sort + "?album_previews=1&client_id=" + utils.Config.ImgurId) if err != nil { return []Submission{}, err } submissions := []Submission{} wg := sync.WaitGroup{} data.Get("data").ForEach( func(key, value gjson.Result) bool { wg.Add(1) go func() { defer wg.Done() submissions = append(submissions, parseSubmission(value)) }() return true }, ) wg.Wait() client.Cache.Set(username+"-submissions", submissions, 15*time.Minute) return submissions, nil } func (client *Client) FetchUserComments(username string) ([]Comment, error) { cacheData, found := client.Cache.Get(username + "-usercomments") if found { return cacheData.([]Comment), nil } req, err := http.NewRequest("GET", "https://api.imgur.com/comment/v1/comments", nil) if err != nil { return []Comment{}, err } utils.SetReqHeaders(req) q := req.URL.Query() q.Add("client_id", client.ClientID) q.Add("filter[account]", "eq:"+username) q.Add("include", "account,post") q.Add("sort", "new") req.URL.RawQuery = q.Encode() res, err := http.DefaultClient.Do(req) if err != nil { return []Comment{}, err } body, err := io.ReadAll(res.Body) if err != nil { return []Comment{}, err } data := gjson.Parse(string(body)) comments := make([]Comment, 0) data.Get("data").ForEach( func(key, value gjson.Result) bool { comments = append(comments, parseComment(value)) return true }, ) client.Cache.Set(username+"-usercomments", comments, cache.DefaultExpiration) return comments, nil } func parseSubmission(value gjson.Result) Submission { var cover Media c := value.Get("cover") coverData := value.Get("images.#(id==\"" + c.String() + "\")") switch { case c.Type == gjson.String && coverData.Exists(): cover = Media{ Id: coverData.Get("id").String(), Description: coverData.Get("description").String(), Type: strings.Split(coverData.Get("type").String(), "/")[0], Url: strings.ReplaceAll(coverData.Get("link").String(), "https://i.imgur.com", ""), } // This case is when fetching comments case c.Type != gjson.Null: cover = Media{ Id: c.Get("id").String(), Url: strings.ReplaceAll(c.Get("url").String(), "https://i.imgur.com", ""), } // Replace with thumbnails here because it's easier. if strings.HasSuffix(cover.Url, ".mp4") { cover.Url = cover.Url[:len(cover.Url)-3] + "webp" } default: cover = Media{ Id: value.Get("id").String(), Description: value.Get("description").String(), Type: strings.Split(value.Get("type").String(), "/")[0], Url: strings.ReplaceAll(value.Get("link").String(), "https://i.imgur.com", ""), } } id := value.Get("id").String() link := "/a/" + id if value.Get("in_gallery").Bool() { link = "/gallery/" + id } return Submission{ Id: id, Link: link, Title: value.Get("title").String(), Cover: cover, Points: value.Get("points").Int(), Upvotes: value.Get("ups").Int(), Downvotes: value.Get("downs").Int(), Comments: value.Get("comment_count").Int(), Views: value.Get("views").Int(), IsAlbum: value.Get("is_album").Bool(), } }