alba 0.13.0 → 1.2.0

data/lib/alba.rb

@@ -1,5 +1,4 @@
 require_relative 'alba/version'
-require_relative 'alba/serializer'
 require_relative 'alba/resource'
 
 # Core module
@@ -10,9 +9,11 @@ module Alba
   # Error class for backend which is not supported
   class UnsupportedBackend < Error; end
 
+  # Error class for type which is not supported
+  class UnsupportedType < Error; end
+
   class << self
-    attr_reader :backend, :encoder
-    attr_accessor :default_serializer
+    attr_reader :backend, :encoder, :inferring, :_on_error, :transforming_root_key
 
     # Set the backend, which actually serializes object into JSON
     #
@@ -28,25 +29,64 @@ module Alba
     # Serialize the object with inline definitions
     #
     # @param object [Object] the object to be serialized
-    # @param with [nil, Proc, Alba::Serializer] selializer
+    # @param key [Symbol]
     # @param block [Block] resource block
     # @return [String] serialized JSON string
     # @raise [ArgumentError] if block is absent or `with` argument's type is wrong
-    def serialize(object, with: nil, &block)
+    def serialize(object, key: nil, &block)
       raise ArgumentError, 'Block required' unless block
 
-      resource_class.class_eval(&block)
-      resource = resource_class.new(object)
-      with ||= @default_serializer
-      resource.serialize(with: with)
+      klass = Class.new
+      klass.include(Alba::Resource)
+      klass.class_eval(&block)
+      resource = klass.new(object)
+      resource.serialize(key: key)
+    end
+
+    # Enable inference for key and resource name
+    def enable_inference!
+      begin
+        require 'active_support/inflector'
+      rescue LoadError
+        raise ::Alba::Error, 'To enable inference, please install `ActiveSupport` gem.'
+      end
+      @inferring = true
+    end
+
+    # Disable inference for key and resource name
+    def disable_inference!
+      @inferring = false
+    end
+
+    # Set error handler
+    #
+    # @param [Symbol] handler
+    # @param [Block]
+    def on_error(handler = nil, &block)
+      raise ArgumentError, 'You cannot specify error handler with both Symbol and block' if handler && block
+      raise ArgumentError, 'You must specify error handler with either Symbol or block' unless handler || block
+
+      @_on_error = handler || block
+    end
+
+    # Enable root key transformation
+    def enable_root_key_transformation!
+      @transforming_root_key = true
+    end
+
+    # Disable root key transformation
+    def disable_root_key_transformation!
+      @transforming_root_key = false
     end
 
     private
 
     def set_encoder
       @encoder = case @backend
-                 when :oj
+                 when :oj, :oj_strict
                    try_oj
+                 when :oj_rails
+                   try_oj(mode: :rails)
                  when :active_support
                    try_active_support
                  when nil, :default, :json
@@ -56,10 +96,11 @@ module Alba
                  end
     end
 
-    def try_oj
+    def try_oj(mode: :strict)
       require 'oj'
-      ->(hash) { Oj.dump(hash, mode: :strict) }
+      ->(hash) { Oj.dump(hash, mode: mode) }
     rescue LoadError
+      Kernel.warn '`Oj` is not installed, falling back to default JSON encoder.'
       default_encoder
     end
 
@@ -67,6 +108,7 @@ module Alba
       require 'active_support/json'
       ->(hash) { ActiveSupport::JSON.encode(hash) }
     rescue LoadError
+      Kernel.warn '`ActiveSupport` is not installed, falling back to default JSON encoder.'
       default_encoder
     end
 
@@ -76,14 +118,9 @@ module Alba
         JSON.dump(hash)
       end
     end
-
-    def resource_class
-      @resource_class ||= begin
-        klass = Class.new
-        klass.include(Alba::Resource)
-      end
-    end
   end
 
   @encoder = default_encoder
+  @_on_error = :raise
+  @transforming_root_key = false # TODO: This will be true since 2.0
 end

data/lib/alba/association.rb

@@ -2,25 +2,40 @@ module Alba
   # Base class for `One` and `Many`
   # Child class should implement `to_hash` method
   class Association
+    attr_reader :object
+
     # @param name [Symbol] name of the method to fetch association
     # @param condition [Proc] a proc filtering data
     # @param resource [Class<Alba::Resource>] a resource class for the association
     # @param block [Block] used to define resource when resource arg is absent
-    def initialize(name:, condition: nil, resource: nil, &block)
+    def initialize(name:, condition: nil, resource: nil, nesting: nil, &block)
       @name = name
       @condition = condition
       @block = block
-      @resource = resource || resource_class
-      raise ArgumentError, 'resource or block is required' if @resource.nil? && @block.nil?
-    end
+      @resource = resource
+      return if @resource
 
-    # @abstract
-    def to_hash
-      :not_implemented
+      if @block
+        @resource = resource_class
+      elsif Alba.inferring
+        const_parent = nesting.nil? ? Object : Object.const_get(nesting)
+        @resource = const_parent.const_get("#{ActiveSupport::Inflector.classify(@name)}Resource")
+      else
+        raise ArgumentError, 'When Alba.inferring is false, either resource or block is required'
+      end
     end
 
     private
 
+    def constantize(resource)
+      case resource # rubocop:disable Style/MissingElse
+      when Class
+        resource
+      when Symbol, String
+        Object.const_get(resource)
+      end
+    end
+
     def resource_class
       klass = Class.new
       klass.include(Alba::Resource)

data/lib/alba/key_transformer.rb

@@ -16,6 +16,7 @@ module Alba
     # @return [String] transformed key
     # @raise [Alba::Error] when transform_type is not supported
     def transform(key, transform_type)
+      key = key.to_s
       case transform_type
       when :camel
         ActiveSupport::Inflector.camelize(key)
@@ -24,7 +25,7 @@ module Alba
       when :dash
         ActiveSupport::Inflector.dasherize(key)
       else
-        raise ::Alba::Error, "Unknown transform_type: #{transform_type}. Supported transform_type are :camel and :dash."
+        raise ::Alba::Error, "Unknown transform_type: #{transform_type}. Supported transform_type are :camel, :lower_camel and :dash."
       end
     end
   end

data/lib/alba/many.rb

@@ -6,12 +6,16 @@ module Alba
     # Recursively converts objects into an Array of Hashes
     #
     # @param target [Object] the object having an association method
+    # @param within [Hash] determines what associations to be serialized. If not set, it serializes all associations.
     # @param params [Hash] user-given Hash for arbitrary data
     # @return [Array<Hash>]
-    def to_hash(target, params: {})
-      objects = target.public_send(@name)
-      objects = @condition.call(objects, params) if @condition
-      objects.map { |o| @resource.new(o, params: params).to_hash }
+    def to_hash(target, within: nil, params: {})
+      @object = target.public_send(@name)
+      @object = @condition.call(@object, params) if @condition
+      return if @object.nil?
+
+      @resource = constantize(@resource)
+      @object.map { |o| @resource.new(o, params: params, within: within).to_hash }
     end
   end
 end

data/lib/alba/one.rb

@@ -6,12 +6,16 @@ module Alba
     # Recursively converts an object into a Hash
     #
     # @param target [Object] the object having an association method
+    # @param within [Hash] determines what associations to be serialized. If not set, it serializes all associations.
     # @param params [Hash] user-given Hash for arbitrary data
     # @return [Hash]
-    def to_hash(target, params: {})
-      object = target.public_send(@name)
-      object = @condition.call(object, params) if @condition
-      @resource.new(object, params: params).to_hash
+    def to_hash(target, within: nil, params: {})
+      @object = target.public_send(@name)
+      @object = @condition.call(object, params) if @condition
+      return if @object.nil?
+
+      @resource = constantize(@resource)
+      @resource.new(object, params: params, within: within).to_hash
     end
   end
 end

data/lib/alba/resource.rb

@@ -1,4 +1,3 @@
-require_relative 'serializer'
 require_relative 'one'
 require_relative 'many'
 
@@ -7,7 +6,7 @@ module Alba
   module Resource
     # @!parse include InstanceMethods
     # @!parse extend ClassMethods
-    DSLS = {_attributes: {}, _serializer: nil, _key: nil, _transform_keys: nil}.freeze
+    DSLS = {_attributes: {}, _key: nil, _transform_keys: nil, _transforming_root_key: false, _on_error: nil}.freeze
     private_constant :DSLS
 
     # @private
@@ -25,32 +24,26 @@ module Alba
 
     # Instance methods
     module InstanceMethods
-      attr_reader :object, :_key, :params
+      attr_reader :object, :params
 
       # @param object [Object] the object to be serialized
       # @param params [Hash] user-given Hash for arbitrary data
-      def initialize(object, params: {})
+      # @param within [Hash] determines what associations to be serialized. If not set, it serializes all associations.
+      def initialize(object, params: {}, within: true)
         @object = object
         @params = params.freeze
+        @within = within
         DSLS.each_key { |name| instance_variable_set("@#{name}", self.class.public_send(name)) }
       end
 
-      # Get serializer with `with` argument and serialize self with it
+      # Serialize object into JSON string
       #
-      # @param with [nil, Proc, Alba::Serializer] selializer
+      # @param key [Symbol]
       # @return [String] serialized JSON string
-      def serialize(with: nil)
-        serializer = case with
-                     when nil
-                       @_serializer || empty_serializer
-                     when ->(obj) { obj.is_a?(Class) && obj <= Alba::Serializer }
-                       with
-                     when Proc
-                       inline_extended_serializer(with)
-                     else
-                       raise ArgumentError, 'Unexpected type for with, possible types are Class or Proc'
-                     end
-        serializer.new(self).serialize
+      def serialize(key: nil)
+        key = key.nil? ? _key : key
+        hash = key && key != '' ? {key.to_s => serializable_hash} : serializable_hash
+        Alba.encoder.call(hash)
       end
 
       # A Hash for serialization
@@ -61,22 +54,67 @@ module Alba
       end
       alias to_hash serializable_hash
 
-      # @return [Symbol]
-      def key
-        @_key || self.class.name.delete_suffix('Resource').downcase.gsub(/:{2}/, '_').to_sym
-      end
-
       private
 
-      # rubocop:disable Style/MethodCalledOnDoEndBlock
+      # @return [String]
+      def _key
+        return @_key.to_s unless @_key == true && Alba.inferring
+
+        resource_name = self.class.name.demodulize.delete_suffix('Resource').underscore
+        key = collection? ? resource_name.pluralize : resource_name
+        transforming_root_key = @_transforming_root_key.nil? ? Alba.transforming_root_key : @_transforming_root_key
+        transforming_root_key ? transform_key(key) : key
+      end
+
       def converter
-        lambda do |resource|
-          @_attributes.map do |key, attribute|
-            [transform_key(key), fetch_attribute(resource, attribute)]
-          end.to_h
+        lambda do |object|
+          arrays = @_attributes.map do |key, attribute|
+            key = transform_key(key)
+            if attribute.is_a?(Array) # Conditional
+              conditional_attribute(object, key, attribute)
+            else
+              [key, fetch_attribute(object, attribute)]
+            end
+          rescue ::Alba::Error, FrozenError, TypeError
+            raise
+          rescue StandardError => e
+            handle_error(e, object, key, attribute)
+          end
+          arrays.reject(&:empty?).to_h
+        end
+      end
+
+      def conditional_attribute(object, key, attribute)
+        condition = attribute.last
+        arity = condition.arity
+        return [] if arity <= 1 && !condition.call(object)
+
+        fetched_attribute = fetch_attribute(object, attribute.first)
+        attr = if attribute.first.is_a?(Alba::Association)
+                 attribute.first.object
+               else
+                 fetched_attribute
+               end
+        return [] if arity >= 2 && !condition.call(object, attr)
+
+        [key, fetched_attribute]
+      end
+
+      def handle_error(error, object, key, attribute)
+        on_error = @_on_error || Alba._on_error
+        case on_error
+        when :raise, nil
+          raise
+        when :nullify
+          [key, nil]
+        when :ignore
+          []
+        when Proc
+          on_error.call(error, object, key, attribute, self.class)
+        else
+          raise ::Alba::Error, "Unknown on_error: #{on_error.inspect}"
         end
       end
-      # rubocop:enable Style/MethodCalledOnDoEndBlock
 
       # Override this method to supply custom key transform method
       def transform_key(key)
@@ -86,29 +124,81 @@ module Alba
         KeyTransformer.transform(key, @_transform_keys)
       end
 
-      def fetch_attribute(resource, attribute)
+      def fetch_attribute(object, attribute)
         case attribute
         when Symbol
-          resource.public_send attribute
+          object.public_send attribute
         when Proc
-          instance_exec(resource, &attribute)
+          instance_exec(object, &attribute)
         when Alba::One, Alba::Many
-          attribute.to_hash(resource, params: params)
+          within = check_within
+          return unless within
+
+          attribute.to_hash(object, params: params, within: within)
+        when Hash # Typed Attribute
+          typed_attribute(object, attribute)
         else
           raise ::Alba::Error, "Unsupported type of attribute: #{attribute.class}"
         end
       end
 
-      def empty_serializer
-        klass = Class.new
-        klass.include Alba::Serializer
-        klass
+      def typed_attribute(object, hash)
+        attr_name = hash[:attr_name]
+        type = hash[:type]
+        type_converter = hash[:type_converter]
+        value, result = type_check(object, attr_name, type)
+        return value if result
+        raise TypeError if !result && !type_converter
+
+        type_converter = type_converter_for(type) if type_converter == true
+        type_converter.call(value)
+      rescue TypeError
+        raise TypeError, "Attribute #{attr_name} is expected to be #{type} but actually #{value.nil? ? 'nil' : value.class.name}."
+      end
+
+      def type_check(object, attr_name, type)
+        value = object.public_send(attr_name)
+        type_correct = case type
+                       when :String, ->(klass) { klass == String }
+                         value.is_a?(String)
+                       when :Integer, ->(klass) { klass == Integer }
+                         value.is_a?(Integer)
+                       when :Boolean
+                         [true, false].include?(attr_name)
+                       else
+                         raise Alba::UnsupportedType, "Unknown type: #{type}"
+                       end
+        [value, type_correct]
       end
 
-      def inline_extended_serializer(with)
-        klass = empty_serializer
-        klass.class_eval(&with)
-        klass
+      def type_converter_for(type)
+        case type
+        when :String, ->(klass) { klass == String }
+          ->(object) { object.to_s }
+        when :Integer, ->(klass) { klass == Integer }
+          ->(object) { Integer(object) }
+        when :Boolean
+          ->(object) { !!object }
+        else
+          raise Alba::UnsupportedType, "Unknown type: #{type}"
+        end
+      end
+
+      def check_within
+        case @within
+        when Hash # Traverse within tree
+          @within.fetch(_key.to_sym, nil)
+        when Array # within tree ends with Array
+          @within.find { |item| item.to_sym == _key.to_sym } # Check if at least one item in the array matches current resource
+        when Symbol # within tree could end with Symbol
+          @within == _key.to_sym # Check if the symbol matches current resource
+        when true # In this case, Alba serializes all associations.
+          true
+        when nil, false # In these cases, Alba stops serialization here.
+          false
+        else
+          raise Alba::Error, "Unknown type for within option: #{@within.class}"
+        end
       end
 
       def collection?
@@ -129,19 +219,32 @@ module Alba
       # Set multiple attributes at once
       #
       # @param attrs [Array<String, Symbol>]
-      def attributes(*attrs)
-        attrs.each { |attr_name| @_attributes[attr_name.to_sym] = attr_name.to_sym }
+      # @param options [Hash] option hash including `if` that is a  condition to render these attributes
+      def attributes(*attrs, if: nil, **attrs_with_types) # rubocop:disable Naming/MethodParameterName
+        if_value = binding.local_variable_get(:if)
+        attrs.each do |attr_name|
+          attr = if_value ? [attr_name.to_sym, if_value] : attr_name.to_sym
+          @_attributes[attr_name.to_sym] = attr
+        end
+        attrs_with_types.each do |attr_name, type_and_converter|
+          attr_name = attr_name.to_sym
+          type, type_converter = type_and_converter
+          typed_attr = {attr_name: attr_name, type: type, type_converter: type_converter}
+          attr = if_value ? [typed_attr, if_value] : typed_attr
+          @_attributes[attr_name] = attr
+        end
       end
 
       # Set an attribute with the given block
       #
       # @param name [String, Symbol] key name
+      # @param options [Hash] option hash including `if` that is a  condition to render
       # @param block [Block] the block called during serialization
       # @raise [ArgumentError] if block is absent
-      def attribute(name, &block)
+      def attribute(name, **options, &block)
         raise ArgumentError, 'No block given in attribute method' unless block
 
-        @_attributes[name.to_sym] = block
+        @_attributes[name.to_sym] = options[:if] ? [block, options[:if]] : block
       end
 
       # Set One association
@@ -150,10 +253,13 @@ module Alba
       # @param condition [Proc]
       # @param resource [Class<Alba::Resource>]
       # @param key [String, Symbol] used as key when given
+      # @param options [Hash] option hash including `if` that is a  condition to render
       # @param block [Block]
       # @see Alba::One#initialize
-      def one(name, condition = nil, resource: nil, key: nil, &block)
-        @_attributes[key&.to_sym || name.to_sym] = One.new(name: name, condition: condition, resource: resource, &block)
+      def one(name, condition = nil, resource: nil, key: nil, **options, &block)
+        nesting = self.name&.rpartition('::')&.first
+        one = One.new(name: name, condition: condition, resource: resource, nesting: nesting, &block)
+        @_attributes[key&.to_sym || name.to_sym] = options[:if] ? [one, options[:if]] : one
       end
       alias has_one one
 
@@ -163,25 +269,27 @@ module Alba
       # @param condition [Proc]
       # @param resource [Class<Alba::Resource>]
       # @param key [String, Symbol] used as key when given
+      # @param options [Hash] option hash including `if` that is a  condition to render
       # @param block [Block]
       # @see Alba::Many#initialize
-      def many(name, condition = nil, resource: nil, key: nil, &block)
-        @_attributes[key&.to_sym || name.to_sym] = Many.new(name: name, condition: condition, resource: resource, &block)
+      def many(name, condition = nil, resource: nil, key: nil, **options, &block)
+        nesting = self.name&.rpartition('::')&.first
+        many = Many.new(name: name, condition: condition, resource: resource, nesting: nesting, &block)
+        @_attributes[key&.to_sym || name.to_sym] = options[:if] ? [many, options[:if]] : many
       end
       alias has_many many
 
-      # Set serializer for the resource
-      #
-      # @param name [Alba::Serializer]
-      def serializer(name)
-        @_serializer = name <= Alba::Serializer ? name : nil
-      end
-
       # Set key
       #
       # @param key [String, Symbol]
       def key(key)
-        @_key = key.to_sym
+        @_key = key.respond_to?(:to_sym) ? key.to_sym : key
+      end
+
+      # Set key to true
+      #
+      def key!
+        @_key = true
       end
 
       # Delete attributes
@@ -196,9 +304,22 @@ module Alba
 
       # Transform keys as specified type
       #
-      # @params type [String, Symbol]
-      def transform_keys(type)
+      # @param type [String, Symbol]
+      # @param root [Boolean] decides if root key also should be transformed
+      def transform_keys(type, root: nil)
         @_transform_keys = type.to_sym
+        @_transforming_root_key = root
+      end
+
+      # Set error handler
+      #
+      # @param [Symbol] handler
+      # @param [Block]
+      def on_error(handler = nil, &block)
+        raise ArgumentError, 'You cannot specify error handler with both Symbol and block' if handler && block
+        raise ArgumentError, 'You must specify error handler with either Symbol or block' unless handler || block
+
+        @_on_error = handler || block
       end
     end
   end