aivory_monitor 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9cd9562d57c9dd817e379e42470aaf15c8961fc17469c20b6fef9ee67ff19f61
-  data.tar.gz: cea2016eab1e3e1cc26f4db1bd7cabdc851e1a11587edf934b71be16301373d1
+  metadata.gz: d72cabd8da9c3b5268f150cc1f02581629ca50f190310a44bd36fd3db6bc4392
+  data.tar.gz: 2c9037abded48539838a9b260d446f4c4e56eb4131c50d8c100d38bc0e3c3c9b
 SHA512:
-  metadata.gz: e7d8da3e4f472c47a40e6c36ae74cab4b67b90ac1ce961b9b9d54ca402ac97e65ed1d871bb882b2f7a51ccd1af8f1df0c69670edb1b5bfe44c3c75e044162767
-  data.tar.gz: 22be0f73a6fcb28375e410812a069de56163fa229e7753357d2caeab7061b1977dcf8b254cc8fa6232a2f6595553752825f06bd52ed2238900ee539aa9d70202
+  metadata.gz: b90dc42a9156bcff498de44fdc6900c8fab8aed53c38c7cbad4e8636d516d21dbb30e713dde18e6f92e51db1697073ef73561919a270ec4ea5918fc3542322ff
+  data.tar.gz: 327a2dbf54eed21b86075672530639cf0e38df3c6c30cf4872674aad540a690dab454f506443d1932aeae3939aa6e59b4011682bc4fdee52c50820a2952beaff

data/lib/aivory_monitor/backend_connection.rb

@@ -34,7 +34,7 @@ module AIVoryMonitor
         uri = URI.parse(@config.backend_url)
         host = uri.host || "api.aivory.net"
         port = uri.port || (uri.scheme == "wss" ? 443 : 80)
-        path = uri.path.empty? ? "/ws/monitor/agent" : uri.path
+        path = uri.path.empty? ? "/monitor/agent" : uri.path
 
         # Create TCP socket
         @socket = TCPSocket.new(host, port)
@@ -122,6 +122,19 @@ module AIVoryMonitor
       send_message("snapshot", payload)
     end
 
+    # Sends a breakpoint hit to the backend.
+    def send_breakpoint_hit(breakpoint_id, payload)
+      payload[:breakpoint_id] = breakpoint_id
+      payload[:agent_id] = @agent_id
+
+      send_message("breakpoint_hit", payload)
+    end
+
+    # Registers a callback for breakpoint commands from the backend.
+    def set_breakpoint_callback(&block)
+      @breakpoint_callback = block
+    end
+
     # Registers an event handler.
     def on(event, &block)
       @event_handlers[event.to_s] = block
@@ -238,8 +251,10 @@ module AIVoryMonitor
       when "error"
         handle_error(message["payload"])
       when "set_breakpoint"
+        @breakpoint_callback&.call("set", message["payload"])
         emit("set_breakpoint", message["payload"])
       when "remove_breakpoint"
+        @breakpoint_callback&.call("remove", message["payload"])
         emit("remove_breakpoint", message["payload"])
       end
     rescue JSON::ParserError => e

data/lib/aivory_monitor/config.rb

@@ -20,7 +20,7 @@ module AIVoryMonitor
       max_reconnect_attempts: nil
     )
       @api_key = api_key || ENV.fetch("AIVORY_API_KEY", "")
-      @backend_url = backend_url || ENV.fetch("AIVORY_BACKEND_URL", "wss://api.aivory.net/ws/monitor/agent")
+      @backend_url = backend_url || ENV.fetch("AIVORY_BACKEND_URL", "wss://api.aivory.net/monitor/agent")
       @environment = environment || ENV.fetch("AIVORY_ENVIRONMENT", "production")
       @application_name = application_name || ENV["AIVORY_APP_NAME"]
       @sampling_rate = (sampling_rate || ENV.fetch("AIVORY_SAMPLING_RATE", "1.0")).to_f

data/lib/aivory_monitor/trace_manager.rb

@@ -0,0 +1,253 @@
+# frozen_string_literal: true
+
+module AIVoryMonitor
+  # Represents a single breakpoint.
+  class BreakpointInfo
+    attr_accessor :backend_id, :file_path, :line_number, :condition, :max_hits, :hit_count, :normalized_path
+
+    def initialize(backend_id, file_path, line_number, condition = nil, max_hits = 1)
+      @backend_id = backend_id
+      @file_path = file_path
+      @line_number = line_number
+      @condition = condition
+      @max_hits = [[max_hits, 1].max, 50].min
+      @hit_count = 0
+      @normalized_path = File.expand_path(file_path).downcase
+    rescue StandardError
+      @normalized_path = file_path.downcase
+    end
+  end
+
+  # Manages tracing for breakpoint support using TracePoint.
+  class TraceManager
+    MAX_CAPTURES_PER_SECOND = 50
+
+    def initialize(config, connection)
+      @config = config
+      @connection = connection
+      @enabled = false
+      @breakpoints = {}
+      @breakpoints_by_file = {}
+      @trace_point = nil
+      @capture_count = 0
+      @capture_window_start = Time.now
+
+      connection.set_breakpoint_callback { |cmd, payload| handle_command(cmd, payload) }
+    end
+
+    def enable
+      return if @enabled
+
+      @trace_point = TracePoint.new(:line) do |tp|
+        trace_callback(tp)
+      end
+      @trace_point.enable
+
+      @enabled = true
+      puts "[AIVory Monitor] Trace manager enabled" if @config.debug
+    end
+
+    def disable
+      return unless @enabled
+
+      @trace_point&.disable
+      @trace_point = nil
+      @breakpoints.clear
+      @breakpoints_by_file.clear
+
+      @enabled = false
+      puts "[AIVory Monitor] Trace manager disabled" if @config.debug
+    end
+
+    def set_breakpoint(backend_id, file_path, line_number, condition = nil, max_hits = 1)
+      bp = BreakpointInfo.new(backend_id, file_path, line_number, condition, max_hits)
+      @breakpoints[backend_id] = bp
+
+      @breakpoints_by_file[bp.normalized_path] ||= []
+      @breakpoints_by_file[bp.normalized_path] << bp
+
+      puts "[AIVory Monitor] Breakpoint set: #{backend_id} at #{file_path}:#{line_number}" if @config.debug
+    end
+
+    def remove_breakpoint(backend_id)
+      bp = @breakpoints.delete(backend_id)
+      return unless bp
+
+      file_bps = @breakpoints_by_file[bp.normalized_path]
+      if file_bps
+        file_bps.reject! { |b| b.backend_id == backend_id }
+        @breakpoints_by_file.delete(bp.normalized_path) if file_bps.empty?
+      end
+
+      puts "[AIVory Monitor] Breakpoint removed: #{backend_id}" if @config.debug
+    end
+
+    private
+
+    def handle_command(command, payload)
+      case command
+      when "set"
+        set_breakpoint(
+          payload["id"] || "",
+          payload["file_path"] || payload["file"] || "",
+          (payload["line_number"] || payload["line"] || 0).to_i,
+          payload["condition"],
+          (payload["max_hits"] || 1).to_i
+        )
+      when "remove"
+        remove_breakpoint(payload["id"] || "")
+      end
+    end
+
+    def trace_callback(tp)
+      return if @breakpoints_by_file.empty?
+
+      file_path = tp.path
+      return unless file_path
+
+      normalized = begin
+        File.expand_path(file_path).downcase
+      rescue StandardError
+        file_path.downcase
+      end
+
+      # Look up breakpoints for this file (exact match, then suffix match)
+      file_bps = @breakpoints_by_file[normalized]
+      unless file_bps
+        @breakpoints_by_file.each do |bp_path, bps|
+          if normalized.end_with?(bp_path) || bp_path.end_with?(normalized)
+            file_bps = bps
+            break
+          end
+        end
+      end
+      return unless file_bps
+
+      line = tp.lineno
+      file_bps.each do |bp|
+        handle_hit(bp, tp) if bp.line_number == line
+      end
+    end
+
+    def handle_hit(bp, tp)
+      return if bp.hit_count >= bp.max_hits
+      return unless rate_limit_ok?
+
+      # Evaluate condition if present
+      if bp.condition && !bp.condition.empty?
+        begin
+          result = tp.binding.eval(bp.condition)
+          return unless result
+        rescue StandardError => e
+          puts "[AIVory Monitor] Condition eval error: #{e.message}" if @config.debug
+          return
+        end
+      end
+
+      bp.hit_count += 1
+
+      puts "[AIVory Monitor] Breakpoint hit: #{bp.backend_id}" if @config.debug
+
+      local_variables = capture_locals(tp.binding)
+      stack_trace = build_stack_trace
+
+      @connection.send_breakpoint_hit(bp.backend_id, {
+        captured_at: (Time.now.to_f * 1000).to_i,
+        file_path: bp.file_path,
+        line_number: bp.line_number,
+        stack_trace: stack_trace,
+        local_variables: local_variables,
+        hit_count: bp.hit_count
+      })
+    end
+
+    def rate_limit_ok?
+      now = Time.now
+      if now - @capture_window_start >= 1.0
+        @capture_count = 0
+        @capture_window_start = now
+      end
+
+      if @capture_count >= MAX_CAPTURES_PER_SECOND
+        puts "[AIVory Monitor] Rate limit reached, skipping capture" if @config.debug
+        return false
+      end
+
+      @capture_count += 1
+      true
+    end
+
+    def capture_locals(binding)
+      variables = {}
+
+      binding.local_variables.each do |name|
+        next if name.to_s.start_with?("_")
+
+        begin
+          value = binding.local_variable_get(name)
+          variables[name.to_s] = capture_variable(name.to_s, value, 0)
+        rescue StandardError
+          # Skip if can't access
+        end
+      end
+
+      variables
+    end
+
+    def capture_variable(name, value, depth)
+      result = { name: name, type: value.class.name }
+
+      if depth > @config.max_variable_depth
+        result[:value] = "<max depth exceeded>"
+        result[:is_truncated] = true
+        return result
+      end
+
+      if value.nil?
+        result[:value] = "nil"
+        result[:is_null] = true
+      elsif value.is_a?(Numeric) || value.is_a?(TrueClass) || value.is_a?(FalseClass)
+        result[:value] = value.to_s
+      elsif value.is_a?(String)
+        if value.length > 500
+          result[:value] = value[0, 500]
+          result[:is_truncated] = true
+        else
+          result[:value] = value
+        end
+      elsif value.is_a?(Symbol)
+        result[:value] = ":#{value}"
+      elsif value.is_a?(Array)
+        result[:value] = "Array(#{value.size})"
+        if depth < @config.max_variable_depth && value.size <= 10
+          result[:children] = value.each_with_index.to_h do |v, i|
+            ["[#{i}]", capture_variable("[#{i}]", v, depth + 1)]
+          end
+        end
+      elsif value.is_a?(Hash)
+        result[:value] = "Hash(#{value.size})"
+        if depth < @config.max_variable_depth && value.size <= 10
+          result[:children] = value.transform_keys(&:to_s).transform_values do |v|
+            capture_variable(v.class.name, v, depth + 1)
+          end
+        end
+      else
+        result[:value] = value.class.name
+      end
+
+      result
+    end
+
+    def build_stack_trace
+      caller_locations(3, 50).map do |loc|
+        {
+          method_name: loc.label,
+          file_path: loc.absolute_path || loc.path,
+          file_name: loc.path ? File.basename(loc.path) : nil,
+          line_number: loc.lineno,
+          is_native: (loc.path || "").start_with?("<") || (loc.path || "").include?("/gems/")
+        }
+      end
+    end
+  end
+end

data/lib/aivory_monitor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AIVoryMonitor
-  VERSION = "1.0.0"
+  VERSION = "0.1.1"
 end

data/lib/aivory_monitor.rb

@@ -5,6 +5,7 @@ require_relative "aivory_monitor/config"
 require_relative "aivory_monitor/models"
 require_relative "aivory_monitor/backend_connection"
 require_relative "aivory_monitor/exception_capture"
+require_relative "aivory_monitor/trace_manager"
 
 # AIVory Monitor Ruby Agent
 #
@@ -81,6 +82,12 @@ module AIVoryMonitor
         # Install exception handlers
         @exception_capture.install
 
+        # Initialize breakpoint support
+        if config.enable_breakpoints
+          @trace_manager = TraceManager.new(config, @connection)
+          @trace_manager.enable
+        end
+
         # Connect to backend
         @connection.connect
 
@@ -131,6 +138,7 @@ module AIVoryMonitor
 
         puts "[AIVory Monitor] Shutting down agent" if @config&.debug
 
+        @trace_manager&.disable
         @exception_capture&.uninstall
         @connection&.disconnect
 
@@ -138,6 +146,7 @@ module AIVoryMonitor
         @config = nil
         @connection = nil
         @exception_capture = nil
+        @trace_manager = nil
       end
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aivory_monitor
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - AIVory
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-02-18 00:00:00.000000000 Z
+date: 2026-02-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: websocket-client-simple
@@ -67,6 +67,7 @@ files:
 - lib/aivory_monitor/config.rb
 - lib/aivory_monitor/exception_capture.rb
 - lib/aivory_monitor/models.rb
+- lib/aivory_monitor/trace_manager.rb
 - lib/aivory_monitor/version.rb
 homepage: https://aivory.net/monitor/
 licenses: