Adds functionality to remove custom actions from the menu, when required

tappytaps · May 22, 2018 · d5ec44f · d5ec44f
1 parent c75e2dc
commit d5ec44f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/DBDebugToolkit/Classes/DBDebugToolkit.h b/DBDebugToolkit/Classes/DBDebugToolkit.h
@@ -108,6 +108,20 @@
  */
 + (void)addCustomActions:(NSArray <DBCustomAction *> *)customActions;
 
+/**
+ Removes a single `DBCustomAction` instance from the array accessible in the menu.
+ 
+ @param customAction The `DBCustomAction` instance that should be removed.
+ */
++ (void)removeCustomAction:(DBCustomAction *)customAction;
+
+/**
+ Removes multiple `DBCustomAction` instances from the array accessible in the menu.
+ 
+ @param customActions An array of `DBCustomAction` instances that should be accessible in the menu.
+ */
++ (void)removeCustomActions:(NSArray <DBCustomAction *> *)customActions;
+
 ///-----------------------
 /// @name Custom variables
 ///-----------------------

diff --git a/DBDebugToolkit/Classes/DBDebugToolkit.m b/DBDebugToolkit/Classes/DBDebugToolkit.m
@@ -217,6 +217,16 @@ + (void)addCustomActions:(NSArray <DBCustomAction *> *)customActions {
     [toolkit.customActions addObjectsFromArray:customActions];
 }
 
++ (void)removeCustomAction:(DBCustomAction *)customAction {
+    DBDebugToolkit *toolkit = [DBDebugToolkit sharedInstance];
+    [toolkit.customActions removeObject:customAction];
+}
+
++ (void)removeCustomActions:(NSArray <DBCustomAction *> *)customActions {
+    DBDebugToolkit *toolkit = [DBDebugToolkit sharedInstance];
+    [toolkit.customActions removeObjectsInArray:customActions];
+}
+
 #pragma mark - Custom variables
 
 - (void)setupCustomVariables {