Sha256: 5296952ec1555e66515839fe1db1c60fc47dcaf2ea7c4c66546ebec70bbe9733

Contents?: true

Size: 1.76 KB

Versions: 20

Compression:

Stored size: 1.76 KB

Contents

# frozen_string_literal: true

Puppet.features.hiera?
require 'hiera/scope'
require_relative 'puppet'

module HieraPuppet
  module_function

  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

  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

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

  def hiera_config
    config = {}

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

    config[:logger] = 'puppet'
    config
  end

  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
end

Version data entries

20 entries across 20 versions & 1 rubygems

Version Path
puppet-8.10.0 lib/hiera_puppet.rb
puppet-8.10.0-x86-mingw32 lib/hiera_puppet.rb
puppet-8.10.0-x64-mingw32 lib/hiera_puppet.rb
puppet-8.10.0-universal-darwin lib/hiera_puppet.rb
puppet-8.9.0 lib/hiera_puppet.rb
puppet-8.9.0-x86-mingw32 lib/hiera_puppet.rb
puppet-8.9.0-x64-mingw32 lib/hiera_puppet.rb
puppet-8.9.0-universal-darwin lib/hiera_puppet.rb
puppet-8.8.1 lib/hiera_puppet.rb
puppet-8.8.1-x86-mingw32 lib/hiera_puppet.rb
puppet-8.8.1-x64-mingw32 lib/hiera_puppet.rb
puppet-8.8.1-universal-darwin lib/hiera_puppet.rb
puppet-8.7.0 lib/hiera_puppet.rb
puppet-8.7.0-x86-mingw32 lib/hiera_puppet.rb
puppet-8.7.0-x64-mingw32 lib/hiera_puppet.rb
puppet-8.7.0-universal-darwin lib/hiera_puppet.rb
puppet-8.6.0 lib/hiera_puppet.rb
puppet-8.6.0-x86-mingw32 lib/hiera_puppet.rb
puppet-8.6.0-x64-mingw32 lib/hiera_puppet.rb
puppet-8.6.0-universal-darwin lib/hiera_puppet.rb