Librarian-Puppet Rake Task for Templating the Puppetfile

librarian-puppet-rake-task

#!/usr/bin/ruby

require 'erb'

desc "Templates the Puppetfile to the specified environment"
task :template_puppetfile, [:env] do |t, args|
  unless args[:env].nil?
    @environment = args[:env]
    read_file = 'Puppetfile.erb'
    write_file = 'Puppetfile'

    erb = ERB.new(File.read(read_file))
    File.open(write_file, 'w') { |file| file.write(erb.result) }
  else
    puts "An environment must be specified."
  end
end