support did_you_mean >= v1.2.0 which has a breaking change on formatters

ruby · Mar 20, 2018 · 7c01069 · 7c01069
1 parent a8ebea2
commit 7c01069
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lib/rake/task_manager.rb b/lib/rake/task_manager.rb
@@ -63,9 +63,16 @@ def generate_message_for_undefined_task(task_name)
       message = "Don't know how to build task '#{task_name}' (see --tasks)"
 
       suggestion_message = \
-        if defined?(::DidYouMean::SpellChecker) && defined?(::DidYouMean::Formatter)
+        if defined?(::DidYouMean::SpellChecker)
           suggestions = ::DidYouMean::SpellChecker.new(dictionary: @tasks.keys).correct(task_name.to_s)
-          ::DidYouMean::Formatter.new(suggestions).to_s
+          puts "DidYouMean v#{DidYouMean::VERSION}"
+          if ::DidYouMean.respond_to?(:formatter)# did_you_mean v1.2.0 or later
+            ::DidYouMean.formatter.message_for(suggestions)
+          elsif defined?(::DidYouMean::Formatter) # before did_you_mean v1.2.0
+            ::DidYouMean::Formatter.new(suggestions).to_s
+          else
+            ""
+          end
         else
           ""
         end