refactor: reduce code duplication of gnupg operations

1 files changed, 20 insertions, 5 deletions

diff --git a/src/ui/struct/GpgOperaResultContext.cpp b/src/ui/struct/GpgOperaResultContext.cpp
index 5640a204..b01f8788 100644
--- a/src/ui/struct/GpgOperaResultContext.cpp
+++ b/src/ui/struct/GpgOperaResultContext.cpp
@@ -45,6 +45,12 @@ auto GpgOperaContextBasement::GetContextOutPath(int category) -> QStringList& {
   return categories[category].o_paths;
 }
 
+auto GpgOperaContextBasement::GetContextBuffer(int category)
+    -> QContainer<GFBuffer>& {
+  if (!categories.contains(category)) categories[category] = {};
+  return categories[category].buffers;
+}
+
 auto GpgOperaContextBasement::GetAllPath() -> QStringList {
   QStringList res;
 
@@ -66,12 +72,21 @@ auto GpgOperaContextBasement::GetAllOutPath() -> QStringList {
 auto GetGpgOperaContextFromBasement(
     const QSharedPointer<GpgOperaContextBasement>& base,
     int category) -> QSharedPointer<GpgOperaContext> {
-  if (base->GetContextPath(category).empty()) return nullptr;
+  if (!base->GetContextPath(category).isEmpty()) {
+    auto context = QSharedPointer<GpgOperaContext>::create(base);
 
-  auto context = QSharedPointer<GpgOperaContext>::create(base);
+    context->paths = base->GetContextPath(category);
+    context->o_paths = base->GetContextOutPath(category);
+    return context;
+  }
+
+  if (!base->GetContextBuffer(category).isEmpty()) {
+    auto context = QSharedPointer<GpgOperaContext>::create(base);
+
+    context->buffers = base->GetContextBuffer(category);
+    return context;
+  }
 
-  context->paths = base->GetContextPath(category);
-  context->o_paths = base->GetContextOutPath(category);
-  return context;
+  return nullptr;
 }
 }  // namespace GpgFrontend::UI
\ No newline at end of file