mentor 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 154403f120fb0038895fd393ed5ee5ade659a97c
-  data.tar.gz: d6463cbbb85424479d49a43c83151f8fa425315d
+  metadata.gz: 9d61b738e627b884222d11ae5e7e01b717a4b473
+  data.tar.gz: 30b3b83534080ed8a0da9eb9d32ba11dde4c6b01
 SHA512:
-  metadata.gz: abcf92fe3f316713f95d3a7d19ed7f1cf6cb997dc60369145bd73829a1c502f4dd79ee848ef8627a38f1a8b776d841d2de44be1b5ecf012561df5a9a69b65c3f
-  data.tar.gz: 0a4cb06f1f27148e5a8f80ade2b0b7c3e29631e59b68defdead01c58d00f964f02f1bbce237f32bbcbf51374b793370cfc17a365748566863602e08653d91891
+  metadata.gz: 974dea0e3cf08d028480c1e4925df058fee38583c7a115d7a83d39bc8b09e6467f8371d47110913f5ed9c5c8655ba8666b84561b86ecabd86625b7756f9fa2a0
+  data.tar.gz: 8b8e7aa5f94c8c3b186c05d5bcc7f5013623b16dd2d6754e059d340197e4134861501729e9b0c5a197aacbffd5342bc6e0d6afc812209488c2252ed685c5ba88

data/.gitignore

@@ -1 +1,3 @@
 /Gemfile.lock
+/.todo.reek
+/.rubocop.yml

data/bin/mentor

@@ -9,6 +9,6 @@ end
 ruby_program = ARGV.first
 ruby_program = Dir.pwd + '/' + ruby_program unless ruby_program[0] == '/'
 
-require 'mentor'
-Mentor.enable
+require_relative '../lib/mentor'
+Mentor::Init.setup
 require_relative ruby_program

data/lib/errors/mentor_error.rb

@@ -20,16 +20,20 @@ module Mentor
       end
     end
 
-    private
+    private_class_method
 
     def self.error_classes
       [
         NoMethodDidYouMeanSuggestionError,
+        NoMethodForNilClassForCommonClassError,
+        NoMethodForNilClassError,
         MentorNoMethodError,
         MentorError
       ]
     end
 
+    private
+
     def sections
       [
         Header.new,

data/lib/errors/no_method_did_you_mean_suggestion_error.rb

@@ -3,7 +3,9 @@ module Mentor
   class NoMethodDidYouMeanSuggestionError < MentorNoMethodError
 
     def self.can_handle?
-      super && Mentor.tp.raised_exception&.corrections.any?
+      super &&
+        Mentor.tp.raised_exception.respond_to?(:corrections) &&
+        Mentor.tp.raised_exception.corrections.any?
     end
 
     def sections

data/lib/errors/no_method_for_nil_class_error.rb

@@ -0,0 +1,35 @@
+module Mentor
+
+  class NoMethodForNilClassError < MentorNoMethodError
+
+    def self.can_handle?
+      super && Mentor.tp.raised_exception.to_s['NilClass']
+    end
+
+    def sections
+      [
+        Header.new,
+        RubyErrorComplete.new,
+        RelativePath.new,
+        LinesOfCode.new,
+        ErrorClassSpecificHelp.new([error_class_specific_help]),
+        Suggestion.new("Try setting #{var_for_method} to a value first and then call the method #{method_name} on it.")
+      ]
+    end
+
+    private
+
+    def error_class_specific_help
+      [
+        "It looks like you're trying to call the method #{method_name} on #{var_for_method}.",
+        '',
+        "It's likely #{var_for_method} was never defined, and therefore is equal to nil, which is how Ruby represents 'nothing'."
+      ]
+    end
+
+  end
+
+end
+
+
+

data/lib/errors/no_method_for_nil_class_for_common_class_error.rb

@@ -0,0 +1,74 @@
+module Mentor
+
+  class NoMethodForNilClassForCommonClassError < MentorNoMethodError
+
+    def self.can_handle?
+      super &&
+        Mentor.tp.raised_exception.to_s['NilClass'] &&
+        common_methods.include?(method_name.to_sym)
+    end
+
+    def sections
+      [
+        Header.new,
+        RubyErrorComplete.new,
+        RelativePath.new,
+        LinesOfCode.new,
+        ErrorClassSpecificHelp.new(error_class_specific_help),
+        Examples.new(examples),
+        Docs.new(docs),
+        Suggestion.new("Try setting #{var_for_method} to #{a_an(common_classes_for_error.keys.first)} #{or_sentence(common_classes_for_error.keys)} first and then call the method #{method_name} on it.")
+      ]
+    end
+
+    private
+
+    def error_class_specific_help
+      [
+        "It looks like you're trying to call the method #{method_name} on #{var_for_method}.",
+        '',
+        "It's likely #{var_for_method} was never defined, and therefore is equal to nil, which is how Ruby represents 'nothing'.",
+        '',
+        "#{and_sentence(pluralize_words(common_classes_for_error.keys))} have the #{method_name} method, so you may want to set #{var_for_method} to #{a_an(common_classes_for_error.keys.first)} #{or_sentence(common_classes_for_error.keys)} first.",
+      ]
+    end
+
+    def examples
+      common_classes_for_error.map { |klass, attrs| "  #{var_for_method} = #{attrs[:example]}  # sets it to #{a_an(klass)} #{klass}" }
+    end
+
+    def docs
+      common_classes_for_error.keys.map { |klass| "ri #{klass}##{method_name}" }
+    end
+
+    def common_classes_for_error
+      self.class.common_classes.select do |klass, attrs|
+        attrs[:methods].include? raised_exception.spell_checker.method_name
+      end
+    end
+
+    private_class_method
+
+    @@string_methods  = String.new.methods
+    @@integer_methods = 1.methods
+    @@float_methods   = 1.0.methods
+    @@array_methods   = Array.new.methods
+    @@hash_methods    = Hash.new.methods
+
+    def self.common_classes
+      {
+        String:  { methods: @@string_methods,  example: "Hello, world!" },
+        Integer: { methods: @@integer_methods, example: 42 },
+        Float:   { methods: @@float_methods,   example: 3.14 },
+        Array:   { methods: @@array_methods,   example: '[]' },
+        Hash:    { methods: @@hash_methods,    example: '{}' }
+      }
+    end
+
+    def self.common_methods
+      common_classes.map { |klass, attrs| attrs[:methods] }.flatten - Object.methods
+    end
+
+  end
+
+end

data/lib/helpers/colorize.rb

@@ -2,10 +2,12 @@ module Mentor
 
   module Colorize
 
+    include OutputHelper
+
     def colorize(line)
 
-      if line_of_code?(line)
-        line = syntax_highlight(line)
+      if line_of_padded_code?(line)
+        line = color_padded_code(line)
       elsif backtrace?(line)
         return rainbow(line, :backtrace_line)
       end
@@ -14,10 +16,7 @@ module Mentor
     end
 
     def colorize_section
-      @lines.map! do |line|
-        line = rainbow(line, :suggestion)
-        colorize(line)
-      end
+      @lines.map! { |line| colorize(line) }
     end
 
     private
@@ -26,84 +25,66 @@ module Mentor
       line['from ']
     end
 
-    def line_of_code?(line)
+    def line_of_padded_code?(line)
       line[padded_lineno_with_colon_regex]
     end
 
-    def apply_colors(line)
-
-      line = color_pattern(line, error_lineno,     "#{file_name}:#{error_lineno}", :error_lineno)
-      line = color_pattern(line, ruby_error_class, "(#{ruby_error_class})", :ruby_error_class)
-      line = color_pattern(line, ruby_error_class, " #{ruby_error_class} ", :ruby_error_class)
-
-      output_types = %i(
+    def output_types
+      %i[
         did_you_mean_text  did_you_mean_word
         message            horizontal_line    ruby_error_text
-        calling_method     method_class       method_name        var_for_method
-        absolute_base_dir  relative_base_dir  app_dir  file_name
-      )
-
-      output_types.reduce(line) do |line, output_type|
-        color_text(line, send(output_type), output_type)
-      end
-
+        calling_method     method_name        var_for_method
+        absolute_base_dir  relative_base_dir  app_dir        file_name
+      ]
     end
 
-    def syntax_highlight(line)
-      padded_lineno = line[padded_lineno_with_colon_regex]
-      code          = line[padded_lineno.size..-1]
-      color_padded_lineno(padded_lineno) + color_code(code)
+    def apply_colors(line)
+      line = apply_pattern_colors(line)
+      line = apply_output_colors(line)
+      line = apply_nil_colors(line)
+      line = apply_common_classes_colors(line)
     end
 
-    def color_code(code)
-      formatter = Rouge::Formatters::Terminal256.new
-      lexer     = Rouge::Lexers::Ruby.new
-      formatter.format(lexer.lex(code))
+    def apply_pattern_colors(line)
+      line = TextToColor.new(line, error_lineno,     :error_lineno,     "#{file_name}:#{error_lineno}").color_pattern
+      line = TextToColor.new(line, ruby_error_class, :ruby_error_class, "(#{ruby_error_class})").color_pattern
+      line = TextToColor.new(line, ruby_error_class, :ruby_error_class, " #{ruby_error_class} ").color_pattern
     end
 
-    def color_padded_lineno(padded_lineno)
-      text_to_color = padded_lineno[padded_lineno_regex]
-      color         = text_to_color['=>'] ? :error_lineno : :subtle
-      color_text(padded_lineno, text_to_color, color)
+    def apply_output_colors(line)
+      output_types.reduce(line) do |new_line, output_type|
+        TextToColor.new(new_line, send(output_type), output_type).colored
+      end
     end
 
-    def color_pattern(full_text, text_to_color, match_pattern, output_type)
-      full_text.gsub(match_pattern, color_text(match_pattern, text_to_color, output_type))
+    def apply_nil_colors(line)
+      line = TextToColor.new(line, ' nil', :nil_text).colored
+      line = TextToColor.new(line, 'NilClass', :nil_text).colored
     end
 
-    def color_text(full_text, text_to_color, output_type)
-
-      return full_text unless text_to_color && text_to_color.size > 0
-
-      colorized_text  = rainbow(text_to_color, output_type)
-      remaining_text = full_text
-      text_processed = ''
-
-      while remaining_text[text_to_color]
-        before_text                 = before_text_to_color(remaining_text, text_to_color)
-        reset_color                 = get_reset_color(before_text).to_s
-        text_processed             += before_text + colorized_text
-        index_after_text_processed  = (before_text + text_to_color).size
-        remaining_text              = reset_color + remaining_text[index_after_text_processed..-1].to_s
+    def apply_common_classes_colors(line)
+      %w(String Integer Float Array Hash [] {}).reduce(line) do |new_line, klass|
+        new_line = TextToColor.new(new_line, pluralize(klass), :common_class).colored
+        new_line = TextToColor.new(new_line, klass.to_s, :common_class).colored
       end
-
-      text_processed + remaining_text
-
     end
 
-    def before_text_to_color(remaining_text, text_to_color)
-      index = remaining_text.index(text_to_color)
-      return '' if !index || index == 0
-      remaining_text[0..index - 1]
+    def color_padded_code(line)
+      padded_lineno = line[padded_lineno_with_colon_regex]
+      code          = line[padded_lineno.size..-1]
+      color_padded_lineno(padded_lineno) + color_code(code)
     end
 
-    def get_reset_color(before_text)
-      index_of_color_to_reset_to = before_text.rindex(color_regex)
-      before_text[index_of_color_to_reset_to..-1][color_regex] if index_of_color_to_reset_to
+    def color_code(code)
+      formatter = Rouge::Formatters::Terminal256.new
+      lexer     = Rouge::Lexers::Ruby.new
+      formatter.format(lexer.lex(code))
     end
 
-    def color_regex
-      /\e\[(\d|;)+m/
+    def color_padded_lineno(padded_lineno)
+      text_to_color_text = padded_lineno[padded_lineno_regex]
+      color              = text_to_color_text['=>'] ? :error_lineno : :subtle
+      TextToColor.new(padded_lineno, text_to_color_text, color).colored
     end
 
     def padded_lineno_regex
@@ -124,6 +105,7 @@ module Mentor
       method_name  = :deepskyblue
       subtle       = :olive
       very_subtle  = :dimgray
+      nil_text     = :orange
 
       {
         app_dir:               subtle,
@@ -131,6 +113,7 @@ module Mentor
         absolute_base_dir:     subtle,
         relative_base_dir:     subtle,
         calling_method:        :green,
+        common_class:          :wheat,
         did_you_mean_text:     :royalblue,
         did_you_mean_word:     method_name,
         error_lineno:          error_lineno,
@@ -141,6 +124,7 @@ module Mentor
         message:               error,
         method_class:          :mediumpurple,
         method_name:           method_name,
+        nil_text:              nil_text,
         other_class_or_module: :darksalmon,
         prominent:             :ivory,
         ruby_error_class:      :orangered,

data/lib/helpers/globals.rb

@@ -2,13 +2,7 @@ module Mentor
 
   module Globals
 
-    DIR = __dir__ + '/../..'
-
-    attr_accessor :cancel_exit_after_error, :tp
-
-    def cancel_exit_after_error?
-      @cancel_exit_after_error
-    end
+    attr_accessor :tp
 
     def enable
       @enabled = true

data/lib/helpers/output_helper.rb

@@ -2,6 +2,26 @@ module Mentor
 
   module OutputHelper
 
+    def a_an(word)
+      %w(A E I O U).include?(word[0]) ? 'an' : 'a'
+    end
+
+    def pluralize(word)
+      word.to_s + (word.to_s == 'Hash' ? 'es' :'s')
+    end
+
+    def pluralize_words(words)
+      words.map { |word| pluralize word }
+    end
+
+    def or_sentence(words)
+      words.join(' or ')
+    end
+
+    def and_sentence(words)
+      words.join(' and ')
+    end
+
     def home_to_tilde(path)
       path.sub(ENV['HOME'], '~')
     end

data/lib/helpers/outputs.rb

@@ -4,31 +4,41 @@ module Mentor
 
     include OutputHelper
 
+    private
+
+    def tp
+      Mentor.tp
+    end
+
+    def raised_exception
+      tp.raised_exception
+    end
+
     def absolute_base_dir
       Dir.pwd + '/'
     end
 
     def app_dir
-      Mentor.tp.path.               # Full path
-        sub(absolute_base_dir, ''). # Remove path up until this file
-        split('/')[0..-2].          # Split path into parts and exclude filename
-        join('/') +                 # Join with '/'
+      tp.path                       # Full path
+        .sub(absolute_base_dir, '') # Remove path up until this file
+        .split('/')[0..-2]          # Split path into parts and exclude filename
+        .join('/') +                # Join with '/'
         '/'                         # Add one more '/' to end of it
     end
 
     def backtrace_lines
 
-      # Mentor.tp.raised_exception.backtrace disappears after accessing it once
+      # raised_exception.backtrace disappears after accessing it once
       # So we just do it once and then save the result to @@backtrace_lines
       if self.class.class_variable_defined?(:@@backtrace_lines)
         return @@backtrace_lines
       end
 
-      bt_lines = Mentor.tp.raised_exception.
-                 backtrace[1..-1].                            # Remove first
-                 grep_v(/bin\/mentor/).                       # Remove mentor involvement
-                 map { |line| line.sub(Dir.pwd + '/', '') }. # Remove path before current dir
-                 map { |line| 'from ' + line }                # Add 'from'
+      bt_lines = raised_exception
+                 .backtrace[1..-1]                            # Remove first
+                 .grep_v(%r{bin/mentor})                       # Remove mentor involvement
+                 .map { |line| line.sub(Dir.pwd + '/', '') }  # Remove path before current dir
+                 .map { |line| 'from ' + line }               # Add 'from'
 
       return bt_lines if bt_lines.empty?
 
@@ -40,13 +50,13 @@ module Mentor
     end
 
     def calling_method
-      (Mentor.tp.method_id || '<main>').to_s
+      (tp.method_id || '<main>').to_s
     end
 
     def did_you_mean_word
-      if Mentor.tp.raised_exception.respond_to? :corrections
-        Mentor.tp.raised_exception.corrections.first&.to_s
-      end
+      return unless raised_exception.respond_to? :corrections
+
+      raised_exception.corrections.first.to_s
     end
 
     def did_you_mean_text
@@ -58,60 +68,35 @@ module Mentor
     end
 
     def error_lineno
-      Mentor.tp.lineno.to_s
+      tp.lineno.to_s
     end
 
     def file_name
-      Mentor.tp.path.split('/').last
+      tp.path.split('/').last
     end
 
     def horizontal_line
       '─' * terminal_width
     end
 
-    def instance_methods
-      Mentor.tp.raised_exception.receiver.class.instance_methods -
-      Mentor.tp.raised_exception.receiver.class.superclass.instance_methods
-    end
-
     def lineno_subtle_padded(width)
       @lineno.to_s.rjust(width)
     end
 
     def message
-      if Mentor.tp.raised_exception.respond_to? :original_message
-        Mentor.tp.raised_exception.original_message
+      if raised_exception.respond_to? :original_message
+        raised_exception.original_message
       else
-        Mentor.tp.raised_exception.message
+        raised_exception.message
       end
     end
 
     def method_name
-      if Mentor.tp.raised_exception.respond_to? :spell_checker
-        if Mentor.tp.raised_exception.spell_checker.respond_to?(:method_name)
-          Mentor.tp.raised_exception.spell_checker.method_name.to_s
-        else
-          raise "cannot determine method name"
-        end
-      end
-    end
-
-    def method_class
-      if Mentor.tp.raised_exception.respond_to? :receiver
-        Mentor.tp.raised_exception.receiver.class.to_s
-      end
-    end
-
-    def method_class_plural
-      pluralize method_class
-    end
-
-    def method_class_superclass
-      Mentor.tp.raised_exception.receiver.class.superclass
-    end
+      return unless
+        raised_exception.respond_to?(:spell_checker) &&
+        raised_exception.spell_checker.respond_to?(:method_name)
 
-    def method_class_superclasses
-      to_sentence(Mentor.tp.raised_exception.receiver.class.ancestors - typical_classes)
+      raised_exception.spell_checker.method_name.to_s
     end
 
     def relative_base_dir
@@ -119,7 +104,7 @@ module Mentor
     end
 
     def ruby_error_class
-      Mentor.tp.raised_exception.class.to_s
+      raised_exception.class.to_s
     end
 
     def ruby_error_text
@@ -127,7 +112,7 @@ module Mentor
     end
 
     def var_for_method
-      culprit_line = lines_from_file[Mentor.tp.lineno]
+      culprit_line = lines_from_file[tp.lineno]
       var = culprit_line[/#{valid_var_name}\.#{method_name}/].to_s.chomp(".#{method_name}")
       if var.empty?
         culprit_line[/#{valid_var_name} method_name/].to_s.chomp(".#{method_name}")

data/lib/helpers/requires.rb

@@ -1,18 +1,18 @@
 module Mentor
-  gems = [
-    'pry',
-    'rainbow',
-    'rouge',
-  ]
 
-  helpers = [
-    'output_helper',
-    'outputs',
-    'colorize'
-  ].map { |file| Globals::DIR + '/lib/helpers/' + file }
+  dir = __dir__ + '/../..'
 
-  sections = Dir.glob(Globals::DIR + '/lib/sections/*.rb')
-  errors   = Dir.glob(Globals::DIR + '/lib/errors/*.rb')
+  gems = %w[pry rainbow rouge]
+
+  helpers = %w[
+    output_helper
+    outputs
+    colorize
+    text_to_color
+  ].map { |file| dir + '/lib/helpers/' + file }
+
+  sections = Dir.glob(dir + '/lib/sections/*.rb')
+  errors   = Dir.glob(dir + '/lib/errors/*.rb')
 
   (gems + helpers + sections + errors).each { |file| require file }
 

data/lib/helpers/text_to_color.rb

@@ -0,0 +1,65 @@
+module Mentor
+
+  class TextToColor
+
+    include Colorize
+
+    def initialize(full_text, text_to_color, output_type, match_pattern = nil)
+      @full_text     = full_text
+      @text_to_color = text_to_color
+      @output_type   = output_type
+      @match_pattern = match_pattern
+    end
+
+    def color_pattern
+      @full_text.gsub(@match_pattern, colored)
+    end
+
+    def colored
+      return @full_text unless text_to_color_exists?
+      color_text
+    end
+
+    private
+
+    def colorized_text
+      rainbow(@text_to_color, @output_type)
+    end
+
+    def text_to_color_exists?
+      @text_to_color && !@text_to_color.empty?
+    end
+
+    def color_text
+      remaining_text = @match_pattern || @full_text
+      text_processed = ''
+
+      while remaining_text[@text_to_color]
+        before_text                  = before_text_to_color(remaining_text)
+        reset_color                  = get_reset_color(before_text).to_s
+        text_processed              += before_text + colorized_text
+        index_after_text_processed   = (before_text + @text_to_color).size
+        remaining_text               = reset_color + remaining_text[index_after_text_processed..-1]
+      end
+
+      text_processed + remaining_text
+    end
+
+    def before_text_to_color(remaining_text)
+      index = remaining_text.index(@text_to_color)
+      return '' if !index || index.zero?
+      remaining_text[0..index - 1]
+    end
+
+    def get_reset_color(before_text)
+      index_of_color_to_reset_to = before_text.rindex(color_regex)
+      before_text[index_of_color_to_reset_to..-1][color_regex] if index_of_color_to_reset_to
+    end
+
+    def color_regex
+      /\e\[(\d|;)+m/
+    end
+
+  end
+
+end

data/lib/main.rb

@@ -13,7 +13,7 @@ module Mentor
         raise error
       rescue *MentorError.error_classes
         error.output
-        exit 1 unless Mentor.cancel_exit_after_error?
+        exit 1
       end
 
     end

data/lib/mentor/version.rb

@@ -1,3 +1,3 @@
 module Mentor
-  VERSION = '0.1.1'
+  VERSION = '0.2.0'.freeze
 end

data/lib/mentor.rb

@@ -1,5 +1,3 @@
-require 'mentor/version'
-
 module Mentor
 
   class Init
@@ -11,19 +9,19 @@ module Mentor
       Mentor.enable
     end
 
-    private
+    private_class_method
 
     def self.setup_trace_point
 
       TracePoint.trace(:raise) do |tp|
         tp.disable
 
-          if Mentor.enabled?
-            Mentor.disable
-            require_relative 'helpers/requires'
-            Mentor.tp = tp
-            Main.new
-          end
+        if Mentor.enabled?
+          Mentor.disable
+          require_relative 'helpers/requires'
+          Mentor.tp = tp
+          Main.new
+        end
 
         tp.enable
       end
@@ -33,5 +31,3 @@ module Mentor
   end
 
 end
-
-Mentor::Init.setup

data/lib/sections/backtrace.rb

@@ -4,10 +4,8 @@ module Mentor
 
     include Outputs
 
-    attr_reader :lines
-
-    def initialize
-      @lines = indent_lines(backtrace_lines, indent: 10)
+    def lines
+      indent_lines(backtrace_lines, indent: 10)
     end
 
   end

data/lib/sections/did_you_mean_correction.rb

@@ -12,7 +12,7 @@ module Mentor
 
     def self.can_handle?
       Mentor.tp.raised_exception.respond_to?(:corrections) &&
-      Mentor.tp.raised_exception.corrections.any?
+        Mentor.tp.raised_exception.corrections.any?
     end
 
   end

data/lib/sections/docs.rb

@@ -0,0 +1,18 @@
+module Mentor
+
+  class Docs
+
+    include Outputs, Colorize
+
+    attr_reader :lines
+
+    def initialize(lines)
+      @lines = ['For docs, type:', '']
+      @lines << indent_lines(lines)
+      @lines = indent_lines(@lines)
+      colorize_section
+    end
+
+  end
+
+end

data/lib/sections/examples.rb

@@ -0,0 +1,28 @@
+module Mentor
+
+  class Examples
+
+    include Outputs, Colorize
+
+    attr_reader :lines
+
+    def initialize(examples)
+      @lines =
+        indent_lines([
+          'For example:',
+          '',
+          examples
+        ]).flatten
+
+      examples.each do |example|
+        @lines.map! do |line|
+          line.sub(example, color_code(example))
+        end
+      end
+
+      colorize_section
+    end
+
+  end
+
+end

data/lib/sections/header.rb

@@ -4,13 +4,11 @@ module Mentor
 
     include Outputs, Colorize
 
-    attr_reader :lines
-
-    def initialize
+    def lines
       @lines = [
         horizontal_line,
         title_indented,
-        horizontal_line,
+        horizontal_line
       ]
 
       colorize_section

data/lib/sections/lines_of_codes.rb

@@ -4,10 +4,8 @@ module Mentor
 
     include Outputs
 
-    attr_accessor :lines
-
-    def initialize
-      @lines = indent_lines(lines_of_code_from_method_or_file, indent: 4)
+    def lines
+      indent_lines(lines_of_code_from_method_or_file, indent: 4)
     end
 
     private
@@ -32,16 +30,19 @@ module Mentor
 
     def set_first_and_last_from_method
       method_text = Pry::Method.from_str("#{Mentor.tp.defined_class}##{calling_method}").source.split("\n")
-      @first_lineno = lines_from_file.select { |lineno, line| line["def #{calling_method}"] }.sort.last.first
+      @first_lineno = lines_from_file.select { |_lineno, line| line["def #{calling_method}"] }.sort.last.first
       @last_lineno  = @first_lineno + method_text.size - 1
     end
 
     def set_first_and_last_from_file
       @first_lineno = Mentor.tp.lineno
-      @first_lineno -= 1 until first_line_at_start_of_file? || first_line_empty? ||
+      @first_lineno -= 1 until first_line_at_start_of_file? ||
+                               first_line_empty? ||
                                line_prior_to_first_line_ends?
+
       @last_lineno = Mentor.tp.lineno
-      @last_lineno += 1 until last_line_at_end_of_file? || last_line_empty?
+      @last_lineno += 1 until last_line_at_end_of_file? ||
+                              last_line_empty?
     end
 
     def first_line_at_start_of_file?
@@ -67,4 +68,3 @@ module Mentor
   end
 
 end
-

data/lib/sections/relative_path.rb

@@ -4,10 +4,8 @@ module Mentor
 
     include Outputs
 
-    attr_accessor :lines
-
-    def initialize
-      @lines = indent_lines(relative_base_dir + app_dir + file_name + ':' + error_lineno)
+    def lines
+      indent_lines(relative_base_dir + app_dir + file_name + ':' + error_lineno)
     end
 
   end

data/lib/sections/ruby_error_complete.rb

@@ -2,18 +2,16 @@ module Mentor
 
   class RubyErrorComplete
 
-    include Outputs, Colorize
+    include Outputs
 
-    attr_reader :lines
-
-    def initialize
-      @lines = []
-      @lines << RubyErrorMain.new.lines
-      @lines << DidYouMeanCorrection.new.line if DidYouMeanCorrection.can_handle?
-      @lines << Backtrace.new.lines
-      @lines = indent_lines(@lines)
-      @lines << ''
-      @lines << horizontal_line
+    def lines
+      lines_array = []
+      lines_array << RubyErrorMain.new.lines
+      lines_array << DidYouMeanCorrection.new.line if DidYouMeanCorrection.can_handle?
+      lines_array << Backtrace.new.lines
+      lines_array = indent_lines(lines_array)
+      lines_array << ''
+      lines_array << horizontal_line
     end
 
   end

data/lib/sections/ruby_error_main.rb

@@ -4,20 +4,14 @@ module Mentor
 
     include Outputs
 
-    attr_reader :lines
-
-    def initialize
-      @lines = ruby_error_main
+    def lines
+      ruby_error_main
     end
 
     private
 
     def single_line
-      parts = []
-      parts << path_and_lineno
-      parts << message
-      parts << bracketed_class
-      parts.flatten.join(' ')
+      [path_and_lineno, message, bracketed_class].flatten.join(' ')
     end
 
     def path_and_lineno
@@ -57,7 +51,7 @@ module Mentor
 
     def fit_on_one_or_two_lines?(part_a, part_b)
       fit_on_one_line?(part_a, part_b) ||
-      does_not_fit_on_one_line?(part_a) && fit_on_two_lines?(part_a, part_b)
+        does_not_fit_on_one_line?(part_a) && fit_on_two_lines?(part_a, part_b)
     end
 
     def fit_on_one_line?(part_a, part_b)

data/lib/sections/suggestion.rb

@@ -15,7 +15,7 @@ module Mentor
 
     def format_lines
       @lines = indent_lines(@lines)
-      colorize_section
+      @lines.map! { |line| rainbow(line, :suggestion) }
     end
 
   end

data/mentor.gemspec

@@ -1,4 +1,3 @@
-# coding: utf-8
 lib = File.expand_path('../lib', __FILE__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'mentor/version'
@@ -12,11 +11,11 @@ Gem::Specification.new do |spec|
   spec.homepage = 'https://gitlab.com/seanlerner/mentor'
   spec.license  = 'MIT'
 
-  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
     f.match(%r{^(test|spec|features)/})
   end
 
-  spec.executables   << 'mentor'
+  spec.executables << 'mentor'
   spec.require_paths = ['lib']
 
   spec.add_dependency 'pry',           '~> 0.10'
@@ -24,6 +23,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'rouge',         '~> 2.0'
 
   spec.add_development_dependency 'bundler',        '~> 1.13'
+  spec.add_development_dependency 'did_you_mean',   '~> 1.1'
   spec.add_development_dependency 'guard',          '~> 2.14'
   spec.add_development_dependency 'guard-minitest', '~> 2.4'
   spec.add_development_dependency 'minitest',       '~> 5.0'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mentor
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Sean Lerner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-06-24 00:00:00.000000000 Z
+date: 2017-06-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry
@@ -66,6 +66,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: did_you_mean
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
 - !ruby/object:Gem::Dependency
   name: guard
   requirement: !ruby/object:Gem::Requirement
@@ -140,17 +154,22 @@ files:
 - lib/errors/mentor_error.rb
 - lib/errors/mentor_no_method_error.rb
 - lib/errors/no_method_did_you_mean_suggestion_error.rb
+- lib/errors/no_method_for_nil_class_error.rb
+- lib/errors/no_method_for_nil_class_for_common_class_error.rb
 - lib/helpers/colorize.rb
 - lib/helpers/globals.rb
 - lib/helpers/output_helper.rb
 - lib/helpers/outputs.rb
 - lib/helpers/requires.rb
+- lib/helpers/text_to_color.rb
 - lib/main.rb
 - lib/mentor.rb
 - lib/mentor/version.rb
 - lib/sections/backtrace.rb
 - lib/sections/did_you_mean_correction.rb
+- lib/sections/docs.rb
 - lib/sections/error_class_specific_help.rb
+- lib/sections/examples.rb
 - lib/sections/header.rb
 - lib/sections/line_of_code.rb
 - lib/sections/lines_of_codes.rb