From cfc8a85fdf72253291001a2291cc9adc37365b22 Mon Sep 17 00:00:00 2001
From: Guillaume Martres <smarter@ubuntu.com>
Date: Mon, 10 Jul 2017 22:39:01 +0200
Subject: [PATCH] Use the new Zinc main class detection

See https://github.com/sbt/zinc/pull/287 for details.
---
 main/src/main/scala/sbt/Defaults.scala | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/main/src/main/scala/sbt/Defaults.scala b/main/src/main/scala/sbt/Defaults.scala
index 857d30b5e..d29dd3858 100755
--- a/main/src/main/scala/sbt/Defaults.scala
+++ b/main/src/main/scala/sbt/Defaults.scala
@@ -78,7 +78,6 @@ import xsbti.compile.IncToolOptionsUtil
 import xsbti.CrossValue
 
 // incremental compiler
-import xsbt.api.Discovery
 import xsbti.compile.{
   ClassFileManagerType,
   ClasspathOptionsUtil,
@@ -1328,11 +1327,10 @@ object Defaults extends BuildCommon {
                                bgCopyClasspath in bgRunMain,
                                runner in run).evaluated
 
-  def discoverMainClasses(analysis: CompileAnalysis): Seq[String] =
-    Discovery
-      .applications(Tests.allDefs(analysis))
-      .collect({ case (definition, discovered) if discovered.hasMain => definition.name })
-      .sorted
+  def discoverMainClasses(analysis: CompileAnalysis): Seq[String] = analysis match {
+    case analysis: Analysis =>
+      analysis.infos.allInfos.values.map(_.getMainClasses).flatten.toSeq.sorted
+  }
 
   def consoleProjectTask =
     Def.task {