lib/shell_shock/logger.rb in shell_shock-0.0.4 vs lib/shell_shock/logger.rb in shell_shock-0.0.5

@@ -1,10 +1,11 @@
 module ShellShock
     module Logger
-    def log message
+    def log message=nil
       return unless ENV['LOG_PATH']
       File.open(ENV['LOG_PATH'], 'a') do |file|
-        file.puts message
+        file.puts message if message
+        file.puts yield if block_given?
       end
     end
   end
 end
\ No newline at end of file