X-Git-Url: http://nitlanguage.org

diff --git a/src/nitls.nit b/src/nitls.nit
index e203ee7..8755bfe 100644
--- a/src/nitls.nit
+++ b/src/nitls.nit
@@ -18,7 +18,6 @@
 module nitls
 
 import modelbuilder
-intrude import loader
 import ordered_tree
 import console
 
@@ -53,6 +52,7 @@ class ProjTree
 				return o.filepath
 			else
 				var d = ""
+				var dd = ""
 				if o.mmodule != null and o.mmodule.mdoc != null then
 					if tc.opt_no_color.value then
 						d = ": {o.mmodule.mdoc.content.first}"
@@ -60,10 +60,25 @@ class ProjTree
 						d = ": {o.mmodule.mdoc.content.first.green}"
 					end
 				end
+				if o.mmodule != null and not o.mmodule.in_importation.direct_greaters.is_empty then
+					var ms = new Array[String]
+					for m in o.mmodule.in_importation.direct_greaters do
+						if m.mgroup.mproject == o.mmodule.mgroup.mproject then
+							ms.add m.name
+						else
+							ms.add m.full_name
+						end
+					end
+					if tc.opt_no_color.value then
+						dd = " ({ms.join(" ")})"
+					else
+						dd = " ({ms.join(" ")})".light_gray
+					end
+				end
 				if tc.opt_no_color.value then
-					return "{o.name.bold}{d} ({o.filepath.to_s})"
+					return "{o.name.bold}{d} ({o.filepath.to_s}){dd}"
 				else
-					return "{o.name.bold}{d} ({o.filepath.yellow})"
+					return "{o.name.bold}{d} ({o.filepath.yellow}){dd}"
 				end
 			end
 		else
@@ -72,6 +87,24 @@ class ProjTree
 	end
 end
 
+class AlphaEntityComparator
+	super Comparator
+	fun nameof(a: COMPARED): String
+	do
+		if a isa MGroup then
+			return a.name
+		else if a isa ModulePath then
+			return a.name
+		else
+			abort
+		end
+	end
+	redef fun compare(a,b)
+	do
+		return nameof(a) <=> nameof(b)
+	end
+end
+
 var tc = new ToolContext
 
 var opt_keep = new OptionBool("Ignore errors and files that are not a Nit source file", "-k", "--keep")
@@ -85,6 +118,7 @@ var opt_paths = new OptionBool("List only path (instead of name + path)", "-p",
 
 tc.option_context.add_option(opt_keep, opt_recursive, opt_tree, opt_source, opt_project, opt_depends, opt_paths, opt_make)
 tc.tooldescription = "Usage: nitls [OPTION]... <file.nit|directory>...\nLists the projects and/or paths of Nit sources files."
+tc.accept_no_arguments = true
 tc.process_options(args)
 
 if opt_make.value then
@@ -95,25 +129,25 @@ end
 
 var sum = opt_tree.value.to_i + opt_source.value.to_i + opt_project.value.to_i
 if sum > 1 then
-	print "Error: options --tree, --source, and --project are exclusives."
+	print "Error: options --tree, --source, and --project are exclusive."
 	print tc.tooldescription
 	exit 1
 end
-
+if sum == 0 then opt_project.value = true
 tc.keep_going = opt_keep.value
 
 var model = new Model
 var mb = new ModelBuilder(model, tc)
 
+if tc.option_context.rest.is_empty then tc.option_context.rest.add "."
 var files
 if opt_recursive.value then
 	files = new Array[String]
 	for d in tc.option_context.rest do
-		var pipe = new IProcess("find", d, "-name", "*.nit")
+		var pipe = new ProcessReader("find", d, "-name", "*.nit")
 		while not pipe.eof do
 			var l = pipe.read_line
 			if l == "" then break # last line
-			l = l.substring(0,l.length-1) # strip last oef
 			files.add l
 		end
 		pipe.close
@@ -124,21 +158,53 @@ else
 	files = tc.option_context.rest
 end
 
+if sum == 0 then
+	# If one of the file is a group, default is `opt_tree` instead of `opt_project`
+	for a in files do
+		var g = mb.get_mgroup(a)
+		if g != null then
+			opt_tree.value = true
+			opt_project.value = false
+			break
+		end
+	end
+end
+
+# Identify all relevant files
 for a in files do
+	var g = mb.get_mgroup(a)
 	var mp = mb.identify_file(a)
-	tc.check_errors
-	if mp != null and not opt_paths.value then
+	if g != null and not opt_project.value then
+		mb.visit_group(g)
+	end
+	if g == null and mp == null then
+		# not a group not a module, then look at files in the directory
+		var fs = a.files
+		for f in fs do
+			g = mb.get_mgroup(a/f)
+			if g != null and not opt_project.value then
+				mb.visit_group(g)
+			end
+			mp = mb.identify_file(a/f)
+			#print "{a/f}: {mp or else "?"}"
+		end
+	end
+end
+
+# Load modules to get more informations
+for mp in mb.identified_files do
+	if not opt_paths.value or opt_depends.value then
 		var mm = mb.load_module(mp.filepath)
 		if mm != null and opt_depends.value then
 			mb.build_module_importation(mm)
 		end
-		tc.check_errors
 	end
 end
+#tc.check_errors
 
-if sum == 0 then opt_project.value = true
 
 var ot = new ProjTree(tc)
+var sorter = new AlphaEntityComparator
 if opt_tree.value then
 	ot.opt_paths = opt_paths.value
 	for p in model.mprojects do
@@ -153,41 +219,54 @@ if opt_tree.value then
 			end
 		end
 	end
-	ot.sort_with(new CachedAlphaComparator)
+	ot.sort_with(sorter)
 	ot.write_to(stdout)
 end
 
 if opt_source.value then
-	var list = new Array[String]
+	var list = new Array[ModulePath]
 	for p in model.mprojects do
 		for g in p.mgroups do
 			for mp in g.module_paths do
-				if opt_paths.value then
-					list.add(mp.filepath)
-				else
-					list.add("{g.full_name}/{ot.display(mp)}")
-				end
+				list.add mp
 			end
 		end
 	end
-	alpha_comparator.sort(list)
-	for l in list do print l
+	sorter.sort(list)
+	for mp in list do
+		if opt_paths.value then
+			print mp.filepath
+		else
+			print "{mp.mgroup.full_name}/{ot.display(mp)}"
+		end
+	end
 end
 
 if opt_project.value then
-	var list = new Array[String]
+	var list = new Array[MGroup]
 	for p in model.mprojects do
-		var path = p.root.filepath.as(not null)
+		list.add p.root.as(not null)
+	end
+	sorter.sort(list)
+	for g in list do
+		var path = g.filepath.as(not null)
 		if opt_paths.value then
-			list.add(path)
+			print path
 		else
+			var d = ""
+			var md = g.mdoc_or_fallback
+			if md != null then
+				if tc.opt_no_color.value then
+					d = ": {md.content.first}"
+				else
+					d = ": {md.content.first.green}"
+				end
+			end
 			if tc.opt_no_color.value then
-				list.add("{p.name} ({path})")
+				print "{g.name}{d} ({path})"
 			else
-				list.add("{p.name} ({path.yellow})")
+				print "{g.name}{d} ({path.yellow})"
 			end
 		end
 	end
-	alpha_comparator.sort(list)
-	for l in list do print l
 end