78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package fa
|
|
|
|
import (
|
|
"context"
|
|
"iter"
|
|
|
|
"github.com/PuerkitoBio/goquery"
|
|
|
|
"git.anthrove.art/public/go-fa-api/internal/urls"
|
|
)
|
|
|
|
// Gallery iterates the submissions in a user's main gallery, newest first.
|
|
//
|
|
// Each yielded *Submission carries only the fields visible on the listing
|
|
// page: ID, Title, Author (for favorites), ThumbURL, and Rating. Call
|
|
// [Client.GetSubmission] with the ID to load the full record.
|
|
func (c *Client) Gallery(ctx context.Context, name string, opts ListOptions, reqOpts ...Option) iter.Seq2[*Submission, error] {
|
|
return c.listGallerySection(ctx, name, urls.Gallery, opts, reqOpts)
|
|
}
|
|
|
|
// Scraps iterates the user's scraps folder. Same yield shape as Gallery.
|
|
func (c *Client) Scraps(ctx context.Context, name string, opts ListOptions, reqOpts ...Option) iter.Seq2[*Submission, error] {
|
|
return c.listGallerySection(ctx, name, urls.Scraps, opts, reqOpts)
|
|
}
|
|
|
|
// Favorites iterates the user's favorited submissions. The yielded
|
|
// *Submission's Author field reflects the original artist (not the user
|
|
// whose favorites we are walking).
|
|
func (c *Client) Favorites(ctx context.Context, name string, opts ListOptions, reqOpts ...Option) iter.Seq2[*Submission, error] {
|
|
return c.listGallerySection(ctx, name, urls.Favorites, opts, reqOpts)
|
|
}
|
|
|
|
// listGallerySection is the shared engine for Gallery / Scraps / Favorites.
|
|
// urlFn picks the section-specific URL builder; the rest of the pagination
|
|
// machinery is identical across all three sections.
|
|
func (c *Client) listGallerySection(
|
|
ctx context.Context,
|
|
name string,
|
|
urlFn func(string, int) string,
|
|
opts ListOptions,
|
|
reqOpts []Option,
|
|
) iter.Seq2[*Submission, error] {
|
|
return func(yield func(*Submission, error) bool) {
|
|
page := opts.firstPage()
|
|
pagesFetched := 0
|
|
for {
|
|
if opts.reachedLimit(pagesFetched) {
|
|
return
|
|
}
|
|
var (
|
|
items []*Submission
|
|
hasNext bool
|
|
)
|
|
err := c.fetch(ctx, urlFn(name, page), func(doc *goquery.Document) error {
|
|
items, hasNext = parseGalleryPage(doc, c.cfg.jsonListings)
|
|
return nil
|
|
}, reqOpts...)
|
|
if err != nil {
|
|
yield(nil, err)
|
|
return
|
|
}
|
|
pagesFetched++
|
|
if len(items) == 0 {
|
|
return
|
|
}
|
|
for _, s := range items {
|
|
if !yield(s, nil) {
|
|
return
|
|
}
|
|
}
|
|
if !hasNext {
|
|
return
|
|
}
|
|
page++
|
|
}
|
|
}
|
|
}
|