class Puppet::Resource::TypeCollection

# File lib/puppet/resource/type_collection.rb, line 42
def <<(thing)
  add(thing)
  self
end

# File lib/puppet/resource/type_collection.rb, line 47
def add(instance)
  if instance.type == :hostclass and other = @hostclasses[instance.name] and other.type == :hostclass
    other.merge(instance)
    return other
  end
  method = "add_#{instance.type}"
  send(method, instance)
  instance.resource_type_collection = self
  instance
end

# File lib/puppet/resource/type_collection.rb, line 104
def add_definition(instance)
  dupe_check(instance, @hostclasses) { |dupe| "'#{instance.name}' is already defined#{dupe.error_context} as a class; cannot redefine as a definition" }
  dupe_check(instance, @definitions) { |dupe| "Definition '#{instance.name}' is already defined#{dupe.error_context}; cannot be redefined" }
  @definitions[instance.name] = instance
end

# File lib/puppet/resource/type_collection.rb, line 58
def add_hostclass(instance)
  dupe_check(instance, @hostclasses) { |dupe| "Class '#{instance.name}' is already defined#{dupe.error_context}; cannot redefine" }
  dupe_check(instance, @definitions) { |dupe| "Definition '#{instance.name}' is already defined#{dupe.error_context}; cannot be redefined as a class" }

  @hostclasses[instance.name] = instance
  instance
end

# File lib/puppet/resource/type_collection.rb, line 70
def add_node(instance)
  dupe_check(instance, @nodes) { |dupe| "Node '#{instance.name}' is already defined#{dupe.error_context}; cannot redefine" }

  @node_list << instance
  @nodes[instance.name] = instance
  instance
end

# File lib/puppet/resource/type_collection.rb, line 11
def clear
  @hostclasses.clear
  @definitions.clear
  @nodes.clear
  @watched_files.clear
  @notfound.clear
end

# File lib/puppet/resource/type_collection.rb, line 110
def definition(name)
  @definitions[munge_name(name)]
end

# File lib/puppet/resource/type_collection.rb, line 122
def find_definition(namespaces, name)
  find_or_load(namespaces, name, :definition)
end

# File lib/puppet/resource/type_collection.rb, line 118
def find_hostclass(namespaces, name, options = {})
  find_or_load(namespaces, name, :hostclass, options)
end

# File lib/puppet/resource/type_collection.rb, line 114
def find_node(namespaces, name)
  @nodes[munge_name(name)]
end

# File lib/puppet/resource/type_collection.rb, line 66
def hostclass(name)
  @hostclasses[munge_name(name)]
end

# File lib/puppet/resource/type_collection.rb, line 32
def import_ast(ast, modname)
  ast.instantiate(modname).each do |instance|
    add(instance)
  end
end

# File lib/puppet/resource/type_collection.rb, line 38
def inspect
  "TypeCollection" + { :hostclasses => @hostclasses.keys, :definitions => @definitions.keys, :nodes => @nodes.keys }.inspect
end

# File lib/puppet/resource/type_collection.rb, line 78
def loader
  @loader ||= Puppet::Parser::TypeLoader.new(environment)
end

# File lib/puppet/resource/type_collection.rb, line 82
def node(name)
  name = munge_name(name)

  if node = @nodes[name]
    return node
  end

  @node_list.each do |node|
    next unless node.name_is_regex?
    return node if node.match(name)
  end
  nil
end

# File lib/puppet/resource/type_collection.rb, line 96
def node_exists?(name)
  @nodes[munge_name(name)]
end

# File lib/puppet/resource/type_collection.rb, line 100
def nodes?
  @nodes.length > 0
end

# File lib/puppet/resource/type_collection.rb, line 132
def require_reparse?
  @parse_failed || stale?
end

# File lib/puppet/resource/type_collection.rb, line 136
def stale?
  @watched_files.changed?
end

# File lib/puppet/resource/type_collection.rb, line 140
def version
  if !defined?(@version)
    if environment.config_version.nil? || environment.config_version == ""
      @version = Time.now.to_i
    else
      @version = Puppet::Util::Execution.execute([environment.config_version]).strip
    end
  end

  @version
rescue Puppet::ExecutionFailure => e
  raise Puppet::ParseError, "Execution of config_version command `#{environment.config_version}` failed: #{e.message}", e.backtrace
end

# File lib/puppet/resource/type_collection.rb, line 154
def watch_file(filename)
  @watched_files.watch(filename)
end

# File lib/puppet/resource/type_collection.rb, line 158
def watching_file?(filename)
  @watched_files.watching?(filename)
end

# File lib/puppet/resource/type_collection.rb, line 19
def initialize(env)
  @environment = env
  @hostclasses = {}
  @definitions = {}
  @nodes = {}
  @notfound = {}

  # So we can keep a list and match the first-defined regex
  @node_list = []

  @watched_files = Puppet::Util::FileWatcher.new
end