summaryrefslogtreecommitdiffstats
path: root/lib/puppet/parser
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/parser')
-rw-r--r--lib/puppet/parser/loaded_code.rb85
1 files changed, 85 insertions, 0 deletions
diff --git a/lib/puppet/parser/loaded_code.rb b/lib/puppet/parser/loaded_code.rb
new file mode 100644
index 000000000..7c918d4c3
--- /dev/null
+++ b/lib/puppet/parser/loaded_code.rb
@@ -0,0 +1,85 @@
+class Puppet::Parser::LoadedCode
+ def initialize
+ @hostclasses = {}
+ @definitions = {}
+ @nodes = {}
+ end
+
+ def add_hostclass(name, code)
+ @hostclasses[munge_name(name)] = code
+ end
+
+ def hostclass(name)
+ @hostclasses[munge_name(name)]
+ end
+
+ def add_node(name, code)
+ @nodes[munge_name(name)] = code
+ end
+
+ def node(name)
+ @nodes[munge_name(name)]
+ end
+
+ def nodes?
+ @nodes.length > 0
+ end
+
+ def add_definition(name, code)
+ @definitions[munge_name(name)] = code
+ end
+
+ def definition(name)
+ @definitions[munge_name(name)]
+ end
+
+ def find(namespace, name, type)
+ if r = find_fully_qualified(name, type)
+ return r
+ end
+
+ ary = namespace.split("::")
+
+ while ary.length > 0
+ tmp_namespace = ary.join("::")
+ if r = find_partially_qualified(tmp_namespace, name, type)
+ return r
+ end
+
+ # Delete the second to last object, which reduces our namespace by one.
+ ary.pop
+ end
+
+ send(type, name)
+ end
+
+ def find_node(name)
+ find("", name, :node)
+ end
+
+ def find_hostclass(namespace, name)
+ find(namespace, name, :hostclass)
+ end
+
+ def find_definition(namespace, name)
+ find(namespace, name, :definition)
+ end
+
+ private
+
+ def find_fully_qualified(name, type)
+ return nil unless name =~ /^::/
+
+ name = name.sub(/^::/, '')
+
+ send(type, name)
+ end
+
+ def find_partially_qualified(namespace, name, type)
+ send(type, [namespace, name].join("::"))
+ end
+
+ def munge_name(name)
+ name.to_s.downcase
+ end
+end