| Current Path : /proc/thread-self/root/usr/bin/ |
Linux gator3171.hostgator.com 4.19.286-203.ELK.el7.x86_64 #1 SMP Wed Jun 14 04:33:55 CDT 2023 x86_64 |
| Current File : //proc/thread-self/root/usr/bin/extlookup2hiera |
#!/usr/bin/ruby
require 'optparse'
require 'csv'
options = {:in => nil, :out => nil, :format => :yaml}
OptionParser.new do |opts|
opts.banner = "Converter for extlookup CSV files into Hiera JSON and YAML files"
opts.on("--in FILE", "-i", "Input CSV file") do |v|
options[:in] = v
end
opts.on("--out FILE", "-o", "Output Hiera file") do |v|
options[:out] = v
end
opts.on("--json", "-j", "Create JSON format file") do |v|
options[:format] = :json
end
end.parse!
if options[:in].nil? || options[:out].nil?
STDERR.puts "Please specify an input and output file with --in and --out"
exit 1
end
unless File.exist?(options[:in])
STDERR.puts "Cannot find input file #{options[:in]}"
exit 1
end
csvdata = CSV.read(options[:in])
hieradata = {}
csvdata.each do |d|
d = d.map{|item| item.to_s}
if d.size > 2
hieradata[d[0]] = d[1, d.size].flatten
else
hieradata[d[0]] = d[1]
end
end
case options[:format]
when :yaml
require 'yaml'
File.open(options[:out], "w") {|f| f.write hieradata.to_yaml}
when :json
require 'rubygems'
require 'json'
File.open(options[:out], "w") {|f| f.write JSON.pretty_generate hieradata}
end