package builder import ( "context" "git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints" "git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model" "git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils" "strconv" ) // PostsBuilder represents a builder for constructing queries to retrieve posts. type PostsBuilder interface { // Tags sets the query to search for posts with specific tags. Tags(tags string) PostsBuilder // PageAfter sets the query to retrieve posts after a specific post ID. PageAfter(postID model.PostID) PostsBuilder // PageBefore sets the query to retrieve posts before a specific post ID. PageBefore(postID model.PostID) PostsBuilder // PageNumber sets the query to retrieve posts on a specific page number. PageNumber(number int) PostsBuilder // SetLimit sets the maximum number of posts to retrieve. SetLimit(limitUser int) PostsBuilder // Execute sends the constructed query and returns a slice of posts and an error, if any. Execute() ([]model.Post, error) } // NewGetPostsBuilder creates a new PostsBuilder with the provided RequestContext. func NewGetPostsBuilder(requestContext model.RequestContext) PostsBuilder { postBuilder := &getPosts{ requestContext: requestContext, query: make(map[string]string), } return postBuilder.SetLimit(utils.E621_MAX_POST_COUNT) } type getPosts struct { requestContext model.RequestContext query map[string]string } // Tags sets the query to search for posts with specific tags. func (g *getPosts) Tags(tags string) PostsBuilder { g.query["tags"] = tags return g } // PageAfter sets the query to retrieve posts after a specific post ID. func (g *getPosts) PageAfter(postID model.PostID) PostsBuilder { g.query["page"] = "a" + strconv.Itoa(int(postID)) return g } // PageBefore sets the query to retrieve posts before a specific post ID. func (g *getPosts) PageBefore(postID model.PostID) PostsBuilder { g.query["page"] = "b" + strconv.Itoa(int(postID)) return g } // PageNumber sets the query to retrieve posts on a specific page number. func (g *getPosts) PageNumber(number int) PostsBuilder { g.query["page"] = strconv.Itoa(number) return g } // SetLimit sets the maximum number of posts to retrieve. func (g *getPosts) SetLimit(limitUser int) PostsBuilder { g.query["limit"] = strconv.Itoa(limitUser) return g } // Execute sends the constructed query and returns a slice of posts and an error, if any. func (g *getPosts) Execute() ([]model.Post, error) { if g.requestContext.RateLimiter != nil { err := g.requestContext.RateLimiter.Wait(context.Background()) if err != nil { return nil, err } } posts, err := endpoints.GetPosts(g.requestContext, g.query) if err != nil { return nil, err } return posts, err }