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

package fix

import scala.meta.Term
import scala.meta.Term.Block
import scala.meta.Term.If
import scala.meta.XtensionCollectionLikeUI
import scalafix.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scalafix.v1.XtensionSeqPatch

object OptionWhenUnless {
  private object SomeValue {
    def unapply(x: Term): Option[Term] = PartialFunction.condOpt(x) {
      case Block(Term.Apply.Initial(Term.Name("Some"), value :: Nil) :: Nil) =>
        value
      case Term.Apply.Initial(Term.Name("Some"), value :: Nil) =>
        value
    }
  }

  private object NoneValue {
    def unapply(x: Term): Boolean = x match {
      case Block(Term.Name("None") :: Nil) =>
        true
      case Term.Name("None") =>
        true
      case _ =>
        false
    }
  }
}

class OptionWhenUnless extends SyntacticRule("OptionWhenUnless") {
  import OptionWhenUnless._

  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case t @ If.After_4_4_0(condition, SomeValue(value), NoneValue(), _) =>
        Patch.replaceTree(t, s"Option.when(${condition})($value)")
      case t @ If.After_4_4_0(condition, NoneValue(), SomeValue(value), _) =>
        Patch.replaceTree(t, s"Option.unless(${condition})($value)")
    }.asPatch
  }
}