// only works in .NET Core. disable in .NET framework

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Runtime.InteropServices;
using System.Xml.Linq;

public static class DllMap
{
    [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    [return: MarshalAs(UnmanagedType.Bool)]
    static extern bool SetDefaultDllDirectories(int directoryFlags);

    [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern void AddDllDirectory(string lpPathName);

    [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    [return: MarshalAs(UnmanagedType.Bool)]
    static extern bool SetDllDirectory(string lpPathName);

    const int LOAD_LIBRARY_SEARCH_DEFAULT_DIRS = 0x00001000;

    public static Dictionary<string, string> MapDictionary;
    public static string OS;
    public static string CPU;
    public static bool Optimise;

    public static void Initialise(bool optimise = true)
    {
        Optimise = optimise;

        // Our executabe needs to know how to find the native libraries
        // For Windows, we can set this to be x86 or x64 directory at runtime (below)
        // For Linux we need to move our native libraries to 'netcoredeps' which is set by .net core
        // For OSX we need to set an environment variable (DYLD_LIBRARY_PATH) outside of the process by a script
        if (Environment.OSVersion.Platform == PlatformID.Win32NT)
        {
            try
            {
                SetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_DEFAULT_DIRS);
                AddDllDirectory(Path.Combine(
                    AppDomain.CurrentDomain.BaseDirectory,
                    Environment.Is64BitProcess ? "x64" : "x86"
                ));
            }
            catch
            {
                // Pre-Windows 7, KB2533623
                SetDllDirectory(Path.Combine(
                    AppDomain.CurrentDomain.BaseDirectory,
                    Environment.Is64BitProcess ? "x64" : "x86"
                ));
            }
        }

        // .NET Core also doesn't use DllImport but we can replicate this using NativeLibrary as per below
        // Uses FNA.dll.config to dictate what the name of the native library is per platform and architecture
        var fnaAssembly = Assembly.GetAssembly(typeof(Microsoft.Xna.Framework.Graphics.ColorWriteChannels));
        DllMap.Register(fnaAssembly);
    }

    // Register a call-back for native library resolution.
    public static void Register(Assembly assembly)
    {
        NativeLibrary.SetDllImportResolver(assembly, MapAndLoad);

        // Do setup so that MapLibraryName is faster than reading the XML each time

        // 1) Get platform & cpu
        OS = GetCurrentPlatform().ToString().ToLowerInvariant();
        CPU = GetCurrentRuntimeArchitecture().ToString().ToLowerInvariant();

        // 2) Setup MapDictionary
        // For Windows use hardcoded values
        // Why?  This is our development platform and we wanted the fastest start time possible (eliminates XML Load)
        if (OS == "windows" && Optimise)
        {
            MapDictionary = new Dictionary<string, string>();
            MapDictionary.Add("SDL2", "SDL2.dll");
            MapDictionary.Add("SDL_image", "SDL_image.dll");
            MapDictionary.Add("FAudio", "FAudio.dll");
        }
        else
        {
            // For every other platform use XML file
            // Read in config XML and only store details we're interested in within MapDictionary
            string xmlPath = Path.Combine(Path.GetDirectoryName(assembly.Location),
               Path.GetFileNameWithoutExtension(assembly.Location) + ".dll.config");

            if (!File.Exists(xmlPath))
            {
                Console.WriteLine($"=== Cannot find XML: " + xmlPath);
                return;
            }

            XElement root = XElement.Load(xmlPath);

            MapDictionary = new Dictionary<string, string>();
            ParseXml(root, true); // Direct match on OS & CPU first
            ParseXml(root, false); // Loose match on CPU second (won't allow duplicates)
        }
    }

    private static void ParseXml(XElement root, bool matchCPU)
    {
        foreach (var el in root.Elements("dllmap"))
        {
            // Ignore entries for other OSs
            if (el.Attribute("os").ToString().IndexOf(OS) < 0)
                continue;

            // Ignore entries for other CPUs
            if (matchCPU)
            {
                if (el.Attribute("cpu") == null)
                    continue;
                if (el.Attribute("cpu").ToString().IndexOf(CPU) < 0)
                    continue;
            }
            else
            {
                if (el.Attribute("cpu") != null && el.Attribute("cpu").ToString().IndexOf(CPU) < 0)
                    continue;
            }

            string oldLib = el.Attribute("dll").Value;
            string newLib = el.Attribute("target").Value;
            if (string.IsNullOrWhiteSpace(oldLib) || string.IsNullOrWhiteSpace(newLib))
                continue;

            // Don't allow duplicates
            if (MapDictionary.ContainsKey(oldLib))
                continue;

            MapDictionary.Add(oldLib, newLib);
        }
    }

    // The callback: which loads the mapped libray in place of the original
    private static IntPtr MapAndLoad(string libraryName, Assembly assembly, DllImportSearchPath? dllImportSearchPath)
    {
        string mappedName = null;
        mappedName = MapLibraryName(assembly.Location, libraryName, out mappedName) ? mappedName : libraryName;
        return NativeLibrary.Load(mappedName, assembly, dllImportSearchPath);
    }

    // Parse the assembly.xml file, and map the old name to the new name of a library.
    private static bool MapLibraryName(string assemblyLocation, string originalLibName, out string mappedLibName)
    {
        if (MapDictionary.TryGetValue(originalLibName, out mappedLibName))
            return true;
        else
            return false;
    }

    // Below pinched from Mono.DllMap project:   https://github.com/Firwood-Software/AdvancedDLSupport/tree/1b7394211a655b2f77649ce3b610a3161215cbdc/Mono.DllMap
    public static DllMapOS GetCurrentPlatform()
    {
        if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
        {
            return DllMapOS.Linux;
        }

        if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
        {
            return DllMapOS.Windows;
        }

        if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
        {
            return DllMapOS.OSX;
        }

        var operatingDesc = RuntimeInformation.OSDescription.ToUpperInvariant();
        foreach (var system in Enum.GetValues(typeof(DllMapOS)).Cast<DllMapOS>()
            .Except(new[] { DllMapOS.Linux, DllMapOS.Windows, DllMapOS.OSX }))
        {
            if (operatingDesc.Contains(system.ToString().ToUpperInvariant()))
            {
                return system;
            }
        }

        throw new PlatformNotSupportedException($"Couldn't detect platform: {RuntimeInformation.OSDescription}");
    }

    public static DllMapArchitecture GetCurrentRuntimeArchitecture()
    {
        switch (RuntimeInformation.ProcessArchitecture)
        {
            case Architecture.Arm:
                {
                    return DllMapArchitecture.ARM;
                }
            case Architecture.X64:
                {
                    return DllMapArchitecture.x86_64;
                }
            case Architecture.X86:
                {
                    return DllMapArchitecture.x86;
                }
        }

        typeof(object).Module.GetPEKind(out _, out var machine);
        switch (machine)
        {
            case ImageFileMachine.I386:
                {
                    return DllMapArchitecture.x86;
                }
            case ImageFileMachine.AMD64:
                {
                    return DllMapArchitecture.x86_64;
                }
            case ImageFileMachine.ARM:
                {
                    return DllMapArchitecture.ARM;
                }
            case ImageFileMachine.IA64:
                {
                    return DllMapArchitecture.IA64;
                }
        }
        throw new PlatformNotSupportedException("Couldn't detect the current architecture.");
    }

    public enum DllMapOS
    {
        Linux = 1 << 0,
        OSX = 1 << 1,
        Solaris = 1 << 2,
        FreeBSD = 1 << 3,
        OpenBSD = 1 << 4,
        NetBSD = 1 << 5,
        Windows = 1 << 6,
        AIX = 1 << 7,
        HPUX = 1 << 8
    }

    public enum DllMapArchitecture
    {
        x86 = 1 << 0,
        x86_64 = 1 << 1,
        SPARC = 1 << 2,
        PPC = 1 << 3,
        S390 = 1 << 4,
        S390X = 1 << 5,
        ARM = 1 << 6,
        ARMV8 = 1 << 7,
        MIPS = 1 << 8,
        Alpha = 1 << 9,
        HPPA = 1 << 10,
        IA64 = 1 << 11
    }
}