From 5efffc6605d4278aa29283762d8d929b427773f0 Mon Sep 17 00:00:00 2001
From: Nav <nav@navnav.me>
Date: Mon, 18 Apr 2022 18:49:49 +0100
Subject: [PATCH] Added member function to deregister TargetController command
 handlers

---
 src/TargetController/TargetControllerComponent.cpp | 4 ++++
 src/TargetController/TargetControllerComponent.hpp | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/src/TargetController/TargetControllerComponent.cpp b/src/TargetController/TargetControllerComponent.cpp
index f0254d77..2077786b 100644
--- a/src/TargetController/TargetControllerComponent.cpp
+++ b/src/TargetController/TargetControllerComponent.cpp
@@ -124,6 +124,10 @@ namespace Bloom::TargetController
         return (response != nullptr) ? std::optional(std::move(response)) : std::nullopt;
     }
 
+    void TargetControllerComponent::deregisterCommandHandler(Commands::CommandType commandType) {
+        this->commandHandlersByCommandType.erase(commandType);
+    }
+
     void TargetControllerComponent::startup() {
         this->setName("TC");
         Logger::info("Starting TargetController");
diff --git a/src/TargetController/TargetControllerComponent.hpp b/src/TargetController/TargetControllerComponent.hpp
index 1c2e7b47..4b1309c6 100644
--- a/src/TargetController/TargetControllerComponent.hpp
+++ b/src/TargetController/TargetControllerComponent.hpp
@@ -132,6 +132,8 @@ namespace Bloom::TargetController
             this->commandHandlersByCommandType.insert(std::pair(CommandType::type, parentCallback));
         }
 
+        void deregisterCommandHandler(Commands::CommandType commandType);
+
         /**
          * Updates the state of the TargetController and emits a state changed event.
          *