lib/departure/logger.rb in departure-1.0.1 vs lib/departure/logger.rb in departure-2.0.0

@@ -3,10 +3,14 @@
   # #write_no_newline to log the migration's status. It's not possible to reuse
   # the from ActiveRecord::Migration because the migration's instance can't be
   # seen from the connection adapter.
   class Logger
 
+    def initialize(sanitizers)
+      @sanitizers = sanitizers
+    end
+
     # Outputs the message through the stdout, following the
     # ActiveRecord::Migration log format
     #
     # @param message [String]
     # @param subitem [Boolean] whether to show message as a nested log item
@@ -16,16 +20,24 @@
 
     # Outputs the text through the stdout adding a new line at the end
     #
     # @param text [String]
     def write(text = '')
-      puts(text)
+      puts(sanitize(text))
     end
 
     # Outputs the text through the stdout without adding a new line at the end
     #
     # @param text [String]
     def write_no_newline(text)
-      print(text)
+      print(sanitize(text))
+    end
+
+    private
+
+    attr_accessor :sanitizers
+
+    def sanitize(text)
+      sanitizers.inject(text) { |memo, sanitizer| sanitizer.execute(memo) }
     end
   end
 end