Merge pull request #761 from deivid-rodriguez/support_latest_did_you_…

…mean Support latest did_you_mean
rails · Jan 4, 2022 · 05a79fc · 05a79fc
2 parents 5c666b4 + 98dbec7
commit 05a79fc
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/lib/thor/error.rb b/lib/thor/error.rb
@@ -102,9 +102,14 @@ class MalformattedArgumentError < InvocationError
   end
 
   if Correctable
-    DidYouMean::SPELL_CHECKERS.merge!(
-      'Thor::UndefinedCommandError' => UndefinedCommandError::SpellChecker,
-      'Thor::UnknownArgumentError' => UnknownArgumentError::SpellChecker
-    )
+    if DidYouMean.respond_to?(:correct_error)
+      DidYouMean.correct_error(Thor::UndefinedCommandError, UndefinedCommandError::SpellChecker)
+      DidYouMean.correct_error(Thor::UnknownArgumentError, UnknownArgumentError::SpellChecker)
+    else
+      DidYouMean::SPELL_CHECKERS.merge!(
+        'Thor::UndefinedCommandError' => UndefinedCommandError::SpellChecker,
+        'Thor::UnknownArgumentError' => UnknownArgumentError::SpellChecker
+      )
+    end
   end
 end