ama-team / ruby-entity-mapper

require_relative '../../exception/mapping_error'

require_relative '../../mixin/errors'

module AMA

  module Entity

    class Mapper

      class Engine

        class Normalizer

          # Special normalizer for registered entities

          class Entity

            include ::AMA::Entity::Mapper::Mixin::Errors

            def normalize(entity, context = nil, target_type = nil)

              type = @registry.find!(entity.class)

              unless type.normalizer

                return normalize_entity(entity, context, target_type)

              end

              type.normalizer.call(entity, context, target_type) do |handle|

                normalize_entity(handle, context, target_type)

              end

            rescue StandardError => e

              message = 'Error while normalizing entity ' \

                "of type #{entity.class}"

              mapping_error(message, parent: e, context: context)

            end

            def supports(value)

              @registry.include?(value.class)

            end

            private

            def normalize_entity(entity, *)

              # TODO: use enumerator

              type = @registry.find!(entity.class)

              type.attributes.values.each_with_object({}) do |attribute, values|

                next values if attribute.sensitive || attribute.virtual

                values[attribute.name] = normalize_attribute(entity, attribute)

                values

              end

            end

            def normalize_attribute(entity, attribute)

              # TODO: add attribute normalizer support

              entity.instance_variable_get("@#{attribute.name}")

            end

          end

        end

      end

    end

  end

end