From af9a92086df8fcfbb572eb3d2bc11eee383ef9e4 Mon Sep 17 00:00:00 2001 From: Luke Kanies Date: Thu, 21 Jan 2010 22:45:43 -0800 Subject: Adding an environment helper Signed-off-by: Luke Kanies --- lib/puppet/node/environment.rb | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'lib/puppet/node/environment.rb') diff --git a/lib/puppet/node/environment.rb b/lib/puppet/node/environment.rb index 537a5dd26..70417449e 100644 --- a/lib/puppet/node/environment.rb +++ b/lib/puppet/node/environment.rb @@ -7,6 +7,20 @@ end # Model the environment that a node can operate in. This class just # provides a simple wrapper for the functionality around environments. class Puppet::Node::Environment + module Helper + def environment + Puppet::Node::Environment.new(@environment) + end + + def environment=(env) + if env.is_a?(String) or env.is_a?(Symbol) + @environment = env + else + @environment = env.name + end + end + end + include Puppet::Util::Cacher @seen = {} -- cgit