trace_viz 0.0.2 → 1.0.1

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/base_formatter.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require_relative "../trace_data_formatter"
+
+module TraceViz
+  module Formatters
+    module Export
+      class BaseFormatter < TraceDataFormatter
+      end
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative "../base_formatter_factory"
+require_relative "method_call_formatter"
+require_relative "method_return_formatter"
+require_relative "summary_group_formatter"
+
+module TraceViz
+  module Formatters
+    module Export
+      class FormatterFactory < BaseFormatterFactory
+        def initialize
+          super(
+            call: Export::MethodCallFormatter.new,
+            return: Export::MethodReturnFormatter.new,
+            summary_group: Export::SummaryGroupFormatter.new
+          )
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "base_formatter"
+
+module TraceViz
+  module Formatters
+    module Export
+      class MethodCallFormatter < BaseFormatter
+        def call(trace_data)
+          [
+            indent_representation(trace_data),
+            depth_representation(trace_data),
+            method_name_representation(trace_data),
+            source_location_representation(trace_data),
+            params_representation(trace_data),
+          ].compact.join(" ")
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative "base_formatter"
+
+module TraceViz
+  module Formatters
+    module Export
+      class MethodReturnFormatter < BaseFormatter
+        def call(trace_data)
+          [
+            indent_representation(trace_data),
+            depth_representation(trace_data),
+            method_name_representation(trace_data),
+            result_representation(trace_data),
+            source_location_representation(trace_data),
+            execution_time_representation(trace_data),
+          ].compact.join(" ")
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require_relative "base_formatter"
+
+module TraceViz
+  module Formatters
+    module Export
+      class SummaryGroupFormatter < BaseFormatter
+        include Helpers::Summary::ParamsHelper
+        include Helpers::Summary::SourceHelper
+
+        def call(trace_data)
+          [
+            indent_representation(trace_data),
+            depth_representation(trace_data),
+            Defaults::Actions.emoji_for(:processing),
+            method_name_representation(trace_data),
+            source_location_representation(trace_data),
+            format_params_template(trace_data, config),
+            summary_info(trace_data),
+          ].compact.join(" ")
+        end
+
+        private
+
+        def summary_info(trace_data)
+          "[Summary: #{trace_data.count} calls | Avg Time: #{format(
+            "in %.6fms",
+            trace_data.duration,
+          )}]"
+        end
+      end
+    end
+  end
+end

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

@@ -4,16 +4,10 @@ module TraceViz
   module Formatters
     module Helpers
       module DepthHelper
-        def depth_representation
+        def depth_representation(trace_data)
           return unless config.general[:show_depth]
 
-          if trace_data.depth.zero?
-            "depth[0]"
-          elsif trace_data.event == :call
-            "depth[#{trace_data.depth}]"
-          elsif trace_data.event == :return
-            "depth[#{trace_data.depth}]"
-          end
+          "depth[#{trace_data.depth}]"
         end
       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/indent_helper.rb

@@ -4,7 +4,7 @@ module TraceViz
   module Formatters
     module Helpers
       module IndentHelper
-        def indent_representation
+        def indent_representation(trace_data)
           return unless config.general[:show_indent] && config.general[:show_depth]
 
           " " * (config.general[:tab_size] * trace_data.depth)

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

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils/colorize"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module ColorHelper
+          def colorize_for(text, action)
+            colors = Defaults::Actions.colors_for(action)
+
+            colorize(text, *colors)
+          end
+
+          def colorize(text, *colors)
+            Utils::Colorize.colorize(text, *colors)
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module DepthHelper
+          def format_depth(trace_data, config)
+            return unless config.general[:show_depth]
+
+            prefix = colorize_for("depth", :trace_depth_prefix)
+            open_block = colorize_for("[", :trace_depth_open)
+            number = colorize_for(trace_data.depth, :trace_depth_value)
+            close_block = colorize_for("]", :trace_depth_close)
+
+            [prefix, open_block, number, close_block].join
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module MethodNameHelper
+          def format_method_name(trace_data, config)
+            return unless config.general[:show_method_name]
+
+            "#{format_class_name(trace_data)}#{format_action_name(trace_data)}"
+          end
+
+          private
+
+          def format_class_name(trace_data)
+            klass = colorize_for(trace_data.klass, :trace_method_class)
+            method_sign = colorize_for("#", :trace_method_sign)
+            [klass, method_sign].join
+          end
+
+          def format_action_name(trace_data)
+            colorize_for(trace_data.action, :trace_method_action)
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module ParamsHelper
+          def format_params(trace_data, config)
+            return unless config.params[:show]
+
+            params = prepare_params(trace_data.params, config)
+            wrap_params(params, config)
+          end
+
+          private
+
+          def prepare_params(params, config)
+            params
+              .then { |p| truncate_values(p, config.params[:truncate_values]) }
+              .then { |p| colorize_keys_and_values(p) }
+              .then { |p| format_as_string(p, config.params[:mode]) }
+          end
+
+          def truncate_values(params, length)
+            return params unless length
+
+            params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: length) }
+          end
+
+          def colorize_keys_and_values(params)
+            params.transform_keys { |key| colorize_for(key.to_s, :trace_params_key) }
+              .transform_values { |value| colorize_for(value.to_s, :trace_params_value) }
+          end
+
+          def format_as_string(params, mode)
+            Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
+          end
+
+          def wrap_params(params_string, config)
+            return unless params_string
+
+            truncated = Utils::Format::ValueTruncator.truncate(
+              params_string,
+              length: config.params[:truncate_length],
+            )
+
+            "(#{truncated})"
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module ResultHelper
+          def format_result(trace_data, config)
+            return unless config.result[:show]
+
+            truncated_result = Utils::Format::ValueTruncator.truncate(
+              trace_data.result.inspect,
+              length: config.result[:truncate_value],
+            )
+
+            prefix = colorize_for("#=>", :trace_result_prefix)
+            result = colorize_for(truncated_result, :trace_result_value)
+
+            [prefix, result].join(" ")
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Log
+        module Summary
+          module ParamsHelper
+            def format_params_template(trace_data, config)
+              return unless config.params[:show]
+
+              params = prepare_params(trace_data.params, config)
+              wrap_params(params, config)
+            end
+
+            private
+
+            def prepare_params(params, config)
+              params
+                .then { |p| truncate_values(p, config.params[:truncate_values]) }
+                .then { |p| colorize_keys_and_values(p) }
+                .then { |p| format_as_string(p, config.params[:mode]) }
+            end
+
+            def truncate_values(params, length)
+              return params unless length
+
+              params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: length) }
+            end
+
+            def colorize_keys_and_values(params)
+              params.transform_keys { |key| colorize_for(key.to_s, :trace_params_key) }
+                .transform_values { |value| colorize("<#{value.class}>", :yellow) }
+            end
+
+            def format_as_string(params, mode)
+              Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
+            end
+
+            def wrap_params(params_string, config)
+              return unless params_string
+
+              truncated = Utils::Format::ValueTruncator.truncate(
+                params_string,
+                length: config.params[:truncate_length],
+              )
+
+              "(#{truncated})"
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -4,7 +4,7 @@ module TraceViz
   module Formatters
     module Helpers
       module MethodDetailsHelper
-        def method_name_representation
+        def method_name_representation(trace_data)
           return unless config.general[:show_method_name]
 
           "#{trace_data.klass}##{trace_data.action}"

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

@@ -1,24 +1,45 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
     module Helpers
       module ParamsHelper
-        def params_representation
+        def params_representation(trace_data)
           return unless config.params[:show]
 
-          truncated_params = trace_data.params.transform_values do |value|
-            Utils::FormatUtils.truncate_value(value, config.params[:truncate_values])
-          end
+          params = process_params(trace_data.params, config)
+          wrap_params_string(params, config)
+        end
+
+        private
+
+        def process_params(params, config)
+          params
+            .then { |p| truncate_param_values(p, config.params[:truncate_values]) }
+            .then { |p| stringify_params(p, config.params[:mode]) }
+        end
+
+        def truncate_param_values(params, max_length)
+          return params unless max_length
+
+          params.transform_values { |value| Utils::Format::ValueTruncator.truncate(value, length: max_length) }
+        end
+
+        def stringify_params(params, mode)
+          Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
+        end
+
+        def wrap_params_string(params_string, config)
+          return unless params_string
 
-          formatted_values = Utils::FormatUtils.format_key_value_pairs(
-            truncated_params,
-            config.params[:mode],
+          truncated_string = Utils::Format::ValueTruncator.truncate(
+            params_string,
+            length: config.params[:truncate_length],
           )
 
-          "(#{formatted_values})"
+          "(#{truncated_string})"
         end
       end
     end

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

@@ -1,18 +1,19 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
     module Helpers
       module ResultHelper
-        def result_representation
+        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,18 +1,20 @@
 # frozen_string_literal: true
 
-require "trace_viz/utils/format_utils"
+require "trace_viz/utils"
 
 module TraceViz
   module Formatters
     module Helpers
       module SourceHelper
-        def source_location_representation
+        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

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Summary
+        module ParamsHelper
+          def format_params_template(trace_data, config)
+            return unless config.params[:show]
+
+            params_template = build_params_template(trace_data.params, config)
+            wrap_params_template(params_template, config)
+          end
+
+          private
+
+          def build_params_template(params, config)
+            params.transform_values do |value|
+              "<#{value.class}>"
+            end.then do |formatted_params|
+              format_as_string(formatted_params, config.params[:mode])
+            end
+          end
+
+          def format_as_string(params, mode)
+            Utils::Format::KeyValueFormatter.format_pairs(params, mode: mode)
+          end
+
+          def wrap_params_template(params_string, config)
+            return unless params_string
+
+            truncated = Utils::Format::ValueTruncator.truncate(
+              params_string,
+              length: config.params[:truncate_length],
+            )
+
+            "(#{truncated})"
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require "trace_viz/utils"
+
+module TraceViz
+  module Formatters
+    module Helpers
+      module Summary
+        module SourceHelper
+          def source_location_representation(trace_data)
+            return unless config.source_location[:show]
+
+            truncated_path = Utils::Format::ValueTruncator.truncate(
+              trace_data.path,
+              length: config.source_location[:truncate_length],
+              direction: :start,
+            )
+            "at #{truncated_path}"
+          end
+        end
+      end
+    end
+  end
+end