Add package_manifest resource.

This patch converts the write_package_names function into
a proper resource. Using the write_package_names only works
if the function comes last in the puppet manifest. By
making the same functionality a custom resource we allow
for it to exist anywhere in the manifest and provide the
same functionality.

 The new syntax would be:

 package_manifest{'/tmp/foo': ensure => present}

Co-Authored-By: Martin Mágr <mmagr@redhat.com>
Change-Id: If3e03b1983fed47082fac8ce63f975557dbc503c
This commit is contained in:
Dan Prince
2015-07-20 16:18:52 -04:00
committed by Martin Mágr
parent c529fce3ef
commit e3abcd6b53
4 changed files with 84 additions and 22 deletions

View File

@@ -1,22 +0,0 @@
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

View File

@@ -0,0 +1,39 @@
require 'set'
Puppet::Type.type(:package_manifest).provide(:flat_file) do
desc "Write package manifest to a flat file"
def exists?
# exists? is always run before create, so we can create package list here
@packages = resource.catalog.resources.collect { |r|
r.name if r.type == :package
}.compact.sort
exists = File.exist?(resource[:path])
if exists
new_content = Set.new @packages
old_content = Set.new(
File.open(resource[:path], 'r').each_line.collect{ |pkg| pkg.strip() }
)
exists = new_content == old_content
end
exists
end
def create
FileUtils.mkdir_p(File.dirname(resource[:path]))
File.open(resource[:path], 'w') do |f|
@packages.each do |pkg_name|
f.puts(pkg_name)
end
end
end
def destroy
File.delete(resource[:path])
end
end

View File

@@ -0,0 +1,8 @@
Puppet::Type.newtype(:package_manifest) do
ensurable
newparam(:path, :namevar => true) do
newvalues(/\S+\/\S+/)
end
end

View File

@@ -0,0 +1,37 @@
require 'puppet'
require 'puppet/type/package_manifest'
describe 'Puppet::Type.type(:package_manifest)' do
before :each do
@manifest = Puppet::Type.type(:package_manifest).new(
:path => '/tmp/test_package_manifest.txt', :ensure => 'present'
)
end
it 'should require a path' do
expect {
Puppet::Type.type(:package_manifest).new({})
}.to raise_error Puppet::Error
end
it 'should not require a value when ensure is absent' do
Puppet::Type.type(:package_manifest).new(
:path => '/tmp/test_package_manifest.txt', :ensure => :absent
)
end
it 'should accept valid ensure values' do
@manifest[:ensure] = :present
expect(@manifest[:ensure]).to eq(:present)
@manifest[:ensure] = :absent
expect(@manifest[:ensure]).to eq(:absent)
end
it 'should not accept invalid ensure values' do
expect {
@manifest[:ensure] = :latest
}.to raise_error(Puppet::Error, /Invalid value/)
end
end