fix block passing

Author: Zeko369

lib/coralogix_logger.rb

@@ -128,7 +128,13 @@ def log severity, message, category: @category, className: "", methodName: "", t
         # For instance, for info severity it will create a method:
         # def info message, category: @category, className: "", methodName: "", threadId: ""
         SEVERITIES.keys.each do |severity|
-            define_method("#{severity}") do |message, category: @category, className: "", methodName: "", threadId: Thread.current.object_id.to_s|
+            define_method("#{severity}") do |message=nil, category: @category, className: "", methodName: "", threadId: Thread.current.object_id.to_s|
+                if block_given?
+                    message = yield
+                elsif message.nil?
+                    raise ArgumentError, "A message must be provided"
+                end
+
                 LoggerManager.add_logline message, SEVERITIES["#{__method__}".to_sym], category, :className => className, :methodName => methodName, :threadId => threadId
             end