Contents

module Nanoc::Int
  # @api private
  class IdentifiableCollection
    include Nanoc::Int::ContractsSupport
    include Enumerable

    extend Forwardable

    def_delegator :@objects, :each
    def_delegator :@objects, :size
    def_delegator :@objects, :<<
    def_delegator :@objects, :concat

    contract C::Or[Hash, C::Named['Nanoc::Int::Configuration']] => C::Any
    def initialize(config)
      @config = config

      @objects = []
    end

    contract C::None => self
    def freeze
      @objects.freeze
      @objects.each(&:freeze)
      build_mapping
      super
    end

    contract C::Any => C::Maybe[C::RespondTo[:identifier]]
    def [](arg)
      case arg
      when Nanoc::Identifier
        object_with_identifier(arg)
      when String
        object_with_identifier(arg) || object_matching_glob(arg)
      when Regexp
        @objects.find { |i| i.identifier.to_s =~ arg }
      else
        raise ArgumentError, "don’t know how to fetch objects by #{arg.inspect}"
      end
    end

    contract C::None => C::ArrayOf[C::RespondTo[:identifier]]
    def to_a
      @objects
    end

    contract C::None => C::Bool
    def empty?
      @objects.empty?
    end

    contract C::Func[C::RespondTo[:identifier] => C::Bool] => self
    def delete_if(&block)
      @objects.delete_if(&block)
      self
    end

    protected

    def object_with_identifier(identifier)
      if frozen?
        @mapping[identifier.to_s]
      else
        @objects.find { |i| i.identifier == identifier }
      end
    end

    def object_matching_glob(glob)
      if use_globs?
        pat = Nanoc::Int::Pattern.from(glob)
        @objects.find { |i| pat.match?(i.identifier) }
      else
        nil
      end
    end

    def build_mapping
      @mapping = {}
      @objects.each do |object|
        @mapping[object.identifier.to_s] = object
      end
      @mapping.freeze
    end

    def use_globs?
      @config[:string_pattern_type] == 'glob'
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version	Path
nanoc-4.6.1	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.6.0	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.5.4	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.5.3	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.5.2	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.5.1	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.5.0	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.4.7	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.4.6	lib/nanoc/base/entities/identifiable_collection.rb
nanoc-4.4.5	lib/nanoc/base/entities/identifiable_collection.rb