kev/Drawer/UCDraw/GeoSigmaDrawLib/NativeLibraryPathRegistrar.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;

namespace GeoSigmaDrawLib
{
    /// <summary>
    /// 负责为本应用程序注册本地原生库（DLL）的搜索路径。
    /// 此类不加载任何 DLL，仅通过 Windows API 设置系统搜索目录，
    /// 确保后续通过 P/Invoke 或 DllImport 加载的库能被正确找到。
    /// 
    /// 注意：必须在首次调用任何 GeoSigmaLib 的 P/Invoke 方法前调用此方法。
    /// </summary>
    public static class NativeLibraryPathRegistrar
    {
        private const string VtkDirectory = "vtk";

        private const string ThirdPartyDirectory = "thirdparty";
        private const string EnvPath = "PATH";

        /// <summary>
        /// 注册三方库目录
        /// </summary>
        public static void Register()
        {
            var baseDirectory = GetExecutableDirectory();

            AddToEnvironmentPath(Path.Combine(baseDirectory, VtkDirectory));
            AddToEnvironmentPath(Path.Combine(baseDirectory, ThirdPartyDirectory));
        }

        /// <summary>
        /// 添加目录到 Path 环境变量中
        /// </summary>
        /// <param name="directory">要添加的目录名，建议使用绝对路径</param>
        public static void AddToEnvironmentPath(string directory)
        {
            if (string.IsNullOrWhiteSpace(directory))
            {
                return;
            }

            string? current = Environment.GetEnvironmentVariable(EnvPath);
            string newPath = AppendToPath(current, directory);
            Environment.SetEnvironmentVariable(EnvPath, newPath);
        }

        public static string GetExecutableDirectory()
        {
            var executablePath = Process.GetCurrentProcess().MainModule?.FileName;
            if (string.IsNullOrEmpty(executablePath))
            {
                throw new InvalidOperationException("无法获取当前进程的可执行文件路径。");
            }

            if(string.IsNullOrWhiteSpace(executablePath))
            {
                executablePath = AppDomain.CurrentDomain.BaseDirectory;
                return executablePath;
            }

            return Path.GetDirectoryName(executablePath) ?? throw new InvalidOperationException("可执行文件路径无效。");
        }

        private static string AppendToPath(string? current, string directory)
        {
            if (string.IsNullOrWhiteSpace(directory))
            {
                return current ?? string.Empty;
            }

            if (string.IsNullOrWhiteSpace(current))
            {
                return directory;
            }

            char pathSeparator = Path.PathSeparator;
            var pathEntries = current.Split(pathSeparator)
                   .Where(p => !string.IsNullOrWhiteSpace(p)) // 过滤空项
                   .ToList();

            if (pathEntries.Contains(directory, StringComparer.OrdinalIgnoreCase))
            {
                return current;
            }

            pathEntries.Add(directory);
            return string.Join(pathSeparator, pathEntries);
        }
    }
}