holes.NamedHolesPhase.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scala-typed-holes_3.6.2 Show documentation

scala-typed-holes

The newest version!

package holes

import dotty.tools.dotc.ast.tpd
import dotty.tools.dotc.ast.untpd
import dotty.tools.dotc.ast.untpd.UntypedTreeMap
import dotty.tools.dotc.core.Contexts.Context
import dotty.tools.dotc.core.Contexts.ctx
import dotty.tools.dotc.core.Names.Name
import dotty.tools.dotc.core.StdNames
import dotty.tools.dotc.parsing.Parser
import dotty.tools.dotc.plugins.PluginPhase
import dotty.tools.dotc.typer.TyperPhase
import dotty.tools.dotc.util.Property

class NamedHolesPhase extends PluginPhase:
  override def phaseName: String = "named-holes"

  override val runsAfter: Set[String] = Set(Parser.name)
  override val runsBefore: Set[String] = Set(TyperPhase.name)

  override def prepareForUnit(tree: tpd.Tree)(using Context): Context =
    val namedHolesTreeMap = new UntypedTreeMap {
      override def transform(tree: tpd.Tree)(using Context): tpd.Tree =
        tree match
          case tpd.Ident(NamedHole(name)) =>
            val copied = cpy.Ident(tree)(StdNames.nme.???)
            copied.putAttachment(NamedHole.NamedHole, HoleName(name))
            copied
          case _ =>
            super.transform(tree)
    }
    ctx.compilationUnit.untpdTree =
      namedHolesTreeMap.transform(ctx.compilationUnit.untpdTree)
    ctx

object NamedHole:
  val NamedHole: Property.Key[HoleName] = Property.StickyKey()
  val pattern = "^__([a-zA-Z0-9_]+)$".r

  def unapply(name: Name): Option[String] =
    pattern.unapplySeq(name.decode.toString).flatMap(_.headOption)