visual_call_graph 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 09183cd012faa7b73fffef22e28ce670e80e1923
-  data.tar.gz: 70f0f7ee21f720f78fed25893a33ffb45b7a5284
+  metadata.gz: e548d83e3737eefca0445354f1a5cbb84c216946
+  data.tar.gz: d01124dd37f6a0867fb57a5a7c93049f0ccfe42b
 SHA512:
-  metadata.gz: b59865f3b7dd246e1ddccafe57616313db5d8238926f2cefd979c9bb954d42c4072a47ef587e16277124d5b76e172a2c26e0c5f5fde229c3bac4c36518a1e295
-  data.tar.gz: f798c59b7283de0f863630f607e21b2e2f507c192204a9f8745e957168017c9ee05f8ee55eb352b4ac9acbaf05087cf6a026c65d781af14f97214d835ea6507f
+  metadata.gz: e4fb3bd2ab4736345b55cb13b2372443e315ce4f3d4d8bc79db6ec25d87c8e57fc41817c151ec166ae8ae73e9c610dc482b9fd754129a48ff3de19500eb406d3
+  data.tar.gz: 96aac7d88ae62fa3db87acffe616d02e82b7ace9c10db308a866ad1a9fff8d6a3fe300004b04e294cfc4e7eebd6ae3103296122b1c2fec2f4ff7c87d5f9d2371

data/lib/graph_manager.rb

@@ -0,0 +1,44 @@
+
+require 'graphviz'
+
+class GraphManager
+  def initialize(options)
+    @stack   = ["start"]
+    @edges   = []
+    @options = options
+
+    @g = GraphViz.new(:G, :type => :digraph)
+
+    @g.add_node("start")
+  end
+
+  def add_edges(event)
+    node =
+    if @options[:show_path]
+      "#{event.defined_class}##{event.method_id}\n#{event.path}".freeze
+    else
+      "#{event.defined_class}##{event.method_id}".freeze
+    end
+
+    edge = [@stack.last, node]
+
+    @stack << node
+
+    return if @edges.include?(edge)
+
+    @edges << edge
+    @g.add_edge(*@edges.last)
+  end
+
+  def output(*args)
+    @g.output(*args)
+  end
+
+  def node_count
+    @g.node_count
+  end
+
+  def pop
+    @stack.pop
+  end
+end

data/lib/visual_call_graph.rb

@@ -1,22 +1,17 @@
 
-require 'graphviz'
+require_relative 'graph_manager'
 
 module VisualCallGraph
   extend self
 
-  def trace
-    stack  = ["start"]
-    edges  = []
-
-    g = GraphViz.new(:G, :type => :digraph)
-
-    g.add_node("start")
+  def trace(options = {})
+    graph = GraphManager.new(options)
 
     trace =
     TracePoint.new(:call, :return) do |event|
       case event.event
-      when :return then stack.pop
-      when :call   then add_edges(event, edges, stack, g)
+      when :return then graph.pop
+      when :call   then graph.add_edges(event)
       end
     end
 
@@ -24,24 +19,8 @@ module VisualCallGraph
     yield
     trace.disable
 
-    g.output(png: "call_graph.png")
-
-    puts "Call graph created with a total of #{g.node_count} #{g.node_count > 1 ? 'nodes' : 'node'}."
-  end
-
-  private
-
-  def add_edges(event, edges, stack, g)
-    node = "#{event.defined_class}##{event.method_id}".freeze
-
-    g.add_node(node)
-    stack << node
-
-    edge = [stack[-2], stack[-1]]
-
-    return if edges.include?(edge)
+    graph.output(png: "#{Dir.pwd}/call_graph.png")
 
-    edges << edge
-    g.add_edge(*edges.last)
+    puts "Call graph created with a total of #{graph.node_count} #{graph.node_count > 1 ? 'nodes' : 'node'}."
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: visual_call_graph
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Jesus Castello
@@ -37,6 +37,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/graph_manager.rb
 - lib/visual_call_graph.rb
 homepage: http://www.blackbytes.info
 licenses: