fix.UnnecessaryMatch.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalafix-rules_2.12 Show documentation

scalafix rules

The newest version!

package fix

import scala.meta.Term
import scala.meta.Token
import scala.meta.XtensionClassifiable
import scala.meta.XtensionCollectionLikeUI
import scalafix.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionSeqPatch

class UnnecessaryMatch extends SyntacticRule("UnnecessaryMatch") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case Term.Apply.Initial(
            _,
            List(
              f @ Term.Function.Initial(
                Term.Param(_, Term.Name(p1), _, _) :: Nil,
                Term.Match.After_4_4_5(Term.Name(p2), cases, _)
              )
            )
          ) if p1 == p2 && cases.forall(_.collect { case Term.Name(n) if n == p2 => () }.isEmpty) =>
        Seq(
          Patch.removeTokens(f.tokens.takeWhile(!_.is[Token.KwMatch])),
          Patch.removeTokens(f.tokens.find(_.is[Token.KwMatch]))
        ).asPatch
      case Term.Apply.Initial(
            _,
            Term.Block(
              List(
                f @ Term.Function.Initial(
                  Term.Param(_, Term.Name(p1), _, _) :: Nil,
                  Term.Match.After_4_4_5(Term.Name(p2), cases, _)
                )
              )
            ) :: Nil
          ) if p1 == p2 && cases.forall(_.collect { case Term.Name(n) if n == p2 => () }.isEmpty) =>
        Seq(
          Patch.removeTokens(f.tokens.takeWhile(!_.is[Token.KwMatch])),
          Patch.removeTokens(f.tokens.find(_.is[Token.KwMatch]))
        ).asPatch
    }
  }.asPatch
}