• Home
• com.github.xuwei-k
• scalafix-rules_2.12
• 0.5.1
• source code
• GroupMap.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.GroupMap.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 GroupMap extends SyntacticRule("GroupMap") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case t @ Term.Select(
            Term.Apply.Initial(
              Term.Select(
                Term.Select(
                  Term.Apply.Initial(
                    Term.Select(
                      obj,
                      Term.Name("groupBy")
                    ),
                    f1 :: Nil
                  ),
                  Term.Name("view")
                ),
                Term.Name("mapValues")
              ),
              List(
                Term.AnonymousFunction(
                  Term.Apply.Initial(
                    Term.Select(
                      Term.Placeholder(),
                      Term.Name("map")
                    ),
                    f2 :: Nil
                  )
                )
              )
            ),
            Term.Name("toMap")
          ) =>
        Patch.replaceTree(t, s"${obj}.groupMap($f1)($f2)")
    }.asPatch
  }
}