diff options
author | Adam Jacob <adam@hjksolutions.com> | 2009-04-24 17:44:02 -0700 |
---|---|---|
committer | Adam Jacob <adam@hjksolutions.com> | 2009-04-24 17:44:02 -0700 |
commit | b00c38b7bd59c68a2900bbf5b49dc98a9620dda4 (patch) | |
tree | 1495f2ac359be36c2c25b6ff1848e30edb54a1ea /chef/lib/chef/util/file_edit.rb | |
parent | ccdc688d4e68769010e19c8df60b7d1cd3437c64 (diff) | |
download | chef-b00c38b7bd59c68a2900bbf5b49dc98a9620dda4.tar.gz |
Fixing the file name for CamelCase, moving the spec to the right directory, ensuring the header is correct
Diffstat (limited to 'chef/lib/chef/util/file_edit.rb')
-rw-r--r-- | chef/lib/chef/util/file_edit.rb | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/chef/lib/chef/util/file_edit.rb b/chef/lib/chef/util/file_edit.rb new file mode 100644 index 0000000000..c09e427741 --- /dev/null +++ b/chef/lib/chef/util/file_edit.rb @@ -0,0 +1,125 @@ +# +# Author:: Nuo Yan (<nuo@opscode.com>) +# Copyright:: Copyright (c) 2009 Opscode, Inc. +# License:: Apache License, Version 2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +require 'ftools' +require 'fileutils' +require 'tempfile' + +class Chef + class Util + class FileEdit + + private + + attr_accessor :original_pathname, :contents, :file_edited + + public + + def initialize(filepath) + @original_pathname = filepath + @file_edited = false + + raise ArgumentError, "File doesn't exist" unless File.exist? @original_pathname + raise ArgumentError, "File is blank" unless (@contents = File.new(@original_pathname).readlines).length > 0 + end + + #search the file line by line and match each line with the given regex + #if matched, replace the whole line with newline. + def search_file_replace_line(regex, newline) + search_match(regex, newline, 'r', 1) + end + + #search the file line by line and match each line with the given regex + #if matched, replace the match (all occurances) with the replace parameter + def search_file_replace(regex, replace) + search_match(regex, replace, 'r', 2) + end + + #search the file line by line and match each line with the given regex + #if matched, delete the line + def search_file_delete_line(regex) + search_match(regex, " ", 'd', 1) + end + + #search the file line by line and match each line with the given regex + #if matched, delete the match (all occurances) from the line + def search_file_delete(regex) + search_match(regex, " ", 'd', 2) + end + + #search the file line by line and match each line with the given regex + #if matched, insert newline after each matching line + def insert_line_after_match(regex, newline) + search_match(regex, newline, 'i', 0) + end + + #Make a copy of old_file and write new file out (only if file changed) + def write_file + + # file_edited is false when there was no match in the whole file and thus no contents have changed. + if file_edited + backup_pathname = original_pathname + ".old" + File.copy(original_pathname, backup_pathname) + Tempfile.open("w") do |newfile| + contents.each do |line| + newfile.puts(line) + end + newfile.flush + FileUtils.mv(newfile.path, original_pathname) + end + end + self.file_edited = false + + end + + private + + #helper method to do the match, replace, delete, and insert operations + #command is the switch of delete, replace, and insert ('d', 'r', 'i') + #method is to control operation on whole line or only the match (1 for line, 2 for match) + def search_match(regex, replace, command, method) + + #convert regex to a Regexp object (if not already is one) and store it in exp. + exp = Regexp.new(regex) + + #loop through contents and do the appropriate operation depending on 'command' and 'method' + new_contents = [] + + contents.each do |line| + if line.match(exp) + self.file_edited = true + case + when command == 'r' + new_contents << ((method == 1) ? replace : line.gsub!(exp, replace)) + when command == 'd' + if method == 2 + new_contents << line.gsub!(exp, "") + end + when command == 'i' + new_contents << line + new_contents << replace + end + else + new_contents << line + end + end + + self.contents = new_contents + end + end + end +end |