1 files changed, 9 insertions, 0 deletions
diff --git a/src/core/model/GpgKeyTreeModel.cpp b/src/core/model/GpgKeyTreeModel.cpp
index cb3ee4ba..bbe5d00b 100644
--- a/src/core/model/GpgKeyTreeModel.cpp
+++ b/src/core/model/GpgKeyTreeModel.cpp
@@ -257,6 +257,15 @@ auto GpgKeyTreeModel::GetAllCheckedSubKey() -> QContainer<GpgSubKey> {
   return ret;
 }
 
+auto GpgKeyTreeModel::GetKeyByIndex(QModelIndex index) -> GpgAbstractKey * {
+  if (!index.isValid()) return nullptr;
+
+  const auto *item =
+      static_cast<const GpgKeyTreeItem *>(index.internalPointer());
+
+  return item->Key();
+}
+
 GpgKeyTreeItem::GpgKeyTreeItem(QSharedPointer<GpgAbstractKey> key,
                                QVariantList data)
     : data_(std::move(data)), key_(std::move(key)) {}