fix.OptionMatchToRight.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.Case
import scala.meta.Pat
import scala.meta.Term
import scala.meta.XtensionCollectionLikeUI
import scalafix.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionSeqPatch

class OptionMatchToRight extends SyntacticRule("OptionMatchToRight") {
  private object SomeToRight {
    def unapply(c: Case): Boolean = PartialFunction.cond(c) {
      case Case(
            Pat.Extract.Initial(Term.Name("Some"), Pat.Var(Term.Name(a1)) :: Nil),
            None,
            Term.Apply.Initial(Term.Name("Right"), Term.Name(a2) :: Nil)
          ) if a1 == a2 =>
        true

    }
  }

  private object NoneToLeft {
    def unapply(c: Case): Option[Term] = PartialFunction.condOpt(c) {
      case Case(
            Term.Name("None") | Pat.Wildcard(),
            None,
            Term.Apply.Initial(Term.Name("Left"), leftArg :: Nil)
          ) =>
        leftArg
    }
  }

  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case t @ Term.Match.After_4_4_5(expr, SomeToRight() :: NoneToLeft(leftArg) :: Nil, _) =>
        Patch.replaceTree(t, s"${expr}.toRight(${leftArg})")
      case t @ Term.Match.After_4_4_5(expr, NoneToLeft(leftArg) :: SomeToRight() :: Nil, _) =>
        Patch.replaceTree(t, s"${expr}.toRight(${leftArg})")
    }
  }.asPatch
}