Merge pull request #257 from jeremytammik/RL-254

WorksharingUtils extensions
jeremytammik · Jun 15, 2024 · b0cd661 · b0cd661
2 parents a4d8cbc + f0f9819
commit b0cd661
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/source/RevitLookup/Core/ComponentModel/Descriptors/DocumentDescriptor.cs b/source/RevitLookup/Core/ComponentModel/Descriptors/DocumentDescriptor.cs
@@ -95,5 +95,12 @@ public void RegisterExtensions(IExtensionManager manager)
                 return FamilySizeTableManager.GetFamilySizeTableManager(context, familyTableId);
             });
         }
+
+        // Disabled: slow performance.
+        // manager.Register(nameof(WorksharingUtils.GetUserWorksetInfo), context =>
+        // {
+        //     var modelPath = context.IsModelInCloud ? context.GetCloudModelPath() : context.GetWorksharingCentralModelPath();
+        //     return WorksharingUtils.GetUserWorksetInfo(modelPath);
+        // });
     }
 }
diff --git a/source/RevitLookup/Core/ComponentModel/Descriptors/ElementDescriptor.cs b/source/RevitLookup/Core/ComponentModel/Descriptors/ElementDescriptor.cs
@@ -286,6 +286,9 @@ public virtual void RegisterExtensions(IExtensionManager manager)
         manager.Register(nameof(SolidSolidCutUtils.GetSolidsBeingCut), _ => SolidSolidCutUtils.GetSolidsBeingCut(_element));
         manager.Register(nameof(SolidSolidCutUtils.IsAllowedForSolidCut), _ => SolidSolidCutUtils.IsAllowedForSolidCut(_element));
         manager.Register(nameof(SolidSolidCutUtils.IsElementFromAppropriateContext), _ => SolidSolidCutUtils.IsElementFromAppropriateContext(_element));
+        manager.Register(nameof(WorksharingUtils.GetCheckoutStatus), context => WorksharingUtils.GetCheckoutStatus(context, _element.Id));
+        manager.Register(nameof(WorksharingUtils.GetWorksharingTooltipInfo), context => WorksharingUtils.GetWorksharingTooltipInfo(context, _element.Id));
+        manager.Register(nameof(WorksharingUtils.GetModelUpdatesStatus), context => WorksharingUtils.GetModelUpdatesStatus(context, _element.Id));
     }
 
     public virtual void RegisterMenu(ContextMenu contextMenu)