annals 0.0.2 → 0.0.3

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.0.3

data/annals.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{annals}
-  s.version = "0.0.2"
+  s.version = "0.0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["geemus (Wesley Beary)"]

data/lib/annals.rb

@@ -4,6 +4,15 @@ class Annals
 
   def initialize(max = 20, &block)
     @max = max
+    if block_given?
+      @trace_func = block
+    else
+      @trace_func = lambda { |event, file, line, id, binding, classname|
+        if event == 'call'
+          unshift(:file => file, :line => line, :method => "in #{id}")
+        end
+      }
+    end
     start
   end
 
@@ -17,7 +26,7 @@ class Annals
     @buffer.map {|event| "#{event[:file]}:#{event[:line]} #{event[:method]}"}
   end
 
-  def puts
+  def print_lines
     for line in lines
       puts(line)
     end
@@ -26,27 +35,23 @@ class Annals
   def start
     @buffer = []
     @size   = 0
-    Kernel.set_trace_func(
-      lambda { |event, file, line, id, binding, classname|
-        if event == 'call'
-          unshift(:file => file, :line => line, :method => "in #{id}")
-        end
-      }
-    )
+    Kernel.set_trace_func(@trace_func)
+    true
   end
 
   def stop
     Kernel.set_trace_func(lambda {})
     @buffer.shift # remove Annals#stop from buffer
+    true
   end
 
   def unshift(line)
-    @buffer.unshift(line)
     if @size == @max
       @buffer.pop
     else
       @size += 1
     end
+    @buffer.unshift(line)
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: annals
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - geemus (Wesley Beary)