Allow custom namespace grouping

src/Lively.Diagrams.Tests/FullNamePlantUmlTests.cs

@@ -145,6 +145,31 @@ class Lively.Diagrams.Tests.FullNamePlantUmlTests.ExampleImplementationWithMetho
 
 Lively.Diagrams.Tests.FullNamePlantUmlTests.ExampleInterfaceWithMethods <--- Lively.Diagrams.Tests.FullNamePlantUmlTests.ExampleImplementationWithMethods
 
+@enduml";
+
+            Assert.Equal(Normalise(expected), Normalise(diagram));
+        }
+
+        [Fact]
+        public void CanApplyCustomNamespaceGrouping()
+        {
+            var assembly = this.GetType().Assembly;
+            var config = new DependencyTreeConfig(assembly);
+            var fullTypeName = "Lively.Diagrams.Tests.FullNamePlantUmlTests+ExampleTypeReferencingOtherAssemblies";
+
+            var tree = new DependencyTree(config);
+            var depTree = tree.GetDependencies(fullTypeName);
+            var diagram = FullNamePlantUml.Create(new[] { depTree }, new[] { "Lively.Diagrams.Tests", "Lively.Diagrams" });
+
+            var expected = @"@startuml
+
+class Lively.Diagrams.Tests.FullNamePlantUmlTests_ExampleTypeReferencingOtherAssemblies {
+}
+class Lively.Diagrams.FullNamePlantUml {
+}
+
+Lively.Diagrams.Tests.FullNamePlantUmlTests_ExampleTypeReferencingOtherAssemblies ---> Lively.Diagrams.FullNamePlantUml
+
 @enduml";
 
             Assert.Equal(Normalise(expected), Normalise(diagram));
@@ -207,6 +232,11 @@ public void One() { }
             public string Beans() => "hello";
         }
 
+        public class ExampleTypeReferencingOtherAssemblies
+        {
+            public ExampleTypeReferencingOtherAssemblies(FullNamePlantUml fullNamePlantUml) { }
+        }
+
         public class Startup
         {
             public void ConfigureServices(IServiceCollection services)

src/Lively.Diagrams/FullNamePlantUml.cs

@@ -1,14 +1,31 @@
 using System.Collections.Generic;
-using Lively.TypeDescriptions;
 
 namespace Lively.Diagrams
 {
     public class FullNamePlantUml
     {
-        public static string Create(IList<DependencyTreeNode> nodes)
+        public static string Create(IList<DependencyTreeNode> nodes, string[] namespaceMatches = null)
         {
-            var renderer = new PlantUmlRenderer(type => type.FullName);
-            return renderer.Create(nodes); 
+            var renderer = new PlantUmlRenderer(type =>
+            {
+                if (namespaceMatches != null)
+                {
+                    foreach (var ns in namespaceMatches)
+                    {
+                        if (type.FullName.StartsWith(ns + "."))
+                            return OverrideNamespaceGrouping(ns, type.FullName);
+                    }
+                }
+
+                return type.FullName;
+            });
+            return renderer.Create(nodes);
+        }
+
+        private static string OverrideNamespaceGrouping(string @namespace, string fullName)
+        {
+            var end = fullName.Remove(0, @namespace.Length + 1);
+            return $"{@namespace}.{end.Replace('.', '_').Replace('+', '_')}";
         }
     }
 }

src/Lively.Diagrams/PlantUml.cs

@@ -1,5 +1,4 @@
 using System.Collections.Generic;
-using Lively.TypeDescriptions;
 
 namespace Lively.Diagrams
 {