lib/alma/loan_set.rb in alma-0.2.8 vs lib/alma/loan_set.rb in alma-0.3.1

@@ -1,34 +1,76 @@
+# frozen_string_literal: true
+
 module Alma
-  class LoanSet
-    extend Forwardable
-    include Enumerable
-    #include Alma::Error
+  class LoanSet < ResultSet
+    class ResponseError < Alma::StandardError
+    end
 
-    attr_reader :response
-    def_delegators :response, :[], :fetch
+    alias :total_records :total_record_count
 
-    def initialize(response_body_hash)
-      @response = response_body_hash
+
+    attr_reader :results, :raw_response
+    def_delegators :results, :empty?
+
+    def initialize(raw_response, search_args={})
+      @raw_response = raw_response
+      @response = raw_response.parsed_response
+      @search_args = search_args
+      validate(raw_response)
+      @results = @response.fetch(key, [])
+        .map { |item| single_record_class.new(item) }
+      # args passed to the search that returned this set
+      # such as limit, expand, order_by, etc
     end
 
-    def each
-      @response.fetch(key, []).map{|item| Alma::Loan.new(item)}
+    def loggable
+      { search_args: @search_args,
+        uri: @raw_response&.request&.uri.to_s
+      }.select { |k, v| !(v.nil? || v.empty?) }
     end
-    alias list each
 
-    def size
-      each.count
+    def validate(response)
+      if response.code != 200
+        error = "Could not find loans info."
+        log = loggable.merge(response.parsed_response)
+        raise ResponseError.new(error, log)
+      end
     end
 
+    def all
+      Enumerator.new do |yielder|
+        offset = 0
+        loop do
+          extra_args = @search_args.merge({limit: 100, offset: offset})
+          r = (offset == 0) ? self : single_record_class.where_user(user_id, extra_args)
+          unless r.empty?
+            r.map { |item| yielder << item }
+            offset += 100
+          else
+            raise StopIteration
+          end
+        end
+      end
+    end
 
+    def each(&block)
+       @results.each(&block)
+    end
+
+    def success?
+      raw_response.response.code.to_s == "200"
+    end
+
     def key
-      'item_loan'
+      "item_loan"
     end
 
-    def total_record_count
-      fetch('total_record_count', 0)
+    def single_record_class
+      Alma::Loan
     end
-    alias :total_records :total_record_count
 
+    private
+      def user_id
+        @user_id ||= results.first.user_id
+      end
   end
 end