DialogBoxResolver to cancel dialog box before revit initialization

CHANGELOG.md

@@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 ## [1.4.0] / 2024-09-05
 ### Features
 - Support `NUnit.Timeout` option to abort application. (default '10' minutes) (Fix: #51)
+- Autoclose `VIEWER MODE` warning dialog box when Revit starts. (Fix: #52)
+### Application
+- Add `DialogBoxResolver` to cancel dialog box before revit initialization.
 ### Console
 - Change `TimeoutMinutesDefault` to 10 minutes in the `RevitTestUtils`.
 - Remove deprecated `wait` option and `RevitDebugUtils`.

Directory.Build.props

@@ -1,5 +1,5 @@
 <Project>
 	<PropertyGroup>
-		<Version>1.4.0-alpha.1</Version>
+		<Version>1.4.0-beta</Version>
 	</PropertyGroup>
 </Project>

ricaun.RevitTest.Application/Revit/App.cs

@@ -23,6 +23,7 @@ public class App : IExternalApplication
         private static PipeTestServer PipeTestServer;
         private static RevitTaskService RevitTask;
         private static RevitBusyService RevitBusyService;
+        private static DialogBoxResolver DialogBoxResolver;
         private static bool IsTestRunning = false;
 
         private const int TestThreadSleepMin = 50;
@@ -38,6 +39,9 @@ public Result OnStartup(UIControlledApplication application)
             RevitTask = new RevitTaskService(application);
             RevitTask.Initialize();
 
+            DialogBoxResolver = new DialogBoxResolver(application);
+            DialogBoxResolver.Initialize();
+
             Log.WriteLine();
             Log.WriteLine($"{AppUtils.GetInfo()}");
 
@@ -219,9 +223,10 @@ public Result OnShutdown(UIControlledApplication application)
             });
 
             ribbonPanel?.Remove();
+
             PipeTestServer?.Dispose();
             RevitBusyService?.Dispose();
-
+            DialogBoxResolver?.Dispose();
             RevitTask?.Dispose();
 
             Log.Finish();

ricaun.RevitTest.Application/Revit/Utils/DialogBoxResolver.cs

@@ -0,0 +1,64 @@
+using Autodesk.Revit.UI;
+using System;
+
+namespace ricaun.RevitTest.Application.Revit.Utils
+{
+    /// <summary>
+    /// DialogBoxResolver resolve the dialog box by DialogId and override the result to cancel.
+    /// </summary>
+    /// <remarks>The resolver only works before Revit initialize.</remarks>
+    public class DialogBoxResolver : IDisposable
+    {
+        private readonly UIControlledApplication application;
+        private readonly string[] TaskDialogIds;
+        public DialogBoxResolver(UIControlledApplication application)
+        {
+            this.application = application;
+            this.TaskDialogIds = new string[]
+            {
+                "TaskDialog_License_Current_Status_Demo_Mode",  // Viewer Mode
+                "TaskDialog_DNSMTemplate"                       // Revit Warning
+            };
+        }
+
+        /// <summary>
+        /// Initialize the events to resolve the dialog box. 
+        /// </summary>
+        /// <remarks>The Initialize is ignored if the <see cref="Autodesk.Revit.UI.UIControlledApplication.IsLateAddinLoading"/>.</remarks>
+        public virtual void Initialize()
+        {
+            if (this.application.IsLateAddinLoading) return;
+            Dispose();
+            this.application.DialogBoxShowing += OnDialogBoxShowing;
+            this.application.ControlledApplication.ApplicationInitialized += OnApplicationInitialized;
+        }
+
+        /// <summary>
+        /// Dispose the events.
+        /// </summary>
+        public virtual void Dispose()
+        {
+            this.application.DialogBoxShowing -= OnDialogBoxShowing;
+            this.application.ControlledApplication.ApplicationInitialized -= OnApplicationInitialized;
+        }
+
+        private void OnDialogBoxShowing(object sender, Autodesk.Revit.UI.Events.DialogBoxShowingEventArgs e)
+        {
+            if (TaskDialogIds is null) return;
+            foreach (var dialogId in TaskDialogIds)
+            {
+                if (dialogId == e.DialogId)
+                {
+                    Console.WriteLine($"{nameof(DialogBoxResolver)}: [{e.DialogId}]");
+                    e.OverrideResult((int)TaskDialogResult.Cancel);
+                    return;
+                }
+            }
+        }
+        private void OnApplicationInitialized(object sender, Autodesk.Revit.DB.Events.ApplicationInitializedEventArgs e)
+        {
+            Dispose();
+        }
+    }
+
+}