From 2573ef1c4c2d472f9cf41ab3a1b67b7408d704b4 Mon Sep 17 00:00:00 2001 From: Luke Kanies Date: Tue, 17 Feb 2009 17:55:57 -0600 Subject: Added support for finding modules from an environment This uses the environment to search for the modules, rather than relying on the Puppet::Module class to know how to handle environments. Signed-off-by: Luke Kanies --- lib/puppet/node/environment.rb | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'lib/puppet/node') diff --git a/lib/puppet/node/environment.rb b/lib/puppet/node/environment.rb index b64b9c2c4..d4c04469c 100644 --- a/lib/puppet/node/environment.rb +++ b/lib/puppet/node/environment.rb @@ -28,4 +28,21 @@ class Puppet::Node::Environment def initialize(name) @name = name end + + def module(name) + Puppet::Module.each_module(self[:modulepath]) do |mod| + return mod if mod.name == name + end + + return nil + end + + # Return all modules from this environment. + def modules + result = [] + Puppet::Module.each_module(self[:modulepath]) do |mod| + result << mod + end + result + end end -- cgit