Untitled

#!/usr/bin/env ruby

require 'yaml'
require 'puppet'

vardir_path = "/var/db/puppet/yaml/facts/"

if File.exists?(vardir_path) && File.directory?(vardir_path)
  puts "Var DB"
else
  vardir_path = "/var/puppet/yaml/facts/"
  puts "Var Puppet"
end

yamlfile = YAML::load_file(vardir_path + ARGV[0] + '.yaml').values

fqdn = yamlfile["fqdn"]

function = %x{echo #{fqdn} | awk -F '-' '{print $2}'}.chomp
building = %x{echo #{fqdn} | awk -F '-' '{print $3}'}.chomp.gsub!(/[.].*/,"")


output = {'classes' => [ 'general_image', building , function ]}
print output.to_yaml