1 files changed, 37 insertions, 0 deletions
diff --git a/main/src/mill/main/VisualizeModule.scala b/main/src/mill/main/VisualizeModule.scala
new file mode 100644
index 00000000..00651a8e
--- /dev/null
+++ b/main/src/mill/main/VisualizeModule.scala
@@ -0,0 +1,37 @@
+package mill.main
+
+import ammonite.ops.Path
+import coursier.core.Repository
+import mill.T
+import mill.eval.{Evaluator, PathRef, Result}
+
+trait VisualizeModule extends mill.define.TaskModule{
+  def repositories: Seq[Repository]
+  def defaultCommandName() = "run"
+  def classpath = T{
+    mill.modules.Util.millProjectModule("MILL_GRAPHVIZ", "mill-main-graphviz", repositories)
+  }
+  /**
+    * Given a set of tasks, prints out the execution plan of what tasks will be
+    * executed in what order, without actually executing them.
+    */
+  def run(evaluator: Evaluator[Any], targets: String*) = mill.T.command{
+    val resolved = RunScript.resolveTasks(
+      mill.main.ResolveTasks, evaluator, targets, multiSelect = true
+    )
+    resolved match{
+      case Left(err) => Result.Failure(err)
+      case Right(rs) =>
+        Result.Success(
+          mill.modules.Jvm.inprocess(classpath().map(_.path), false, isolated = false, cl => {
+            cl.loadClass("mill.main.graphviz.GraphvizTools")
+              .getMethod("apply", classOf[Seq[_]], classOf[Path])
+              .invoke(null, rs, T.ctx().dest)
+              .asInstanceOf[Seq[PathRef]]
+          })
+        )
+
+    }
+  }
+
+}