Minecraft-Console-Client/MinecraftClient/Scripting/DynamicRun/Builder/CompileRunner.cs

/*
MIT License
Copyright (c) 2019 Laurent Kempé
https://github.com/laurentkempe/DynamicRun/blob/master/LICENSE
*/

using System;
using System.Collections.Generic;
using System.IO;
using System.Runtime.CompilerServices;
using MinecraftClient;

namespace DynamicRun.Builder
{
    internal class CompileRunner
    {
        public object? Execute(byte[] compiledAssembly, string[] args, Dictionary<string, object> localVars, ChatBot apiHandler)
        {
            var assemblyLoadContextWeakRef = LoadAndExecute(compiledAssembly, args, localVars, apiHandler);

            for (var i = 0; i < 8 && assemblyLoadContextWeakRef.Item1.IsAlive; i++)
            {
                GC.Collect();
                GC.WaitForPendingFinalizers();
            }

            ConsoleIO.WriteLogLine(assemblyLoadContextWeakRef.Item1.IsAlive ? "Script failed to clean-up" : "Script finished!");
            return assemblyLoadContextWeakRef.Item2;
        }

        [MethodImpl(MethodImplOptions.NoInlining)]
        private static Tuple<WeakReference, object?> LoadAndExecute(byte[] compiledAssembly, string[] args, Dictionary<string, object> localVars, ChatBot apiHandler)
        {
            using (var asm = new MemoryStream(compiledAssembly))
            {
                var assemblyLoadContext = new SimpleUnloadableAssemblyLoadContext();

                var assembly = assemblyLoadContext.LoadFromStream(asm);
                var compiledScript = assembly.CreateInstance("ScriptLoader.Script");
                var execResult  = compiledScript.GetType().GetMethod("__run").Invoke(compiledScript, new object[] { new CSharpAPI(apiHandler, localVars), args });
                
                assemblyLoadContext.Unload();

                return new (new WeakReference(assemblyLoadContext), execResult);
            }
        }
    }
}