1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
package dotty.tools
package dottydoc
package core
/** Dotty and Dottydoc imports */
import dotc.ast.Trees._
import dotc.CompilationUnit
import dotc.config.Printers.dottydoc
import dotc.core.Contexts.Context
import dotc.core.Phases.Phase
import dotc.core.Symbols.Symbol
object Phases {
class DocPhase extends Phase {
import model.comment.Comment
import model.CommentParsers.wikiParser
import model.Entities._
import model.EntityFactories._
import dotty.tools.dotc.core.Flags
import dotty.tools.dotc.ast.tpd._
import util.Traversing._
def phaseName = "docphase"
private[this] var commentCache: Map[Entity, (Entity, Map[String, Package]) => Option[Comment]] = Map.empty
/** Saves the commentParser function for later evaluation, for when the AST has been filled */
def track(symbol: Symbol, ctx: Context)(op: => Entity) = {
val entity = op
val commentParser = { (entity: Entity, packs: Map[String, Package]) =>
wikiParser.parseHtml(symbol, entity, packs)(ctx)
}
commentCache = commentCache + (entity -> commentParser)
entity
}
/** Build documentation hierarchy from existing tree */
def collect(tree: Tree)(implicit ctx: Context): Entity = track(tree.symbol, ctx) {
def collectList(xs: List[Tree])(implicit ctx: Context): List[Entity] =
xs.map(collect).filter(_ != NonEntity)
def collectPackageMembers(xs: List[Tree])(implicit ctx: Context): List[PackageMember] =
collectList(xs).asInstanceOf[List[PackageMember]]
def collectMembers(tree: Tree)(implicit ctx: Context): List[Entity] = tree match {
case t: Template => collectList(t.body)
case _ => Nil
}
tree match {
/** package */
case p @ PackageDef(pid, st) =>
val name = pid.name.toString
Package(name, collectPackageMembers(st), path(p))
/** trait */
case t @ TypeDef(n, rhs) if t.symbol.is(Flags.Trait) =>
val name = n.toString
Trait(name, collectMembers(rhs), flags(t), path(t))
/** objects, on the format "Object$" so drop the last letter */
case o @ TypeDef(n, rhs) if o.symbol.is(Flags.Module) =>
val name = n.toString.dropRight(1)
Object(name, collectMembers(rhs), flags(o), path(o))
/** class / case class */
case c @ TypeDef(name, rhs) if c.symbol.isClass =>
(name.toString, collectMembers(rhs), flags(c), path(c), None) match {
case x if c.symbol.is(Flags.CaseClass) => CaseClass.tupled(x)
case x => Class.tupled(x)
}
/** def */
case d: DefDef =>
Def(d.name.toString, flags(d), path(d))
/** val */
case v: ValDef if !v.symbol.is(Flags.ModuleVal) =>
Val(v.name.toString, flags(v), path(v))
case x => {
//dottydoc.println(s"Found unwanted entity: $x (${x.pos}, ${comment})\n${x.show}")
NonEntity
}
}
}
var packages: Map[String, Package] = Map.empty
def addEntity(p: Package): Unit = {
val path = p.path.mkString(".")
packages = packages + (path -> packages.get(path).map { ex =>
val children = (ex.children ::: p.children).distinct.sortBy(_.name)
Package(p.name, children, p.path, None)
}.getOrElse(p))
}
override def run(implicit ctx: Context): Unit =
collect(ctx.compilationUnit.tpdTree) match {
case p: Package => addEntity(p)
case _ => ()
}
override def runOn(units: List[CompilationUnit])(implicit ctx: Context): List[CompilationUnit] = {
// (1) Create package structure for all `units`, this will give us a complete structure
val compUnits = super.runOn(units)
// (2) Set parent of all package children
def setParent(ent: Entity, to: Entity): Unit =
ent match {
case e: Class =>
e.parent = Some(to)
e.members.foreach(setParent(_, e))
case e: CaseClass =>
e.parent = Some(to)
e.members.foreach(setParent(_, e))
case e: Object =>
e.parent = Some(to)
e.members.foreach(setParent(_, e))
case e: Trait =>
e.parent = Some(to)
e.members.foreach(setParent(_, e))
case e: Val =>
e.parent = Some(to)
case e: Def =>
e.parent = Some(to)
case _ => ()
}
for {
parent <- packages.values
child <- parent.children
} setParent(child, to = parent)
// (3) Create documentation template from docstrings, with internal links
packages.values.foreach { p =>
mutateEntities(p) {
case e: Package => e.comment = commentCache(e)(e, packages)
case e: Class => e.comment = commentCache(e)(e, packages)
case e: CaseClass => e.comment = commentCache(e)(e, packages)
case e: Object => e.comment = commentCache(e)(e, packages)
case e: Trait => e.comment = commentCache(e)(e, packages)
case e: Val => e.comment = commentCache(e)(e, packages)
case e: Def => e.comment = commentCache(e)(e, packages)
case _ => ()
}
}
// (4) Write the finished model to JSON
util.IndexWriters.writeJs(packages, "../js/out")
// (5) Clear caches
commentCache = Map.empty
// Return super's result
compUnits
}
}
}
|