feat(tracer): inject debug mode via constructor

Author: zah

gems/pure-ruby-tracer/lib/recorder.rb

@@ -4,8 +4,6 @@
 
 require 'fileutils'
 
-DEBUG_TRACER = ENV['CODETRACER_RUBY_TRACER_DEBUG'] == '1'
-
 CallRecord = Struct.new(:function_id, :args) do
   def to_data_for_json
     res = to_h
@@ -115,6 +113,7 @@ class TraceRecord
   attr_accessor :stack, :step_stack, :exprs, :tracing
   attr_accessor :t1, :t2, :t3, :t4 # tracepoints
   attr_accessor :codetracer_id
+  attr_accessor :debug
 
   def initialize
     @events = []
@@ -139,6 +138,7 @@ def initialize
     @t2 = nil
     @t3 = nil
     @codetracer_id = 0
+    @debug = false
   end
 
   def load_flow(path, line, binding)
@@ -248,7 +248,7 @@ def load_exprs(path)
   end
 
   def serialize(program, out_dir = nil)
-    pp @events if DEBUG_TRACER
+    pp @events if @debug
 
     output = @events.map { |kind, event| [[kind, event.to_data_for_json]].to_h }
 
@@ -277,7 +277,7 @@ def serialize(program, out_dir = nil)
     File.write(trace_metadata_path, metadata_json_output)
     File.write(trace_paths_path, paths_json_output)
 
-    if DEBUG_TRACER
+    if @debug
       $stderr.write("=================================================\n")
       $stderr.write("codetracer ruby tracer: saved trace to #{trace_folder}\n")
     end
@@ -355,7 +355,7 @@ def to_value(v, depth=10)
   end
   $VALUE_COUNT += 1
   if $VALUE_COUNT % 10_000 == 0
-    $stderr.write("value #{$VALUE_COUNT}\n") if DEBUG_TRACER
+    $stderr.write("value #{$VALUE_COUNT}\n") if $codetracer_record.debug
   end
   case v
   when Integer

gems/pure-ruby-tracer/lib/trace.rb

@@ -6,7 +6,6 @@
 require 'optparse'
 require_relative 'recorder'
 
-DEBUG_TRACER = ENV['CODETRACER_RUBY_TRACER_DEBUG'] == '1'
 
 # Warning:
 # probably related to our development env:
@@ -71,13 +70,15 @@ class Tracer
   attr_accessor :calls_tracepoint, :return_tracepoint,
                 :line_tracepoint, :raise_tracepoint, :tracing
 
-  attr_reader :ignore_list, :record
+  attr_reader :ignore_list, :record, :debug
 
-  def initialize(record)
+  def initialize(record, debug: ENV['CODETRACER_RUBY_TRACER_DEBUG'] == '1')
     @tracing = false
     @trace_stopped = false
     @record = record
     @ignore_list = []
+    @debug = debug
+    @record.debug = debug if @record.respond_to?(:debug=)
     setup_tracepoints
   end
 
@@ -160,7 +161,7 @@ def record_call(tp)
       method_name_prefix = module_name == 'Object' ? '' :  "#{module_name}#"
       method_name = "#{method_name_prefix}#{tp.method_id}"
 
-      old_puts "call #{method_name} with #{tp.parameters}" if DEBUG_TRACER
+      old_puts "call #{method_name} with #{tp.parameters}" if @debug
 
       arg_records = prepare_args(tp)
 
@@ -172,7 +173,7 @@ def record_call(tp)
 
   def record_return(tp)
     if self.tracks_call?(tp)
-      old_puts "return" if DEBUG_TRACER
+      old_puts "return" if @debug
       return_value = to_value(tp.return_value)
       @record.register_step(tp.path, tp.lineno)
       # return value support inspired by existing IDE-s/envs like
@@ -283,7 +284,7 @@ def load_variables(binding)
   begin
     Kernel.load(program)
   rescue Exception => e
-    if DEBUG_TRACER
+    if @debug
       old_puts ''
       old_puts '==== trace.rb error while tracing program ==='
       old_puts 'ERROR'