kyo.Trees.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kyo-direct_sjs1_3 Show documentation

kyo-direct

There is a newer version: 3.0.7

package kyo

import scala.quoted.*

object Trees:
    def traverse(using
        Quotes
    )(tree: quotes.reflect.Tree)(pf: PartialFunction[quotes.reflect.Tree, Unit]): Unit =
        import quotes.reflect.*
        (new TreeTraverser:
            override def traverseTree(tree: Tree)(owner: Symbol): Unit =
                pf.lift(tree).getOrElse(super.traverseTree(tree)(owner))
        ).traverseTree(tree)(Symbol.spliceOwner)
    end traverse

    def transform(using
        Quotes
    )(tree: quotes.reflect.Tree)(pf: PartialFunction[quotes.reflect.Tree, quotes.reflect.Term]): quotes.reflect.Tree =
        import quotes.reflect.*
        (new TreeMap:
            override def transformTerm(tree: Term)(owner: Symbol): Term =
                pf.lift(tree).getOrElse(super.transformTerm(tree)(owner))
        ).transformTree(tree)(Symbol.spliceOwner)
    end transform

    def exists(using
        Quotes
    )(tree: quotes.reflect.Tree)(pf: PartialFunction[quotes.reflect.Tree, Boolean]) =
        var r = false
        traverse(tree) {
            case t if pf.isDefinedAt(t) && !r => r = pf(t)
        }
        r
    end exists
end Trees