Fixed total number of records in the collection

following this pull request fix:
PubToolBox@f849f21

Getting the total number of records in the collection PubToolBox#71

Author: Chaiavi

src/main/java/com/afrozaar/wordpress/wpapi/v2/Strings.java

@@ -3,6 +3,7 @@
 public interface Strings {
     String HEADER_TOTAL_PAGES = "X-WP-TotalPages";
     String HEADER_LINK = "Link";
+    String HEADER_TOTAL = "x-wp-total";
     String NEXT = "next";
     String PREV = "prev";
 }

src/main/java/com/afrozaar/wordpress/wpapi/v2/response/PagedResponse.java

@@ -30,14 +30,33 @@ public class PagedResponse<T> {
     final String previous;
     final int pages;
     final List<T> list;
-
-    public PagedResponse(Class<T> clazz, String self, String next, String previous, int pages, List<T> list) {
+    final int total;
+
+//    public PagedResponse(Class<T> clazz, String self, String next, String previous, int pages, List<T> list) {
+//        this.clazz = clazz;
+//        this.self = self;
+//        this.next = next;
+//        this.previous = previous;
+//        this.pages = pages;
+//        this.list = list;
+//    }
+
+    public PagedResponse(Class<T> clazz, String self, String next, String previous, int pages, List<T> list, int total) {
         this.clazz = clazz;
         this.self = self;
         this.next = next;
         this.previous = previous;
         this.pages = pages;
         this.list = list;
+        this.total = total;
+    }
+
+    public int getPages() {
+        return pages;
+    }
+
+    public int getTotal() {
+        return total;
     }
 
     public boolean hasNext() {
@@ -88,6 +107,7 @@ public static class Builder<BT> {
         private List<BT> posts;
         private int pages;
         private Class<BT> t1;
+        private int total;
 
         private Builder(Class<BT> t1) {
             this.t1 = t1;
@@ -118,21 +138,29 @@ public Builder<BT> withPosts(List<BT> posts) {
         }
 
         public PagedResponse<BT> build() {
-            return new PagedResponse<>(t1, self, next, previous, pages, posts);
+            return new PagedResponse<>(t1, self, next, previous, pages, posts, total);
         }
 
         public Builder<BT> withPages(int pages) {
             this.pages = pages;
             return this;
         }
 
+        public Builder<BT> withPages(int pages, int total) {
+            this.pages = pages;
+            this.total = total;
+            return this;
+        }
+
         public Builder<BT> withPages(HttpHeaders headers) {
             Optional<String> totalPages = headers.keySet().stream().filter(x -> Strings.HEADER_TOTAL_PAGES.compareToIgnoreCase(x) == 0).findFirst();
-            LOG.debug("found pages {} from headers {}", totalPages, headers);
+            Optional<String> totalOptional = headers.keySet().stream().filter(x -> Strings.HEADER_TOTAL.compareToIgnoreCase(x) == 0).findFirst();
+            LOG.debug("found pages {}, total {} from headers {}", totalPages, totalOptional, headers);
             Stream<String> totalPageHeader = totalPages.map(x -> headers.get(x)).map(x -> x.stream()).orElse(Stream.empty());
-            totalPageHeader.findFirst().ifPresent(pages -> Builder.this.withPages(Integer.valueOf(pages)));
+            int page = Integer.valueOf(totalPageHeader.findFirst().get());
+            Stream<String> totalHeader = totalOptional.map(x -> headers.get(x)).map(x -> x.stream()).orElse(Stream.empty());
+            totalHeader.findFirst().ifPresent(total -> Builder.this.withPages(page, Integer.valueOf(total)));
             return this;
         }
-
     }
 }