codegram/hyperclient

module Hyperclient
  # Public: A wrapper class to easily acces the links in a  Resource.
  #
  # Examples
  #
  #   resource.links['author']
  #   resource.links.author
  #
  class LinkCollection < Collection
    attr_reader :_curies

    # Public: Initializes a LinkCollection.
    #
    # collection  - The Hash with the links.
    # curies      - The Hash with link curies.
    # entry_point - The EntryPoint object to inject the configuration.
    def initialize(collection, curies, entry_point)
      if collection && !collection.respond_to?(:collect)
        raise "Invalid response for LinkCollection. The response was: #{collection.inspect}"
      end

      @_curies = (curies || {}).reduce({}) do |hash, curie_hash|
        curie = build_curie(curie_hash, entry_point)
        hash.update(curie.name => curie)
      end

      @collection = (collection || {}).reduce({}) do |hash, (name, link)|
        hash.update(name => build_link(name, link, @_curies, entry_point))
      end
    end

    private

    # Internal: Creates links from the response hash.
    #
    # name          - A String to identify the link's name.
    # link_or_links - A Hash or an Array of hashes with the links to build.
    # curies        - An Array of Curies for templated links.
    # entry_point   - The EntryPoint object to inject the configuration.
    #
    # Returns a Link or an Array of Links when given an Array.
    def build_link(name, link_or_links, curies, entry_point)
      return unless link_or_links

      if link_or_links.respond_to?(:to_ary)
        link_or_links.map do |link|
          build_link(name, link, curies, entry_point)
        end
      else
        Link.new(name, link_or_links, entry_point)
      end
    end

    # Internal: Creates a curie from the response hash.
    #
    # curie_hash - A Hash with the curie.
    # entry_point - The EntryPoint object to inject the configuration.
    #
    # Returns a Link or an array of Links when given an Array.
    def build_curie(curie_hash, entry_point)
      Curie.new(curie_hash, entry_point)
    end
  end
end