e621-sdk-go/pkg/e621/builder/dmail.go

package builder

import (
	"context"
	"strconv"

	"git.anthrove.art/Anthrove/e621-sdk-go/pkg/e621/endpoints"
	"git.anthrove.art/Anthrove/e621-sdk-go/pkg/e621/model"
	"git.anthrove.art/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
}