Module: HieraPuppet

Defined in:
lib/hiera_puppet.rb

Class Method Summary collapse

Class Method Details

.hiera ⇒ Object 



53
54
55
 
# File 'lib/hiera_puppet.rb', line 53

def hiera
  @hiera ||= Hiera.new(:config => hiera_config)
end

.hiera_config ⇒ Object 



57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/hiera_puppet.rb', line 57

def hiera_config
  config = {}

  config_file = hiera_config_file
  if config_file
    config = Hiera::Config.load(config_file)
  end

  config[:logger] = 'puppet'
  config
end

.hiera_config_file ⇒ Object 



69
70
71
72
73
74
75
76
77
 
# File 'lib/hiera_puppet.rb', line 69

def hiera_config_file
  hiera_config = Puppet.settings[:hiera_config]
  if Puppet::FileSystem.exist?(hiera_config)
    hiera_config
  else
    Puppet.warning _("Config file %{hiera_config} not found, using Hiera defaults") % { hiera_config: hiera_config }
    nil
  end
end

.lookup(key, default, scope, override, resolution_type) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/hiera_puppet.rb', line 10

def lookup(key, default, scope, override, resolution_type)
  scope = Hiera::Scope.new(scope)

  answer = hiera.lookup(key, default, scope, override, resolution_type)

  if answer.nil?
    raise Puppet::ParseError, _("Could not find data item %{key} in any Hiera data file and no default supplied") % { key: key }
  end

  answer
end

.parse_args(args) ⇒ Object 



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
 
# File 'lib/hiera_puppet.rb', line 22

def parse_args(args)
  # Functions called from Puppet manifests like this:
  #
  #   hiera("foo", "bar")
  #
  # Are invoked internally after combining the positional arguments into a
  # single array:
  #
  #   func = function_hiera
  #   func(["foo", "bar"])
  #
  # Functions called from templates preserve the positional arguments:
  #
  #   scope.function_hiera("foo", "bar")
  #
  # Deal with Puppet's special calling mechanism here.
  if args[0].is_a?(Array)
    args = args[0]
  end

  if args.empty?
    raise Puppet::ParseError, _("Please supply a parameter to perform a Hiera lookup")
  end

  key      = args[0]
  default  = args[1]
  override = args[2]

  [key, default, override]
end