lib/trace_location/collector.rb in trace_location-0.9.1 vs lib/trace_location/collector.rb in trace_location-0.9.2

@@ -5,57 +5,80 @@
 module TraceLocation
   module Collector # :nodoc:
     require_relative 'event'
     Result = Struct.new(:events, :return_value)
 
-    def self.collect(match:, ignore:, &block)
-      events = []
-      hierarchy = 0
-      id = 0
-      cache = {}
+    class << self
+      def collect(match:, ignore:, &block)
+        events = []
+        hierarchy = 0
+        id = 0
+        cache = {}
 
-      tracer = TracePoint.new(:call, :return) do |trace_point|
-        next if match && !trace_point.path.to_s.match?(/#{match}/)
-        next if ignore && trace_point.path.to_s.match?(/#{ignore}/)
+        tracer = TracePoint.new(:call, :return) do |trace_point|
+          next if match && !trace_point.path.to_s.match?(/#{match}/)
+          next if ignore && trace_point.path.to_s.match?(/#{ignore}/)
 
-        id += 1
-        caller_path, caller_lineno = trace_point.binding.of_caller(2).source_location
-        location_cache_key = "#{caller_path}:#{caller_lineno}"
+          id += 1
+          caller_path, caller_lineno = trace_point.binding.of_caller(2).source_location
+          location_cache_key = "#{caller_path}:#{caller_lineno}"
 
-        case trace_point.event
-        when :call
-          cache[location_cache_key] = hierarchy
+          mes = extract_method_from(trace_point)
 
-          events << Event.new(
-            id: id,
-            event: trace_point.event,
-            path: trace_point.path,
-            lineno: trace_point.lineno,
-            caller_path: caller_path,
-            caller_lineno: caller_lineno,
-            method_id: trace_point.method_id,
-            defined_class: trace_point.defined_class,
-            hierarchy: hierarchy
-          )
+          case trace_point.event
+          when :call
+            cache[location_cache_key] = hierarchy
 
-          hierarchy += 1
-        when :return
-          hierarchy = cache[location_cache_key] || hierarchy
+            events << Event.new(
+              id: id,
+              event: trace_point.event,
+              path: trace_point.path,
+              lineno: trace_point.lineno,
+              caller_path: caller_path,
+              caller_lineno: caller_lineno,
+              owner: mes.owner,
+              name: mes.name,
+              source: remove_indent(mes.source),
+              hierarchy: hierarchy,
+              is_module: trace_point.self.is_a?(Module)
+            )
 
-          events << Event.new(
-            id: id,
-            event: trace_point.event,
-            path: trace_point.path,
-            lineno: trace_point.lineno,
-            caller_path: caller_path,
-            caller_lineno: caller_lineno,
-            method_id: trace_point.method_id,
-            defined_class: trace_point.defined_class,
-            hierarchy: hierarchy
-          )
+            hierarchy += 1
+          when :return
+            hierarchy = cache[location_cache_key] || hierarchy
+
+            events << Event.new(
+              id: id,
+              event: trace_point.event,
+              path: trace_point.path,
+              lineno: trace_point.lineno,
+              caller_path: caller_path,
+              caller_lineno: caller_lineno,
+              owner: mes.owner,
+              name: mes.name,
+              source: remove_indent(mes.source),
+              hierarchy: hierarchy,
+              is_module: trace_point.self.is_a?(Module)
+            )
+          end
         end
+        return_value = tracer.enable { block.call }
+        Result.new(events, return_value)
       end
-      return_value = tracer.enable { block.call }
-      Result.new(events, return_value)
+
+      private
+
+      def extract_method_from(trace_point)
+        if trace_point.self.is_a?(Module)
+          ::Module.instance_method(:method).bind(trace_point.self).call(trace_point.method_id)
+        else
+          ::Kernel.instance_method(:method).bind(trace_point.self).call(trace_point.method_id)
+        end
+      end
+
+      def remove_indent(source)
+        indent = source.split("\n").first.match(/\A(\s+).+\z/)[1]
+        source.split("\n").map { |line| line.gsub(/\A#{indent}/, '') }.join("\n")
+      end
     end
   end
 end