inital commit

gallery.go

@@ -0,0 +1,76 @@
+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) iter.Seq2[*Submission, error] {
+	return c.listGallerySection(ctx, name, urls.Gallery, opts)
+}
+
+// Scraps iterates the user's scraps folder. Same yield shape as Gallery.
+func (c *Client) Scraps(ctx context.Context, name string, opts ListOptions) iter.Seq2[*Submission, error] {
+	return c.listGallerySection(ctx, name, urls.Scraps, opts)
+}
+
+// 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) iter.Seq2[*Submission, error] {
+	return c.listGallerySection(ctx, name, urls.Favorites, opts)
+}
+
+// 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,
+) 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
+			})
+			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++
+		}
+	}
+}