1 files changed, 11 insertions, 2 deletions
diff --git a/src/main/headers/Weapon.hpp b/src/main/headers/Weapon.hpp
index 40eb7a3..f59fe4d 100644
--- a/src/main/headers/Weapon.hpp
+++ b/src/main/headers/Weapon.hpp
@@ -2,17 +2,26 @@
 #define WEAPON_HPP
 
 #include "Item.hpp"
-#include "Actor.hpp"
 
 #include <string>
 
+class Actor;
+
 class Weapon : public Item
 {
+public:
+    enum class Class
+    {
+        SWORD, AXE, LANCE, RANGED
+    };
+
+    const Class weaponClass;
+
 protected:
     unsigned _damage;
     unsigned _requiredLevel;
 
-    Weapon(Item::Type t, const std::string &name) : Item(t, name) {}
+    Weapon(Item::Type t, Class _weaponClass, const std::string &name) : Item(t, name), weaponClass(_weaponClass) {}
 };
 
 #endif