From 60a5a98ec5ed20320224a841922f735c9532ce95 Mon Sep 17 00:00:00 2001
From: Dominic Masters <dominic@domsplace.com>
Date: Fri, 7 Apr 2023 07:46:41 -0700
Subject: [PATCH] Example of allowing multiple components of the same type to
 be queried.

---
 src/dawn/scene/Scene.hpp                      |  3 +++
 src/dawn/scene/SceneItem.hpp                  | 21 +++++++++++++++++++
 src/dawn/scene/SceneItemComponent.hpp         |  7 +++++--
 .../scene/components/entity/EntityMove.cpp    |  7 +++++++
 .../scene/components/entity/EntityMove.hpp    |  1 +
 5 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/dawn/scene/Scene.hpp b/src/dawn/scene/Scene.hpp
index 5b91cab9..4fbd197f 100644
--- a/src/dawn/scene/Scene.hpp
+++ b/src/dawn/scene/Scene.hpp
@@ -22,6 +22,9 @@ namespace Dawn {
   template<class T>
   T * _sceneForwardGetComponent(SceneItem *item);
 
+  template<class T>
+  std::vector<T*> _sceneForwardGetComponents(SceneItem *item);
+
   class Scene {
     private:
       sceneitemid_t nextId;
diff --git a/src/dawn/scene/SceneItem.hpp b/src/dawn/scene/SceneItem.hpp
index b950eab9..06c7a4e2 100644
--- a/src/dawn/scene/SceneItem.hpp
+++ b/src/dawn/scene/SceneItem.hpp
@@ -78,6 +78,27 @@ namespace Dawn {
         }
         return nullptr;
       }
+      
+      /**
+       * Returns all components attached to this SceneItem that match the 
+       * queried component type. This method will return an array of shared
+       * pointers to the components, or an empty array if the item does not have
+       * any components of the queried type.
+       * 
+       * @tparam T Type of component to be fetched.
+       * @return An array of pointers to the components.
+       */
+      template<class T>
+      std::vector<T*> getComponents() {
+        std::vector<T*> components;
+        auto it = this->components.begin();
+        while(it != this->components.end()) {
+          T *castedAs = dynamic_cast<T*>(*it);
+          if(castedAs != nullptr) components.push_back(castedAs);
+          ++it;
+        }
+        return components;
+      }
 
       /**
        * Finds a (direct) child of this component that has a matching component.
diff --git a/src/dawn/scene/SceneItemComponent.hpp b/src/dawn/scene/SceneItemComponent.hpp
index 15156560..2139c5b8 100644
--- a/src/dawn/scene/SceneItemComponent.hpp
+++ b/src/dawn/scene/SceneItemComponent.hpp
@@ -79,10 +79,13 @@ namespace Dawn {
       virtual ~SceneItemComponent();
   };
 
-  
-
   template<class T>
   T * _sceneForwardGetComponent(SceneItem *item) {
     return item->getComponent<T>();
   }
+
+  template<class T>
+  T * _sceneForwardGetComponents(SceneItem *item) {
+    return item->getComponents<T>();
+  }
 }
\ No newline at end of file
diff --git a/src/dawnrose/scene/components/entity/EntityMove.cpp b/src/dawnrose/scene/components/entity/EntityMove.cpp
index dc233de4..09b07a30 100644
--- a/src/dawnrose/scene/components/entity/EntityMove.cpp
+++ b/src/dawnrose/scene/components/entity/EntityMove.cpp
@@ -23,11 +23,18 @@ void EntityMove::onStart() {
   assertNotNull(this->entityHealth);
 
   useEvent([&](float_t delta) {
+    
     // Don't move if stunned or attacking.
     if(entityHealth->isStunned()) return;
 
     // Don't move if no move.
     if(direction.x == 0 && direction.y == 0) return;
+
+    // If attacking, don't move? Will likely change this later.
+    auto attacks = item->getComponents<EntityAttackBase>();
+    for(auto attack : attacks) {
+      if(attack->isAttacking()) return;
+    }
     
     // Move
     float_t angle = atan2(direction.y, direction.x);
diff --git a/src/dawnrose/scene/components/entity/EntityMove.hpp b/src/dawnrose/scene/components/entity/EntityMove.hpp
index 9b754841..2786353c 100644
--- a/src/dawnrose/scene/components/entity/EntityMove.hpp
+++ b/src/dawnrose/scene/components/entity/EntityMove.hpp
@@ -7,6 +7,7 @@
 #include "scene/SceneItemComponent.hpp"
 #include "scene/components/physics/2d/CharacterController2D.hpp"
 #include "scene/components/entity/EntityHealth.hpp"
+#include "scene/components/entity/EntityAttackBase.hpp"
 
 namespace Dawn {
   class EntityMove : public SceneItemComponent {