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
|
/* NSC -- new Scala compiler
* Copyright 2005-2013 LAMP/EPFL
*/
package scala.reflect.macros
package contexts
import scala.reflect.internal.Flags._
import scala.reflect.internal.util.BatchSourceFile
import scala.reflect.io.VirtualFile
trait Synthetics {
self: Context =>
import global._
import mirror.wrapMissing
// getClassIfDefined and getModuleIfDefined cannot be used here
// because they don't work for stuff declared in the empty package
// (as specified in SLS, code inside non-empty packages cannot see
// declarations from the empty package, so compiler internals
// default to ignoring contents of the empty package)
// to the contrast, staticModule and staticClass are designed
// to be a part of the reflection API and, therefore, they
// correctly resolve all names
private def topLevelSymbol(name: Name): Symbol = wrapMissing {
if (name.isTermName) mirror.staticModule(name.toString)
else mirror.staticClass(name.toString)
}
def topLevelDef(name: Name): Tree =
enclosingRun.units.toList.map(_.body).flatMap {
// it's okay to check `stat.symbol` here, because currently macros expand strictly after namer
// which means that by the earliest time one can call this method all top-level definitions will have already been entered
case PackageDef(_, stats) => stats filter (stat => stat.symbol != NoSymbol && stat.symbol == topLevelSymbol(name))
case _ => Nil // should never happen, but better be safe than sorry
}.headOption getOrElse EmptyTree
def topLevelRef(name: Name): Tree = {
if (topLevelDef(name).nonEmpty) gen.mkUnattributedRef(name)
else EmptyTree
}
def introduceTopLevel[T: PackageSpec](packagePrototype: T, definition: universe.ImplDef): RefTree =
introduceTopLevel(packagePrototype, List(definition)).head
def introduceTopLevel[T: PackageSpec](packagePrototype: T, definitions: universe.ImplDef*): List[RefTree] =
introduceTopLevel(packagePrototype, definitions.toList)
private def introduceTopLevel[T: PackageSpec](packagePrototype: T, definitions: List[universe.ImplDef]): List[RefTree] = {
val code @ PackageDef(pid, _) = implicitly[PackageSpec[T]].mkPackageDef(packagePrototype, definitions)
universe.currentRun.compileLate(code)
definitions map (definition => Select(pid, definition.name))
}
protected def mkPackageDef(name: String, stats: List[Tree]) = gen.mkPackageDef(name, stats)
protected def mkPackageDef(name: TermName, stats: List[Tree]) = gen.mkPackageDef(name.toString, stats)
protected def mkPackageDef(tree: RefTree, stats: List[Tree]) = PackageDef(tree, stats)
protected def mkPackageDef(sym: Symbol, stats: List[Tree]) = {
assert(sym hasFlag PACKAGE, s"expected a package or package class symbol, found: $sym")
gen.mkPackageDef(sym.fullName.toString, stats)
}
}
|