summaryrefslogtreecommitdiffstats
path: root/lib/puppet/parser/functions/write_package_names.rb
blob: 8f99674e06ebf0b5994924fe5e03e768eca67ffb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
require 'fileutils'

module Puppet::Parser::Functions
  newfunction(:write_package_names, :doc => "Write package names which are managed via this puppet run to a file.") do |arg|
    if arg[0].class == String
      begin
        output_file = arg[0]
        packages = catalog.resources.collect { |r| r.title if r.type == 'Package' }.compact
        FileUtils.mkdir_p(File.dirname(output_file))
        File.open(output_file, 'w') do |f|
            packages.each do |pkg_name|
                f.write(pkg_name + "\n")
            end
        end
      rescue JSON::ParserError
        raise Puppet::ParseError, "Syntax error: #{arg[0]} is invalid"
      end
    else
      raise Puppet::ParseError, "Syntax error: #{arg[0]} is not a String"
    end
  end
end