summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions/ensure_resources.rb
blob: 30d57a88a81e34cabda62fa11dc0a037528bbb51 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
require 'puppet/parser/functions'

Puppet::Parser::Functions.newfunction(:ensure_resources,
                                      :type => :statement,
                                      :doc => <<-'ENDOFDOC'
Takes a resource type, title (only hash), and a list of attributes that describe a
resource.

    user { 'dan':
      gid => 'mygroup',
      ensure => present,
    }

An hash of resources should be passed in and each will be created with
the type and parameters specified if it doesn't already exist.

    ensure_resources('user', {'dan' => { gid => 'mygroup', uid => '600' } ,  'alex' => { gid => 'mygroup' }}, {'ensure' => 'present'})

From Hiera Backend:

userlist:
  dan:
    gid: 'mygroup'
 uid: '600'
  alex:
 gid: 'mygroup'

Call:
ensure_resources('user', hiera_hash('userlist'), {'ensure' => 'present'})

ENDOFDOC
) do |vals|
  type, title, params = vals
  raise(ArgumentError, 'Must specify a type') unless type
  raise(ArgumentError, 'Must specify a title') unless title
  params ||= {}

  if title.is_a?(Hash)
    resource_hash = Hash(title)
    resources = resource_hash.keys

    Puppet::Parser::Functions.function(:ensure_resource)
    resources.each { |resource_name|
    if resource_hash[resource_name]
        params_merged = params.merge(resource_hash[resource_name])
    else
        params_merged = params
    end
    function_ensure_resource([ type, resource_name, params_merged ])
    }
  else
       raise(Puppet::ParseError, 'ensure_resources(): Requires second argument to be a Hash')
  end
end