taglog 0.0.1 → 0.0.2

data/lib/taglog.rb

@@ -29,25 +29,32 @@ class Taglog < Module
     end
 
     LEVELS.each do |level|
-      define_method(level) do |message|
-        delegate(level, tagged(message))
-      end
+      class_eval %Q{
+        def #{level}(arg=nil, &block)
+          if block_given?
+            delegate(:#{level}, arg) { tagged(block.call) }
+          else
+            delegate(:#{level}, tagged(arg))
+          end
+        end
+      }
     end
 
-    def method_missing(name, *args)
-      delegate(name, *args)
+    def method_missing(name, *args, &block)
+      delegate(name, *args, &block)
     end
 
-    private
-
     attr_reader :context, :tag
+    private :context, :tag
+
+    private
 
     def tagged(message)
       "[#{tag}] #{message}"
     end
 
-    def delegate(method_name, *args)
-      logger.send(method_name, *args)
+    def delegate(method_name, *args, &block)
+      logger.send(method_name, *args, &block)
     end
 
     def logger

data/lib/taglog/version.rb

@@ -1,3 +1,3 @@
 class Taglog < Module
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/taglog_spec.rb

@@ -18,7 +18,7 @@ describe Taglog do
   end
 
   let(:formatter) do
-    proc {|_, _, _, message| message }
+    proc {|_, _, program_name, message| "#{program_name}: #{message}" }
   end
 
   let(:io) do
@@ -44,9 +44,19 @@ describe Taglog do
     end
 
     describe "#info" do
-      it "is wrapped by tag" do
+      it "is wrapped by tag for an argument" do
         klass.new.logger.info("message")
-        result.should == "[tag] message"
+        result.should == ": [tag] message"
+      end
+
+      it "is wrapped by tag for return value of block" do
+        klass.new.logger.info { "block" + "message" }
+        result.should == ": [tag] blockmessage"
+      end
+
+      it "is wrapped by tag for an argument with return value of block" do
+        klass.new.logger.info("program_name") { "block" + "message" }
+        result.should == "program_name: [tag] blockmessage"
       end
     end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: taglog
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: 
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Ryo Nakamura