131 lines
3.4 KiB
Go
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)
|
|
}
|