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:
@@ -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
|
39
lib/puppet/provider/package_manifest/flat_file.rb
Normal file
39
lib/puppet/provider/package_manifest/flat_file.rb
Normal 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
|
8
lib/puppet/type/package_manifest.rb
Normal file
8
lib/puppet/type/package_manifest.rb
Normal file
@@ -0,0 +1,8 @@
|
||||
Puppet::Type.newtype(:package_manifest) do
|
||||
|
||||
ensurable
|
||||
newparam(:path, :namevar => true) do
|
||||
newvalues(/\S+\/\S+/)
|
||||
end
|
||||
|
||||
end
|
37
spec/unit/type/package_manifest_spec.rb
Normal file
37
spec/unit/type/package_manifest_spec.rb
Normal 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
|
Reference in New Issue
Block a user