From 9fd34aec7fe9fa1359418c92b0a14526f837a4a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Thieriot?= <a.thieriot@gmail.com>
Date: Sun, 12 Aug 2012 12:25:14 +0200
Subject: [PATCH] Paging GHPullRequests getter and allow to transform iterator
 to a list

---
 .../java/org/kohsuke/github/GHOrganization.java |  2 +-
 .../java/org/kohsuke/github/GHRepository.java   | 17 ++++++++++++-----
 .../java/org/kohsuke/github/PagedIterator.java  | 15 ++++++++++++++-
 src/test/java/org/kohsuke/AppTest.java          | 13 +++++++++++++
 4 files changed, 40 insertions(+), 7 deletions(-)

diff --git a/src/main/java/org/kohsuke/github/GHOrganization.java b/src/main/java/org/kohsuke/github/GHOrganization.java
index f15626370c..884e165b32 100644
--- a/src/main/java/org/kohsuke/github/GHOrganization.java
+++ b/src/main/java/org/kohsuke/github/GHOrganization.java
@@ -129,7 +129,7 @@ public List<GHRepository> getRepositoriesWithOpenPullRequests() throws IOExcepti
     public List<GHPullRequest> getPullRequests() throws IOException {
         List<GHPullRequest> all = new ArrayList<GHPullRequest>();
         for (GHRepository r : getRepositoriesWithOpenPullRequests()) {
-            all.addAll(r.getPullRequests(GHIssueState.OPEN));
+            all.addAll(r.getPullRequests(GHIssueState.OPEN).iterator().asList());
         }
         return all;
     }
diff --git a/src/main/java/org/kohsuke/github/GHRepository.java b/src/main/java/org/kohsuke/github/GHRepository.java
index 6ff0553580..811209656c 100644
--- a/src/main/java/org/kohsuke/github/GHRepository.java
+++ b/src/main/java/org/kohsuke/github/GHRepository.java
@@ -359,11 +359,18 @@ public GHPullRequest getPullRequest(int i) throws IOException {
     /**
      * Retrieves all the pull requests of a particular state.
      */
-    public List<GHPullRequest> getPullRequests(GHIssueState state) throws IOException {
-        GHPullRequest[] r = root.retrieveWithAuth("/repos/" + owner.login + '/' + name + "/pulls?state=" + state.name().toLowerCase(Locale.ENGLISH), GHPullRequest[].class);
-        for (GHPullRequest p : r)
-            p.wrapUp(this);
-        return new ArrayList<GHPullRequest>(Arrays.asList(r));
+    public PagedIterable<GHPullRequest> getPullRequests(final GHIssueState state) {
+        return new PagedIterable<GHPullRequest>() {
+            public PagedIterator<GHPullRequest> iterator() {
+                return new PagedIterator<GHPullRequest>(root.retrievePaged(String.format("/repos/%s/%s/pulls?state=%s", owner.login,name,state.name().toLowerCase(Locale.ENGLISH)), GHPullRequest[].class, false)) {
+                    @Override
+                    protected void wrapUp(GHPullRequest[] page) {
+                        for (GHPullRequest pr : page)
+                            pr.wrap(GHRepository.this);
+                    }
+                };
+            };
+        };
     }
 
     /**
diff --git a/src/main/java/org/kohsuke/github/PagedIterator.java b/src/main/java/org/kohsuke/github/PagedIterator.java
index 74c197aa2e..60cffa9019 100644
--- a/src/main/java/org/kohsuke/github/PagedIterator.java
+++ b/src/main/java/org/kohsuke/github/PagedIterator.java
@@ -1,5 +1,6 @@
 package org.kohsuke.github;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Iterator;
 import java.util.List;
@@ -8,7 +9,8 @@
  * Iterator over a pagenated data source.
  *
  * Aside from the normal iterator operation, this method exposes {@link #nextPage()}
- * that allows the caller to retrieve items per page.
+ * that allows the caller to retrieve items per page and {@link #asList()}
+ * that allows the caller to retrieve all items at once.
  *
  * @author Kohsuke Kawaguchi
  */
@@ -61,4 +63,15 @@ public List<T> nextPage() {
         pos = 0;
         return r;
     }
+
+    /**
+     * Gets a list of all items
+     */
+    public List<T> asList() {
+        List<T> r = new ArrayList<T>();
+        for(Iterator i = this; i.hasNext();) {
+            r.addAll(nextPage());
+        }
+        return r;
+    }
 }
diff --git a/src/test/java/org/kohsuke/AppTest.java b/src/test/java/org/kohsuke/AppTest.java
index 120ef0e7c1..b79ce584f2 100644
--- a/src/test/java/org/kohsuke/AppTest.java
+++ b/src/test/java/org/kohsuke/AppTest.java
@@ -14,11 +14,13 @@
 import org.kohsuke.github.GHMyself;
 import org.kohsuke.github.GHOrganization;
 import org.kohsuke.github.GHOrganization.Permission;
+import org.kohsuke.github.GHPullRequest;
 import org.kohsuke.github.GHRepository;
 import org.kohsuke.github.GHTeam;
 import org.kohsuke.github.GHUser;
 import org.kohsuke.github.GitHub;
 import org.kohsuke.github.PagedIterable;
+import org.kohsuke.github.PagedIterator;
 
 import java.io.IOException;
 import java.net.URL;
@@ -64,6 +66,17 @@ public void testFetchPullRequest() throws Exception {
         r.getPullRequests(GHIssueState.OPEN);
     }
 
+    public void testFetchPullRequestAsList() throws Exception {
+        GitHub gh = GitHub.connect();
+        GHRepository r = gh.getOrganization("symfony").getRepository("symfony-docs");
+        assertEquals("master", r.getMasterBranch());
+        PagedIterator<GHPullRequest> i = r.getPullRequests(GHIssueState.CLOSED).iterator();
+        List<GHPullRequest> prs = i.asList();
+        assertNotNull(prs);
+        assertTrue(prs.size() > 0);
+        assertFalse(i.hasNext());
+    }
+
     public void testRepoPermissions() throws Exception {
         GitHub gh = GitHub.connect();
         GHRepository r = gh.getOrganization("jenkinsci").getRepository("jenkins");