From d360eaf8d18801a2109e0a2a2c23362c14821c9f Mon Sep 17 00:00:00 2001
From: yazgoo <yazgoo@gmail.com>
Date: Mon, 4 Nov 2024 13:56:28 +0100
Subject: [PATCH] add depeendency graph detailed

---
 .../AnalyzeDependencyGraphDetailed.scala      | 111 ++++++++++++++++++
 .../scala/GithubDependencyGraphPlugin.scala   |   2 +-
 2 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100644 sbt-plugin/src/main/scala/ch/epfl/scala/AnalyzeDependencyGraphDetailed.scala

diff --git a/sbt-plugin/src/main/scala/ch/epfl/scala/AnalyzeDependencyGraphDetailed.scala b/sbt-plugin/src/main/scala/ch/epfl/scala/AnalyzeDependencyGraphDetailed.scala
new file mode 100644
index 0000000..974d309
--- /dev/null
+++ b/sbt-plugin/src/main/scala/ch/epfl/scala/AnalyzeDependencyGraphDetailed.scala
@@ -0,0 +1,111 @@
+package ch.epfl.scala
+
+import ch.epfl.scala.githubapi._
+import sbt._
+import sbt.internal.util.complete._
+
+object AnalyzeDependencyGraphDetailed {
+
+  object Model {
+    trait DetailedAnalysisAction {
+      def name: String
+      def help: String
+    }
+    object DetailedAnalysisAction {
+      case object Get extends DetailedAnalysisAction {
+        val name = "get"
+        val help = "search for a pattern in the dependencies (requires githubGenerateSnapshot)"
+      }
+      case object List extends DetailedAnalysisAction {
+        val name = "list"
+        val help = "list all dependencies matching a pattern (requires githubGenerateSnapshot)"
+      }
+      val values: Seq[DetailedAnalysisAction] = Seq(Get, List)
+      def fromString(str: String): Option[DetailedAnalysisAction] = values.find(_.name == str)
+      case class AnalysisParams(action: DetailedAnalysisAction, arg: Option[String])
+    }
+    case class AnalysisParams(action: DetailedAnalysisAction, arg: Option[String])
+  }
+  import Model._
+  import AnalyzeDependencyGraph.getGithubManifest
+
+  val AnalyzeDependenciesDetailed = "githubAnalyzeDependenciesDetailed"
+  val AnalyzeDependenciesUsage: String =
+    s"""$AnalyzeDependenciesDetailed [${DetailedAnalysisAction.values.map(_.name).mkString("|")}] [pattern]"""
+  val AnalyzeDependenciesDetail: String = s"""Analyze the dependencies based on a search pattern:
+  ${DetailedAnalysisAction.values.map(a => s"${a.name}: ${a.help}").mkString("\n  ")}
+  """
+
+  private def highlight(string: String, pattern: String): String =
+    string.replaceAll(pattern, s"\u001b[32m${pattern}\u001b[0m")
+
+  private def analyzeDependenciesInternal(
+      state: State,
+      action: DetailedAnalysisAction,
+      pattern: String,
+      originalPattern: String
+  ): Unit = {
+    def getDeps(dependencies: Seq[String], pattern: String): Seq[String] =
+      dependencies.filter(_.contains(pattern)).map(highlight(_, originalPattern))
+
+    def blue(str: String): String = s"\u001b[34m${str}\u001b[0m"
+
+    def resolvedDeps(
+        tabs: String,
+        acc: Seq[String],
+        resolvedByName: Map[String, DependencyNode],
+        pattern: String,
+        originalPattern: String
+    ): Seq[String] =
+      acc ++ resolvedByName.toSeq.flatMap {
+        case (name, resolved) =>
+          val matchingDependencies = getDeps(resolved.dependencies, pattern)
+          if (matchingDependencies.isEmpty) {
+            if (name.contains(pattern)) Seq(tabs + highlight(name, originalPattern)) else Nil
+          } else {
+            matchingDependencies.flatMap { matchingDependency =>
+              resolvedDeps("  " + tabs, acc :+ (tabs + matchingDependency), resolvedByName, name, originalPattern)
+            }
+          }
+      }
+
+    val matches = getGithubManifest(state)
+      .flatMap { manifests =>
+        manifests.map {
+          case (name, manifest) =>
+            manifest -> resolvedDeps("", Nil, manifest.resolved, pattern, originalPattern = pattern)
+        }
+      }
+      .toMap
+
+    action match {
+      case DetailedAnalysisAction.Get =>
+        matches.foreach {
+          case (manifest, deps) =>
+            println(s"📁 ${blue(manifest.name)}")
+            println(deps.map(dep => s"  $dep").mkString("\n"))
+        }
+      case DetailedAnalysisAction.List =>
+        println(matches.values.flatten.filter(_.contains(pattern)).toSet.mkString("\n"))
+    }
+  }
+
+  private def extractPattern(state: State): Parser[AnalysisParams] =
+    Parsers.any.*.map { raw =>
+      raw.mkString.trim.split(" ").toSeq match {
+        case Seq(action, arg) => AnalysisParams(DetailedAnalysisAction.fromString(action).get, Some(arg))
+        case Seq(action)      => AnalysisParams(DetailedAnalysisAction.fromString(action).get, None)
+      }
+    }.failOnException
+
+  val commands: Seq[Command] = Seq(
+    Command(AnalyzeDependenciesDetailed, (AnalyzeDependenciesUsage, AnalyzeDependenciesDetail), AnalyzeDependenciesDetail)(
+      extractPattern
+    )(analyzeDependencies)
+  )
+
+  private def analyzeDependencies(state: State, params: AnalysisParams): State = {
+    params.arg.foreach(pattern => analyzeDependenciesInternal(state, params.action, pattern, pattern))
+    state
+  }
+}
diff --git a/sbt-plugin/src/main/scala/ch/epfl/scala/GithubDependencyGraphPlugin.scala b/sbt-plugin/src/main/scala/ch/epfl/scala/GithubDependencyGraphPlugin.scala
index 448a48f..3c31b80 100644
--- a/sbt-plugin/src/main/scala/ch/epfl/scala/GithubDependencyGraphPlugin.scala
+++ b/sbt-plugin/src/main/scala/ch/epfl/scala/GithubDependencyGraphPlugin.scala
@@ -49,7 +49,7 @@ object GithubDependencyGraphPlugin extends AutoPlugin {
 
   override def globalSettings: Seq[Setting[_]] = Def.settings(
     githubStoreDependencyManifests := storeManifestsTask.evaluated,
-    Keys.commands ++= SubmitDependencyGraph.commands ++ AnalyzeDependencyGraph.commands
+    Keys.commands ++= SubmitDependencyGraph.commands ++ AnalyzeDependencyGraph.commands ++ AnalyzeDependencyGraphDetailed.commands
   )
 
   override def projectSettings: Seq[Setting[_]] = Def.settings(