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 | |
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')
-rw-r--r-- | chef/lib/chef/util/file_edit.rb | 125 | ||||
-rw-r--r-- | chef/lib/chef/util/fileedit.rb | 121 |
2 files changed, 125 insertions, 121 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 diff --git a/chef/lib/chef/util/fileedit.rb b/chef/lib/chef/util/fileedit.rb deleted file mode 100644 index a6ac99352b..0000000000 --- a/chef/lib/chef/util/fileedit.rb +++ /dev/null @@ -1,121 +0,0 @@ -# -# 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 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 |