Merge pull request #428 from RichiCoder1/feature/Paging

(GH-427) Add paging to ListCommand
chocolatey · Sep 24, 2015 · 9c430c0 · 9c430c0
2 parents 44b1f32 + 4c550a8
commit 9c430c0
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/src/chocolatey/infrastructure.app/commands/ChocolateyListCommand.cs b/src/chocolatey/infrastructure.app/commands/ChocolateyListCommand.cs
@@ -15,6 +15,7 @@
 
 namespace chocolatey.infrastructure.app.commands
 {
+    using System;
     using System.Collections.Generic;
     using System.Linq;
     using attributes;
@@ -61,6 +62,22 @@ public void configure_argument_parser(OptionSet optionSet, ChocolateyConfigurati
                 .Add("p=|password=",
                      "Password - the user's password to the source. Defaults to empty.",
                      option => configuration.SourceCommand.Password = option.remove_surrounding_quotes())
+                .Add("page=",
+                     "Page - the 'page' of results to return. Defaults to return all results.", option =>
+                         {
+                             int page;
+                             if (int.TryParse(option, out page))
+                             {
+                                 configuration.ListCommand.Page = page;
+                             }
+                             else
+                             {
+                                 configuration.ListCommand.Page = null;
+                             }
+                         })
+                .Add("page-size=",
+                     "Page Size - the amount of package results to return per page. Defaults to 25.",
+                     option => configuration.ListCommand.PageSize = int.Parse(option))
                 ;
             //todo exact name
         }
@@ -94,6 +111,7 @@ choco list <filter> [<options/switches>]
     choco list --local-only
     choco list -li
     choco list -lai
+    choco list --page=0 --page-size=25
     choco search git
     choco search git -s ""https://somewhere/out/there""
     choco search bob -s ""https://somewhere/protected"" -u user -p pass

diff --git a/src/chocolatey/infrastructure.app/configuration/ChocolateyConfiguration.cs b/src/chocolatey/infrastructure.app/configuration/ChocolateyConfiguration.cs
@@ -342,9 +342,16 @@ public sealed class FeaturesConfiguration
     [Serializable]
     public sealed class ListCommandConfiguration
     {
+        public ListCommandConfiguration()
+        {
+            PageSize = 25;
+        }
+
         // list
         public bool LocalOnly { get; set; }
         public bool IncludeRegistryPrograms { get; set; }
+        public int? Page { get; set; }
+        public int PageSize { get; set; }
     }
 
     [Serializable]

diff --git a/src/chocolatey/infrastructure.app/nuget/NugetList.cs b/src/chocolatey/infrastructure.app/nuget/NugetList.cs
@@ -43,6 +43,11 @@ public static IEnumerable<IPackage> GetPackages(ChocolateyConfiguration configur
                 results = results.Where(p => p.IsLatestVersion);
             }
 
+            if (configuration.ListCommand.Page.HasValue)
+            {
+                results = results.Skip(configuration.ListCommand.PageSize * configuration.ListCommand.Page.Value).Take(configuration.ListCommand.PageSize);
+            }
+
             return results.OrderBy(p => p.Id)
                         .AsEnumerable()
                         .Where(PackageExtensions.IsListed)