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

package fix

import scala.meta.Importee
import scala.meta.Pkg
import scala.meta.transversers._
import scalafix.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionOptionPatch
import scalafix.v1.XtensionSeqPatch

class RemoveSamePackageImport extends SyntacticRule("RemoveSamePackageImport") {

  override def fix(implicit doc: SyntacticDocument): Patch = {
    val pkgOpt = doc.tree.collect { case pkg: Pkg =>
      pkg
    }.headOption

    pkgOpt.map { pkg =>
      val ref = s"${pkg.ref.toString()}"
      doc.tree.collect {
        case _: Importee.Rename =>
          Patch.empty
        case t: Importee =>
          val p = t.parent.get
          val children = p.children
          if (ref == children.head.toString()) {
            val last = children.last.toString()
            if (last.matches("^[a-z]+$")) {
              // Keep sub package import
              Patch.empty
            } else {
              Patch.replaceTree(p.parent.get, "")
            }
          } else {
            Patch.empty
          }
      }.asPatch
    }.asPatch
  }
}