3rdparty/modules/stdlib/lib/puppet/parser/functions/ensure_packages.rb

   1 #
   2 # ensure_packages.rb
   3 #
   4
   5 module Puppet::Parser::Functions
   6   newfunction(:ensure_packages, :type => :statement, :doc => <<-EOS
   7 Takes a list of packages and only installs them if they don't already exist.
   8 It optionally takes a hash as a second parameter that will be passed as the
   9 third argument to the ensure_resource() function.
  10     EOS
  11   ) do |arguments|
  12
  13     if arguments.size > 2 or arguments.size == 0
  14       raise(Puppet::ParseError, "ensure_packages(): Wrong number of arguments given (#{arguments.size} for 1 or 2)")
  15     elsif arguments.size == 2 and !arguments[1].is_a?(Hash)
  16       raise(Puppet::ParseError, 'ensure_packages(): Requires second argument to be a Hash')
  17     end
  18
  19     if arguments[0].is_a?(Hash)
  20       if arguments[1]
  21         defaults = { 'ensure' => 'present' }.merge(arguments[1])
  22         if defaults['ensure'] == 'installed'
  23           defaults['ensure'] = 'present'
  24         end
  25       else
  26         defaults = { 'ensure' => 'present' }
  27       end
  28
  29       Puppet::Parser::Functions.function(:ensure_resources)
  30       function_ensure_resources(['package', arguments[0].dup, defaults ])
  31     else
  32       packages = Array(arguments[0])
  33
  34       if arguments[1]
  35         defaults = { 'ensure' => 'present' }.merge(arguments[1])
  36         if defaults['ensure'] == 'installed'
  37           defaults['ensure'] = 'present'
  38         end
  39       else
  40         defaults = { 'ensure' => 'present' }
  41       end
  42
  43       Puppet::Parser::Functions.function(:ensure_resource)
  44       packages.each { |package_name|
  45       raise(Puppet::ParseError, 'ensure_packages(): Empty String provided for package name') if package_name.length == 0
  46       if !findresource("Package[#{package_name}]")
  47         function_ensure_resource(['package', package_name, defaults ])
  48       end
  49     }
  50     end
  51   end
  52 end
  53
  54 # vim: set ts=2 sw=2 et :