diff options
author | Dan Bode <dan@puppetlabs.com> | 2012-10-10 17:40:38 -0700 |
---|---|---|
committer | Dan Bode <dan@puppetlabs.com> | 2012-10-10 17:40:38 -0700 |
commit | 1106d70e881028ee2dfa476307444780c9c4cbaa (patch) | |
tree | d58ff7e2e0e2a96f8bed7c26c09c5962626d5dd0 /lib/puppet | |
parent | cbc90d38347f8ca61de2208b16ffc76f9fd5938c (diff) | |
download | puppet-inifile-1106d70e881028ee2dfa476307444780c9c4cbaa.tar.gz puppet-inifile-1106d70e881028ee2dfa476307444780c9c4cbaa.tar.bz2 |
Add support for removing lines
This commit adds the ability to ensure that lines are absent.
Diffstat (limited to 'lib/puppet')
-rw-r--r-- | lib/puppet/provider/ini_setting/ruby.rb | 6 | ||||
-rw-r--r-- | lib/puppet/util/ini_file.rb | 23 | ||||
-rw-r--r-- | lib/puppet/util/ini_file/section.rb | 6 |
3 files changed, 33 insertions, 2 deletions
diff --git a/lib/puppet/provider/ini_setting/ruby.rb b/lib/puppet/provider/ini_setting/ruby.rb index 946685e..2c13847 100644 --- a/lib/puppet/provider/ini_setting/ruby.rb +++ b/lib/puppet/provider/ini_setting/ruby.rb @@ -12,6 +12,12 @@ Puppet::Type.type(:ini_setting).provide(:ruby) do @ini_file = nil end + def destroy + ini_file.remove_setting(section, setting) + ini_file.save + @ini_file = nil + end + def value ini_file.get_value(section, setting) end diff --git a/lib/puppet/util/ini_file.rb b/lib/puppet/util/ini_file.rb index b2e554f..107924d 100644 --- a/lib/puppet/util/ini_file.rb +++ b/lib/puppet/util/ini_file.rb @@ -42,17 +42,28 @@ module Util end end + def remove_setting(section_name, setting) + section = @sections_hash[section_name] + if (section.has_existing_setting?(setting)) + remove_line(section, setting) + section.remove_existing_setting(setting) + end + end + def save File.open(@path, 'w') do |fh| @section_names.each do |name| + section = @sections_hash[name] if section.start_line.nil? fh.puts("\n[#{section.name}]") elsif ! section.end_line.nil? (section.start_line..section.end_line).each do |line_num| - fh.puts(lines[line_num]) + if lines[line_num] + fh.puts(lines[line_num]) + end end end @@ -113,6 +124,16 @@ module Util end end + def remove_line(section, setting) + (section.start_line..section.end_line).each do |line_num| + if (match = SETTING_REGEX.match(lines[line_num])) + if (match[1] == setting) + lines.delete_at(line_num) + end + end + end + end + def create_line_iter ExternalIterator.new(lines) end diff --git a/lib/puppet/util/ini_file/section.rb b/lib/puppet/util/ini_file/section.rb index 39f2959..e192d5b 100644 --- a/lib/puppet/util/ini_file/section.rb +++ b/lib/puppet/util/ini_file/section.rb @@ -24,6 +24,10 @@ class IniFile @existing_settings[setting_name] = value end + def remove_existing_setting(setting_name) + @existing_settings.delete(setting_name) + end + def set_additional_setting(setting_name, value) @additional_settings[setting_name] = value end @@ -31,4 +35,4 @@ class IniFile end end end -end
\ No newline at end of file +end |