lib/arstotzka/class_methods.rb in arstotzka-1.1.0 vs lib/arstotzka/class_methods.rb in arstotzka-1.2.0

@@ -2,10 +2,36 @@
 
 module Arstotzka
   # As Arstotzka extends ActiveSupport::Concern, Arstotzka::ClassMethods define
   # methods that will be available when defining a class that includes Arstotka
   module ClassMethods
+    # @api private
+    #
+    # Create builder that will be used to create Fetchers
+    #
+    # @param attribute [Symbol,String] attribute key
+    # @param options [Arstotzka::Options] fetcher options
+    #
+    # @return [Artotzka::FetcherBuilder]
+    def add_fetcher(attribute, options)
+      fetcher_builders[attribute.to_sym] = FetcherBuilder.new(options.merge(key: attribute))
+    end
+
+    # @api private
+    #
+    # Return the fetcher for an attribute and instance
+    #
+    # a new fetcher is built everytime this method is called
+    #
+    # @param attribute [Symbol,String] Name of method that will use this Fetcher
+    # @param instance [Object] instance that will contain the Hash needed by fetcher
+    #
+    # @return [Arstotzka::Fetcher]
+    def fetcher_for(attribute, instance)
+      fetcher_builders[attribute.to_sym].build(instance)
+    end
+
     private
 
     # @api public
     # @!visibility public
     #
@@ -35,15 +61,25 @@
     #   instance.age        # returns 20
     #   instance.cars       # returns 2
     #
     # @return [Array<Sinclair::MethodDefinition>]
     #
-    # @see Builder Arstotzka::Builder
+    # @see MethodBuilder Arstotzka::MethodBuilder
     # @see
     #   https://www.rubydoc.info/gems/activesupport/5.2.2/ActiveSupport/Concern
     #   ActiveSupport::Concern
     def expose(*attr_names, **options_hash)
       options = Options.new(options_hash.symbolize_keys)
-      Builder.new(attr_names, self, options).build
+      MethodBuilder.new(attr_names, self, options).build
+    end
+
+    # @private
+    # @api private
+    #
+    # Map of FetcherBuilders
+    #
+    # @return [Hash<FetcherBuilder>]
+    def fetcher_builders
+      @fetcher_builders ||= {}
     end
   end
 end