pry--pry/lib/pry/commands/cat/file_formatter.rb

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

      def initialize(file_with_embedded_line, _pry_, opts)
        raise CommandError, "Must provide a filename, --in, or --ex." if !file_with_embedded_line

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

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

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

        [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_.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