File Manager

Viewing File: file_formatter.rb

# frozen_string_literal: true

class Pry
  class Command
    class Cat
      class FileFormatter < AbstractFormatter
        attr_reader :file_with_embedded_line
        attr_reader :opts
        attr_reader :pry_instance

        def initialize(file_with_embedded_line, pry_instance, opts)
          unless file_with_embedded_line
            raise CommandError, "Must provide a filename, --in, or --ex."
          end

          @file_with_embedded_line = file_with_embedded_line
          @opts = opts
          @pry_instance = pry_instance
          @code_from_file = Pry::Code.from_file(file_name)
        end

        def format
          set_file_and_dir_locals(file_name, pry_instance, pry_instance.current_context)
          decorate(@code_from_file)
        end

        def file_and_line
          file_name, line_num = file_with_embedded_line.split(%r{:(?!/|\\)})

          [file_name, line_num ? line_num.to_i : nil]
        end

        private

        def file_name
          file_and_line.first
        end

        def line_number
          file_and_line.last
        end

        def code_window_size
          pry_instance.config.default_window_size || 7
        end

        def decorate(content)
          if line_number
            super(content.around(line_number, code_window_size))
          else
            super
          end
        end

        def code_type
          opts[:type] || detect_code_type_from_file(file_name)
        end

        def detect_code_type_from_file(file_name)
          code_type = @code_from_file.code_type

          if code_type == :unknown
            name = File.basename(file_name).split('.', 2).first
            case name
            when "Rakefile", "Gemfile"
              :ruby
            else
              :text
            end
          else
            code_type
          end
        end
      end
    end
  end
end