luke
/
sbt
mirror of https://github.com/sbt/sbt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add option to show inferred type of an expression

This commit is contained in:
Mark Harrah 2010-12-19 11:37:37 -05:00
parent 0b3c2dada5
commit 20a5e3b3c7
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
compile/Eval.scala
View File

@ -23,8 +23,8 @@ final class Eval(options: Seq[String], mkReporter: Settings => Reporter, parent:
import global._ import global._
import definitions._ import definitions._
def eval[T](expression: String)(implicit mf: Manifest[T]): T = eval(expression, mf.toString).asInstanceOf[T] def eval[T](expression: String)(implicit mf: Manifest[T]): T = eval(expression, Some(mf.toString)).asInstanceOf[T]
def eval(expression: String, tpeName: String): Any = def eval(expression: String, tpeName: Option[String]): Any =
{ {
reporter.reset reporter.reset
val unit = mkUnit(expression) val unit = mkUnit(expression)
@ -36,7 +36,7 @@ final class Eval(options: Seq[String], mkReporter: Settings => Reporter, parent:
try { eval0(expression, tpeName, run, unit) } finally { unlinkAll() } try { eval0(expression, tpeName, run, unit) } finally { unlinkAll() }
} }
def eval0(expression: String, tpeName: String, run: Run, unit: CompilationUnit): Any = def eval0(expression: String, tpeName: Option[String], run: Run, unit: CompilationUnit): (String, Any) =
{ {
val virtualDirectory = new VirtualDirectory("<virtual>", None) val virtualDirectory = new VirtualDirectory("<virtual>", None)
settings.outputDirs setSingleOutput virtualDirectory settings.outputDirs setSingleOutput virtualDirectory
@ -46,10 +46,15 @@ final class Eval(options: Seq[String], mkReporter: Settings => Reporter, parent:
parser.accept(EOF) parser.accept(EOF)
checkError("Error parsing expression.") checkError("Error parsing expression.")
val tpeParser = new syntaxAnalyzer.UnitParser(mkUnit(tpeName)) val tpt: Tree = tpeName match {
val tpt: Tree = tpeParser.typ() case Some(tpe) =>
val tpeParser = new syntaxAnalyzer.UnitParser(mkUnit(tpe))
val tpt0: Tree = tpeParser.typ()
tpeParser.accept(EOF) tpeParser.accept(EOF)
checkError("Error parsing type.") checkError("Error parsing type.")
tpt0
case None => TypeTree(NoType)
}
unit.body = augment(parser, tree, tpt) unit.body = augment(parser, tree, tpt)
@ -69,9 +74,10 @@ final class Eval(options: Seq[String], mkReporter: Settings => Reporter, parent:
compile(run.namerPhase) compile(run.namerPhase)
checkError("Type error.") checkError("Type error.")
val tpe = atPhase(run.typerPhase.next) { (new TypeExtractor).getType(unit.body) }
val loader = new AbstractFileClassLoader(virtualDirectory, parent) val loader = new AbstractFileClassLoader(virtualDirectory, parent)
getValue(loader) (tpe, getValue(loader))
} }
val WrapObjectName = "$sbtobj" val WrapObjectName = "$sbtobj"
val WrapValName = "$sbtdef" val WrapValName = "$sbtdef"
@ -103,6 +109,14 @@ final class Eval(options: Seq[String], mkReporter: Settings => Reporter, parent:
value.asInstanceOf[T] value.asInstanceOf[T]
} }
final class TypeExtractor extends Traverser {
private[this] var result = ""
def getType(t: Tree) = { result = ""; traverse(t); result }
override def traverse(tree: Tree): Unit = tree match {
case d: DefDef if d.symbol.nameString == WrapValName => result = d.symbol.tpe.finalResultType.toString
case _ => super.traverse(tree)
}
}
def mkUnit(s: String) = new CompilationUnit(new BatchSourceFile("<setting>", s)) def mkUnit(s: String) = new CompilationUnit(new BatchSourceFile("<setting>", s))
def checkError(label: String) = if(reporter.hasErrors) throw new EvalException(label) def checkError(label: String) = if(reporter.hasErrors) throw new EvalException(label)