Add posibility to exclude namespaces (#92)

SwissLife-OSS · Jan 26, 2024 · c83bf4e · c83bf4e
1 parent d64ffbc
commit c83bf4e
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 13 deletions.
diff --git a/src/Context.Tests/Internal/DependencyTypesResolverTests.cs b/src/Context.Tests/Internal/DependencyTypesResolverTests.cs
@@ -14,8 +14,7 @@ public void GetAllowedTypesByDependencies_All_Successful()
 
         // Act
         IEnumerable<string> knownNamespaces = DependencyTypesResolver
-            .GetAllowedTypesByDependencies()
-            .Except(new[] { "Coverlet" })
+            .GetAllowedTypesByDependencies(new[] { "Coverlet" })
             .OrderBy(x => x);
 
         // Assert

diff --git a/src/Context/IMongoDatabaseBuilder.cs b/src/Context/IMongoDatabaseBuilder.cs
@@ -34,6 +34,6 @@ public interface IMongoDatabaseBuilder
 
         IMongoDatabaseBuilder AddAllowedTypes(params string[] allowedNamespaces);
 
-        IMongoDatabaseBuilder AddAllowedTypesOfAllDependencies();
+        IMongoDatabaseBuilder AddAllowedTypesOfAllDependencies(params string[] excludeNamespaces);
     }
 }
diff --git a/src/Context/Internal/DependencyTypesResolver.cs b/src/Context/Internal/DependencyTypesResolver.cs
@@ -6,34 +6,34 @@
 
 namespace MongoDB.Extensions.Context;
 
-internal class DependencyTypesResolver
+internal static class DependencyTypesResolver
 {
     private static readonly HashSet<string> _notAllowedNames =
         new HashSet<string>{ "System", "Microsoft" };
 
-    public static HashSet<string> GetAllowedTypesByDependencies()
+    internal static HashSet<string> GetAllowedTypesByDependencies(string[] excludeNamespaces)
     {
         Assembly[] assemblies = AppDomain.CurrentDomain.GetAssemblies();
 
         IEnumerable<Assembly> allowedAssemblies = assemblies
-            .Where(assembly => IsNamespaceAllowed(assembly.GetName().Name));
+            .Where(assembly => IsNamespaceAllowed(assembly.GetName().Name, excludeNamespaces));
 
         IEnumerable<string> namespaces = allowedAssemblies
             .SelectMany(a => a.GetTypes())
             .Select(type => type.GetRootNamespace())
-            .Where(name => IsNamespaceAllowed(name));
+            .Where(name => IsNamespaceAllowed(name, excludeNamespaces));
 
         return new HashSet<string>(namespaces);
     }
 
-    private static bool IsNamespaceAllowed(string name)
+    private static bool IsNamespaceAllowed(string name, string [] excludeNamespaces)
     {
         if(string.IsNullOrEmpty(name))
         {
             return false;
         }
 
-        return !_notAllowedNames
+        return !_notAllowedNames.Concat(excludeNamespaces)
             .Any(entry => name.StartsWith(entry));
     }
 }
diff --git a/src/Context/Internal/MongoDatabaseBuilder.cs b/src/Context/Internal/MongoDatabaseBuilder.cs
@@ -71,9 +71,9 @@ public IMongoDatabaseBuilder AddAllowedTypes(params string[] allowedNamespaces)
         return this;
     }
 
-    public IMongoDatabaseBuilder AddAllowedTypesOfAllDependencies()
+    public IMongoDatabaseBuilder AddAllowedTypesOfAllDependencies(params string[] excludeNamespaces)
     {
-        TypeObjectSerializer.AddAllowedTypesOfAllDependencies();
+        TypeObjectSerializer.AddAllowedTypesOfAllDependencies(excludeNamespaces);
         return this;
     }
 

diff --git a/src/Context/TypeObjectSerializer.cs b/src/Context/TypeObjectSerializer.cs
@@ -55,10 +55,10 @@ public static void AddAllowedTypes(params string[] allowedNamespaces)
         }
     }
 
-    public static void AddAllowedTypesOfAllDependencies()
+    public static void AddAllowedTypesOfAllDependencies(params string[] excludeNamespaces)
     {
         _allowedTypesByDependencies
-            .UnionWith(DependencyTypesResolver.GetAllowedTypesByDependencies());
+            .UnionWith(DependencyTypesResolver.GetAllowedTypesByDependencies(excludeNamespaces));
     }
 
     internal static void Clear()