trace_viz 0.0.2 → 1.0.1

data/lib/trace_viz/collectors/matchers/within_depth_matcher.rb

@@ -1,12 +1,13 @@
 # frozen_string_literal: true
 
+require "trace_viz/helpers/config_helper"
+require_relative "base_matcher"
+
 module TraceViz
   module Collectors
     module Matchers
       class WithinDepthMatcher
-        def initialize
-          @config = Context.for(:config).configuration
-        end
+        include Helpers::ConfigHelper
 
         def matches?(depth)
           depth <= max_depth
@@ -14,11 +15,11 @@ module TraceViz
 
         private
 
-        attr_reader :config
-
         def max_depth
           config.general[:max_display_depth]
         end
+
+        private :config
       end
     end
   end

data/lib/trace_viz/collectors/steps/assign_depth_for_call_step.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require_relative "base_step"
+
+module TraceViz
+  module Collectors
+    module Steps
+      class AssignDepthForCallStep < BaseStep
+        def call(trace_data)
+          assign_depth(trace_data) if valid?(trace_data)
+          trace_data
+        rescue StandardError => e
+          logger.error("Depth assignment failed for trace_data ID: #{trace_data.id} - #{e.message}")
+          nil
+        end
+
+        private
+
+        def valid?(trace_data)
+          trace_data.event == :call
+        end
+
+        def assign_depth(trace_data)
+          trace_data.assign_depth(current_depth)
+          active_call_stack.push(trace_data)
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/steps/assign_depth_for_return_step.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative "base_step"
+
+module TraceViz
+  module Collectors
+    module Steps
+      class AssignDepthForReturnStep < BaseStep
+        def call(trace_data)
+          assign_depth(trace_data) if valid?(trace_data)
+          trace_data
+        rescue StandardError => e
+          logger.error("Depth assignment failed for trace_data ID: #{trace_data.id} - #{e.message}")
+          nil
+        end
+
+        private
+
+        def valid?(trace_data)
+          trace_data.event == :return
+        end
+
+        def call_matches?(trace_data)
+          trace_data.action_id == current_call&.action_id
+        end
+
+        def assign_depth(trace_data)
+          if call_matches?(trace_data)
+            active_call_stack.pop
+          else
+            logger.warn("Unmatched return event for trace_data ID #{trace_data.id}")
+          end
+
+          trace_data.assign_depth(current_depth)
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/steps/base_step.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require "trace_viz/helpers/tracking_helper"
+require "trace_viz/helpers/config_helper"
+
+module TraceViz
+  module Collectors
+    module Steps
+      class BaseStep
+        include Helpers::TrackingHelper
+        include Helpers::ConfigHelper
+
+        def initialize
+          @logger = TraceViz.logger
+        end
+
+        def call
+          raise NotImplementedError
+        end
+
+        private
+
+        attr_reader :logger
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/steps/build_hierarchy_step.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative "base_step"
+
+module TraceViz
+  module Collectors
+    module Steps
+      class BuildHierarchyStep < BaseStep
+        def call(trace_data)
+          link_to_parent(trace_data) if valid?(trace_data)
+
+          trace_data
+        rescue StandardError => e
+          logger.error("Hierarchy building failed for trace_data ID: #{trace_data.id} - #{e.message}")
+          nil
+        end
+
+        private
+
+        def valid?(trace_data)
+          trace_data.event == :call
+        end
+
+        def link_to_parent(trace_data)
+          return unless current_call
+
+          current_call.add_child(trace_data)
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/{evaluators/hidden_evaluator.rb → steps/hidden_step.rb}

@@ -1,17 +1,21 @@
 # frozen_string_literal: true
 
-require_relative "base_evaluator"
+require_relative "base_step"
 
 module TraceViz
   module Collectors
-    module Evaluators
-      class HiddenEvaluator < BaseEvaluator
-        def pass?(trace_data)
-          !allowed_events.include?(trace_data.event)
+    module Steps
+      class HiddenStep < BaseStep
+        def call(trace_data)
+          trace_data unless pass?(trace_data)
         end
 
         private
 
+        def pass?(trace_data)
+          !allowed_events.include?(trace_data.event)
+        end
+
         def allowed_events
           config.execution[:show_trace_events]
         end

data/lib/trace_viz/collectors/steps/linking_step.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require "trace_viz/context"
+require_relative "base_step"
+
+module TraceViz
+  module Collectors
+    module Steps
+      class LinkingStep < BaseStep
+        def call(trace_data)
+          valid?(trace_data) ? linking_trace(trace_data) : trace_data
+        rescue StandardError => e
+          logger.error("Linking failed for trace_data ID: #{trace_data.id} - #{e.message}")
+          nil
+        end
+
+        private
+
+        def valid?(trace_data)
+          return false unless current_call
+
+          [
+            trace_data.event == :return,
+            current_call.event == :call,
+            trace_data.action_id == current_call.action_id,
+          ].all?
+        end
+
+        def linking_trace(trace_data)
+          trace_data.link(current_call)
+          trace_data
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/{evaluators/filter_evaluator.rb → steps/validation_step.rb}

@@ -1,26 +1,29 @@
 # frozen_string_literal: true
 
 require "trace_viz/collectors/filters/registry"
-require_relative "base_evaluator"
+require_relative "base_step"
 
 module TraceViz
   module Collectors
-    module Evaluators
-      class FilterEvaluator < BaseEvaluator
+    module Steps
+      class ValidationStep < BaseStep
         def initialize
           super()
-
           @filters = build_filters.freeze
         end
 
-        def pass?(trace_data)
-          filters.all? { |filter| filter.apply?(trace_data) }
+        def call(trace_data)
+          trace_data if pass?(trace_data)
         end
 
         private
 
         attr_reader :filters
 
+        def pass?(trace_data)
+          filters.all? { |filter| filter.apply?(trace_data) }
+        end
+
         def build_filters
           Collectors::Filters::Registry.build(config.filters)
         end

data/lib/trace_viz/collectors/steps.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+Dir[File.join(__dir__, "steps/**/*.rb")].each { |file| require_relative file }
+
+module TraceViz
+  module Collectors
+    module Steps
+    end
+  end
+end

data/lib/trace_viz/collectors/trace_pipeline.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Collectors
+    class TracePipeline
+      def initialize
+        @steps = []
+      end
+
+      def add_step(step)
+        @steps << step
+      end
+
+      def process(trace_data)
+        @steps.each do |step|
+          trace_data = step.call(trace_data)
+          return nil unless trace_data
+        end
+        trace_data
+      rescue StandardError => e
+        TraceViz.logger.error("Pipeline processing failed for trace_data ID: #{trace_data.id} - #{e.message}")
+        nil
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/trace_pipeline_builder.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require_relative "trace_pipeline"
+require_relative "steps"
+
+module TraceViz
+  module Collectors
+    class TracePipelineBuilder
+      class << self
+        def build
+          TracePipeline.new.tap do |pipeline|
+            pipeline.add_step(Steps::ValidationStep.new)
+
+            #
+            # Those actions require access to the current call,
+            # so it needs to be run before current call is updated
+            #
+            pipeline.add_step(Steps::BuildHierarchyStep.new)
+            pipeline.add_step(Steps::LinkingStep.new)
+
+            pipeline.add_step(Steps::AssignDepthForCallStep.new)
+            pipeline.add_step(Steps::AssignDepthForReturnStep.new)
+            pipeline.add_step(Steps::HiddenStep.new)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/collectors/trace_point_collector.rb

@@ -4,8 +4,6 @@ require "trace_viz/trace_data/trace_point_builder"
 require_relative "base_collector"
 require_relative "matchers/trace_point_action_matcher"
 require_relative "matchers/within_depth_matcher"
-require_relative "depth_manager"
-require_relative "trace_stats"
 
 module TraceViz
   module Collectors
@@ -21,7 +19,7 @@ module TraceViz
 
       attr_reader :action_matcher, :within_depth_matcher
 
-      def collectible?(trace_point)
+      def can_collect?(trace_point)
         within_depth? || match_action?(trace_point)
       end
 
@@ -29,10 +27,6 @@ module TraceViz
         TraceData::TracePointBuilder.build(trace_point)
       end
 
-      def update_trace_depth(trace_data)
-        depth_manager.align(trace_data)
-      end
-
       def match_action?(trace_point)
         action_matcher.matches?(trace_point)
       end
@@ -42,10 +36,6 @@ module TraceViz
 
         within_depth_matcher.matches?(depth)
       end
-
-      def depth_manager
-        @depth_manager ||= DepthManager.new
-      end
     end
   end
 end

data/lib/trace_viz/config/validator.rb

@@ -27,8 +27,9 @@ module TraceViz
 
       def validate_params(value)
         mode = value[:mode]
-        unless Defaults.valid_param_modes.include?(mode)
-          raise ArgumentError, "Invalid param mode: #{mode}. Valid modes are #{Defaults.valid_param_modes.join(", ")}."
+        unless Defaults::Config.valid_param_mode?(mode)
+          raise ArgumentError,
+            "Invalid param mode: #{mode}. Valid modes are #{Defaults::Config.valid_param_modes.join(", ")}."
         end
 
         if value[:truncate_values] && (!value[:truncate_values].is_a?(Integer) || value[:truncate_values] <= 0)
@@ -37,7 +38,7 @@ module TraceViz
       end
 
       def validate_result(value)
-        if value[:truncate_length] && (!value[:truncate_length].is_a?(Integer) || value[:truncate_length] <= 0)
+        if value[:truncate_value] && !value[:truncate_value].is_a?(Integer)
           raise ArgumentError, "Truncate values must be a positive integer."
         end
       end
@@ -64,9 +65,9 @@ module TraceViz
         end
 
         format = value[:format]
-        unless Defaults.valid_export_formats.include?(format)
+        unless Defaults::Config.valid_export_format?(format)
           raise ArgumentError,
-            "Invalid export format: #{format}. Valid formats are #{Defaults.valid_export_formats.join(", ")}."
+            "Invalid export format: #{format}. Valid formats are #{Defaults::Config.valid_export_formats.join(", ")}."
         end
       end
     end

data/lib/trace_viz/configuration.rb

@@ -7,12 +7,12 @@ require "trace_viz/config/copier"
 module TraceViz
   class Configuration
     attr_reader :logger, :settings
-    attr_reader(*Defaults.fetch_defaults.keys)
+    attr_reader(*Defaults::Config.fetch.keys)
 
     def initialize
       @logger = Logger.new
       @validator = Config::Validator.new
-      @settings = Defaults.fetch_defaults
+      @settings = Defaults::Config.fetch
       define_dynamic_accessors
     end
 
@@ -32,7 +32,7 @@ module TraceViz
     end
 
     def reset_defaults
-      @settings = Defaults.fetch_defaults
+      @settings = Defaults::Config.fetch
     end
 
     def dup

data/lib/trace_viz/context/tracking_context.rb

@@ -15,6 +15,10 @@ module TraceViz
         @active_calls = Tracking::ActiveCalls.new
       end
 
+      def current_call
+        active_calls.current
+      end
+
       def current_depth
         active_calls.size
       end

data/lib/trace_viz/core/tracer.rb

@@ -10,6 +10,8 @@ module TraceViz
   module Core
     class Tracer
       def trace(**options, &block)
+        Defaults::Themes.apply_solarized_theme
+
         Context::Manager.with_contexts(config: options, tracking: {}) do
           adapter = Adapters::TracePointAdapter.new
           adapter.trace(&block)

data/lib/trace_viz/defaults/actions.rb

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Defaults
+    class Actions
+      EMOJIS = {
+        default: "",
+        info: "ℹ️",
+        success: "✅",
+        error: "❌",
+        warn: "⚠️",
+        start: "🚀",
+        processing: "🔄",
+        finish: "🏁",
+        exported: "📤",
+        skipped: "⏩",
+        stats: "📊",
+      }.freeze
+
+      COLORS = {
+        # Default actions
+        default: :reset,
+        info: :cyan,
+        success: :green,
+        error: :bright_red,
+        warn: :yellow,
+
+        # Processing actions
+        start: :bright_cyan,
+        processing: [:dim, :bright_white],
+        finish: :bright_magenta,
+
+        # Export & stats actions
+        exported: :bright_green,
+        skipped: :bright_white,
+        stats: [:bold, :underline, :bright_white],
+
+        # Trace data actions
+        trace_indent: :dim,
+        trace_depth: :blue,
+        trace_depth_prefix: [:dim, :italic, :blue],
+        trace_depth_open: [:dim, :bright_blue],
+        trace_depth_value: :bright_red,
+        trace_depth_close: [:dim, :bright_blue],
+        trace_method_name: :bright_cyan,
+        trace_method_class: :bright_green,
+        trace_method_sign: :blue,
+        trace_method_action: [:bold, :bright_cyan],
+        trace_source_location: [:dim, :bright_white],
+        trace_params_key: :bright_yellow,
+        trace_params_value: [:dim, :bright_yellow],
+        trace_result_prefix: [:italic, :bright_blue],
+        trace_result_value: :bright_white,
+        trace_execution_time: [:dim, :bright_red],
+      }.freeze
+
+      class << self
+        def emojis
+          EMOJIS
+        end
+
+        def colors
+          COLORS
+        end
+
+        def keys
+          colors.keys
+        end
+
+        def emoji_for(action)
+          EMOJIS.fetch(action, EMOJIS[:default])
+        end
+
+        def source_colors
+          Defaults.action_colors || COLORS
+        end
+
+        def colors_for(action)
+          Array(source_colors.fetch(action, source_colors[:default]))
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/defaults/colors.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Defaults
+    class Colors
+      COLORS = {
+        reset: "\e[0m",
+        bold: "\e[1m",
+        dim: "\e[2m",
+        italic: "\e[3m",
+        underline: "\e[4m",
+        reverse: "\e[7m",
+        hidden: "\e[8m",
+        strikethrough: "\e[9m",
+      }
+
+      # Reference ASCII color codes
+      # https://talyian.github.io/ansicolors/
+
+      # Map 256-color ANSI codes
+      (0..255).each do |i|
+        COLORS["color_#{i}".to_sym] = "\e[38;5;#{i}m"
+      end
+
+      # To add background colors
+      (0..255).each do |i|
+        COLORS["bg_color_#{i}".to_sym] = "\e[48;5;#{i}m"
+      end
+
+      # Predefined names
+      COLORS.merge!({
+        black: "\e[30m",
+        red: "\e[31m",
+        green: "\e[32m",
+        yellow: "\e[33m",
+        blue: "\e[34m",
+        magenta: "\e[35m",
+        cyan: "\e[36m",
+        white: "\e[37m",
+        bright_black: "\e[90m",
+        bright_red: "\e[91m",
+        bright_green: "\e[92m",
+        bright_yellow: "\e[93m",
+        bright_blue: "\e[94m",
+        bright_magenta: "\e[95m",
+        bright_cyan: "\e[96m",
+        bright_white: "\e[97m",
+      })
+
+      class << self
+        def all
+          COLORS
+        end
+
+        def fetch(color)
+          COLORS.fetch(color, COLORS[:reset])
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/defaults/config.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Defaults
+    class Config
+      DEFAULTS = {
+        general: {
+          tab_size: 2,
+          mode: :summary, # :summary, :verbose
+          group_keys: [:event, :klass, :action],
+          show_indent: true,
+          show_depth: true,
+          max_display_depth: 3,
+          show_method_name: true,
+        },
+        source_location: {
+          show: false,
+          truncate_length: 100,
+        },
+        params: {
+          show: true,
+          mode: :name_and_value,
+          truncate_values: 10,
+          truncate_length: -1,
+        },
+        result: {
+          show: true,
+          truncate_value: 50,
+          truncate_length: 30,
+        },
+        execution: {
+          show_time: true,
+          show_trace_events: [:call, :return],
+        },
+        filters: [
+          :exclude_internal_call,
+          # :exclude_default_classes,
+          # :exclude_rails_framework,
+          # include_classes: {
+          #   classes: [],
+          # },
+          # exclude_classes: {
+          #   classes: [],
+          # },
+          # include_gems: {
+          #   app_running: true,
+          #   app_path: Dir.pwd,
+          #   gems: [],
+          # },
+          # exclude_gems: {
+          #   gems: [],
+          # },
+        ],
+        log: {
+          enabled: true,
+          runtime: false,
+          post_collection: true,
+          stats: true,
+        },
+        export: {
+          enabled: true,
+          path: "tmp",
+          format: :txt,
+          overwrite: false,
+        },
+      }.freeze
+
+      class << self
+        def fetch
+          DEFAULTS.dup
+        end
+
+        def valid_param_modes
+          [:name_and_value, :name_only, :value_only].freeze
+        end
+
+        def valid_param_mode?(mode)
+          valid_param_modes.include?(mode)
+        end
+
+        def valid_export_formats
+          [:txt, :json, :yaml, :mermaid].freeze
+        end
+
+        def valid_export_format?(format)
+          valid_export_formats.include?(format)
+        end
+      end
+    end
+  end
+end