Make pagination great again

Author: Casao

lib/cortex/result.rb

@@ -1,13 +1,12 @@
 require 'hashie'
 module Cortex
   class Result
-    attr_reader :raw_headers, :contents, :total_items, :page, :per_page, :errors, :range_start, :range_end, :range, :status
+    attr_reader :raw_headers, :contents, :total_items, :page, :per_page, :errors, :range_start, :range_end, :range, :status, :total_pages, :next_page, :prev_page
 
     def initialize(body, headers, status)
       @contents = parse(body)
       @raw_headers = headers
       @status = status
-      @total_items = headers['x-total-items'] unless headers['x-total-items'].nil?
       parse_headers(headers)
       @errors = find_errors
     end
@@ -20,15 +19,16 @@ def is_error?
 
     def parse_headers(headers)
       if headers['X-Total']
+        @total_items = headers['X-Total']
         @count = headers['X-Total'].to_i
-      end
-      if headers['X-Total']
         @page = headers['X-Page'].to_i
         @per_page = headers['X-Per-Page'].to_i
         @range_start = (@page-1) * @per_page
         @range_end = @per_page * @page - 1
         @range = "#{@range_start}-#{@range_end}"
-
+        @total_pages = headers['X-Total-Pages']
+        @next_page = headers['X-Next-Page']
+        @prev_page = headers['X-Prev-Page']
       end
     end