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 <luke@madstop.com>

1 files changed, 17 insertions, 0 deletions

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