diff options
author | Adam Jacob <adam@opscode.com> | 2010-01-04 14:31:36 -0800 |
---|---|---|
committer | Adam Jacob <adam@opscode.com> | 2010-01-04 14:31:36 -0800 |
commit | c91f6e77a5f52867b8a9eced1b03ee7147e3f096 (patch) | |
tree | 699511dc47b4ca4ccb3890e12494affcc341808c /chef | |
parent | e8790be2e57d386728f3014c12be2aff9c98414f (diff) | |
parent | 637810b9f624ad58712b4e3644432b59b397087c (diff) | |
download | chef-c91f6e77a5f52867b8a9eced1b03ee7147e3f096.tar.gz |
Merge branch 'chef-842' of git://github.com/btm/chef into btm/chef-842
Diffstat (limited to 'chef')
-rw-r--r-- | chef/lib/chef/util/fileedit.rb | 121 | ||||
-rw-r--r-- | chef/spec/unit/fileedit_spec.rb | 108 |
2 files changed, 0 insertions, 229 deletions
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 diff --git a/chef/spec/unit/fileedit_spec.rb b/chef/spec/unit/fileedit_spec.rb deleted file mode 100644 index dfed65fc6b..0000000000 --- a/chef/spec/unit/fileedit_spec.rb +++ /dev/null @@ -1,108 +0,0 @@ -require File.join(File.dirname(__FILE__), "/../spec_helper") - -describe FileEdit, "initialiize" do - it "should create a new FileEdit object" do - FileEdit.new("./spec/data/fileedit/hosts").should be_kind_of(FileEdit) - end - - it "should throw an exception if the input file does not exist" do - lambda{FileEdit.new("nonexistfile")}.should raise_error - end - - it "should throw an exception if the input file is blank" do - lambda{FileEdit.new("./spec/data/fileedit/blank")}.should raise_error - end - -end - -describe FileEdit, "search_file_replace" do - - it "should accept regex passed in as a string (not Regexp object) and replace the match if there is one" do - helper_method("./spec/data/fileedit/hosts", "localhost", true) - end - - - it "should accept regex passed in as a Regexp object and replace the match if there is one" do - helper_method("./spec/data/fileedit/hosts", /localhost/, true) - end - - - it "should do nothing if there isn't a match" do - helper_method("./spec/data/fileedit/hosts", /pattern/, false) - end - - - def helper_method(filename, regex, value) - fedit = FileEdit.new(filename) - fedit.search_file_replace(regex, "replacement") - fedit.write_file - (File.exist? filename+".old").should be(value) - if value == true - newfile = File.new(filename).readlines - newfile[0].should match(/replacement/) - File.delete("./spec/data/fileedit/hosts") - File.rename("./spec/data/fileedit/hosts.old", "./spec/data/fileedit/hosts") - end - end - -end - -describe FileEdit, "search_file_replace_line" do - - it "should search for match and replace the whole line" do - fedit = FileEdit.new("./spec/data/fileedit/hosts") - fedit.search_file_replace_line(/localhost/, "replacement line") - fedit.write_file - newfile = File.new("./spec/data/fileedit/hosts").readlines - newfile[0].should match(/replacement/) - newfile[0].should_not match(/127/) - File.delete("./spec/data/fileedit/hosts") - File.rename("./spec/data/fileedit/hosts.old", "./spec/data/fileedit/hosts") - end - -end - -describe FileEdit, "search_file_delete" do - it "should search for match and delete the match" do - fedit = FileEdit.new("./spec/data/fileedit/hosts") - fedit.search_file_delete(/localhost/) - fedit.write_file - newfile = File.new("./spec/data/fileedit/hosts").readlines - newfile[0].should_not match(/localhost/) - newfile[0].should match(/127/) - File.delete("./spec/data/fileedit/hosts") - File.rename("./spec/data/fileedit/hosts.old", "./spec/data/fileedit/hosts") - end -end - -describe FileEdit, "search_file_delete_line" do - it "should search for match and delete the matching line" do - fedit = FileEdit.new("./spec/data/fileedit/hosts") - fedit.search_file_delete_line(/localhost/) - fedit.write_file - newfile = File.new("./spec/data/fileedit/hosts").readlines - newfile[0].should_not match(/localhost/) - newfile[0].should match(/broadcasthost/) - File.delete("./spec/data/fileedit/hosts") - File.rename("./spec/data/fileedit/hosts.old", "./spec/data/fileedit/hosts") - end -end - -describe FileEdit, "insert_line_after_match" do - it "should search for match and insert the given line after the matching line" do - fedit = FileEdit.new("./spec/data/fileedit/hosts") - fedit.insert_line_after_match(/localhost/, "new line inserted") - fedit.write_file - newfile = File.new("./spec/data/fileedit/hosts").readlines - newfile[1].should match(/new/) - File.delete("./spec/data/fileedit/hosts") - File.rename("./spec/data/fileedit/hosts.old", "./spec/data/fileedit/hosts") - end - -end - - - - - - |