lib/azeroth/model.rb in azeroth-0.4.0 vs lib/azeroth/model.rb in azeroth-0.5.0

@@ -5,16 +5,19 @@
   # @author Darthjee
   #
   # Model responsible for making the conection to the resource model class
   class Model
     # @param name [String,Symbol] name of the resource
-    def initialize(name)
+    # @param options [Azeroth::Options] resource options
+    def initialize(name, options)
       if name.is_a?(Class)
         @klass = name
       else
         @name = name.to_s
       end
+
+      @options = options
     end
 
     # Returns the name of the resource represented by the model
     #
     # @return [String]
@@ -44,21 +47,49 @@
     # receive an +#as_json+ call instead
     #
     # @return [Hash]
     def decorate(object)
       decorator_class.new(object).as_json
-    rescue NameError
-      object.as_json
     end
 
     private
 
+    attr_reader :options
+    # @method options
+    # @api private
     # @private
     #
+    # Returns options
+    #
+    # @return [Azeroth::Options]
+
+    # @private
+    #
     # Returns decorator class for the object
     #
     # @return [Class] subclass of {Decorator}
     def decorator_class
-      @decorator_class ||= klass::Decorator
+      @decorator_class ||= calculate_decorator_class
+    end
+
+    # @private
+    #
+    # Calculates decorator class
+    #
+    # When options.decorator is false return DummyDecorator
+    #
+    # when options.decorator is a class, returns it
+    #
+    # When finding the decorator from the model fails,
+    # returns DummyDecorator
+    #
+    # @return [Azeroth::Decorator,DummyDecorator]
+    def calculate_decorator_class
+      return DummyDecorator unless options.decorator
+      return options.decorator if options.decorator.is_a?(Class)
+
+      klass::Decorator
+    rescue NameError
+      DummyDecorator
     end
   end
 end