kurangu 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fbd6e9e2d7412e8274e1c9ac079589bb0103ca27
-  data.tar.gz: 322c6cc9d6d1091454ce77b3881c144d82f773e1
+  metadata.gz: b865407c12e193a82d44ff98340ad395b4cf1fd8
+  data.tar.gz: dbef6cd12b4b4187e070b1c0951e29836b467fbc
 SHA512:
-  metadata.gz: 4b44400e6dd7b468721562c8549c517f26ae23a7b92333c850b42a183809109767561e96055e8db2d751843a8cae1b3a8edc6e07b6c4cca2ed4fcfff76cf038f
-  data.tar.gz: 789087dd384888c2719cf089ae4e5c7ff16b550ee045f4828588b23763c860a43eb50f087d249210a2d0248f0bb3c6ea6bc3ac26f20f2a86ed26f48ce95b062a
+  metadata.gz: f40a07094d28f21f337183ad5405dcf6172049532cca8949f24cb05d805d173114376d4611688073a39f24af9bfcc0da2b915f1386848f75d49aff1665826995
+  data.tar.gz: c8b15b45e951120b729bad1163706a448b32ab74eb5137e9e650e42328de72c5c20c4bcd0765b999509062e0467d46cec1d3d8dc790027dd91d50854ce9fa8cc

data/lib/signature.rb

@@ -18,14 +18,17 @@ class MethodSignature
   end
 
   def to_s
-    signatures = @signatures.to_a.map { |a| 
-      "#{@line} type '(#{a[0]}) -> #{a[1].to_a.join(" or ")}'" 
+    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)
+  def add(parameter_types, return_type, owner)
     parameter_signature = generate_parameter_signature(parameter_types)
+    if return_type.to_s == owner.class.name
+      return_type = "self"
+    end
     @signatures[parameter_signature].add(return_type)
   end
 

data/lib/trace.rb

@@ -23,9 +23,10 @@ trace_return = TracePoint.new(:return) do |t|
     if args
       line = t.self.method(t.method_id).source_location[1]
       if !signatures.key?(s)
-        signatures[s] = MethodSignature.new(line, t.self.method(t.method_id).parameters)
+        parameters = t.self.method(t.method_id).parameters
+        signatures[s] = MethodSignature.new(line, parameters) 
       end
-      signatures[s].add(args, t.return_value.class)
+      signatures[s].add(args, t.return_value.class, t.self)
       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.7
+  version: 0.0.8
 platform: ruby
 authors:
 - Arpith Siromoney