SPT-Server-Build/SPTarkov.Server/Modding/ModDllLoader.cs

using System.Reflection;
using System.Runtime.Loader;
using System.Text.Json;
using SPTarkov.Server.Core.Models.Spt.Mod;

namespace SPTarkov.Server.Modding;

public class ModDllLoader
{
    private const string ModPath = "./user/mods/";

    public static List<SptMod> LoadAllMods()
    {
        if (!Directory.Exists(ModPath))
        {
            Directory.CreateDirectory(ModPath);
        }

        var mods = new List<SptMod>();

        // foreach directory in /user/mods/
        // treat this as the MOD
        // should contain a dll and Package.json
        // load both
        // if either is missing Throw Warning and skip

        var modDirectories = Directory.GetDirectories(ModPath);

        // Load mods found in dir
        foreach (var modDirectory in modDirectories)
        {
            try
            {
                mods.Add(LoadMod(modDirectory));
            }
            catch (Exception e)
            {
                Console.WriteLine(e.Message);
            }
        }

        return mods;
    }

    /// <summary>
    /// Check the provided directory path for a dll and .json file, load into memory
    /// </summary>
    /// <param name="path">Directory path that contains mod files</param>
    /// <returns>SptMod</returns>
    private static SptMod LoadMod(string path)
    {
        var result = new SptMod
        {
            Directory = path,
            Assemblies = []
        };
        var asmCount = 0;
        var packCount = 0;
        foreach (var file in new DirectoryInfo(path).GetFiles()) // only search top level
        {
            if (file.Name.ToLower() == "package.json")
            {
                packCount++;

                // Handle package.json
                var rawJson = File.ReadAllText(file.FullName);
                result.PackageJson = JsonSerializer.Deserialize<PackageJsonData>(rawJson);
                if (packCount > 1)
                {
                    throw new Exception($"More than one package.json file found in path: {path}");
                }
            }

            if (file.Extension.ToLower() == ".dll")
            {
                asmCount++;
                result.Assemblies.Add(AssemblyLoadContext.Default.LoadFromAssemblyPath(Path.GetFullPath(file.FullName)));
            }
        }

        if (asmCount == 0 && packCount == 0)
        {
            throw new Exception($"No Assembly or package.json found in: {Path.GetFullPath(path)}");
        }

        if (packCount == 0)
        {
            throw new Exception($"No package.json found in path: {Path.GetFullPath(path)}");
        }

        if (asmCount == 0)
        {
            throw new Exception($"No Assemblies found in path: {Path.GetFullPath(path)}");
        }

        if (result.PackageJson?.Name == null || result.PackageJson?.Author == null ||
            result.PackageJson?.Version == null || result.PackageJson?.Licence == null ||
            result.PackageJson?.SptVersion == null)
        {
            throw new Exception($"The package.json file for {path} is missing one of these properties: name, author, licence, version or sptVersion");
        }

        if (result.Assemblies is not null && result.PackageJson is not null)
        {
            Console.WriteLine($"Loaded: {result.PackageJson.Name} Version: {result.PackageJson.Version} by: {result.PackageJson.Author}");
        }

        return result;
    }
}