kurangu 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7752f5414ccab463481b345ff7200e814fb5859a
-  data.tar.gz: b47108ea9f4cd426311d641d6807a6cf48bd2ac5
+  metadata.gz: 5cfb85b2a2f72f69b01f58ffaef7c0eae1fe01c5
+  data.tar.gz: 1d17f5760af0cfc989c7adc7628f01a0fb0a80c9
 SHA512:
-  metadata.gz: a07563548226e345a01eda0373b3ab2becf85f9bfa43be17882a7ddaa4c91e46179c02f2486a841dbb6f16f1a9701be5267abb8dc2efacb163e6fcdd09fb5c42
-  data.tar.gz: c0fee84605fa788d50f1a59e7e4d5e1a30b965b31aa213dfa262dbdcd789a6a9af5169e6a1d5d14528b7cc6671c9700246cc925d5b52b634bf877666b116a9bb
+  metadata.gz: f9a05c8bb9f336e8f520b998fd7a43bbe54112b72f70da38758e197bd5109a42ba5af505c632e1f58fa0e74628598bced5bc3eadcc114fd2df825c5e64a5c27f
+  data.tar.gz: 1d5ef8c578cbb9c8f605fee4f7e0ede8c344e359057ca37c07f035aadca524fe2117b45fa922bf3886c8fae9e77dabd64f1ae864f62335b1431756756f5ab180

data/lib/signature.rb

@@ -0,0 +1,26 @@
+class MethodSignature
+  def initialize(line, parameters)
+    @line = line
+    @parameters = parameters
+    # For each parameter signature (key), store a set of return types (value)
+    @signatures = Hash.new { | h, k | h[k] = Set.new }
+  end
+ 
+  def generate_parameter_signature(parameter_types)
+    joined_parameters = @parameters.map { |arg| parameter_types[arg] }
+    joined_parameters.join(", ")
+  end
+
+  def to_s
+    signatures = @signatures.to_a.map { |a| 
+      "#{@line} type '(#{a[0]}) -> #{a[1].to_a.join(" or ")}'" 
+    }
+    signatures.join("\n")
+  end
+
+  def add(parameter_types, return_type)
+    parameter_signature = generate_parameter_signature(parameter_types)
+    @signatures[parameter_signature].add(return_type)
+  end
+
+end

data/lib/trace.rb

@@ -1,24 +1,19 @@
 require 'set'
+require 'signature'
 
 INPUT_FILE = $stdin.readline.chomp
 
 stack = Hash.new { |h, k| h[k] = [] }
-parameter_list = Hash.new { |h, k| h[k] = [] }
-signatures = Array.new
+signatures = Hash.new
 paths = Set.new
 
-def generate_signature(line, parameters, parameter_types, return_type)
-  joined_parameters = parameters.map { |arg| parameter_types[arg] }
-  "#{line} type '(#{joined_parameters.join(", ")}) -> #{return_type}'"
-end
-
 def write_annotations_paths(dir, paths)
   paths_file = "#{dir}/annotations_paths.txt"
   IO.write(paths_file, paths.to_a.join("\n"))
 end
 
 def write_annotations(path, signatures)
-  IO.write(path, signatures.join("\n"))
+  IO.write(path, signatures.values.join("\n"))
 end
 
 trace_return = TracePoint.new(:return) do |t|
@@ -28,7 +23,10 @@ trace_return = TracePoint.new(:return) do |t|
     if args
       parameter_list = t.self.method(t.method_id).parameters.map { |a | a[1] }
       line = t.self.method(t.method_id).source_location[1]
-      signatures.push generate_signature(line, parameter_list, args, t.return_value.class)
+      if !signatures.key?(s)
+        signatures[s] = MethodSignature.new(line, parameter_list)
+      end
+      signatures[s].add(args, t.return_value.class)
       path = "#{t.path}.annotations"
       dir = File.dirname(t.path)
       write_annotations_paths(dir, paths.add(path))

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kurangu
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Arpith Siromoney
@@ -19,6 +19,7 @@ extra_rdoc_files: []
 files:
 - bin/kurangu
 - lib/kurangu.rb
+- lib/signature.rb
 - lib/trace.rb
 homepage: https://github.com/arpith/kurangu
 licenses: