X-Git-Url: https://git.openstreetmap.org/chef.git/blobdiff_plain/7b2e5c20165878df65a244b7eb34bf6ac1b35782..48debd1ec47bbada97c4ef9ec7ae7e4daf3ded53:/cookbooks/chef/libraries/edit_file.rb

diff --git a/cookbooks/chef/libraries/edit_file.rb b/cookbooks/chef/libraries/edit_file.rb
index e632bcfd4..6260c8a40 100644
--- a/cookbooks/chef/libraries/edit_file.rb
+++ b/cookbooks/chef/libraries/edit_file.rb
@@ -1,22 +1,17 @@
 class Chef
-  class Util
-    class EditedFile
-      def initialize(file, block)
-        @file = file
-        @block = block
-      end
-
-      def to_s
-        ::File.new(@file).collect do |line|
-          line = @block.call(line)
-        end.join("")
+  module Mixin
+    module EditFile
+      def edit_file(file, &block)
+        Chef::DelayedEvaluator.new do
+          ::File.new(file).collect do |line|
+            block.call(line)
+          end.join("")
+        end
       end
     end
   end
 
   class Recipe
-    def edit_file(file, &block)
-      Chef::Util::EditedFile.new(file, block).to_s
-    end
+    include Chef::Mixin::EditFile
   end
 end