diff options
author | Travis Fields <travis@puppetlabs.com> | 2014-10-24 14:30:01 -0700 |
---|---|---|
committer | Travis Fields <travis@puppetlabs.com> | 2014-10-24 14:30:01 -0700 |
commit | f0b207c03587f5c92915f86051dc139b939c53c0 (patch) | |
tree | 9e75a36b80885e255b1aa9dc5cb93de962bda3a8 /lib/puppet/provider/file_line | |
parent | 0c0f7e13cdb407ca57d2808de9e6dc6ace9d856b (diff) | |
parent | cb6c7d8e49cdd392dfa55c27cf78795125163c70 (diff) |
Merge branch '4.3.x'
Conflicts:
README.markdown
lib/puppet/parser/functions/has_interface_with.rb
Diffstat (limited to 'lib/puppet/provider/file_line')
-rw-r--r-- | lib/puppet/provider/file_line/ruby.rb | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/lib/puppet/provider/file_line/ruby.rb b/lib/puppet/provider/file_line/ruby.rb index 94e7fac..ae1a8b3 100644 --- a/lib/puppet/provider/file_line/ruby.rb +++ b/lib/puppet/provider/file_line/ruby.rb @@ -34,7 +34,7 @@ Puppet::Type.type(:file_line).provide(:ruby) do def handle_create_with_match() regex = resource[:match] ? Regexp.new(resource[:match]) : nil - match_count = lines.select { |l| regex.match(l) }.size + match_count = count_matches(regex) if match_count > 1 && resource[:multiple].to_s != 'true' raise Puppet::Error, "More than one line in file '#{resource[:path]}' matches pattern '#{resource[:match]}'" end @@ -51,9 +51,7 @@ Puppet::Type.type(:file_line).provide(:ruby) do def handle_create_with_after regex = Regexp.new(resource[:after]) - - count = lines.count {|l| l.match(regex)} - + count = count_matches(regex) case count when 1 # find the line to put our line after File.open(resource[:path], 'w') do |fh| @@ -71,6 +69,10 @@ Puppet::Type.type(:file_line).provide(:ruby) do end end + def count_matches(regex) + lines.select{|l| l.match(regex)}.size + end + ## # append the line to the file. # |