codebeacon-tracer 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 834e2095be6b5d5a919b96bbb1395dc33bfc569cfb189d67b7e95434574c8641
-  data.tar.gz: 1215e606cf263cc3cf77954e4f60e175bd8df43ad9adbe1ad17859bc03003c4b
+  metadata.gz: 47ec3d248b6f0358c3c36310a7b39c506982695565e453d11580ae20b1f28582
+  data.tar.gz: 8f6da9ed15ca33105c73fead83856a014c718853cbf30db5641719e5d6fdfc36
 SHA512:
-  metadata.gz: 7f7871f11660157413fecdb33a856d9f06465690570226b201e9c5ff2a31dd3a6dbea826b2c4a0d4729a15ebadfab4d07357e9ff8978f76314ed43ef63844471
-  data.tar.gz: c2d72f58559f32136268687dfaea900344a4141d6beb4f252821f9cb838e7e2b296cd304c0a9e51387589611068dff1e34f9a88139c56c5a85ebe2022389cb2a
+  metadata.gz: 69b782857204580baf4b8251f478d0e16652d015fcbf5f79c65a47f691f9623a52ee9271c973d5539ed4d680d4c37c21f9571299225ce16ecab446e9bbd43ffd
+  data.tar.gz: 27dd8bdcb652b7db338e0ac8aba0862b6cb3106c00a76524265cec238714849d046c416c43c62448aa3ca4ca84be19a3c0651b53b347cdd79b9561fd3b7dca99

data/lib/codebeacon/tracer/src/configuration.rb

@@ -14,6 +14,7 @@ module Codebeacon
       RETURN_VAL_MAX_LENGTH = 1000
       MAX_CALL_COUNT = 100000000
       MAX_DEPTH = 99999
+      SERIALIZATION_TIMEOUT_MS = 10
 
       def initialize()
         @query = ""
@@ -69,7 +70,7 @@ module Codebeacon
       end
 
       def data_dir
-        ".code-beacon"
+        File.join(root_path, ".code-beacon")
       end
 
       def db_path
@@ -115,7 +116,7 @@ module Codebeacon
       end
 
       def root_path
-        @root_path ||= defined?(Rails) ? Rails.root.to_s : Dir.pwd
+        @root_path ||= @config_root_path || (defined?(Rails) ? Rails.root.to_s : Dir.pwd)
       end
 
       def rubylib_path
@@ -145,9 +146,15 @@ module Codebeacon
 
       def reload_tracer_config
         @trace_enabled = load_tracer_config_enabled
+        @config_root_path = load_tracer_config_root_path
         @recording_meta_exclude_patterns = nil  # Force reload of exclusion patterns
       end
 
+      def reload_main_config
+        @root_path = nil  # Force reload of root_path
+        load_main_config
+      end
+
       def recording_meta_exclude_patterns
         if @donotcache
           load_recording_meta_exclude_patterns
@@ -229,6 +236,10 @@ module Codebeacon
         MAX_DEPTH
       end
 
+      def serialization_timeout_ms
+        SERIALIZATION_TIMEOUT_MS
+      end
+
       def logger
         @logger ||= Codebeacon::Tracer::Logger.new()
       end
@@ -301,6 +312,15 @@ module Codebeacon
         logger.warn("Error loading tracer config: #{e.message}")
         true
       end
+
+      def load_tracer_config_root_path
+        if File.exist?(tracer_config_path)
+          config_data = YAML.load_file(tracer_config_path)
+          @config_root_path = config_data['root_path'] if config_data['root_path']
+        end
+      rescue => e
+        logger.warn("Error loading tracer config root_path: #{e.message}")
+      end
     end
   end
 end

data/lib/codebeacon/tracer/src/data/metadata_mapper.rb

@@ -19,7 +19,8 @@ module Codebeacon
             start_time TEXT,
             end_time TEXT,
             duration_ms REAL,
-            trigger_type TEXT
+            trigger_type TEXT,
+            language TEXT
           );
         SQL
       end
@@ -34,9 +35,9 @@ module Codebeacon
           INSERT INTO metadata (
             name, description, caller_file, caller_method, caller_line,
             caller_class, caller_defined_class, start_time, end_time,
-            duration_ms, trigger_type
+            duration_ms, trigger_type, language
           ) VALUES (
-            ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
+            ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
           )
         SQL
           metadata_hash[:name],
@@ -49,7 +50,8 @@ module Codebeacon
           metadata_hash[:start_time]&.iso8601,
           metadata_hash[:end_time]&.iso8601,
           metadata_hash[:duration_ms],
-          metadata_hash[:trigger_type]
+          metadata_hash[:trigger_type],
+          metadata_hash[:language]
         )
       end
     end

data/lib/codebeacon/tracer/src/data/persistence_manager.rb

@@ -1,27 +1,13 @@
 require_relative 'tree_node_mapper'
 require_relative 'node_source_mapper'
 require_relative 'metadata_mapper'
+require_relative 'type_detector'
+require_relative 'safe_serializer'
 
 module Codebeacon
   module Tracer
     class PersistenceManager
 
-      def self.marshal(name, value, tree_node)
-        begin
-          return value.inspect[0..Codebeacon::Tracer.config.max_value_length]
-        rescue => e
-          begin
-            if Codebeacon::Tracer.config.debug?
-              Codebeacon::Tracer.logger.warn "Marshal inspect failure - attempting to_s fallback for: \"#{name}\", located at: \"#{tree_node.file}:#{tree_node.line}\"\nerror message: \"#{e.message}\", error_location: \"#{e.backtrace[0]}\""
-            end
-            return value.to_s[0..Codebeacon::Tracer.config.max_value_length]
-          rescue => e
-            Codebeacon::Tracer.logger.error "Marshal failure for: \"#{name}\", located at: \"#{tree_node.file}:#{tree_node.line}\"\nerror message: \"#{e.message}\", error_location: \"#{e.backtrace[0]}\""
-            return "--Codebeacon::Tracer ERROR-- could not marshall value. See logs."
-          end
-        end
-      end
-
       def initialize(database)
         @database = database
         @tree_node_mapper = TreeNodeMapper.new(database)
@@ -69,6 +55,7 @@ module Codebeacon
           node_id = @tree_node_mapper.insert(
             tree_node.file,
             tree_node.line,
+            tree_node.called_method&.to_s, # Convert symbols to strings, keep nil as nil
             tree_node.method.to_s,
             tree_node.tp_class.to_s,
             tree_node.tp_defined_class.to_s,
@@ -80,7 +67,8 @@ module Codebeacon
             parent_id,
             tree_node.block,
             tree_node.node_source&.id,
-            _return_value(tree_node)
+            return_type(tree_node),
+            return_value(tree_node)
           )
 
           unless tree_node.depth_truncated?
@@ -95,12 +83,14 @@ module Codebeacon
         end
       end
 
-      def _return_value(node)
-        if node.method == :initialize
-          return nil
-        else
-          PersistenceManager.marshal(node.method, node.return_value, node)
-        end
+      private def return_type(node)
+        return nil if node.method == :initialize
+        node.return_value.class.name
+      end
+
+      private def return_value(node)
+        return nil if node.method == :initialize
+        SafeSerializer.serialize(node.return_value, Codebeacon::Tracer.config.max_value_length)
       end
     end
   end

data/lib/codebeacon/tracer/src/data/safe_serializer.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require_relative 'type_detector'
+require 'timeout'
+
+module Codebeacon
+  module Tracer
+    class SafeSerializer
+      def self.serialize(value, max_length)
+        return nil unless TypeDetector.serializable_type?(value)
+        return "..." if max_length <= 0
+        
+        timeout_ms = Codebeacon::Tracer.config.serialization_timeout_ms
+        
+        begin
+          Timeout.timeout(timeout_ms / 1000.0) do
+            serialized = case value
+            when String
+              value
+            when Integer, Float
+              value.to_s
+            when Symbol, TrueClass, FalseClass
+              value.to_s
+            when NilClass
+              "nil"
+            else
+              nil
+            end
+
+            return nil if serialized.nil?
+
+            serialized.length <= max_length ? serialized : serialized[0...max_length] + "..."
+          end
+        rescue Timeout::Error
+          return "(serialization timeout)"
+        end
+      end
+    end
+  end
+end 

data/lib/codebeacon/tracer/src/data/trace_metadata.rb

@@ -3,7 +3,7 @@ module Codebeacon
     class TraceMetadata
       attr_reader :name, :description, :caller_file, :caller_method, :caller_line, 
                   :caller_class, :caller_defined_class, :start_time, 
-                  :end_time, :duration_ms, :trigger_type
+                  :end_time, :duration_ms, :trigger_type, :language
 
       def initialize(name: nil, description: nil, caller_location: nil, trigger_type: nil)
         @name = name
@@ -12,6 +12,7 @@ module Codebeacon
         @end_time = nil
         @duration_ms = nil
         @trigger_type = trigger_type
+        @language = "ruby"
         
         if caller_location
           capture_caller_info_from_location(caller_location)
@@ -35,7 +36,8 @@ module Codebeacon
           start_time: @start_time,
           end_time: @end_time,
           duration_ms: @duration_ms,
-          trigger_type: @trigger_type
+          trigger_type: @trigger_type,
+          language: @language
         }
       end
 

data/lib/codebeacon/tracer/src/data/tree_node_mapper.rb

@@ -8,20 +8,20 @@ module Codebeacon
         @db = database
       end
 
-      def insert(file, line, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, gem_entry, parent_id, block, node_source_id, return_value)
+      def insert(file, line, called_method, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, gem_entry, parent_id, block, node_source_id, return_type, return_value)
         @db.execute(<<-SQL, 
           INSERT INTO treenodes 
           (
-              file, line, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, 
-              gemEntry, parent_id, block, node_source_id, return_value
+              file, line, called_method, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, 
+              gemEntry, parent_id, block, node_source_id, return_type, return_value
           )
           VALUES 
           (
-              ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
+              ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
           )
         SQL
-        file, line, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, 
-        gem_entry ? 1 : 0, parent_id, block ? 1 : 0, node_source_id, return_value)
+        file, line, called_method, method, tp_class, tp_defined_class, tp_class_name, self_type, depth, caller, 
+        gem_entry ? 1 : 0, parent_id, block ? 1 : 0, node_source_id, return_type, return_value)
         
         @db.last_insert_row_id
       end
@@ -32,6 +32,7 @@ module Codebeacon
             id INTEGER PRIMARY KEY,
             file TEXT,
             line INTEGER,
+            called_method TEXT,
             method TEXT,
             tp_class TEXT,
             tp_defined_class TEXT,
@@ -43,6 +44,7 @@ module Codebeacon
             parent_id INTEGER,
             block INTEGER,
             node_source_id INTEGER,
+            return_type TEXT,
             return_value TEXT,
             FOREIGN KEY (parent_id) REFERENCES treenodes(id),
             FOREIGN KEY (node_source_id) REFERENCES node_sources(id)
@@ -54,6 +56,7 @@ module Codebeacon
         database.execute("CREATE INDEX IF NOT EXISTS IDX_treenode_parent_id ON treenodes(parent_id)")
         database.execute("CREATE INDEX IF NOT EXISTS IDX_treenode_node_source_id ON treenodes(node_source_id)")
         database.execute("CREATE INDEX IF NOT EXISTS IDX_treenode_file ON treenodes(file)")
+        database.execute("CREATE INDEX IF NOT EXISTS IDX_treenode_called_method ON treenodes(called_method)")
       end
     end
   end

data/lib/codebeacon/tracer/src/data/type_detector.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Codebeacon
+  module Tracer
+    class TypeDetector
+      BASIC_TYPES = [
+        String, Integer, Float, Symbol, TrueClass, FalseClass, NilClass
+      ].freeze
+
+      def self.basic_type?(value)
+        return true if value.nil?
+        BASIC_TYPES.any? { |type| value.is_a?(type) }
+      end
+
+      def self.serializable_type?(value)
+        basic_type?(value)
+      end
+    end
+  end
+end 

data/lib/codebeacon/tracer/src/models/node_builder.rb

@@ -37,6 +37,10 @@ module Codebeacon
           current_context.line = tp.lineno
           current_context.object_id = tp.self.object_id
           current_context.method = tp.method_id
+          if tp.callee_id != tp.method_id
+            current_context.called_method = tp.callee_id
+          end
+
           klass = TPKlass.new(tp)
 
           current_context.tp_class = klass.tp_class.to_s

data/lib/codebeacon/tracer/src/models/tree_node.rb

@@ -31,9 +31,9 @@ module Codebeacon
       #   @children = []
       # end
 
-      attr_accessor :file, :line, :method, :object_id, :tp_class, :tp_defined_class, :tp_class_name, :self_type, :depth, :caller, :gem_entry, :children, :parent, :block, :locals, :return_value, :linevars, :node_source, :trace_status, :script, :backtrace_count, :backtrace_location, :script_binding, :script_self
+      attr_accessor :file, :line, :method, :object_id, :tp_class, :tp_defined_class, :tp_class_name, :self_type, :depth, :caller, :gem_entry, :children, :parent, :block, :locals, :return_value, :linevars, :node_source, :trace_status, :script, :backtrace_count, :backtrace_location, :script_binding, :script_self, :called_method
 
-      def initialize(file: nil, line: nil, object_id: nil, method: nil, tp_class: nil, tp_defined_class: nil, tp_class_name: nil, self_type: nil, depth: 0, caller: "", gem_entry: false, parent: nil, block: false, locals: [], return_value: nil, node_source: nil, script: false)
+      def initialize(file: nil, line: nil, object_id: nil, method: nil, tp_class: nil, tp_defined_class: nil, tp_class_name: nil, self_type: nil, depth: 0, caller: "", gem_entry: false, parent: nil, block: false, locals: [], return_value: nil, node_source: nil, script: false, called_method: nil)
         @file = file
         @line = line
         @method = method
@@ -58,6 +58,7 @@ module Codebeacon
         @backtrace_location = nil
         @script_binding = nil
         @script_self = nil
+        @called_method = called_method
       end
 
       def add_line(lineno, variables)

data/lib/codebeacon/tracer/src/tracer.rb

@@ -73,7 +73,7 @@ module Codebeacon
 
       def trace_call
         trace(:call) do |tp|
-          NodeBuilder.trace_method_call(call_tree, tp, Kernel.caller[2..])
+          NodeBuilder.trace_method_call(call_tree, tp, "")
         ensure
           @progress_logger.increment()
         end
@@ -81,7 +81,7 @@ module Codebeacon
 
       def trace_b_call
         trace(:b_call) do |tp|
-          NodeBuilder.trace_block_call(call_tree, tp, Kernel.caller[2..])
+          NodeBuilder.trace_block_call(call_tree, tp, "")
         ensure
           @progress_logger.increment()
         end
@@ -101,24 +101,16 @@ module Codebeacon
 
       def trace(type)
         TracePoint.new(type) do |tp|
-          paths = [tp.path]
-          # capture calls and returns to skipped paths from non skipped paths. All I need is the return value to display in recorded files, but the code doesn't yet support this without tracing the entire call and return.
-          if [:call, :b_call, :return, :b_return].include?(type)
-            paths << Kernel.caller(1..1)[0]
-          end
-          paths.uniq!
-          next if skip_methods?(paths)
+          next if skip_methods?(tp.path)
           yield tp
         rescue => e
           Codebeacon::Tracer.logger.error("TracePoint(#{type}) #{tp.path} #{e.message}")
         end
       end
 
-      def skip_methods?(paths)
-        paths.all? do |path|
-          path.nil? || Codebeacon::Tracer.config.exclude_paths.any?{ |exclude_path| path.start_with?(exclude_path) } ||
-            Codebeacon::Tracer.config.local_methods_only? && !path.start_with?(Codebeacon::Tracer.config.root_path)
-        end
+      def skip_methods?(path)
+        path.nil? || Codebeacon::Tracer.config.exclude_paths.any?{ |exclude_path| path.start_with?(exclude_path) } ||
+          Codebeacon::Tracer.config.local_methods_only? && !path.start_with?(Codebeacon::Tracer.config.root_path)
       end
     end
   end

data/lib/codebeacon/tracer/version.rb

@@ -2,6 +2,6 @@
 
 module Codebeacon
   module Tracer
-    VERSION = "0.3.0"
+    VERSION = "0.4.0"
   end
 end 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: codebeacon-tracer
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Jonathan Conley
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-07-16 00:00:00.000000000 Z
+date: 2025-07-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser
@@ -172,8 +172,10 @@ files:
 - lib/codebeacon/tracer/src/data/metadata_mapper.rb
 - lib/codebeacon/tracer/src/data/node_source_mapper.rb
 - lib/codebeacon/tracer/src/data/persistence_manager.rb
+- lib/codebeacon/tracer/src/data/safe_serializer.rb
 - lib/codebeacon/tracer/src/data/trace_metadata.rb
 - lib/codebeacon/tracer/src/data/tree_node_mapper.rb
+- lib/codebeacon/tracer/src/data/type_detector.rb
 - lib/codebeacon/tracer/src/logger.rb
 - lib/codebeacon/tracer/src/models/call_tree.rb
 - lib/codebeacon/tracer/src/models/node_builder.rb