pry--pry/lib/pry/commands/ls/local_names.rb

class Pry
  class Command::Ls < Pry::ClassCommand
    class LocalNames < Pry::Command::Ls::Formatter

      def initialize(target, no_user_opts, sticky_locals, args)
        super(target)
        @no_user_opts = no_user_opts
        @sticky_locals = sticky_locals
        @args = args
      end

      def correct_opts?
        super || (@no_user_opts && @args.empty?)
      end

      def output_self
        local_vars = grep.regexp[@target.eval('local_variables')]
        output_section('locals', format(local_vars))
      end

      private

      def format(locals)
        locals.sort_by(&:downcase).map do |name|
          if @sticky_locals.include?(name.to_sym)
            color(:pry_var, name)
          else
            color(:local_var, name)
          end
        end
      end

    end
  end
end