trace_viz 1.0.0 → 1.0.1

data/lib/trace_viz/extractors/diagram/processors/message_processor.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+require "trace_viz/formatters/diagram/sequence/message_formatter"
+require "trace_viz/builders/diagram/message_builder"
+
+module TraceViz
+  module Extractors
+    module Diagram
+      module Processors
+        class MessageProcessor
+          def initialize(participants)
+            @formatter = Formatters::Diagram::Sequence::MessageFormatter.new
+            @message_builder = Builders::Diagram::MessageBuilder.new(@formatter, participants)
+          end
+
+          def process_node(node, caller_node = nil)
+            trace = node.data
+            caller_trace = caller_node&.data
+
+            [].tap do |messages|
+              # Handle participant transitions
+              messages << @message_builder.build_call_message(
+                caller_trace,
+                trace,
+              ) if caller_trace
+
+              # Process the current node
+              messages << handle_loop_start(trace)
+              messages << handle_internal_message(trace)
+              messages << handle_note(trace)
+              messages << handle_activation(trace)
+              messages.concat(process_children(node))
+              messages << handle_deactivation(trace)
+              messages << handle_loop_end(trace)
+
+              # Update the current node after processing
+              messages << @message_builder.build_return_message(
+                trace,
+                caller_trace,
+              ) if caller_trace
+            end.compact
+          end
+
+          private
+
+          def handle_loop_start(trace)
+            return unless loop?(trace)
+
+            @message_builder.build_loop_start_message(trace)
+          end
+
+          def handle_internal_message(trace)
+            @message_builder.build_internal_message(trace)
+          end
+
+          def handle_note(trace)
+            @message_builder.build_note(trace)
+          end
+
+          def handle_activation(trace)
+            return unless node_has_children?(trace)
+
+            @message_builder.build_activate_message(trace)
+          end
+
+          def handle_deactivation(trace)
+            return unless node_has_children?(trace)
+
+            @message_builder.build_deactivate_message(trace)
+          end
+
+          def handle_loop_end(trace)
+            return unless loop?(trace)
+
+            @message_builder.build_loop_end_message
+          end
+
+          def process_children(node)
+            node.children.flat_map { |child| process_node(child, node) }
+          end
+
+          def loop?(trace)
+            trace.key == :summary_group
+          end
+
+          def node_has_children?(trace)
+            trace.children.any?
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/base_formatter.rb

@@ -7,6 +7,7 @@ module TraceViz
     class BaseFormatter
       include Helpers::ConfigHelper
 
+      # Format the data to a line
       def call
         raise NotImplementedError
       end

data/lib/trace_viz/formatters/base_formatter_factory.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Formatters
+    class BaseFormatterFactory
+      def initialize(formatters)
+        @formatters = formatters.freeze
+      end
+
+      def fetch_formatter(key)
+        @formatters.fetch(key) do
+          raise ArgumentError, "Unsupported formatter key: #{key}. Available keys: #{available_keys.join(", ")}"
+        end
+      end
+
+      private
+
+      def available_keys
+        @formatters.keys
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/diagram/sequence/base_formatter.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require_relative "../../diagram_formatter"
+
+module TraceViz
+  module Formatters
+    module Diagram
+      module Sequence
+        class BaseFormatter < Formatters::DiagramFormatter
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/diagram/sequence/message_formatter.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative "base_formatter"
+require "trace_viz/formatters/helpers"
+
+module TraceViz
+  module Formatters
+    module Diagram
+      module Sequence
+        class MessageFormatter < BaseFormatter
+          include Helpers::ParamsHelper
+          include Helpers::Diagram::ActionHelper
+          include Helpers::Diagram::ResultHelper
+
+          def format_internal_message(trace)
+            [
+              action_representation(trace),
+              params_representation(trace),
+            ].join("")
+          end
+
+          def format_result(trace)
+            result_representation(trace)
+          end
+
+          def format_call
+            "Calling"
+          end
+
+          def format_return
+            "Returning"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/diagram_formatter.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require_relative "base_formatter"
+
+module TraceViz
+  module Formatters
+    class DiagramFormatter < BaseFormatter
+    end
+  end
+end

data/lib/trace_viz/formatters/export/formatter_factory.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require_relative "../base_formatter_factory"
 require_relative "method_call_formatter"
 require_relative "method_return_formatter"
 require_relative "summary_group_formatter"
@@ -7,19 +8,13 @@ require_relative "summary_group_formatter"
 module TraceViz
   module Formatters
     module Export
-      class FormatterFactory
-        FORMATTERS = {
-          call: MethodCallFormatter.new,
-          return: MethodReturnFormatter.new,
-          summary_group: SummaryGroupFormatter.new,
-        }.freeze
-
-        class << self
-          def fetch_formatter(key)
-            FORMATTERS.fetch(key) do
-              raise ArgumentError, "Unsupported factory key: #{key}"
-            end
-          end
+      class FormatterFactory < BaseFormatterFactory
+        def initialize
+          super(
+            call: Export::MethodCallFormatter.new,
+            return: Export::MethodReturnFormatter.new,
+            summary_group: Export::SummaryGroupFormatter.new
+          )
         end
       end
     end

data/lib/trace_viz/formatters/export/summary_group_formatter.rb

@@ -26,7 +26,7 @@ module TraceViz
         def summary_info(trace_data)
           "[Summary: #{trace_data.count} calls | Avg Time: #{format(
             "in %.6fms",
-            trace_data.average_duration,
+            trace_data.duration,
           )}]"
         end
       end

data/lib/trace_viz/formatters/helpers/digram/action_helper.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Diagram
+        module ActionHelper
+          def action_representation(trace_data)
+            return unless config.general[:show_method_name]
+
+            trace_data.action
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/helpers/digram/result_helper.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Diagram
+        module ResultHelper
+          def result_representation(trace_data)
+            return unless config.result[:show]
+
+            result = truncate_structure(trace_data.result)
+            result = sinitize_result(result)
+            result
+          end
+
+          private
+
+          def truncate_structure(input)
+            Utils::Format::ValueTruncator.truncate(
+              input,
+              length: config.result[:truncate_value],
+              hash_length: config.result[:truncate_length],
+            )
+          end
+
+          def sinitize_result(input)
+            input.to_s
+              # .gsub('"', "")
+              # .gsub("'", "")
+              .gsub("#", "")
+              .strip
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trace_viz/formatters/helpers/log/params_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -26,7 +26,7 @@ module TraceViz
           def truncate_values(params, length)
             return params unless length
 
-            params.transform_values { |value| Utils::FormatUtils.truncate_value(value, length) }
+            params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: length) }
           end
 
           def colorize_keys_and_values(params)
@@ -35,13 +35,17 @@ module TraceViz
           end
 
           def format_as_string(params, mode)
-            Utils::FormatUtils.format_key_value_pairs(params, mode)
+            Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
           end
 
           def wrap_params(params_string, config)
             return unless params_string
 
-            truncated = Utils::FormatUtils.truncate_value(params_string, config.params[:truncate_length])
+            truncated = Utils::Format::ValueTruncator.truncate(
+              params_string,
+              length: config.params[:truncate_length],
+            )
+
             "(#{truncated})"
           end
         end

data/lib/trace_viz/formatters/helpers/log/result_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -10,9 +10,9 @@ module TraceViz
           def format_result(trace_data, config)
             return unless config.result[:show]
 
-            truncated_result = Utils::FormatUtils.truncate_value(
+            truncated_result = Utils::Format::ValueTruncator.truncate(
               trace_data.result.inspect,
-              config.result[:truncate_length],
+              length: config.result[:truncate_value],
             )
 
             prefix = colorize_for("#=>", :trace_result_prefix)

data/lib/trace_viz/formatters/helpers/log/summary/params_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -27,7 +27,7 @@ module TraceViz
             def truncate_values(params, length)
               return params unless length
 
-              params.transform_values { |value| Utils::FormatUtils.truncate_value(value, length) }
+              params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: length) }
             end
 
             def colorize_keys_and_values(params)
@@ -36,13 +36,17 @@ module TraceViz
             end
 
             def format_as_string(params, mode)
-              Utils::FormatUtils.format_key_value_pairs(params, mode)
+              Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
             end
 
             def wrap_params(params_string, config)
               return unless params_string
 
-              truncated = Utils::FormatUtils.truncate_value(params_string, config.params[:truncate_length])
+              truncated = Utils::Format::ValueTruncator.truncate(
+                params_string,
+                length: config.params[:truncate_length],
+              )
+
               "(#{truncated})"
             end
           end

data/lib/trace_viz/formatters/helpers/params_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -24,17 +24,21 @@ module TraceViz
         def truncate_param_values(params, max_length)
           return params unless max_length
 
-          params.transform_values { |value| Utils::FormatUtils.truncate_value(value, max_length) }
+          params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: max_length) }
         end
 
         def stringify_params(params, mode)
-          Utils::FormatUtils.format_key_value_pairs(params, mode)
+          Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
         end
 
         def wrap_params_string(params_string, config)
           return unless params_string
 
-          truncated_string = Utils::FormatUtils.truncate_value(params_string, config.params[:truncate_length])
+          truncated_string = Utils::Format::ValueTruncator.truncate(
+            params_string,
+            length: config.params[:truncate_length],
+          )
+
           "(#{truncated_string})"
         end
       end

data/lib/trace_viz/formatters/helpers/result_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -9,10 +9,11 @@ module TraceViz
         def result_representation(trace_data)
           return unless config.result[:show]
 
-          truncated_result = Utils::FormatUtils.truncate_value(
+          truncated_result = Utils::Format::ValueTruncator.truncate(
             trace_data.result.inspect,
-            config.result[:truncate_length],
+            length: config.result[:truncate_value],
           )
+
           "#=> #{truncated_result}"
         end
       end

data/lib/trace_viz/formatters/helpers/source_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -9,11 +9,12 @@ module TraceViz
         def source_location_representation(trace_data)
           return unless config.source_location[:show]
 
-          truncated_path = Utils::FormatUtils.truncate_value(
+          truncated_path = Utils::Format::ValueTruncator.truncate(
             trace_data.path,
-            config.source_location[:truncate_length],
+            length: config.source_location[:truncate_length],
             direction: :start,
           )
+
           "at #{truncated_path}:#{trace_data.line_number}"
         end
       end

data/lib/trace_viz/formatters/helpers/summary/params_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -25,13 +25,17 @@ module TraceViz
           end
 
           def format_as_string(params, mode)
-            Utils::FormatUtils.format_key_value_pairs(params, mode)
+            Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
           end
 
           def wrap_params_template(params_string, config)
             return unless params_string
 
-            truncated = Utils::FormatUtils.truncate_value(params_string, config.params[:truncate_length])
+            truncated = Utils::Format::ValueTruncator.truncate(
+              params_string,
+              length: config.params[:truncate_length],
+            )
+
             "(#{truncated})"
           end
         end

data/lib/trace_viz/formatters/helpers/summary/source_helper.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
@@ -10,9 +10,9 @@ module TraceViz
           def source_location_representation(trace_data)
             return unless config.source_location[:show]
 
-            truncated_path = Utils::FormatUtils.truncate_value(
+            truncated_path = Utils::Format::ValueTruncator.truncate(
               trace_data.path,
-              config.source_location[:truncate_length],
+              length: config.source_location[:truncate_length],
               direction: :start,
             )
             "at #{truncated_path}"

data/lib/trace_viz/formatters/log/formatter_factory.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require_relative "../base_formatter_factory"
 require_relative "method_call_formatter"
 require_relative "method_return_formatter"
 require_relative "summary_group_formatter"
@@ -7,19 +8,13 @@ require_relative "summary_group_formatter"
 module TraceViz
   module Formatters
     module Log
-      class FormatterFactory
-        FORMATTERS = {
-          call: MethodCallFormatter.new,
-          return: MethodReturnFormatter.new,
-          summary_group: SummaryGroupFormatter.new,
-        }.freeze
-
-        class << self
-          def fetch_formatter(key)
-            FORMATTERS.fetch(key) do
-              raise ArgumentError, "Unsupported factory key: #{key}"
-            end
-          end
+      class FormatterFactory < BaseFormatterFactory
+        def initialize
+          super(
+            call: Log::MethodCallFormatter.new,
+            return: Log::MethodReturnFormatter.new,
+            summary_group: Log::SummaryGroupFormatter.new
+          )
         end
       end
     end

data/lib/trace_viz/formatters/log/summary_group_formatter.rb

@@ -27,7 +27,7 @@ module TraceViz
           colorize(
             "[Summary: #{trace_data.count} calls | Avg Time: #{format(
               "in %.6fms",
-              trace_data.average_duration,
+              trace_data.duration,
             )}]",
             :bold,
             :italic,

data/lib/trace_viz/helpers/config_helper.rb

@@ -8,6 +8,10 @@ module TraceViz
       def config
         Context.for(:config).configuration
       end
+
+      def fetch_general_config(key)
+        config.general[key]
+      end
     end
   end
 end

data/lib/trace_viz/loggers/post_collection_logger.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "trace_viz/formatters/log/formatter_factory"
+require "trace_viz/renderers/renderer_builder"
 require "trace_viz/shared"
 require_relative "base_logger"
 require_relative "log_level_resolver"
@@ -8,13 +9,19 @@ require_relative "log_level_resolver"
 module TraceViz
   module Loggers
     class PostCollectionLogger < BaseLogger
+      include Helpers::ConfigHelper
       include Shared::RendererHelper
 
       def initialize(collector)
         super()
 
         @collector = collector
-        @renderer = build_renderer(collector, Formatters::Log::FormatterFactory)
+
+        @renderer = Renderers::RendererBuilder.build(
+          collector,
+          key: fetch_general_config(:mode),
+          formatter_factory: Formatters::Log::FormatterFactory.new,
+        )
       end
 
       def log
@@ -26,9 +33,7 @@ module TraceViz
       attr_reader :collector, :renderer
 
       def log_line(line)
-        log_message(resolve_log_level(line[:trace_data]), line[:line])
-
-        process_lines(line[:nested_lines]) { |nested| log_line(nested) }
+        log_message(resolve_log_level(line[:data]), line[:line])
       end
 
       def resolve_log_level(trace_data)

data/lib/trace_viz/loggers/trace_logger.rb

@@ -11,6 +11,7 @@ module TraceViz
         super()
 
         @trace_data = trace_data
+        @formatter_factory = Formatters::Log::FormatterFactory.new
       end
 
       def log
@@ -19,18 +20,14 @@ module TraceViz
 
       private
 
-      attr_reader :trace_data
+      attr_reader :trace_data, :formatter_factory
 
       def log_level
         LogLevelResolver.resolve(trace_data)
       end
 
       def formatted_message
-        fetch_formatter(trace_data).call(trace_data)
-      end
-
-      def fetch_formatter(trace_data)
-        Formatters::Log::FormatterFactory.fetch_formatter(trace_data.event)
+        formatter_factory.fetch_formatter(trace_data.key).call(trace_data)
       end
     end
   end

data/lib/trace_viz/managers/diagram/participant_manager.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Managers
+    module Diagram
+      class ParticipantsManager
+        def initialize(participants)
+          @participants_map = build_participants_map(participants)
+        end
+
+        def find(name)
+          @participants_map[name.to_s]
+        end
+
+        private
+
+        def build_participants_map(participants)
+          participants.each_with_object({}) { |participant, memo| memo[participant.name] = participant }
+        end
+      end
+    end
+  end
+end