dotty.tools.dottydoc.core.UsecasePhase.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dotty-doc_0.20 Show documentation

dotty-doc-bootstrapped

The newest version!

package dotty.tools
package dottydoc
package core

import dotc.core.Contexts.Context
import dotc.ast.tpd

import transform.DocMiniPhase
import model.internal._
import model.factories._
import dotty.tools.dotc.core.Symbols.Symbol
import util.syntax._

class UsecasePhase extends DocMiniPhase {
  private def defdefToDef(d: tpd.DefDef, sym: Symbol)(implicit ctx: Context) = {
    val name = d.name.show.split("\\$").head
    DefImpl(
      sym,
      annotations(sym),
      name,
      flags(d),
      path(d.symbol).init :+ name,
      returnType(d.tpt.tpe),
      typeParams(d.symbol),
      paramLists(d.symbol.info)
    )
  }

  override def transformDef(implicit ctx: Context) = { case df: DefImpl =>
    val defdefs =
      ctx.docbase.docstring(df.symbol)
        .map(_.usecases.flatMap(_.tpdCode))
        .getOrElse(Nil)

    if (defdefs.isEmpty) df :: Nil
    else defdefs.map(defdefToDef(_, df.symbol))
  }
}