summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions/dig44.rb
blob: 1e7c318c0f9213bdffdbd6e1bbc8e5b25f6171d4 (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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#
# dig44.rb
#

module Puppet::Parser::Functions
  newfunction(
      :dig44,
      :type => :rvalue,
      :arity => -2,
      :doc => <<-eos
DEPRECATED: This function has been replaced in puppet 4.5.0.

Looks up into a complex structure of arrays and hashes and returns a value
or the default value if nothing was found.

Key can contain slashes to describe path components. The function will go down
the structure and try to extract the required value.

$data = {
  'a' => {
    'b' => [
      'b1',
      'b2',
      'b3',
    ]
  }
}

$value = dig44($data, ['a', 'b', '2'], 'not_found')
=> $value = 'b3'

a -> first hash key
b -> second hash key
2 -> array index starting with 0

not_found -> (optional) will be returned if there is no value or the path
did not match. Defaults to nil.

In addition to the required "key" argument, the function accepts a default
argument. It will be returned if no value was found or a path component is
missing. And the fourth argument can set a variable path separator.
  eos
  ) do |arguments|
    # Two arguments are required
    raise(Puppet::ParseError, "dig44(): Wrong number of arguments " +
                              "given (#{arguments.size} for at least 2)") if arguments.size < 2

    data, path, default = *arguments

    unless data.is_a?(Hash) or data.is_a?(Array)
      raise(Puppet::ParseError, "dig44(): first argument must be a hash or an array, " <<
                                "given #{data.class.name}")
    end

    unless path.is_a? Array
      raise(Puppet::ParseError, "dig44(): second argument must be an array, " <<
                                "given #{path.class.name}")
    end

    value = path.reduce(data) do |structure, key|
      if structure.is_a? Hash or structure.is_a? Array
        if structure.is_a? Array
          key = Integer key rescue break
        end
        break if structure[key].nil? or structure[key] == :undef
        structure[key]
      else
        break
      end
    end
    value.nil? ? default : value
  end
end