• Home
• com.github.xuwei-k
• scalafix-rules_2.12
• 0.6.4
• source code
• CollectHeadOption.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

fix.CollectHeadOption.scala Maven / Gradle / Ivy

package fix

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

class CollectHeadOption extends SyntacticRule("CollectHeadOption") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case t @ Term.Select(
            Term.Apply.After_4_6_0(
              Term.Select(obj, Term.Name("collect")),
              Term.ArgClause(func :: Nil, _)
            ),
            Term.Name("headOption")
          ) =>
        func match {
          case _: Term.PartialFunction =>
            Patch.replaceTree(t, s"${obj}.collectFirst${func}")
          case _ =>
            Patch.replaceTree(t, s"${obj}.collectFirst(${func})")
        }
    }
  }.asPatch
}