Strip comments and whitespace from assembly export

MerlinVR · Mar 25, 2021 · 205e3ad · 205e3ad
1 parent f2de786
commit 205e3ad
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/Assets/UdonSharp/Editor/UdonSharpEditorUtility.cs b/Assets/UdonSharp/Editor/UdonSharpEditorUtility.cs
@@ -5,6 +5,7 @@
 using System.IO;
 using System.Linq;
 using System.Reflection;
+using System.Text;
 using UdonSharp;
 using UdonSharp.Serialization;
 using UnityEditor;
@@ -42,6 +43,25 @@ public static UdonAssemblyProgramAsset UdonSharpProgramToAssemblyProgram(UdonSha
 
             string programAssembly = UdonSharpEditorCache.Instance.GetUASMStr(udonSharpProgramAsset);
 
+            // Strip comments/inline code
+            StringBuilder asmBuilder = new StringBuilder();
+
+            using (StringReader reader = new StringReader(programAssembly))
+            {
+                string line = reader.ReadLine();
+
+                while (line != null)
+                {
+                    if (!string.IsNullOrWhiteSpace(line) &&
+                        !line.TrimStart().StartsWith("#", System.StringComparison.Ordinal))
+                        asmBuilder.AppendFormat("{0}\n", line);
+
+                    line = reader.ReadLine();
+                }
+            }
+
+            programAssembly = asmBuilder.ToString();
+
             FieldInfo assemblyField = typeof(UdonAssemblyProgramAsset).GetField("udonAssembly", BindingFlags.NonPublic | BindingFlags.Instance);
             assemblyField.SetValue(newProgramAsset, programAssembly);