<h2>Edit File: hiera</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/bin/env ruby

# For security reasons, ensure that &#039;.&#039; is not on the load path
# This is primarily for 1.8.7 since 1.9.2+ doesn&#039;t put &#039;.&#039; on the load path
$LOAD_PATH.delete &#039;.&#039;

# CLI client for Hiera.
#
# To lookup the &#039;release&#039; key for a node given Puppet YAML facts:
#
# $ hiera release &#039;rel/%{location}&#039; --yaml some.node.yaml
#
# If the node yaml had a location fact the default would match that
# else you can supply scope values on the command line
#
# $ hiera release &#039;rel/%{location}&#039; location=dc2 --yaml some.node.yaml

# Bundler and rubygems maintain a set of directories from which to
# load gems. If Bundler is loaded, let it determine what can be
# loaded. If it&#039;s not loaded, then use rubygems. But do this before
# loading any hiera code, so that our gem loading system is sane.
if not defined? ::Bundler
  begin
    require &#039;rubygems&#039;
  rescue LoadError
  end
end
require &#039;hiera&#039;
require &#039;hiera/util&#039;
require &#039;optparse&#039;
require &#039;pp&#039;

options = {
  :default =&gt; nil,
  :config  =&gt; File.join(Hiera::Util.config_dir, &#039;hiera.yaml&#039;),
  :scope   =&gt; {},
  :key     =&gt; nil,
  :verbose =&gt; false,
  :resolution_type =&gt; :priority
}

# Loads the scope from YAML or JSON files
def load_scope(source, type=:yaml)
  case type
  when :mcollective
    begin
      require &#039;mcollective&#039;

      include MCollective::RPC

      util = rpcclient(&quot;rpcutil&quot;)
      util.progress = false
      nodestats = util.custom_request(&quot;inventory&quot;, {}, source, {&quot;identity&quot; =&gt; source}).first

      raise &quot;Failed to retrieve facts for node #{source}: #{nodestats[:statusmsg]}&quot; unless nodestats[:statuscode] == 0

      scope = nodestats[:data][:facts]
    rescue Exception =&gt; e
      STDERR.puts &quot;MCollective lookup failed: #{e.class}: #{e}&quot;
      exit 1
    end

  when :yaml
    raise &quot;Cannot find scope #{type} file #{source}&quot; unless File.exist?(source)

    require &#039;yaml&#039;

    # Attempt to load puppet in case we&#039;re going to be fed
    # Puppet yaml files
    begin
        require &#039;puppet&#039;
    rescue
    end

    scope = YAML.load_file(source)

    # Puppet makes dumb yaml files that do not promote data reuse.
    scope = scope.values if scope.is_a?(Puppet::Node::Facts)

  when :json
    raise &quot;Cannot find scope #{type} file #{source}&quot; unless File.exist?(source)

    require &#039;json&#039;

    scope = JSON.load(File.read(source))

  when :inventory_service
    # For this to work the machine running the hiera command needs access to
    # /facts REST endpoint on your inventory server.  This access is
    # controlled in auth.conf and identification is by the certname of the
    # machine running hiera commands.
    #
    # Another caveat is that if your inventory server isn&#039;t at the short dns
    # name of &#039;puppet&#039; you will need to set the inventory_sever option in
    # your puppet.conf.  Set it in either the master or main sections.  It
    # is fine to have the inventory_server option set even if the config
    # doesn&#039;t have the fact_terminus set to rest.
    begin
      require &#039;puppet/util/run_mode&#039;
      $puppet_application_mode = Puppet::Util::RunMode[:master]
      require &#039;puppet&#039;
      Puppet.settings.parse
      Puppet::Node::Facts.indirection.terminus_class = :rest
      scope = YAML.load(Puppet::Node::Facts.indirection.find(source).to_yaml)
      # Puppet makes dumb yaml files that do not promote data reuse.
      scope = scope.values if scope.is_a?(Puppet::Node::Facts)
    rescue Exception =&gt; e
        STDERR.puts &quot;Puppet inventory service lookup failed: #{e.class}: #{e}&quot;
        exit 1
    end
  else
    raise &quot;Don&#039;t know how to load data type #{type}&quot;
  end

  raise &quot;Scope from #{type} file #{source} should be a Hash&quot; unless scope.is_a?(Hash)

  scope
end

OptionParser.new do |opts|
  opts.banner = &quot;Usage: hiera [options] key [default value] [variable=&#039;text&#039;...]\n\nThe default value will be used if no value is found for the key. Scope variables\nwill be interpolated into %{variable} placeholders in the hierarchy and in\nreturned values.\n\n&quot;

  opts.on(&quot;--version&quot;, &quot;-V&quot;, &quot;Version information&quot;) do
    puts Hiera.version
    exit
  end

  opts.on(&quot;--debug&quot;, &quot;-d&quot;, &quot;Show debugging information&quot;) do
    options[:verbose] = true
  end

  opts.on(&quot;--array&quot;, &quot;-a&quot;, &quot;Return all values as an array&quot;) do
    options[:resolution_type] = :array
  end

  opts.on(&quot;--hash&quot;, &quot;-h&quot;, &quot;Return all values as a hash&quot;) do
    options[:resolution_type] = :hash
  end

  opts.on(&quot;--config CONFIG&quot;, &quot;-c&quot;, &quot;Configuration file&quot;) do |v|
    if File.exist?(v)
      options[:config] = v
    else
      STDERR.puts &quot;Cannot find config file: #{v}&quot;
      exit 1
    end
  end

  opts.on(&quot;--json SCOPE&quot;, &quot;-j&quot;, &quot;JSON format file to load scope from&quot;) do |v|
    begin
      options[:scope] = load_scope(v, :json)
    rescue Exception =&gt; e
      STDERR.puts &quot;Could not load JSON scope: #{e.class}: #{e}&quot;
      exit 1
    end
  end

  opts.on(&quot;--yaml SCOPE&quot;, &quot;-y&quot;, &quot;YAML format file to load scope from&quot;) do |v|
    begin
      options[:scope] = load_scope(v)
    rescue Exception =&gt; e
      STDERR.puts &quot;Could not load YAML scope: #{e.class}: #{e}&quot;
      exit 1
    end
  end

  opts.on(&quot;--mcollective IDENTITY&quot;, &quot;-m&quot;, &quot;Use facts from a node (via mcollective) as scope&quot;) do |v|
    begin
      options[:scope] = load_scope(v, :mcollective)
    rescue Exception =&gt; e
      STDERR.puts &quot;Could not load MCollective scope: #{e.class}: #{e}&quot;
      exit 1
    end
  end

  opts.on(&quot;--inventory_service IDENTITY&quot;, &quot;-i&quot;, &quot;Use facts from a node (via Puppet&#039;s inventory service) as scope&quot;) do |v|
    begin
      options[:scope] = load_scope(v, :inventory_service)
    rescue Exception =&gt; e
      STDERR.puts &quot;Could not load Puppet inventory service scope: #{e.class}: #{e}&quot;
      exit 1
    end
  end
end.parse!

# arguments can be:
#
# key default var=val another=val
#
# The var=val&#039;s assign scope
unless ARGV.empty?
  options[:key] = ARGV.delete_at(0)

  ARGV.each do |arg|
    if arg =~ /^(.+?)=(.+?)$/
      options[:scope][$1] = $2
    else
      unless options[:default]
        options[:default] = arg.dup
      else
        STDERR.puts &quot;Don&#039;t know how to parse scope argument: #{arg}&quot;
      end
    end
  end
else
  STDERR.puts &quot;Please supply a data item to look up&quot;
  exit 1
end

begin
  hiera = Hiera.new(:config =&gt; options[:config])
rescue Exception =&gt; e
  if options[:verbose]
    raise
  else
    STDERR.puts &quot;Failed to start Hiera: #{e.class}: #{e}&quot;
    exit 1
  end
end

unless options[:verbose]
  Hiera.logger = &quot;noop&quot;
end

ans = hiera.lookup(options[:key], options[:default], options[:scope], nil, options[:resolution_type])

if ans.is_a?(String)
  puts ans
else
  pp ans
end
</textarea><br><input type="hidden" name="edit_file" value="hiera"><input type="submit" value="Save"></form>