fix.NoElse.scala Maven / Gradle / Ivy

Go to download
package fix

import scala.meta.Lit
import scala.meta.Term
import scala.meta.XtensionCollectionLikeUI
import scala.meta.inputs.Position
import scalafix.Diagnostic
import scalafix.Patch
import scalafix.lint.LintSeverity
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionSeqPatch

class NoElse extends SyntacticRule("NoElse") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect { case t @ Term.If.After_4_4_0(_, _, Lit.Unit(), _) =>
      Patch.lint(
        NoElseWarn(t.pos)
      )
    }.asPatch
  }
}

case class NoElseWarn(override val position: Position) extends Diagnostic {
  override def message = "add `else`"
  override def severity: LintSeverity = LintSeverity.Warning
}