don't swallow stack traces on unchecked exceptions

scalacenter · Jun 24, 2022 · 82187f1 · 82187f1
1 parent dd84703
commit 82187f1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/scala/scalafix/internal/sbt/ScalafixInterface.scala b/src/main/scala/scalafix/internal/sbt/ScalafixInterface.scala
@@ -88,7 +88,9 @@ class ScalafixInterface private (
   def withArgs(args: Arg*): ScalafixInterface = {
     val newScalafixArguments = args.foldLeft(scalafixArguments) { (acc, arg) =>
       try arg(acc)
-      catch { case NonFatal(e) => throw new InvalidArgument(e.getMessage) }
+      catch {
+        case e: ScalafixException => throw new InvalidArgument(e.getMessage)
+      }
     }
     new ScalafixInterface(newScalafixArguments, this.args ++ args)
   }
@@ -101,7 +103,9 @@ class ScalafixInterface private (
 
   def rulesThatWillRun(): Seq[ScalafixRule] =
     try scalafixArguments.rulesThatWillRun().asScala
-    catch { case NonFatal(e) => throw new InvalidArgument(e.getMessage) }
+    catch {
+      case e: ScalafixException => throw new InvalidArgument(e.getMessage)
+    }
 
   def validate(): Option[ScalafixException] =
     Option(scalafixArguments.validate().orElse(null))