From 11c7dd33fb1e4640ec57e510cd4fcbe432a76fdd Mon Sep 17 00:00:00 2001
From: Matthew Leibowitz <mattleibow@live.com>
Date: Wed, 1 Jun 2022 21:37:56 +0200
Subject: [PATCH] Add support for excluding "namespaces"

---
 utils/SkiaSharpGenerator/BaseTool.cs                  | 11 +++++++++++
 .../SkiaSharpGenerator/ConfigJson/NamespaceMapping.cs |  3 +++
 utils/SkiaSharpGenerator/Generate/Generator.cs        |  4 ++++
 3 files changed, 18 insertions(+)

diff --git a/utils/SkiaSharpGenerator/BaseTool.cs b/utils/SkiaSharpGenerator/BaseTool.cs
index 474dbfc29de..4c1b1ed2019 100644
--- a/utils/SkiaSharpGenerator/BaseTool.cs
+++ b/utils/SkiaSharpGenerator/BaseTool.cs
@@ -330,6 +330,17 @@ static string RemovePrefixSuffix(string member, string type)
 			}
 		}
 
+		protected bool IncludeNamespace(string name)
+		{
+			foreach (var ns in config.Namespaces)
+			{
+				if (name.StartsWith(ns.Key) && ns.Value.Exclude == true)
+					return false;
+			}
+
+			return true;
+		}
+
 		protected string GetNamespace(string name)
 		{
 			foreach (var ns in config.Namespaces)
diff --git a/utils/SkiaSharpGenerator/ConfigJson/NamespaceMapping.cs b/utils/SkiaSharpGenerator/ConfigJson/NamespaceMapping.cs
index d6589af3c33..3d867ff4cee 100644
--- a/utils/SkiaSharpGenerator/ConfigJson/NamespaceMapping.cs
+++ b/utils/SkiaSharpGenerator/ConfigJson/NamespaceMapping.cs
@@ -9,5 +9,8 @@ public class NamespaceMapping
 
 		[JsonPropertyName("prefix")]
 		public string? Prefix { get; set; }
+
+		[JsonPropertyName("exclude")]
+		public bool? Exclude { get; set; }
 	}
 }
diff --git a/utils/SkiaSharpGenerator/Generate/Generator.cs b/utils/SkiaSharpGenerator/Generate/Generator.cs
index 018bc1e03e2..cea06f3eb5a 100644
--- a/utils/SkiaSharpGenerator/Generate/Generator.cs
+++ b/utils/SkiaSharpGenerator/Generate/Generator.cs
@@ -72,6 +72,7 @@ private void WriteDelegates(TextWriter writer)
 
 			var delegates = compilation.Typedefs
 				.Where(t => t.ElementType.TypeKind == CppTypeKind.Pointer)
+				.Where(t => IncludeNamespace(t.GetDisplayName()))
 				.OrderBy(t => t.GetDisplayName())
 				.GroupBy(t => GetNamespace(t.GetDisplayName()));
 
@@ -149,6 +150,7 @@ private void WriteStructs(TextWriter writer)
 
 			var classes = compilation.Classes
 				.Where(c => c.SizeOf != 0)
+				.Where(c => IncludeNamespace(c.GetDisplayName()))
 				.OrderBy(c => c.GetDisplayName())
 				.GroupBy(c => GetNamespace(c.GetDisplayName()));
 
@@ -313,6 +315,7 @@ private void WriteEnums(TextWriter writer)
 			writer.WriteLine($"#region Enums");
 
 			var enums = compilation.Enums
+				.Where(e => IncludeNamespace(e.GetDisplayName()))
 				.OrderBy(e => e.GetDisplayName())
 				.GroupBy(e => GetNamespace(e.GetDisplayName()));
 
@@ -437,6 +440,7 @@ private void WriteFunctions(TextWriter writer)
 			Log?.LogVerbose("  Writing p/invokes...");
 
 			var functionGroups = compilation.Functions
+				.Where(f => IncludeNamespace(f.Name))
 				.OrderBy(f => f.Name)
 				.GroupBy(f => f.Span.Start.File.ToLower().Replace("\\", "/"))
 				.OrderBy(g => Path.GetDirectoryName(g.Key) + "/" + Path.GetFileName(g.Key));