Untitled

#!/opt/puppet/bin/ruby

require 'puppet'

require 'puppet/face'

require 'yaml'

# assumes that this credential file has been laid down

# should contain credentials with cloudformation:DescribeStackResource

# access to the resource being classified

credential_file='/var/lib/cfn-init/data/cfn-credentials'

 

# figure out which facts terminus to use

# is this the best way to set runmode out side of an application?

$puppet_application_mode = Puppet::Util::RunMode[:master]

Puppet[:config]='/etc/puppetlabs/puppet/puppet.conf'

Puppet.parse_config

facts = Puppet::Face[:facts, :current].find(ARGV.first)

if facts

  # right now this is hard coded for region us-east-1

  # the stackname and resource id facts are currently laid

  # down by the cloud formation script

  if facts.values['cfn_stack_name'] and facts.values['cfn_resource_id'] and File.exists?(credential_file)

    command = "/opt/aws/bin/cfn-get-metadata --region #{facts.values['cfn_region']} -s #{facts.values['cfn_stack_name']} -r #{facts.values['cfn_resource_id']} -f #{credential_file} --key Puppet"

    classification_meta_data = `#{command}`.downcase

    unless classification_meta_data == ''

      puts PSON.parse(classification_meta_data).to_yaml

      exit 0

    end

  end

end

puts({'classes' => []}.to_yaml)