• Home
• com.github.xuwei-k
• scalafix-rules_2.12
• 0.5.1
• source code
• RepeatedRewrite.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.RepeatedRewrite.scala Maven / Gradle / Ivy

package fix

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

class RepeatedRewrite extends SyntacticRule("RepeatedRewrite") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect { case t @ Term.Repeated(x) =>
      val a = t.tokens.dropWhile(!_.is[Token.Colon])
      if (a.nonEmpty) {
        a.dropWhile(!_.is[Token.Underscore]).drop(1).headOption match {
          case Some(Token.Ident("*")) =>
            Seq(
              if (x.is[Term.ApplyInfix]) {
                Patch.addAround(x, "(", ")")
              } else {
                Patch.empty
              },
              Patch.removeToken(a.head),
              a.find(_.is[Token.Underscore]).map(Patch.removeToken).asPatch,
            ).asPatch
          case _ =>
            Patch.empty
        }
      } else {
        Patch.empty
      }
    }.asPatch
  }
}