1 files changed, 78 insertions, 0 deletions
diff --git a/railties/lib/vendor/thor-0.11.5/lib/thor/actions/inject_into_file.rb b/railties/lib/vendor/thor-0.11.5/lib/thor/actions/inject_into_file.rb
new file mode 100644
index 0000000000..66dd1f5fc1
--- /dev/null
+++ b/railties/lib/vendor/thor-0.11.5/lib/thor/actions/inject_into_file.rb
@@ -0,0 +1,78 @@
+require 'thor/actions/empty_directory'
+
+class Thor
+  module Actions
+
+    # Injects the given content into a file. Different from append_file,
+    # prepend_file and gsub_file, this method is reversible. By this reason,
+    # the flag can only be strings. gsub_file is your friend if you need to
+    # deal with more complex cases.
+    #
+    # ==== Parameters
+    # destination<String>:: Relative path to the destination root
+    # data<String>:: Data to add to the file. Can be given as a block.
+    # config<Hash>:: give :verbose => false to not log the status and the flag
+    #                for injection (:after or :before).
+    # 
+    # ==== Examples
+    #
+    #   inject_into_file "config/environment.rb", "config.gem thor", :after => "Rails::Initializer.run do |config|\n"
+    #
+    #   inject_into_file "config/environment.rb", :after => "Rails::Initializer.run do |config|\n" do
+    #     gems = ask "Which gems would you like to add?"
+    #     gems.split(" ").map{ |gem| "  config.gem #{gem}" }.join("\n")
+    #   end
+    #
+    def inject_into_file(destination, *args, &block)
+      if block_given?
+        data, config = block, args.shift
+      else
+        data, config = args.shift, args.shift
+      end
+
+      log_status = args.empty? || args.pop
+      action InjectIntoFile.new(self, destination, data, config)
+    end
+
+    class InjectIntoFile < EmptyDirectory #:nodoc:
+      attr_reader :flag, :replacement
+
+      def initialize(base, destination, data, config)
+        super(base, destination, { :verbose => true }.merge(config))
+
+        data = data.call if data.is_a?(Proc)
+
+        @replacement = if @config.key?(:after)
+          @flag = @config.delete(:after)
+          @flag + data
+        else
+          @flag = @config.delete(:before)
+          data + @flag
+        end
+      end
+
+      def invoke!
+        say_status :inject, config[:verbose]
+        replace!(flag, replacement)
+      end
+
+      def revoke!
+        say_status :deinject, config[:verbose]
+        replace!(replacement, flag)
+      end
+
+      protected
+
+        # Adds the content to the file.
+        #
+        def replace!(regexp, string)
+          unless base.options[:pretend]
+            content = File.read(destination)
+            content.gsub!(regexp, string)
+            File.open(destination, 'wb') { |file| file.write(content) }
+          end
+        end
+
+    end
+  end
+end