fix.RemoveParameterUntuplingCase.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

There is a newer version: 0.6.7

package fix

import scala.meta.Case
import scala.meta.Pat
import scala.meta.Term
import scala.meta.XtensionClassifiable
import scala.meta.XtensionCollectionLikeUI
import scala.meta.tokens.Token
import scalafix.v1.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionOptionPatch
import scalafix.v1.XtensionSeqPatch

/**
 * [[https://docs.scala-lang.org/scala3/reference/other-new-features/parameter-untupling.html]]
 * [[https://docs.scala-lang.org/scala3/reference/other-new-features/parameter-untupling-spec.html]]
 */
class RemoveParameterUntuplingCase extends SyntacticRule("RemoveParameterUntuplingCase") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case Term.PartialFunction(
            List(
              c @ Case(
                Pat.Tuple(
                  values
                ),
                None,
                _
              )
            )
          ) if values.forall {
            case Pat.Var(_: Term.Name) | Pat.Wildcard() => true
            case _ => false
          } =>
        Seq(
          PartialFunction
            .condOpt(
              (
                c.tokens.filter(x => x.is[Token.LeftParen] && (x.pos.start < values.head.pos.start)),
                c.tokens.filter(x =>
                  x.is[Token.RightParen] && (values.last.pos.start < x.pos.start) && (x.pos.end < c.body.pos.start)
                )
              )
            ) { case (l +: _ +: _, r +: _ +: _) =>
              Patch.removeTokens(Seq(l, r))
            }
            .asPatch,
          c.tokens
            .find(_.is[Token.KwCase])
            .map(
              Patch.removeToken
            )
            .asPatch
        ).asPatch
    }.asPatch
  }
}