log-block 0.1.0 → 0.2.0

data/lib/log_block.rb

@@ -1,32 +1,44 @@
 require 'logger'
 
-class Logger
-  def with( prefix, indent_size = 2 )
-    yield get_block_logger( prefix, indent_size )
-  end
-  
-  def get_block_logger( prefix, indent_size = 2)
-    BlockLogger.new( self, prefix, indent_size )
-  end
-end
+module LogBlock
+  module InstanceMethods
+    def with( prefix, indent_size = 2 )
+      yield get_block_logger( prefix, indent_size )
+    end
 
-class BlockLogger
-  def initialize( logger, prefix, indent_size = 2 )
-    @logger = logger
-    @indent_size = indent_size
-    @prefix = prefix
+    def get_block_logger( prefix, indent_size = 2)
+      ::LogBlock::BlockLogger.new( self, prefix, indent_size )
+    end    
   end
 
-  def method_missing( meth, *args )
-    if [:debug, :info, :warn, :error, :fatal, :unknown].include?(meth)
-      @logger.send(meth, message(args.first))
-    else
-      super
+  class BlockLogger
+    def initialize( logger, prefix, indent_size = 2 )
+      @logger = logger
+      @indent_size = indent_size
+      @prefix = prefix
     end
+
+    def method_missing( meth, *args )
+      if [:debug, :info, :warn, :error, :fatal, :unknown].include?(meth)
+        @logger.send(meth, message(args.first))
+      else
+        super
+      end
+    end
+
+    def message( msg )
+      "#{' ' * @indent_size}#{@prefix}#{msg}"
+    end
+
   end
-  
-  def message( msg )
-    "#{' ' * @indent_size}#{@prefix}#{msg}"
-  end
-  
+
+end
+
+if defined?(Logger)
+  Logger.send(:include, LogBlock::InstanceMethods)
+end
+
+if defined?(ActiveSupport) and defined?(ActiveSupport::BufferedLogger)
+  ActiveSupport::BufferedLogger.send(:include, LogBlock::InstanceMethods)
 end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: log-block
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Jon Moses