feat: library pagination

Signed-off-by: ygelfand <yuri@shlitz.com>

Author: ygelfand

cmd/library.go

@@ -14,6 +14,12 @@ import (
 	"github.com/ygelfand/plexctl/internal/ui"
 )
 
+var (
+	libraryCount int
+	libraryPage  int
+	libraryAll   bool
+)
+
 var libraryCmd = &cobra.Command{
 	Use:     "library",
 	Short:   "Manage libraries",
@@ -52,25 +58,21 @@ var libraryShowCmd = &cobra.Command{
 		libraryID := args[0]
 		slog.Debug("SDK: Fetching library content", "library_id", libraryID)
 
-		res, err := client.SDK.Content.ListContent(ctx, operations.ListContentRequest{
-			SectionID: libraryID,
-		})
+		allMetadata, err := plex.WalkContent(ctx, libraryAll, libraryPage, libraryCount, plex.LibraryWalker(client, libraryID))
 		if err != nil {
-			slog.Error("SDK: Failed to get library items", "library_id", libraryID, "error", err)
-			return fmt.Errorf("failed to get library items: %w", err)
+			return err
 		}
 
-		if res.MediaContainerWithMetadata == nil || res.MediaContainerWithMetadata.MediaContainer == nil || len(res.MediaContainerWithMetadata.MediaContainer.Metadata) == 0 {
-			slog.Debug("SDK: No items found", "library_id", libraryID)
+		if len(allMetadata) == 0 {
 			fmt.Println("No items found in this library.")
 			return nil
 		}
 
-		slog.Debug("SDK: Found items", "library_id", libraryID, "count", len(res.MediaContainerWithMetadata.MediaContainer.Metadata))
+		slog.Debug("SDK: Found items", "library_id", libraryID, "count", len(allMetadata))
 		return commands.Print(&presenters.LibraryItemsPresenter{
 			SectionID: libraryID,
-			Items:     presenters.MapMetadata(res.MediaContainerWithMetadata.MediaContainer.Metadata),
-			RawData:   res.MediaContainerWithMetadata.MediaContainer.Metadata,
+			Items:     presenters.MapMetadata(allMetadata),
+			RawData:   allMetadata,
 		}, opts)
 	}),
 }
@@ -104,4 +106,8 @@ func init() {
 	libraryCmd.AddCommand(libraryListCmd)
 	libraryCmd.AddCommand(libraryShowCmd)
 	libraryCmd.AddCommand(libraryRefreshCmd)
+
+	libraryShowCmd.Flags().IntVar(&libraryCount, "count", 50, "Number of items to return per page")
+	libraryShowCmd.Flags().IntVar(&libraryPage, "page", 1, "Page number to return")
+	libraryShowCmd.Flags().BoolVar(&libraryAll, "all", false, "Return all items (overrides count/page)")
 }

internal/commands/options.go

@@ -5,4 +5,7 @@ type PlexCtlOptions struct {
 	OutputFormat string
 	Verbosity    int
 	Sort         string
+	Count        int
+	Page         int
+	All          bool
 }

internal/commands/wrapper.go

@@ -17,13 +17,17 @@ import (
 // RunnerFunc defines the signature for a command handler that receives a Plex client
 type RunnerFunc func(ctx context.Context, client *plex.Client, cmd *cobra.Command, args []string, opts *PlexCtlOptions) error
 
-// RunWithClient wraps a cobra command RunE function to inject a configured Plex client
+// RunWithClient wraps a cobra command RunE function to inject a configured Plex client.
+// This is for commands that only need a token (like login or discovery).
 func RunWithClient(runner RunnerFunc) func(cmd *cobra.Command, args []string) error {
 	return func(cmd *cobra.Command, args []string) error {
 		opts := &PlexCtlOptions{
 			OutputFormat: viper.GetString("output"),
 			Verbosity:    viper.GetInt("verbose"),
 			Sort:         viper.GetString("sort"),
+			Count:        viper.GetInt("count"),
+			Page:         viper.GetInt("page"),
+			All:          viper.GetBool("all"),
 		}
 
 		client, err := plex.NewClient()
@@ -34,22 +38,10 @@ func RunWithClient(runner RunnerFunc) func(cmd *cobra.Command, args []string) er
 	}
 }
 
-// RunWithServer wraps a cobra command RunE function to inject a configured Plex client
-// and ensures that a default server has been selected.
+// RunWithServer wraps RunWithClient but ensures a default server is selected first.
+// This is for commands that interact with specific media or server library sections.
 func RunWithServer(runner RunnerFunc) func(cmd *cobra.Command, args []string) error {
-	return func(cmd *cobra.Command, args []string) error {
-		opts := &PlexCtlOptions{
-			OutputFormat: viper.GetString("output"),
-			Verbosity:    viper.GetInt("verbose"),
-			Sort:         viper.GetString("sort"),
-		}
-
-		client, err := plex.NewClient()
-		if err != nil {
-			return err
-		}
-		return runner(cmd.Context(), client, cmd, args, opts)
-	}
+	return RunWithClient(runner)
 }
 
 // EnsureActiveServer checks if a server is configured and triggers discovery if not

internal/plex/pagination.go

@@ -0,0 +1,76 @@
+package plex
+
+import (
+	"context"
+
+	"github.com/LukeHagar/plexgo/models/components"
+	"github.com/LukeHagar/plexgo/models/operations"
+	"github.com/ygelfand/plexctl/internal/ui"
+)
+
+// ContentWalker is a function that fetches a single page of results
+type ContentWalker func(ctx context.Context, start, size int) ([]components.Metadata, int64, error)
+
+// WalkContent handles either fetching a single page or walking all pages based on the provided options
+func WalkContent(ctx context.Context, all bool, page, count int, walker ContentWalker) ([]components.Metadata, error) {
+	var allMetadata []components.Metadata
+	start := 0
+	size := count
+
+	if all {
+		size = 100 // doesn't seem to make much of a difference how many per page once it gets here
+	} else {
+		start = (page - 1) * count
+	}
+
+	for {
+		metadata, totalSize, err := walker(ctx, start, size)
+		if err != nil {
+			return nil, err
+		}
+
+		allMetadata = append(allMetadata, metadata...)
+
+		if !all {
+			break
+		}
+
+		start += len(metadata)
+		if totalSize > 0 && int64(start) >= totalSize {
+			break
+		}
+		if len(metadata) == 0 {
+			break
+		}
+	}
+
+	return allMetadata, nil
+}
+
+// LibraryWalker returns a ContentWalker for a specific library section
+func LibraryWalker(client *Client, sectionID string) ContentWalker {
+	return func(ctx context.Context, start, size int) ([]components.Metadata, int64, error) {
+		req := operations.ListContentRequest{
+			SectionID:           sectionID,
+			XPlexContainerStart: ui.Ptr(start),
+			XPlexContainerSize:  ui.Ptr(size),
+		}
+
+		res, err := client.SDK.Content.ListContent(ctx, req)
+		if err != nil {
+			return nil, 0, err
+		}
+
+		if res.MediaContainerWithMetadata == nil || res.MediaContainerWithMetadata.MediaContainer == nil {
+			return nil, 0, nil
+		}
+
+		mc := res.MediaContainerWithMetadata.MediaContainer
+		total := int64(0)
+		if mc.TotalSize != nil {
+			total = *mc.TotalSize
+		}
+
+		return mc.Metadata, total, nil
+	}
+}