Add Roslyn native preprocessor directive support

Author: austinv900

src/Compilation.cs

@@ -7,7 +7,6 @@
 using System.IO;
 using System.Linq;
 using System.Reflection;
-using System.Runtime.InteropServices.ComTypes;
 using System.Text.RegularExpressions;
 using System.Threading;
 
@@ -16,7 +15,6 @@ namespace Oxide.Plugins
     internal class Compilation
     {
         public static Compilation Current;
-        private static readonly Regex SymbolEscapeRegex = new Regex(@"[^\w\d]", RegexOptions.Compiled);
 
         internal int id;
         internal string name;
@@ -32,9 +30,6 @@ internal class Compilation
 
         private string includePath;
         private string[] extensionNames;
-        private string gameExtensionNamespace;
-        private readonly string gameExtensionName;
-        private readonly string gameExtensionBranch;
 
         internal Compilation(int id, Action<Compilation> callback, CompilablePlugin[] plugins)
         {
@@ -55,10 +50,6 @@ internal Compilation(int id, Action<Compilation> callback, CompilablePlugin[] pl
 
             includePath = Path.Combine(Interface.Oxide.PluginDirectory, "include");
             extensionNames = Interface.Oxide.GetAllExtensions().Select(ext => ext.Name).ToArray();
-            Core.Extensions.Extension gameExtension = Interface.Oxide.GetAllExtensions().SingleOrDefault(ext => ext.IsGameExtension);
-            gameExtensionName = gameExtension?.Name.ToUpper();
-            gameExtensionNamespace = gameExtension?.GetType().Namespace;
-            gameExtensionBranch = gameExtension?.Branch?.ToUpper();
         }
 
         internal void Started()
@@ -488,18 +479,6 @@ private bool CacheScriptLines(CompilablePlugin plugin)
                                 lines.Add(reader.ReadLine());
                             }
 
-                            if (!string.IsNullOrEmpty(gameExtensionName))
-                            {
-                                string gameExtensionNameEscaped = EscapeSymbolName(gameExtensionName);
-                                lines.Insert(0, $"#define {gameExtensionNameEscaped}");
-
-                                if (!string.IsNullOrEmpty(gameExtensionBranch) && gameExtensionBranch != "public")
-                                {
-                                    string gameExtensionBranchEscaped = EscapeSymbolName(gameExtensionBranch);
-                                    lines.Insert(0, $"#define {gameExtensionNameEscaped}{gameExtensionBranchEscaped}");
-                                }
-                            }
-
                             plugin.ScriptLines = lines.ToArray();
                             plugin.ScriptEncoding = reader.CurrentEncoding;
                         }
@@ -560,15 +539,5 @@ private void RemovePlugin(CompilablePlugin plugin)
                 }
             }
         }
-
-        /// <summary>
-        /// This allows to handle cases where injected symbols have inappropriate characters (e.g. git branches can have "/", "-" etc. while #define disallows them)
-        /// </summary>
-        /// <param name="name"></param>
-        /// <returns></returns>
-        private string EscapeSymbolName(string name)
-        {
-            return SymbolEscapeRegex.Replace(name, "_");
-        }
     }
 }

src/CompilerService.cs

@@ -24,6 +24,7 @@ namespace Oxide.CSharp
 {
     internal class CompilerService
     {
+        private static readonly Regex SymbolEscapeRegex = new Regex(@"[^\w\d]", RegexOptions.Compiled);
         private const string baseUrl = "https://downloads.oxidemod.com/artifacts/Oxide.Compiler/{0}/";
         private Hash<int, Compilation> compilations;
         private Queue<CompilerMessage> messageQueue;
@@ -38,6 +39,7 @@ internal class CompilerService
         private static Regex fileErrorRegex = new Regex(@"^\[(?'Severity'\S+)\]\[(?'Code'\S+)\]\[(?'File'\S+)\] (?'Message'.+)$", RegexOptions.Compiled);
         public bool Installed => File.Exists(filePath);
         private float startTime;
+        private string[] preprocessor = null;
 
         public CompilerService(Extension extension)
         {
@@ -97,16 +99,49 @@ private void ExpireFileCache()
                 }
 
                 ArrayPool.Free(toRemove);
-
-                if (index <= 0) return;
-
-                //Interface.Oxide.LogWarning($"[CSharp] Released {index} cached compiler dependencies, running garbage collection. . .");
-                GC.Collect();
             }
         }
 
         internal bool Precheck()
         {
+            List<string> preprocessorList = new List<string>()
+            {
+                "OXIDE",
+                "OXIDEMOD"
+            };
+
+            Extension game = Interface.Oxide.GetAllExtensions().SingleOrDefault(e => e.IsGameExtension);
+
+            if (game != null)
+            {
+                string name = game.Name.ToUpperInvariant();
+                string branch = game.Branch?.ToUpperInvariant() ?? "PUBLIC";
+                preprocessorList.Add(EscapeSymbolName(name));
+                preprocessorList.Add(EscapeSymbolName(name + "_" + branch));
+
+                if (game.Version != default)
+                {
+                    preprocessorList.Add(EscapeSymbolName(name + "_" + game.Version));
+                    preprocessorList.Add(EscapeSymbolName(name + "_" + game.Version + "_" + branch));
+                }
+            }
+
+#if DEBUG
+            preprocessorList.Add("DEBUG");
+#endif
+
+            if (Interface.Oxide.Config.Compiler.PreprocessorDirectives.Count > 0)
+            {
+                preprocessorList.AddRange(Interface.Oxide.Config.Compiler.PreprocessorDirectives);
+            }
+
+            preprocessor = preprocessorList.Distinct().ToArray();
+
+#if DEBUG
+            Log(LogType.Debug, $"Preprocessors are: {string.Join(", ", preprocessor)}");
+#endif
+
+
             if (!DownloadFile(remoteName, filePath, 3))
             {
                 return false;
@@ -511,12 +546,13 @@ private void EnqueueCompilation(Compilation compilation)
             {
                 OutputFile = compilation.name,
                 SourceFiles = sourceFiles.ToArray(),
-                ReferenceFiles = compilation.references.Values.ToArray()
+                ReferenceFiles = compilation.references.Values.ToArray(),
+                Preprocessor = preprocessor
                 #if DEBUG
                 , Debug = true
                 #endif
             };
-
+            
             CompilerMessage message = new CompilerMessage { Id = compilation.id, Data = data, Type = CompilerMessageType.Compile };
             if (ready)
             {
@@ -779,5 +815,15 @@ private static string GenerateFileHash(string file)
                 return BitConverter.ToString(md5.ComputeHash(stream)).Replace("-", string.Empty).ToLowerInvariant();
             }
         }
+
+        /// <summary>
+        /// This allows to handle cases where injected symbols have inappropriate characters (e.g. git branches can have "/", "-" etc. while #define disallows them)
+        /// </summary>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        private string EscapeSymbolName(string name)
+        {
+            return SymbolEscapeRegex.Replace(name, "_");
+        }
     }
 }

src/ObjectStream/Data/CompilerData.cs

@@ -28,5 +28,6 @@ public CompilerData()
         public CompilerLanguageVersion Version { get; set; }
         public string Encoding { get; set; }
         public bool Debug { get; set; }
+        public string[] Preprocessor { get; set; }
     }
 }