fix: VisualizationServerController start

Author: trgiangv

source/RevitDevTool/Application.cs

@@ -12,6 +12,12 @@ public override void OnStartup()
         ExternalEventController.Register();
         AddButton(Application);
         AddDockable(Application);
+        VisualizationServerController.Start();
+    }
+    
+    public override void OnShutdown()
+    {
+        VisualizationServerController.Stop();
     }
 
     private static void AddDockable(UIControlledApplication application)

source/RevitDevTool/Geometry/TraceGeometry.cs

@@ -0,0 +1,56 @@
+using System.Diagnostics;
+
+namespace RevitDevTool.Models;
+
+public static class TraceGeometry
+{
+    public static readonly TraceListener TraceListener = new TraceGeometryListener();
+
+    private static void Trace(object geometryObject)
+    {
+        VisualizationServerController.Add(geometryObject);
+    }
+
+    private static void Trace(IEnumerable<object> geometries)
+    {
+        VisualizationServerController.Add(geometries);
+    }
+
+    private class TraceGeometryListener : TraceListener
+    {
+        public override void Write(object? o)
+        {
+            switch (o)
+            {
+                case IEnumerable<GeometryObject> geometries:
+                    Trace(geometries);
+                    break;
+                case GeometryObject geometryObject:
+                    Trace(geometryObject);
+                    break;
+                case XYZ xyz:
+                    Trace(xyz);
+                    break;
+                case BoundingBoxXYZ boundingBoxXyz:
+                    Trace(boundingBoxXyz);
+                    break;
+                case IEnumerable<XYZ> xyzs:
+                    Trace(xyzs);
+                    break;
+                case IEnumerable<BoundingBoxXYZ> boundingBoxXyzs:
+                    Trace(boundingBoxXyzs);
+                    break;
+                default:
+                    base.Write(o);
+                    break;
+            }
+        }
+        public override void Write(string? message)
+        {
+        }
+
+        public override void WriteLine(string? message)
+        {
+        }
+    }
+}

source/RevitDevTool/Models/TraceGeometry.cs

@@ -2,7 +2,6 @@
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Media;
-using RevitDevTool.Geometry;
 using RevitDevTool.Models;
 using Serilog;
 using Serilog.Core;
@@ -75,37 +74,7 @@ [RelayCommand] private void Clear()
 
     [RelayCommand] private static void ClearGeometry()
     {
-        ExternalEventController.ActionEventHandler.Raise(app =>
-        {
-            var doc = app.ActiveUIDocument.Document;
-            if (doc == null)
-            {
-                Trace.TraceWarning("No active document");
-                return;
-            }
-            
-            var hashKey = doc.GetHashCode();
-
-            if (!TraceGeometry.DocGeometries.TryGetValue(hashKey, out var value)) 
-                return;
-        
-            var transaction = new Transaction(doc, "RemoveTransient");
-            try
-            {
-                transaction.Start();
-                doc.Delete(value);
-                transaction.Commit();
-            }
-            catch (Exception e)
-            {
-                Trace.TraceWarning($"Remove Transient Geometry Failed : [{e.Message}]");
-                transaction.RollBack();
-            }
-            finally
-            {
-                TraceGeometry.DocGeometries.Remove(hashKey);
-            }
-        });
+        VisualizationServerController.Clear();
     }
 
     public void Dispose()