summaryrefslogtreecommitdiff
path: root/lib/pry/commands/ls/local_vars.rb
blob: 45605e692a56f95fd8fc2acd9960a537da5aeebf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# frozen_string_literal: true

class Pry
  class Command
    class Ls < Pry::ClassCommand
      class LocalVars < Pry::Command::Ls::Formatter
        def initialize(opts, pry_instance)
          super(pry_instance)
          @default_switch = opts[:locals]
          @sticky_locals = pry_instance.sticky_locals
        end

        def output_self
          locals = @target.eval('local_variables').reject do |e|
            @sticky_locals.key?(e.to_sym)
          end
          name_value_pairs = locals.map do |name|
            [name, @target.eval(name.to_s)]
          end
          format(name_value_pairs).join('')
        end

        private

        def format(name_value_pairs)
          sorted = name_value_pairs.sort_by do |_name, value|
            value.to_s.size
          end
          sorted.reverse.map do |name, value|
            colorized_assignment_style(name, format_value(value))
          end
        end

        def colorized_assignment_style(lhs, rhs, desired_width = 7)
          colorized_lhs = color(:local_var, lhs)
          color_escape_padding = colorized_lhs.size - lhs.size
          pad = desired_width + color_escape_padding
          Kernel.format(
            "%-#{pad}<name>s = %<value>s",
            name: color(:local_var, colorized_lhs),
            value: rhs
          )
        end
      end
    end
  end
end