• Home
• com.github.xuwei-k
• scalafix-rules_2.12
• 0.5.1
• source code
• MapFlattenFlatMap.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.MapFlattenFlatMap.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 MapFlattenFlatMap extends SyntacticRule("MapFlattenFlatMap") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case t1 @ Term.Select(
            t2 @ Term.Apply.Initial(
              Term.Select(_, map @ Term.Name("map")),
              _ :: Nil
            ),
            flatten @ Term.Name("flatten")
          ) =>
        Seq(
          Patch.replaceTree(map, "flatMap"),
          t1.tokens.reverseIterator
            .filter(_.is[Token.Dot])
            .find(x => t2.pos.end <= x.pos.start)
            .map(Patch.removeToken)
            .asPatch,
          Patch.removeTokens(flatten.tokens)
        ).asPatch
    }.asPatch
  }
}