active_model_serializers/lib/active_model/serializer/adapter/json_api.rb

module ActiveModel
  class Serializer
    class Adapter
      class JsonApi < Adapter
        def initialize(serializer, options = {})
          super
          serializer.root = true
        end

        def serializable_hash(options = {})
          @root = (options[:root] || serializer.json_key).to_s.pluralize.to_sym
          @hash = {}
          @fieldset = options[:fieldset]

          if serializer.respond_to?(:each)
            opt = @fieldset ? {fieldset: @fieldset} : {}
        
            @hash[@root] = serializer.map{|s| self.class.new(s).serializable_hash(opt)[@root] }
          else
            @hash[@root] = attributes_for_serializer(serializer, {})

            serializer.each_association do |name, association, opts|
              @hash[@root][:links] ||= {}
              unless options[:embed] == :ids
                @hash[:linked] ||= {}
              end

              if association.respond_to?(:each)
                add_links(name, association, opts)
              else
                add_link(name, association, opts)
              end
            end
          end

          @hash
        end

        def add_links(name, serializers, options)
          @hash[@root][:links][name] ||= []
          @hash[@root][:links][name] += serializers.map{|serializer| serializer.id.to_s }

          unless options[:embed] == :ids
            @hash[:linked][name] ||= []
            @hash[:linked][name] += serializers.map { |item| attributes_for_serializer(item, options) }
          end
        end

        def add_link(name, serializer, options)
          @hash[@root][:links][name] = serializer.id.to_s

          unless options[:embed] == :ids
            plural_name = name.to_s.pluralize.to_sym

            @hash[:linked][plural_name] ||= []
            @hash[:linked][plural_name].push attributes_for_serializer(serializer, options)
          end
        end

        private

        def attributes_for_serializer(serializer, options)

          if fields = @fieldset && @fieldset.fields_for(serializer)
            options[:fields] = fields
          end

          attributes = serializer.attributes(options)
          attributes[:id] = attributes[:id].to_s if attributes[:id]
          attributes
        end
      end
    end
  end
end