lib/yard/logging.rb in yard-0.2.3.2 vs lib/yard/logging.rb in yard-0.2.3.3

@@ -1,12 +1,17 @@
-require "logger"
+require 'logger'
 
 module YARD
   class Logger < ::Logger
+    def self.instance(pipe = STDERR)
+      @logger ||= new(pipe)
+    end
+    
     def initialize(*args)
       super
       self.level = INFO
+      self.formatter = method(:format_log)
     end
     
     def debug(*args)
       self.level = DEBUG if $DEBUG
       super
@@ -15,13 +20,11 @@
     def enter_level(new_level = level, &block) 
       old_level, self.level = level, new_level
       yield
       self.level = old_level
     end
+    
+    def format_log(sev, time, prog, msg)
+      "[#{sev.downcase}]: #{msg}\n"
+    end
   end
-  
-  def self.logger
-    @logger ||= YARD::Logger.new(STDERR)
-  end
 end
-
-def log; YARD.logger end
\ No newline at end of file