Puppet Function: tp_lookup

Defined in:
lib/puppet/parser/functions/tp_lookup.rb
Function type:
Ruby 3.x API

Overview

tp_lookup() ⇒ Any

Looks for tp data. Usage: $tp_settings=tp_lookup($title,‘settings’,‘site’,deep_merge’)

Returns:

  • (Any) 


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
74
75
76
77
78
79
80
81
82
 
# File 'lib/puppet/parser/functions/tp_lookup.rb', line 4

newfunction(:tp_lookup, :type => :rvalue, :doc => "
Looks for tp data. Usage:
$tp_settings=tp_lookup($title,'settings','site',deep_merge')
") do |args|
  unless args.length >= 3
    raise Puppet::ParseError, ("tp_lookup(): wrong number of arguments (#{args.length}; must be 2 or 3)")
  end

  app = args[0]
  res = args[1]
  datamodule = args[2]
  args[3].to_s.length!=0 ? look = args[3] : look = 'direct'
  key = app + "::" + res

  value = { }

  if mp = Puppet::Module.find(datamodule, compiler.environment.to_s)

    hiera_file_path  = mp.path + '/data/' + app + '/hiera.yaml'

    unless File.exist?(hiera_file_path)
      function_warning(["No tinydata found for: #{app} in #{hiera_file_path}. Trying to install package #{app}"])
      default_fallback = true
      hiera_file_path  = mp.path + '/data/default/hiera.yaml'
    end

    hiera_file = File.open(hiera_file_path)
    hiera = YAML::load(hiera_file)
    if lookupvar("upstream_repo")
      repo = 'upstream'
    elsif lookupvar("repo")
      repo = lookupvar("repo")
    else
      repo = ''
    end

    os = lookupvar("::os")

    model = {
      :title => app,
      :osfamily => os['family'],
      :operatingsystem => os['name'],
      :operatingsystemmajrelease => os['release']['major'],
      :operatingsystemrelease => os['release']['full'],
      :repo => repo,
    }

    hiera[:hierarchy].reverse!.each { | p |
      conf_file_path = mp.path + '/data/' + p % model + '.yaml'

      if File.exist?(conf_file_path)
        conf_file = File.open(conf_file_path)
        got_value = YAML::load(conf_file)

        if res == 'settings'
          got_value = got_value.include?(key) ? got_value[key] : got_value['default::settings']
        else
          got_value = got_value.include?(key) ? got_value[key] : {}
        end

        unless got_value.nil?
          value = function_deep_merge([value,got_value]) if look=='deep_merge'
          value.merge!(got_value) if look=='merge'
          value=got_value if look=='direct'
        end
        conf_file.close
      end
    }
    hiera_file.close

    value.merge!({'package_name' => app}) if default_fallback

  else
    raise(Puppet::ParseError, "Could not find module #{datamodule} in environment #{compiler.environment}")
  end

  return value

end