visualitypl/jsonapi_parameters

require 'active_support/inflector'

module JsonApi::Parameters
  include ActiveSupport::Inflector

  def jsonapify(params, naming_convention: :snake, custom_stack_limit: stack_limit)
    initialize_stack(custom_stack_limit)

    jsonapi_translate(params, naming_convention: naming_convention)
  end

  private

  def jsonapi_translate(params, naming_convention:)
    params = params.to_unsafe_h if params.is_a?(ActionController::Parameters)

    return params if params.nil? || params.empty?

    @jsonapi_unsafe_hash = if naming_convention != :snake || JsonApi::Parameters.ensure_underscore_translation
                             params = params.deep_transform_keys { |key| key.to_s.underscore.to_sym }
                             params[:data][:type] = params[:data][:type].underscore if params.dig(:data, :type)
                             params
                           else
                             params.deep_symbolize_keys
                           end

    formed_parameters
  end

  def formed_parameters
    @formed_parameters ||= {}.tap do |param|
      param[jsonapi_main_key.to_sym] = jsonapi_main_body
    end
  end

  def jsonapi_main_key
    @jsonapi_unsafe_hash.dig(:data, :type)&.singularize || ''
  end

  def jsonapi_main_body
    jsonapi_unsafe_params.tap do |param|
      jsonapi_relationships.each do |relationship_key, relationship_value|
        param = handle_relationships(param, relationship_key, relationship_value)
      end
    end
  ensure
    reset_stack_level
  end

  def jsonapi_unsafe_params
    @jsonapi_unsafe_params ||= (@jsonapi_unsafe_hash.dig(:data, :attributes) || {}).tap do |param|
      id = @jsonapi_unsafe_hash.dig(:data, :id)

      param[:id] = id if id.present?
    end
  end

  def jsonapi_relationships
    @jsonapi_relationships ||= @jsonapi_unsafe_hash.dig(:data, :relationships) || []
  end

  def jsonapi_included
    @jsonapi_included ||= @jsonapi_unsafe_hash[:included] || []
  end

  def handle_relationships(param, relationship_key, relationship_value)
    increment_stack_level!

    relationship_value = relationship_value[:data]
    handler_args = [relationship_key, relationship_value, jsonapi_included]
    handler = if Handlers.resource_handlers.key?(relationship_key)
                Handlers.handlers[Handlers.resource_handlers[relationship_key]]
              else
                case relationship_value
                when Array
                  Handlers.handlers[:to_many]
                when Hash
                  Handlers.handlers[:to_one]
                when nil
                  Handlers.handlers[:nil]
                else
                  raise NotImplementedError.new('relationship resource linkage has to be a type of Array, Hash or nil')
                end
              end

    key, val = handler.call(*handler_args)

    param[key] = handle_nested_relationships(val)

    param
  ensure
    decrement_stack_level
  end

  def handle_nested_relationships(val)
    # We can only consider Hash relationships (which imply to-one relationship) and Array relationships (which imply to-many).
    # Each type has a different handling method, though in both cases we end up passing the nested relationship recursively to handle_relationship
    # (and yes, this may go on indefinitely, basically we're going by the relationship levels, deeper and deeper)
    case val
    when Array
      relationships_with_nested_relationships = val.select { |rel| rel.is_a?(Hash) && rel.dig(:relationships) }
      relationships_with_nested_relationships.each do |relationship_with_nested_relationship|
        relationship_with_nested_relationship.delete(:relationships).each do |rel_key, rel_val|
          relationship_with_nested_relationship = handle_relationships(relationship_with_nested_relationship, rel_key, rel_val)
        end
      end
    when Hash
      if val.key?(:relationships)
        val.delete(:relationships).each do |rel_key, rel_val|
          val = handle_relationships(val, rel_key, rel_val)
        end
      end
    end

    val
  end
end