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

diff --git a/src/nitls.nit b/src/nitls.nit
index 1709c56..62fc09f 100644
--- a/src/nitls.nit
+++ b/src/nitls.nit
@@ -17,13 +17,15 @@
 # Simple tool to list Nit source files
 module nitls
 
-intrude import modelbuilder
+import modelbuilder
 import ordered_tree
+import console
 
 class ProjTree
-	super OrderedTree[Object]
+	super OrderedTree[MConcern]
 
 	var opt_paths = false
+	var tc: ToolContext
 
 	redef fun display(o)
 	do
@@ -31,13 +33,53 @@ class ProjTree
 			if opt_paths then
 				return o.filepath.as(not null)
 			else
-				return "{o.name} ({o.filepath})"
+				var d = ""
+				if o.mdoc != null then
+					if tc.opt_no_color.value then
+						d = ": {o.mdoc.content.first}"
+					else
+						d = ": {o.mdoc.content.first.green}"
+					end
+				end
+				if tc.opt_no_color.value then
+					return "{o.name}{d} ({o.filepath.to_s})"
+				else
+					return "{o.name}{d} ({o.filepath.yellow})"
+				end
 			end
-		else if o isa ModulePath then
+		else if o isa MModule then
 			if opt_paths then
-				return o.filepath
+				return o.filepath.as(not null)
 			else
-				return "{o.name} ({o.filepath})"
+				var d = ""
+				var dd = ""
+				if o.mdoc != null then
+					if tc.opt_no_color.value then
+						d = ": {o.mdoc.content.first}"
+					else
+						d = ": {o.mdoc.content.first.green}"
+					end
+				end
+				if not o.in_importation.direct_greaters.is_empty then
+					var ms = new Array[String]
+					for m in o.in_importation.direct_greaters do
+						if m.mgroup.mpackage == o.mgroup.mpackage 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}){dd}"
+				else
+					return "{o.name.bold}{d} ({o.filepath.yellow}){dd}"
+				end
 			end
 		else
 			abort
@@ -46,20 +88,19 @@ class ProjTree
 end
 
 var tc = new ToolContext
-var model = new Model
-var mb = new ModelBuilder(model, tc)
 
 var opt_keep = new OptionBool("Ignore errors and files that are not a Nit source file", "-k", "--keep")
-var opt_recursive = new OptionBool("Process directories recussively", "-r", "--recursive")
-var opt_tree = new OptionBool("List source files in their groups and projects", "-t", "--tree")
-var opt_source = new OptionBool("List source files", "-s", "--source")
-var opt_project = new OptionBool("List projects paths (default)", "-P", "--project")
+var opt_recursive = new OptionBool("Process directories recursively", "-r", "--recursive")
+var opt_tree = new OptionBool("List source files in their groups and packages", "-t", "--tree")
+var opt_source = new OptionBool("List source files in a flat list", "-s", "--source")
+var opt_package = new OptionBool("List packages in a flat list (default)", "-P", "--package")
 var opt_depends = new OptionBool("List dependencies of given modules", "-d", "--depends")
-var opt_make = new OptionBool("List dependencies suitable for a rule in a Makefile. Alias for -d, -p and -s", "-M")
-var opt_paths = new OptionBool("List only path (instead of name + path)", "-p", "--path")
+var opt_make = new OptionBool("List dependencies suitable for a rule in a Makefile (alias for -d, -p and -s)", "-M")
+var opt_paths = new OptionBool("List only path (instead of name + path)", "-p", "--path-only")
 
-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.option_context.add_option(opt_keep, opt_recursive, opt_tree, opt_source, opt_package, opt_depends, opt_paths, opt_make)
+tc.tooldescription = "Usage: nitls [OPTION]... <file.nit|directory>...\nLists the packages and/or paths of Nit sources files."
+tc.accept_no_arguments = true
 tc.process_options(args)
 
 if opt_make.value then
@@ -68,92 +109,132 @@ if opt_make.value then
 	opt_source.value = true
 end
 
-var sum = opt_tree.value.to_i + opt_source.value.to_i + opt_project.value.to_i
+var sum = opt_tree.value.to_i + opt_source.value.to_i + opt_package.value.to_i
 if sum > 1 then
-	print "Error: options --tree, --source, and --project are exclusives."
+	print "Error: options --tree, --source, and --package are exclusive."
 	print tc.tooldescription
 	exit 1
 end
+if sum == 0 then opt_package.value = true
+tc.keep_going = opt_keep.value
 
-if opt_depends.value then
-	if opt_recursive.value then
-		print "-M incompatible with -r"
-		exit 1
-	end
+var model = new Model
+var mb = new ModelBuilder(model, tc)
 
-	mb.parse(tc.option_context.rest)
-else
-	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")
-			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
-			pipe.wait
-			if pipe.status != 0 and not opt_keep.value then exit 1
+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 ProcessReader("find", d, "-name", "*.nit")
+		while not pipe.eof do
+			var l = pipe.read_line
+			if l == "" then break # last line
+			files.add l
 		end
-	else
-		files = tc.option_context.rest
+		pipe.close
+		pipe.wait
+		if pipe.status != 0 and not opt_keep.value then exit 1
 	end
+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_package`
 	for a in files do
-		var mp = mb.identify_file(a)
-		if mp == null then
-			if not opt_keep.value then tc.check_errors
+		var g = mb.identify_group(a)
+		if g != null then
+			opt_tree.value = true
+			opt_package.value = false
+			break
 		end
 	end
 end
 
-if sum == 0 then opt_project.value = true
+var mmodules = mb.scan_full(files)
 
+# Load modules to get more informations
+for mmodule in mmodules do
+	if not opt_paths.value or opt_depends.value then
+		var ast = mmodule.parse(mb)
+		if ast != null and opt_depends.value then
+			mb.build_module_importation(ast)
+		end
+	end
+end
+#tc.check_errors
+
+if opt_depends.value then
+	# Extends the list of module with the loaded ones
+	mmodules = mb.parsed_modules.to_a
+end
+
+var ot = new ProjTree(tc)
 if opt_tree.value then
-	var ot = new ProjTree
 	ot.opt_paths = opt_paths.value
-	for p in model.mprojects do
-		for g in p.mgroups do
-			ot.add(g.parent, g)
-			for mp in g.module_paths do
-				ot.add(g, mp)
-			end
+	var mgroups = new HashSet[MGroup]
+	for mp in mmodules do
+		var pa = mp.mgroup
+		while pa != null and not pa.is_interesting do pa = pa.parent
+		ot.add(pa, mp)
+		while pa != null do
+			mgroups.add pa
+			pa = pa.parent
+		end
+	end
+	for g in mgroups do
+		var pa = g.parent
+		if g.is_interesting then
+			ot.add(pa, g)
 		end
 	end
-	ot.sort_with(new CachedAlphaComparator)
+	ot.sort_with(alpha_comparator)
 	ot.write_to(stdout)
 end
 
 if opt_source.value then
-	var list = new Array[String]
-	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}/{mp.name} ({mp.filepath})")
-				end
-			end
+	alpha_comparator.sort(mmodules)
+	for mp in mmodules do
+		if opt_make.value then
+			print mp.filepath.escape_to_mk
+		else if opt_paths.value then
+			print mp.filepath.as(not null)
+		else
+			print "{mp.mgroup.full_name}{ot.display(mp)}"
 		end
 	end
-	alpha_comparator.sort(list)
-	for l in list do print l
 end
 
-if opt_project.value then
-	var list = new Array[String]
-	for p in model.mprojects do
-		var path = p.root.filepath.as(not null)
+if opt_package.value then
+	var mpackages = new Array[MPackage]
+	for m in mmodules do
+		var p = m.mgroup.mpackage
+		if mpackages.has(p) then continue
+		mpackages.add p
+	end
+
+	alpha_comparator.sort(mpackages)
+	for p in mpackages do
+		var g = p.root.as(not null)
+		var path = g.filepath.as(not null)
 		if opt_paths.value then
-			list.add(path)
+			print path
 		else
-			list.add("{p.name} ({path})")
+			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
+				print "{g.name}{d} ({path})"
+			else
+				print "{g.name}{d} ({path.yellow})"
+			end
 		end
 	end
-	alpha_comparator.sort(list)
-	for l in list do print l
 end