summaryrefslogtreecommitdiff
path: root/lib/puppet/parser
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/parser')
-rw-r--r--lib/puppet/parser/functions/ensure_packages.rb13
1 files changed, 7 insertions, 6 deletions
diff --git a/lib/puppet/parser/functions/ensure_packages.rb b/lib/puppet/parser/functions/ensure_packages.rb
index 450ea02..1e0f225 100644
--- a/lib/puppet/parser/functions/ensure_packages.rb
+++ b/lib/puppet/parser/functions/ensure_packages.rb
@@ -1,7 +1,6 @@
#
# ensure_packages.rb
#
-require 'puppet/parser/functions'
module Puppet::Parser::Functions
newfunction(:ensure_packages, :type => :statement, :doc => <<-EOS
@@ -9,13 +8,15 @@ Takes a list of packages and only installs them if they don't already exist.
EOS
) do |arguments|
- raise(Puppet::ParseError, "ensure_packages(): Wrong number of arguments " +
- "given (#{arguments.size} for 1)") if arguments.size != 1
- raise(Puppet::ParseError, "ensure_packages(): Requires array " +
- "given (#{arguments[0].class})") if !arguments[0].kind_of?(Array)
+ if arguments.size != 1
+ raise(Puppet::ParseError, "ensure_packages(): Wrong number of arguments " +
+ "given (#{arguments.size} for 1)")
+ end
+
+ packages = Array(arguments[0])
Puppet::Parser::Functions.function(:ensure_resource)
- arguments[0].each { |package_name|
+ packages.each { |package_name|
function_ensure_resource(['package', package_name, {'ensure' => 'present' } ])
}
end