diff options
author | Christopher Vogt <oss.nsp@cvogt.org> | 2017-03-28 09:35:24 -0400 |
---|---|---|
committer | Christopher Vogt <oss.nsp@cvogt.org> | 2017-03-28 10:09:15 -0400 |
commit | a3e5f304e99ff47165af6ed20182f06700e15b33 (patch) | |
tree | ecd5ab13a2a2c5a11415fa9d4f64f3634e02309a /stage1/resolver.scala | |
parent | 497fff26705be6a21e6099a562ff14bfe1d90003 (diff) | |
download | cbt-a3e5f304e99ff47165af6ed20182f06700e15b33.tar.gz cbt-a3e5f304e99ff47165af6ed20182f06700e15b33.tar.bz2 cbt-a3e5f304e99ff47165af6ed20182f06700e15b33.zip |
minor reflection related refactor
Diffstat (limited to 'stage1/resolver.scala')
-rw-r--r-- | stage1/resolver.scala | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/stage1/resolver.scala b/stage1/resolver.scala index 0e5d221..be4d278 100644 --- a/stage1/resolver.scala +++ b/stage1/resolver.scala @@ -77,24 +77,28 @@ trait DependencyImplementation extends Dependency{ ) } */ - def flatClassLoader: Boolean = false - def runMain( className: String, args: Seq[String] ): ExitCode = lib.trapExitCode{ - lib.runMain( classLoader.loadClass( className ), args ) + def runMain( className: String, args: Seq[String] ): ExitCode = { + lib.getMain( classLoader.loadClass( className ) )( args ) } - def runMain( args: Seq[String] ): ExitCode = lib.trapExitCode{ - mainMethod.getOrElse( + def runMain( args: Seq[String] ): ExitCode = { + val c = mainClass.getOrElse( throw new RuntimeException( "No main class found in " + this ) - )( args ) + ) + runMain( c.getName, args ) } - def mainMethod = lib.pickOne( "Which one do you want to run?", mainMethods )( _.name ) + def mainClass = lib.pickOne( + "Which one do you want to run?", + classes.filter( lib.findMain(_).nonEmpty ) + )( _.name.stripSuffix( "$" ) ) def classes = exportedClasspath.files.flatMap( - lib.iterateClasses( _, classLoader, false ) + lib.topLevelClasses( _, classLoader, false ) ) - def mainMethods = classes.flatMap( lib.discoverMain ) + + def flatClassLoader: Boolean = false def classLoader: ClassLoader = taskCache[DependencyImplementation]( "classLoader" ).memoize{ if( flatClassLoader ){ |