From de16fc43f39b86679135aec2ae33e498100ca457 Mon Sep 17 00:00:00 2001
From: ChavXO <mchavinda@colgate.edu>
Date: Sun, 14 Aug 2016 21:25:20 -0400
Subject: Added run with package discovery

---
 stage2/BasicBuild.scala | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'stage2/BasicBuild.scala')

diff --git a/stage2/BasicBuild.scala b/stage2/BasicBuild.scala
index bcc3646..baa6d26 100644
--- a/stage2/BasicBuild.scala
+++ b/stage2/BasicBuild.scala
@@ -136,7 +136,12 @@ trait BaseBuild extends DependencyImplementation with BuildInterface with Trigge
     )
   }
 
-  def runClass: String = "Main"
+  def runClass: String = lib.mainClasses ( compileTarget, classLoader(context.classLoaderCache) ) match {
+    case f if f.length == 1 => f.head
+    case f if f.length > 2  => System.err.println( " Multiple main classes defined in project." ); "None"
+    case _                  => "Main"
+  }
+
   def run: ExitCode = lib.runMainIfFound( runClass, context.args, classLoader(context.classLoaderCache) )
 
   def clean = {
-- 
cgit v1.2.3