Shrlalgo.RvKits/ShrlAlgoToolkit.RevitCore/Base/BaseApplication.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Reflection;
using System.Text;

using ShrlAlgoToolkit.RevitCore.Base;

namespace ShrlAlgoToolkit.RevitCore.Assists
{
    public abstract class BaseApplication : IExternalApplication
    {
        public UIApplication UiApplication { get; private set; }
       
        public Result OnStartup(UIControlledApplication application)
        {
            var flag = System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.InvokeMethod;

            UiApplication = (Autodesk.Revit.UI.UIApplication)application.GetType().InvokeMember("getUIApplication", flag, Type.DefaultBinder, application, null);


            // 1. 注册程序集依赖解析事件 (解决第三方库找不到的问题)
            AppDomain.CurrentDomain.AssemblyResolve += OnAssemblyResolve;

            // 2. 初始化全局异步外部事件处理器 (必须在主线程创建)
            ExternalEventHelper.Initialize();

            try
            {
                OnApplicationStartup();
                return Result.Succeeded;
            }
            catch (Exception ex)
            {
                TaskDialog.Show("Startup Error", ex.ToString());
                return Result.Failed;
            }
        }

        public Result OnShutdown(UIControlledApplication application)
        {
            try
            {
                OnApplicationShutdown();
                return Result.Succeeded;
            }
            finally
            {
                AppDomain.CurrentDomain.AssemblyResolve -= OnAssemblyResolve;
            }
        }

        // 供子类重写的方法
        public abstract void OnApplicationStartup();
        public virtual void OnApplicationShutdown() { }

        // 依赖自动解析逻辑
        private Assembly OnAssemblyResolve(object sender, ResolveEventArgs args)
        {
            var assemblyName = new AssemblyName(args.Name).Name;
            var pluginDirectory = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
            var assemblyPath = Path.Combine(pluginDirectory, $"{assemblyName}.dll");

            return File.Exists(assemblyPath) ? Assembly.LoadFrom(assemblyPath) : null;
        }
    }
}