This repository has been archived on 2024-07-22. You can view files and clone it, but cannot push or open issues or pull requests.
e621-sdk-go/pkg/e621/client.go
SoXX 06ba87d1f2 feat: added getting n & all favs & some fixes
Signed-off-by: SoXX <soxx@fenpa.ws>
2023-11-06 15:26:47 +01:00

131 lines
3.4 KiB
Go

package e621
import (
"fmt"
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/builder"
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
_ "github.com/joho/godotenv/autoload"
"golang.org/x/time/rate"
"math"
"net/http"
"strconv"
)
type Client struct {
RequestContext model.RequestContext
}
func NewE621Client(username string, apikey string) Client {
return Client{
RequestContext: model.RequestContext{
Client: http.Client{},
RateLimiter: rate.NewLimiter(1, 2),
Host: "https://e621.net",
UserAgent: fmt.Sprintf("Go-e621-SDK used by %s | (made by the Anthrove Team)", username),
Username: username,
APIKey: apikey,
},
}
}
func (c *Client) SetHost(host string) *Client {
c.RequestContext.Host = host
return c
}
func (c *Client) SetAgentName(userAgent string) *Client {
c.RequestContext.UserAgent = userAgent
return c
}
func (c *Client) GetUserByName(username string) builder.UserBuilder {
return builder.NewGetUserBuilder(c.RequestContext).SetUsername(username)
}
func (c *Client) GetUserByID(userID model.UserID) builder.UserBuilder {
return builder.NewGetUserBuilder(c.RequestContext).SetUsername(strconv.FormatInt(int64(userID), 10))
}
func (c *Client) GetFavoritesForUser(username string) (builder.FavoritesBuilder, error) {
user, err := builder.NewGetUserBuilder(c.RequestContext).SetUsername(username).Execute()
if err != nil {
return nil, err
}
favoritesBuilder := builder.NewGetFavoritesBuilder(c.RequestContext).SetUserID(user.ID)
return favoritesBuilder, nil
}
func (c *Client) GetNFavoritesForUser(n int, favoriteBuilder builder.FavoritesBuilder) ([]model.Post, error) {
if n < utils.E621_MAX_POST_COUNT {
favoriteBuilder.SetLimit(n)
}
var favorites []model.Post
var page = 1
for len(favorites) < n {
favoriteBuilder.Page(page)
favoriteBuilder.SetLimit(n - len(favorites))
newFavorites, err := favoriteBuilder.Execute()
if err != nil {
return nil, err
}
if len(newFavorites) == 0 {
break
}
favorites = append(favorites, newFavorites...)
page = page + 1
}
return favorites, nil
}
func (c *Client) GetAllFavoritesForUser(favoriteBuilder builder.FavoritesBuilder) ([]model.Post, error) {
return c.GetNFavoritesForUser(math.MaxInt, favoriteBuilder)
}
func (c *Client) GetFavoritesForUserWithTags(username string, tags string) builder.PostsBuilder {
favoritesBuilder := builder.NewGetPostsBuilder(c.RequestContext).Tags(fmt.Sprintf("fav:%s %s", username, tags))
return favoritesBuilder
}
func (c *Client) GetPostByID(id model.PostID) builder.PostBuilder {
return builder.NewGetPostBuilder(c.RequestContext).SetPostID(id)
}
func (c *Client) GetPosts() builder.PostsBuilder {
return builder.NewGetPostsBuilder(c.RequestContext)
}
func (c *Client) GetNPosts(n int, postBuilder builder.PostsBuilder) ([]model.Post, error) {
if n < utils.E621_MAX_POST_COUNT {
postBuilder.SetLimit(n)
}
posts, err := postBuilder.Execute()
if err != nil {
return nil, err
}
for len(posts) < n {
lastPostID := posts[len(posts)-1].ID
postBuilder.PageBefore(lastPostID)
postBuilder.SetLimit(n - len(posts))
newPosts, err := postBuilder.Execute()
if err != nil {
return nil, err
}
if len(newPosts) == 0 {
break
}
posts = append(posts, newPosts...)
}
return posts, nil
}
func (c *Client) GetAllPosts(postBuilder builder.PostsBuilder) ([]model.Post, error) {
return c.GetNPosts(math.MaxInt, postBuilder)
}