scala.meta.contrib.implicits.TreeExtensions.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalameta_2.12 Show documentation

Scalameta umbrella module that includes all public APIs

There is a newer version: 4.11.2

package scala.meta.contrib.implicits

import scala.meta.Tree
import scala.meta.contrib.TreeOps
import scala.meta.contrib.equality.Equal
import scala.meta.contrib.equality.TreeEquality

import scala.language.higherKinds

trait TreeExtensions {
  implicit class XtensionTreeOps[A <: Tree](a: A) {
    @inline
    def ancestors: List[Tree] = TreeOps.ancestors(a)

    @inline
    def descendants: List[Tree] = TreeOps.descendants(a)

    @inline
    def foreach(f: Tree => Unit): Unit = TreeOps.foreach(a)(f)

    @inline
    def collectFirst[B](pf: PartialFunction[Tree, B]): Option[B] = TreeOps.collectFirst(a)(pf)

    @inline
    def find(f: Tree => Boolean): Option[Tree] = TreeOps.find(a)(f)

    @inline
    def forall(f: Tree => Boolean): Boolean = TreeOps.forall(a)(f)

    @inline
    def exists(f: Tree => Boolean): Boolean = TreeOps.exists(a)(f)

    @inline
    def contains[F[x <: Tree] <: TreeEquality[x]](
        toFind: Tree
    )(implicit conv: Tree => F[Tree], eqEv: Equal[F[Tree]]): Boolean = TreeOps.contains(a)(toFind)
  }
}

object TreeExtensions extends TreeExtensions