From 454bc6095bd31025fe75e179583cf148aef2e218 Mon Sep 17 00:00:00 2001
From: Mark Harrah <dmharrah@gmail.com>
Date: Tue, 13 Dec 2011 17:29:08 -0500
Subject: [PATCH] show definitions and uses of a key (not a full scoped key,
 just a key)

---
 main/Command.scala |  5 ++++-
 main/Main.scala    | 46 +++++++++++++++++++++++++++++++++-------------
 main/Project.scala | 11 +++++++++++
 3 files changed, 48 insertions(+), 14 deletions(-)

diff --git a/main/Command.scala b/main/Command.scala
index 49eb1ee53..cb45a6590 100644
--- a/main/Command.scala
+++ b/main/Command.scala
@@ -146,7 +146,10 @@ sealed trait InspectOption
 object InspectOption
 {
 	final case class Details(actual: Boolean) extends InspectOption
-	case object DependencyTree extends InspectOption
+	private[this] final class Opt(override val toString: String) extends InspectOption
+	val DependencyTree: InspectOption = new Opt("tree")
+	val Uses: InspectOption = new Opt("inspect")
+	val Definitions: InspectOption = new Opt("definitions")
 }
 
 trait Help
diff --git a/main/Main.scala b/main/Main.scala
index ee0cc0a1a..29a2a5901 100644
--- a/main/Main.scala
+++ b/main/Main.scala
@@ -395,16 +395,26 @@ object BuiltinCommands
 		val newSession = session.appendSettings( append map (a => (a, arg)))
 		reapply(newSession, structure, s)
 	}
-	def inspect = Command(InspectCommand, inspectBrief, inspectDetailed)(inspectParser) {
-		case (s, (InspectOption.DependencyTree, sk)) =>
-		  val basedir = new File(Project.session(s).current.build)
-			val treeString = Project.settingGraph(Project.structure(s), basedir, sk)( Project.showContextKey(s) ).dependsAscii
-			logger(s).info(treeString)
-			s
-		case (s, (InspectOption.Details(actual), sk)) =>
-			val detailString = Project.details(Project.structure(s), actual, sk.scope, sk.key)( Project.showContextKey(s) )
-			logger(s).info(detailString)
-			s
+	def inspect = Command(InspectCommand, inspectBrief, inspectDetailed)(inspectParser) { case (s, (option, sk)) => 
+		logger(s).info(inspectOutput(s, option, sk))
+		s
+	}
+	def inspectOutput(s: State, option: InspectOption, sk: Project.ScopedKey[_]): String =
+	{
+		val extracted = Project.extract(s)
+			import extracted._
+		option match
+		{
+			case InspectOption.Details(actual) =>
+				Project.details(structure, actual, sk.scope, sk.key)
+			case InspectOption.DependencyTree => 
+				val basedir = new File(Project.session(s).current.build)
+				Project.settingGraph(structure, basedir, sk).dependsAscii
+			case InspectOption.Uses =>
+				Project.showUses(Project.usedBy(structure, true, sk.key))
+			case InspectOption.Definitions =>
+				Project.showDefinitions(sk.key, Project.definitions(structure, true, sk.key))
+		}
 	}
 	def lastGrep = Command(LastGrepCommand, lastGrepBrief, lastGrepDetailed)(lastGrepParser) {
 		case (s, (pattern,Some(sk))) =>
@@ -420,12 +430,22 @@ object BuiltinCommands
 		val ext = Project.extract(s)
 		(ext.structure, Select(ext.currentRef), ext.showKey)
 	}
-	def inspectParser = (s: State) => spacedInspectOptionParser(s) ~ spacedKeyParser(s)
-	val spacedInspectOptionParser: (State => Parser[InspectOption]) = (s: State) => {
 		import InspectOption._
+	def inspectParser = (s: State) => spacedInspectOptionParser(s) flatMap {
+		case opt @ (Uses | Definitions) => allKeyParser(s).map(key => (opt, Project.ScopedKey(Global, key)))
+		case opt @ (DependencyTree | Details(_)) => spacedKeyParser(s).map(key => (opt, key))
+	}
+	val spacedInspectOptionParser: (State => Parser[InspectOption]) = (s: State) => {
 		val actual = "actual" ^^^ Details(true)
 		val tree = "tree" ^^^ DependencyTree
-		token(Space ~> (tree | actual)) ?? Details(false)
+		val uses = "uses" ^^^ Uses
+		val definitions = "definitions" ^^^ Definitions
+		token(Space ~> (tree | actual | uses | definitions)) ?? Details(false)
+	}
+	def allKeyParser(s: State): Parser[AttributeKey[_]] =
+	{
+		val keyMap = Project.structure(s).index.keyMap
+		token((Space ~> ID) !!! "Expected key" examples keyMap.keySet) flatMap { key => Act.getKey(keyMap, key, idFun) }
 	}
 	val spacedKeyParser = (s: State) => Act.requireSession(s, token(Space) ~> Act.scopedKeyParser(s))
 	val optSpacedKeyParser = (s: State) => spacedKeyParser(s).?
diff --git a/main/Project.scala b/main/Project.scala
index e5f02438e..3882634d3 100755
--- a/main/Project.scala
+++ b/main/Project.scala
@@ -303,6 +303,17 @@ object Project extends Init[Scope] with ProjectExtra
 			r + (key, value.dependencies)
 		}
 	}
+
+	def showDefinitions(key: AttributeKey[_], defs: Seq[Scope])(implicit display: Show[ScopedKey[_]]): String =
+		defs.map(scope => display(ScopedKey(scope, key))).sorted.mkString("\n\t", "\n\t", "\n\n")
+	def showUses(defs: Seq[ScopedKey[_]])(implicit display: Show[ScopedKey[_]]): String =
+		defs.map(display.apply).sorted.mkString("\n\t", "\n\t", "\n\n")
+
+	def definitions(structure: BuildStructure, actual: Boolean, key: AttributeKey[_])(implicit display: Show[ScopedKey[_]]): Seq[Scope] =
+		relation(structure, actual)(display)._1s.toSeq flatMap { sk => if(sk.key == key) sk.scope :: Nil else Nil }
+	def usedBy(structure: BuildStructure, actual: Boolean, key: AttributeKey[_])(implicit display: Show[ScopedKey[_]]): Seq[ScopedKey[_]] =
+		relation(structure, actual)(display).all.toSeq flatMap { case (a,b) => if(b.key == key) List[ScopedKey[_]](a) else Nil }
+
 	def reverseDependencies(cMap: Map[ScopedKey[_],Flattened], scoped: ScopedKey[_]): Iterable[ScopedKey[_]] =
 		for( (key,compiled) <- cMap; dep <- compiled.dependencies if dep == scoped)  yield  key