From d5ec44ff69f3c5b49d4d9c1334d18b34fe2919ae Mon Sep 17 00:00:00 2001
From: Dan Silk <dsilk@nextfaze.com>
Date: Wed, 23 May 2018 09:20:35 +0930
Subject: [PATCH] Adds functionality to remove custom actions from the menu,
 when required

---
 DBDebugToolkit/Classes/DBDebugToolkit.h | 14 ++++++++++++++
 DBDebugToolkit/Classes/DBDebugToolkit.m | 10 ++++++++++
 2 files changed, 24 insertions(+)

diff --git a/DBDebugToolkit/Classes/DBDebugToolkit.h b/DBDebugToolkit/Classes/DBDebugToolkit.h
index db680c2..90af75a 100644
--- 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
index 9a0d155..c31e19d 100644
--- 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 {