# frozen_string_literal: true

module Mongoid

  # This module provides the extra behavior for including associations in JSON
  # and XML serialization.
  module Serializable
    extend ActiveSupport::Concern

    # We need to redefine where the JSON configuration is getting defined,
    # similar to +ActiveRecord+.
    included do

      class << self
        def include_root_in_json
          @include_root_in_json.nil? ? ::Mongoid.include_root_in_json : @include_root_in_json
        end

        def include_root_in_json=(new_value)
          @include_root_in_json = new_value
        end
      end
    end

    # Gets the document as a serializable hash, used by ActiveModel's JSON
    # serializer.
    #
    # @example Get the serializable hash.
    #   document.serializable_hash
    #
    # @example Get the serializable hash with options.
    #   document.serializable_hash(:include => :addresses)
    #
    # @param [ Hash ] options The options to pass.
    #
    # @option options [ Symbol | String | Array<Symbol | String> ] :except
    #   Do not include these field(s).
    # @option options [ Symbol | String | Array<Symbol | String> ] :include
    #   Which association(s) to include.
    # @option options [ Symbol | String | Array<Symbol | String> ] :only
    #   Limit the field(s) to only these.
    # @option options [ Symbol | String | Array<Symbol | String> ] :methods
    #   What methods to include.
    #
    # @return [ Hash ] The document, ready to be serialized.
    def serializable_hash(options = nil)
      options ||= {}
      attrs = {}

      names = field_names(options)

      method_names = Array.wrap(options[:methods]).map do |name|
        name.to_s if respond_to?(name)
      end.compact

      (names + method_names).each do |name|
        without_autobuild do
          serialize_attribute(attrs, name, names, options)
        end
      end
      serialize_relations(attrs, options) if options[:include]
      attrs
    end

    private

    # Get the names of all fields that will be serialized.
    #
    # @api private
    #
    # @example Get all the field names.
    #   document.send(:field_names)
    #
    # @return [ Array<String> ] The names of the fields.
    def field_names(options)
      names = (as_attributes.keys + attribute_names).uniq.sort

      only = Array.wrap(options[:only]).map(&:to_s)
      except = Array.wrap(options[:except]).map(&:to_s)
      except |= [self.class.discriminator_key] unless Mongoid.include_type_for_serialization

      if !only.empty?
        names &= only
      elsif !except.empty?
        names -= except
      end
      names
    end

    # Serialize a single attribute. Handles associations, fields, and dynamic
    # attributes.
    #
    # @api private
    #
    # @example Serialize the attribute.
    #   document.serialize_attribute({}, "id" , [ "id" ])
    #
    # @param [ Hash ] attrs The attributes.
    # @param [ String ] name The attribute name.
    # @param [ Array<String> ] names The names of all attributes.
    # @param [ Hash ] options The options.
    #
    # @return [ Object ] The attribute.
    def serialize_attribute(attrs, name, names, options)
      if relations.key?(name)
        value = send(name)
        attrs[name] = value ? value.serializable_hash(options) : nil
      elsif names.include?(name) && !fields.key?(name)
        attrs[name] = read_raw_attribute(name)
      elsif !attribute_missing?(name)
        attrs[name] = send(name)
      end
    end

    # For each of the provided include options, get the association needed and
    # provide it in the hash.
    #
    # @example Serialize the included associations.
    #   document.serialize_relations({}, :include => :addresses)
    #
    # @param [ Hash ] attributes The attributes to serialize.
    # @param [ Hash ] options The serialization options.
    #
    # @option options [ Symbol ] :include What associations to include
    # @option options [ Symbol ] :only Limit the fields to only these.
    # @option options [ Symbol ] :except Dont include these fields.
    def serialize_relations(attributes = {}, options = {})
      inclusions = options[:include]
      relation_names(inclusions).each do |name|
        association = relations[name.to_s]
        if association && relation = send(association.name)
          attributes[association.name.to_s] =
            relation.serializable_hash(relation_options(inclusions, options, name))
        end
      end
    end

    # Since the inclusions can be a hash, symbol, or array of symbols, this is
    # provided as a convenience to parse out the names.
    #
    # @example Get the association names.
    #   document.relation_names(:include => [ :addresses ])
    #
    # @param [ Hash | Symbol | Array<Symbol> ] inclusions The inclusions.
    #
    # @return [ Array<Symbol> ] The names of the included associations.
    def relation_names(inclusions)
      inclusions.is_a?(Hash) ? inclusions.keys : Array.wrap(inclusions)
    end

    # Since the inclusions can be a hash, symbol, or array of symbols, this is
    # provided as a convenience to parse out the options.
    #
    # @example Get the association options.
    #   document.relation_names(:include => [ :addresses ])
    #
    # @param [ Hash | Symbol | Array<Symbol> ] inclusions The inclusions.
    # @param [ Hash ] options The options.
    # @param [ Symbol ] name The name of the association.
    #
    # @return [ Hash ] The options for the association.
    def relation_options(inclusions, options, name)
      if inclusions.is_a?(Hash)
        inclusions[name]
      else
        { except: options[:except], only: options[:only] }
      end
    end
  end
end