From: Alexandre Terrasa <alexandre@moz-code.org>
Date: Mon, 4 May 2015 14:46:36 +0000 (-0400)
Subject: nitx: redo search for MType in new calls
X-Git-Tag: v0.7.5~29^2~4
X-Git-Url: http://nitlanguage.org

nitx: redo search for MType in new calls

Signed-off-by: Alexandre Terrasa <alexandre@moz-code.org>
---

diff --git a/src/doc/doc_phases/doc_console.nit b/src/doc/doc_phases/doc_console.nit
index ba8dc04..0d1509c 100644
--- a/src/doc/doc_phases/doc_console.nit
+++ b/src/doc/doc_phases/doc_console.nit
@@ -62,6 +62,7 @@ class Nitx
 		print "\tdoc: <name::space>\tdisplay the documentation page of 'namespace'"
 		print "\tparam: <Type>\tlookup methods using the corresponding 'Type' as parameter"
 		print "\treturn: <Type>\tlookup methods returning the corresponding 'Type'"
+		print "\tnew: <Type>\tlookup methods creating new instances of 'Type'"
 		print "\t:h\t\tdisplay this help message"
 		print "\t:q\t\tquit interactive mode"
 		print ""
@@ -119,6 +120,8 @@ interface NitxQuery
 			return new ParamQuery(query_string)
 		else if query_string.has_prefix("return:") then
 			return new ReturnQuery(query_string)
+		else if query_string.has_prefix("new:") then
+			return new NewQuery(query_string)
 		end
 		return new CommentQuery("comment: {query_string}")
 	end
@@ -259,6 +262,26 @@ class ReturnQuery
 	end
 end
 
+# A query to search methods creating new instances of a specific `MType`.
+class NewQuery
+	super MetaQuery
+
+	redef fun perform(nitx, doc) do
+		var res = new Array[NitxMatch]
+		var mtype_name = args.first
+		for mpropdef in doc.mpropdefs do
+			var visitor = new TypeInitVisitor(mtype_name)
+			var npropdef = nitx.ctx.modelbuilder.mpropdef2node(mpropdef)
+			if npropdef == null then continue
+			visitor.enter_visit(npropdef)
+			for i in visitor.inits do
+				res.add new MEntityMatch(self, mpropdef)
+			end
+		end
+		return res
+	end
+end
+
 # A query to search a Nitdoc documentation page by its name.
 class DocQuery
 	super MetaQuery
@@ -362,6 +385,26 @@ redef class DocModel
 	end
 end
 
+# Visitor looking for initialized `MType` (new T).
+#
+# See `NewQuery`.
+private class TypeInitVisitor
+	super Visitor
+
+	# `MType` name to look for.
+	var mtype_name: String
+
+	var inits = new HashSet[MType]
+	redef fun visit(node)
+	do
+		node.visit_all(self)
+		# look for init
+		if not node isa ANewExpr then return
+		var mtype = node.n_type.mtype
+		if mtype != null and mtype.name == mtype_name then inits.add(mtype)
+	end
+end
+
 # display
 
 # A `DocArticle` that displays query results.
diff --git a/src/nitx.nit b/src/nitx.nit
index 4c08b29..8536fea 100644
--- a/src/nitx.nit
+++ b/src/nitx.nit
@@ -18,6 +18,7 @@
 #
 # * Display comment from name/namespace
 # * Display documentation page from Nitdoc in console
+# * Find type usage in parameters, returns and news.
 module nitx
 
 import modelbuilder