1 files changed, 17 insertions, 0 deletions
diff --git a/pokemodr/models/AbilityModel.cpp b/pokemodr/models/AbilityModel.cpp
index a76a6c87..687fdd2d 100644
--- a/pokemodr/models/AbilityModel.cpp
+++ b/pokemodr/models/AbilityModel.cpp
@@ -18,6 +18,9 @@
 // Header include
 #include "AbilityModel.h"
 
+// Model includes
+#include "GroupModel.h"
+
 // PokeModr includes
 #include "../AbilityUI.h"
 
@@ -27,6 +30,9 @@
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 AbilityModel::AbilityModel(BaseModel* parent, Ability* ability) :
         ObjectModel(parent, ability)
 {
@@ -51,6 +57,12 @@ QVariant AbilityModel::data(int role) const
         QWidget* widget = new AbilityUI(static_cast<Ability*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Delete Ability", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return ObjectModel::data(role);
 }
 
@@ -75,3 +87,8 @@ bool AbilityModel::setData(const QVariant& value, int role)
     }
     return false;
 }
+
+void AbilityModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}