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

package fix

import scala.meta.Term
import scala.meta.XtensionCollectionLikeUI
import scalafix.Patch
import scalafix.v1.SemanticDocument
import scalafix.v1.SemanticRule
import scalafix.v1.TypeRef
import scalafix.v1.ValueSignature
import scalafix.v1.XtensionOptionPatch
import scalafix.v1.XtensionSeqPatch
import scalafix.v1.XtensionTreeScalafix

class SizeToLength extends SemanticRule("SizeToLength") {
  override def fix(implicit doc: SemanticDocument): Patch = {
    doc.tree.collect {
      case Term.Select(
            obj,
            size @ Term.Name("size")
          ) =>
        val p = Patch.replaceTree(size, "length")
        obj.symbol.info.flatMap { i =>
          PartialFunction.condOpt(i.signature) {
            case ValueSignature(TypeRef(_, symbol, Nil)) if symbol.normalized.value == "scala.Predef.String." =>
              p
            case ValueSignature(TypeRef(_, symbol, _ :: Nil)) if symbol.normalized.value == "scala.Array." =>
              p
          }
        }.asPatch
    }.asPatch
  }
}