Some initial CSharp scripting functionality

2022-04-09 20:47:55 +03:00
parent 86a88c6e20
commit 429557ad7d
12 changed files with 466 additions and 1 deletions
@@ -0,0 +1,175 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using Microsoft.CodeAnalysis.Scripting;
+using System.Reflection;
+using Microsoft.CodeAnalysis.CSharp;
+using System.Linq;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+using Microsoft.CodeAnalysis;
+using System.Collections.Immutable;
+using System.Runtime.Loader;
+using static NetScript;
+using Microsoft.CodeAnalysis.Emit;
+using System.Reflection.PortableExecutable;
+using System.Reflection.Metadata;
+
+namespace Barotrauma
+{
+	partial class NetSetup
+	{
+
+		public class NetScriptLoader : AssemblyLoadContext
+		{
+			public NetSetup net;
+			private List<MetadataReference> defaultReferences;
+			private List<SyntaxTree> syntaxTrees;
+			public Assembly Assembly { get; private set; }
+
+			public NetScriptLoader(NetSetup net)
+			{
+				this.net = net;
+
+				defaultReferences = AppDomain.CurrentDomain.GetAssemblies()
+					.Where(a => !(a.IsDynamic || string.IsNullOrEmpty(a.Location) || a.Location.Contains("xunit")))
+					.Select(a => MetadataReference.CreateFromFile(a.Location) as MetadataReference)
+					.ToList();
+
+				syntaxTrees = new List<SyntaxTree>();
+				Assembly = null;
+			}
+
+			public void SearchFolders()
+            {
+				foreach(ContentPackage cp in GameMain.Config.AllEnabledPackages)
+                {
+					var path = Path.GetDirectoryName(cp.Path);
+					RunFolder(path);
+                }
+			}
+
+            public void RunFolder(string folder)
+			{
+				var scriptFiles = new List<string>();
+				foreach (var str in DirSearch(folder))
+				{
+					var s = str.Replace("\\", "/");
+
+					if (s.EndsWith(".cs"))
+					{
+						NetSetup.PrintMessage(s);
+						scriptFiles.Add(s);
+					}
+				}
+
+				try
+				{
+					if (scriptFiles.Count <= 0) return;
+
+					var mainFile = scriptFiles.Find(s => s.EndsWith("Main.cs"));
+					if (mainFile == null) throw new Exception("Mod folder has no Main.cs file");
+					scriptFiles.Remove(mainFile);
+					scriptFiles.Add(mainFile);
+
+					// Check file content for prohibited stuff
+					foreach (var file in scriptFiles)
+					{
+						var tree = SyntaxFactory.ParseSyntaxTree(File.ReadAllText(file), CSharpParseOptions.Default, file);
+						var error = NetScriptFilter.FilterSyntaxTree(tree as CSharpSyntaxTree);
+						if (error != null) throw new Exception(error);
+
+						syntaxTrees.Add(tree);
+					}
+                }
+				catch (CompilationErrorException ex)
+				{
+					string errStr = "Cmopilation Error in '" + folder + "':";
+					foreach (var diag in ex.Diagnostics)
+					{
+						errStr += "\n" + diag.ToString();
+					}
+					NetSetup.PrintMessage(errStr);
+				}
+				catch (Exception ex)
+                {
+					NetSetup.PrintMessage("Error loading '" + folder + "':\n" + ex.Message + "\n" + ex.StackTrace);
+				}
+			}
+
+            public List<Type> Compile()
+            {
+				var options = new CSharpCompilationOptions(OutputKind.DynamicallyLinkedLibrary)
+					.WithMetadataImportOptions(MetadataImportOptions.All)
+					.WithOptimizationLevel(OptimizationLevel.Release)
+					.WithAllowUnsafe(false);
+				var compilation = CSharpCompilation.Create("NetScriptAssembly",syntaxTrees, defaultReferences, options);
+
+				using (var mem = new MemoryStream())
+				{
+					var result = compilation.Emit(mem);
+					if (!result.Success)
+					{
+						IEnumerable<Diagnostic> failures = result.Diagnostics.Where(d => d.IsWarningAsError || d.Severity == DiagnosticSeverity.Error);
+
+						string errStr = "NET MODS NOT LOADED | Mod cmopilation errors:";
+						foreach (Diagnostic diagnostic in failures)
+						{
+							errStr = $"\n{diagnostic}";
+						}
+						NetSetup.PrintMessage(errStr);
+					}
+					else
+					{
+						mem.Seek(0, SeekOrigin.Begin);
+						var reader = new PEReader(mem);
+						var mdReader = reader.GetMetadataReader();
+						mdReader.AssemblyReferences.ToList().ForEach(a =>
+						{
+							var aRef = mdReader.GetAssemblyReference(a);
+							Console.WriteLine(aRef.GetAssemblyName() + " " + aRef.Version);
+						});
+						Console.WriteLine();
+						mdReader.TypeReferences.ToList().ForEach(t =>
+						{
+							var tRef = mdReader.GetTypeReference(t);
+							Console.WriteLine(mdReader.GetString(tRef.Namespace) + "  -  " + mdReader.GetString(tRef.Name));
+						});
+						mem.Seek(0, SeekOrigin.Begin);
+						Assembly = LoadFromStream(mem);
+					}
+				}
+				syntaxTrees.Clear();
+
+				if (Assembly != null)
+					return Assembly.GetTypes().Where(t => t.IsSubclassOf(typeof(ANetMod))).ToList();
+				else
+					throw new Exception("Unable to create net mods assembly.");
+			}
+
+			static string[] DirSearch(string sDir)
+			{
+				List<string> files = new List<string>();
+
+				try
+				{
+					foreach (string f in Directory.GetFiles(sDir))
+					{
+						files.Add(f);
+					}
+
+					foreach (string d in Directory.GetDirectories(sDir))
+					{
+						files.AddRange(DirSearch(d));
+					}
+				}
+				catch (System.Exception excpt)
+				{
+					Console.WriteLine(excpt.Message);
+				}
+
+				return files.ToArray();
+			}
+
+		}
+	}
+}