nitls: correctly sort by names with --tree

[nit.git] / src / nitls.nit

diff --git a/src/nitls.nit b/src/nitls.nit
index 64131b7..9bfafd7 100644 (file)
--- a/src/nitls.nit
+++ b/src/nitls.nit
@@ -53,6 +53,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 +61,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 +88,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")
@@ -143,13 +177,17 @@ if opt_tree.value then
        ot.opt_paths = opt_paths.value
        for p in model.mprojects do
                for g in p.mgroups do
-                       ot.add(g.parent, g)
+                       var pa = g.parent
+                       if g.is_interesting then
+                               ot.add(pa, g)
+                               pa = g
+                       end
                        for mp in g.module_paths do
-                               ot.add(g, mp)
+                               ot.add(pa, mp)
                        end
                end
        end
-       ot.sort_with(new CachedAlphaComparator)
+       ot.sort_with(new AlphaEntityComparator)
        ot.write_to(stdout)
 end