move VisualizeModule out into it's own ExternalModule; for some reason it wasn't getting picked up in the build discovery before

2 files changed, 15 insertions, 7 deletions

diff --git a/main/src/mill/main/MainModule.scala b/main/src/mill/main/MainModule.scala
index c7efbd21..00ddcde1 100644
--- a/main/src/mill/main/MainModule.scala
+++ b/main/src/mill/main/MainModule.scala
@@ -220,12 +220,7 @@ trait MainModule extends mill.Module{
     }
   }
 
-  val visualize: VisualizeModule = new VisualizeModule {
-    def repositories = Seq(
-      Cache.ivy2Local,
-      MavenRepository("https://repo1.maven.org/maven2"),
-      MavenRepository("https://oss.sonatype.org/content/repositories/releases")
-    )
+  def visualize(evaluator: Evaluator[Any], targets: String*) = mill.T.command{
+    VisualizeModule.run(evaluator, targets:_*)
   }
-
 }
diff --git a/main/src/mill/main/VisualizeModule.scala b/main/src/mill/main/VisualizeModule.scala
index 00651a8e..e939bd86 100644
--- a/main/src/mill/main/VisualizeModule.scala
+++ b/main/src/mill/main/VisualizeModule.scala
@@ -1,10 +1,23 @@
 package mill.main
 
 import ammonite.ops.Path
+import coursier.Cache
 import coursier.core.Repository
+import coursier.maven.MavenRepository
 import mill.T
+import mill.define.{Discover, ExternalModule}
 import mill.eval.{Evaluator, PathRef, Result}
 
+object VisualizeModule extends ExternalModule with VisualizeModule {
+  def repositories = Seq(
+    Cache.ivy2Local,
+    MavenRepository("https://repo1.maven.org/maven2"),
+    MavenRepository("https://oss.sonatype.org/content/repositories/releases")
+  )
+
+  implicit def millScoptEvaluatorReads[T] = new mill.main.EvaluatorScopt[T]()
+  lazy val millDiscover = Discover[this.type]
+}
 trait VisualizeModule extends mill.define.TaskModule{
   def repositories: Seq[Repository]
   def defaultCommandName() = "run"