80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package builder
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"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"
|
|
)
|
|
|
|
type DMailsBuilder interface {
|
|
ByTitle(title string) DMailsBuilder
|
|
ByBody(body string) DMailsBuilder
|
|
ByToName(toName string) DMailsBuilder
|
|
ByFromName(fromName string) DMailsBuilder
|
|
SetLimit(limit int) DMailsBuilder
|
|
PageNumber(number int) DMailsBuilder
|
|
Execute() ([]model.DMail, error)
|
|
}
|
|
|
|
type getDMails struct {
|
|
requestContext model.RequestContext
|
|
query map[string]string
|
|
id int
|
|
}
|
|
|
|
func NewGetDMailsBuilder(requestContext model.RequestContext) DMailsBuilder {
|
|
dMailsBuilder := &getDMails{
|
|
requestContext: requestContext,
|
|
query: make(map[string]string),
|
|
}
|
|
|
|
return dMailsBuilder.SetLimit(utils.E621_MAX_POST_COUNT)
|
|
}
|
|
|
|
func (g getDMails) ByTitle(title string) DMailsBuilder {
|
|
g.query["search[title_matches]"] = title
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) ByBody(body string) DMailsBuilder {
|
|
g.query["search[message_matches]"] = body
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) ByToName(toName string) DMailsBuilder {
|
|
g.query["search[to_name]"] = toName
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) ByFromName(fromName string) DMailsBuilder {
|
|
g.query["search[from_name]"] = fromName
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) SetLimit(limit int) DMailsBuilder {
|
|
g.query["limit"] = strconv.Itoa(limit)
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) PageNumber(number int) DMailsBuilder {
|
|
g.query["page"] = strconv.Itoa(number)
|
|
return g
|
|
}
|
|
|
|
func (g getDMails) Execute() ([]model.DMail, error) {
|
|
if g.requestContext.RateLimiter != nil {
|
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
favorites, err := endpoints.GetAllDmails(g.requestContext, g.query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return favorites, nil
|
|
}
|