Add addKeyPressScript method to IEngine

Author: adazem009

include/scratchcpp/iengine.h

@@ -210,6 +210,9 @@ class LIBSCRATCHCPP_EXPORT IEngine
         /* Registers the given "when I start as clone" script. */
         virtual void addCloneInitScript(std::shared_ptr<Block> hatBlock) = 0;
 
+        /* Registers the given "when key pressed" script. */
+        virtual void addKeyPressScript(std::shared_ptr<Block> hatBlock, std::string keyName) = 0;
+
         /*! Returns the list of targets. */
         virtual const std::vector<std::shared_ptr<Target>> &targets() const = 0;
 

src/engine/internal/engine.cpp

@@ -620,6 +620,22 @@ void Engine::addCloneInitScript(std::shared_ptr<Block> hatBlock)
     }
 }
 
+void Engine::addKeyPressScript(std::shared_ptr<Block> hatBlock, std::string keyName)
+{
+    std::transform(keyName.begin(), keyName.end(), keyName.begin(), ::tolower);
+    Script *script = m_scripts[hatBlock].get();
+    auto it = m_whenKeyPressedScripts.find(keyName);
+
+    if (it == m_whenKeyPressedScripts.cend())
+        m_whenKeyPressedScripts[keyName] = { script };
+    else {
+        auto &scripts = it->second;
+
+        if (std::find(scripts.begin(), scripts.end(), script) == scripts.cend())
+            scripts.push_back(script);
+    }
+}
+
 const std::vector<std::shared_ptr<Target>> &Engine::targets() const
 {
     return m_targets;

src/engine/internal/engine.h

@@ -83,8 +83,8 @@ class Engine : public IEngine
         int findBroadcastById(const std::string &broadcastId) const override;
 
         void addBroadcastScript(std::shared_ptr<Block> whenReceivedBlock, Broadcast *broadcast) override;
-
         void addCloneInitScript(std::shared_ptr<Block> hatBlock) override;
+        void addKeyPressScript(std::shared_ptr<Block> hatBlock, std::string keyName) override;
 
         const std::vector<std::shared_ptr<Target>> &targets() const override;
         void setTargets(const std::vector<std::shared_ptr<Target>> &newTargets) override;
@@ -120,6 +120,7 @@ class Engine : public IEngine
         std::unordered_map<Broadcast *, std::vector<Script *>> m_broadcastMap;
         std::unordered_map<Broadcast *, std::vector<std::pair<VirtualMachine *, VirtualMachine *>>> m_runningBroadcastMap; // source script, "when received" script
         std::unordered_map<Target *, std::vector<Script *>> m_cloneInitScriptsMap;                                         // target (no clones), "when I start as a clone" scripts
+        std::unordered_map<std::string, std::vector<Script *>> m_whenKeyPressedScripts;                                    // key name, "when key pressed" scripts
         std::vector<std::string> m_extensions;
         std::vector<std::shared_ptr<VirtualMachine>> m_runningScripts;
         std::vector<VirtualMachine *> m_scriptsToRemove;

test/mocks/enginemock.h

@@ -69,8 +69,8 @@ class EngineMock : public IEngine
         MOCK_METHOD(int, findBroadcastById, (const std::string &), (const, override));
 
         MOCK_METHOD(void, addBroadcastScript, (std::shared_ptr<Block>, Broadcast *), (override));
-
         MOCK_METHOD(void, addCloneInitScript, (std::shared_ptr<Block>), (override));
+        MOCK_METHOD(void, addKeyPressScript, (std::shared_ptr<Block>, std::string), (override));
 
         MOCK_METHOD(const std::vector<std::shared_ptr<Target>> &, targets, (), (const, override));
         MOCK_METHOD(void, setTargets, (const std::vector<std::shared_ptr<Target>> &), (override));