kurangu 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 98ec8f13829b78125818e91a6b78163f40b295a6
-  data.tar.gz: f06990a2e564dd4531e7813fa415911ddc5b40f1
+  metadata.gz: c878be455c92153388c28789fa52dd75cf8af897
+  data.tar.gz: 11199180b1f492e70c92e479c0435ad4e30c594f
 SHA512:
-  metadata.gz: b9a82fe613d3926dc7acbd4a71bd4b88c910bbdf9b62a010d3d5eaf7f31dfb06ee6512aec26c531466fd40a74ff1350da7528a636ddb583e0f5179d0f3185274
-  data.tar.gz: 4594ab0b230bad1bd51a65ec995de2c42ba049eba823fdcbd7816800544c8e204235529c602ee67f6c470574816127f14461e673a40dbd243303352cd8912258
+  metadata.gz: c965377a60f1e88885d69b9a5b092534441ab008c3ed3c439c2534e47f83735b53625c7e588254168c26439ef4a0850fe96045371a1440d9bb76fe75a6a70c32
+  data.tar.gz: 860f00f76c6513fff33f7711a85355388339a69cace2a469664442891edd7ac2dc61e4695f844d85dad71031cf4944fcf880f81de3c1f6365435605f97efb2ce

data/lib/kurangu.rb

@@ -9,9 +9,9 @@ class Kurangu
     system(ruby, "-r", trace_file, input_file)
   end
 
-  def print_annotations
+  def print_annotations(paths_file)
     puts "\nthe annotations generated are:\n"
-    File.open("annotations_paths.txt", "r") do |f|
+    File.open(paths_file, "r") do |f|
       f.each_line do |annotation_file|
         puts annotation_file
         File.open(annotation_file, "r") do |f|
@@ -25,8 +25,8 @@ class Kurangu
     end
   end
 
-  def generate_annotated_files
-    File.open("annotations_paths.txt", "r") do |f|
+  def generate_annotated_files(paths_file)
+    File.open(paths_file, "r") do |f|
       f.each_line do |annotation_path|
         original_path = annotation_path.chomp('.annotations')
         annotated_path = "#{original_path}.annotated"
@@ -61,8 +61,8 @@ class Kurangu
     FileUtils.mv(annotated_path, original_path)
   end
 
-  def apply_annotations
-    File.open("annotations_paths.txt", "r") do |f|
+  def apply_annotations(paths_file)
+    File.open(paths_file, "r") do |f|
       f.each_line do |annotation_path|
         original_path = annotation_path.chomp('.annotations')
         annotated_path = "#{original_path}.annotated"
@@ -79,10 +79,11 @@ class Kurangu
 
   def run(input_file_path)
     input_file = File.expand_path(input_file_path)
+    paths_file = "#{File.dirname(input_file)}/annotations_paths.txt"
     self.generate_annotations(input_file)
-    self.print_annotations
-    self.generate_annotated_files
-    self.apply_annotations
+    self.print_annotations(paths_file)
+    self.generate_annotated_files(paths_file)
+    self.apply_annotations(paths_file)
     self.run_rdl(input_file)
   end
 end

data/lib/trace.rb

@@ -13,8 +13,9 @@ def generate_signature(line, parameters, parameter_types, return_types)
   "#{line} type '(#{joined_parameters.join(", ")}) -> #{return_types.join(" or ")}'"
 end
 
-def write_annotations_paths(paths)
-  IO.write("annotations_paths.txt", paths.to_a.join("\n"))
+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)
@@ -33,7 +34,8 @@ trace_return = TracePoint.new(:return) do |t|
     line = t.self.method(t.method_id).source_location[1]
     signatures[s] = generate_signature(line, parameter_list[s], parameter_types[s], return_types[s])
     path = "#{t.path}.annotations"
-    write_annotations_paths(paths.add(path))
+    dir = File.dirname(t.path)
+    write_annotations_paths(dir, paths.add(path))
     write_annotations(path, signatures)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kurangu
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Arpith Siromoney