Merge pull request #4 from blackopseng3/master

Bugfix and improvements

Author: eric-martin

README.md

@@ -34,6 +34,12 @@ stackify-api-ruby starts with start of Rails. Every error, which occurs within y
 
     Rails.logger.info "Some log message"
 
+If you want to redefine <b>logger</b>, you should add
+
+    config.logger = ::Stackify::LoggerProxy.new(ActiveSupport::Logger.new(File.join(Rails.root, 'log', "#{Rails.env}.log")))
+
+to your config/environments/%environment%. <b>Note:</b> in this case Stackify#config.log_level will affect entire system.
+
 ### Other Environment
 
 For using stackify-api-ruby gem within any Ruby application add to top of your main file:

lib/stackify/logger_proxy.rb

@@ -7,9 +7,9 @@ def initialize logger
       %w( debug info warn error fatal unknown).each do |level|
         LoggerProxy.class_eval do
           define_method level.to_sym do |*args , &block|
-            msg = message(args, block)
+            msg = message(*args, &block)
             Stackify.logger_client.log(level.downcase, msg, caller)
-            @logger.send(level.to_sym, args, &block)
+            @logger.send(level.to_sym, msg, &block)
           end
         end
       end
@@ -24,10 +24,12 @@ def method_missing(name, *args, &block)
     private
 
     def message *args, &block
-      if block
+      if block_given?
         block.call
       else
-        args.flatten[0]
+        args = args.flatten.compact
+        args = (args.count == 1 ? args[0] : args)
+        args.is_a?(Proc) ? args.call : args
       end
     end