From 67b8bef4143d269f0e6c982eac21b51d6e598f23 Mon Sep 17 00:00:00 2001
From: jessikitty <jess.rogerson.29@outlook.com>
Date: Mon, 27 Apr 2026 10:18:18 +1000
Subject: [PATCH] Register UIExtension features on Initialize for Device and
 User pages

---
 ADComparePlugin.cs | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/ADComparePlugin.cs b/ADComparePlugin.cs
index 31cc48a..7339cdd 100644
--- a/ADComparePlugin.cs
+++ b/ADComparePlugin.cs
@@ -14,13 +14,31 @@ namespace Disco.Plugins.ADCompare
     {
         public override void Initialize(DiscoDataContext Database)
         {
-            // Plugin initialisation - nothing required at startup
+            // Register UIExtension features so they appear on Device and User pages
+            foreach (var feature in Manifest.Features)
+            {
+                using (var instance = feature.CreateInstance())
+                {
+                    var deviceExt = instance as Features.DeviceUIExtension;
+                    if (deviceExt != null)
+                    {
+                        deviceExt.Register();
+                        continue;
+                    }
+
+                    var userExt = instance as Features.UserUIExtension;
+                    if (userExt != null)
+                    {
+                        userExt.Register();
+                        continue;
+                    }
+                }
+            }
         }
 
         public override void Install(DiscoDataContext Database, ScheduledTaskStatus Status)
         {
             Status.UpdateStatus(0, "Installing AD Compare Plugin...");
-            // No database schema changes needed - we read-only compare
             Status.UpdateStatus(100, "Installation Complete");
         }