From b9ff1eaca3f76ff9afbee1560bcd73a3739b72f0 Mon Sep 17 00:00:00 2001
From: Gres <gres@gres.biz>
Date: Mon, 6 Apr 2020 18:18:01 +0300
Subject: [PATCH] Add items to run new capture to result menu

---
 src/represent/representer.cpp  |  3 ++-
 src/represent/resultwidget.cpp | 14 ++++++++++++--
 src/represent/resultwidget.h   |  4 ++--
 3 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/represent/representer.cpp b/src/represent/representer.cpp
index d6f97a6..dd465b0 100644
--- a/src/represent/representer.cpp
+++ b/src/represent/representer.cpp
@@ -146,7 +146,8 @@ void Representer::showWidget(const TaskPtr &task)
   }
 
   if (index == count) {
-    widgets_.emplace_back(std::make_unique<ResultWidget>(*this, settings_));
+    widgets_.emplace_back(
+        std::make_unique<ResultWidget>(manager_, *this, settings_));
     widgets_.back()->installEventFilter(this);
   }
 
diff --git a/src/represent/resultwidget.cpp b/src/represent/resultwidget.cpp
index 06ec068..289c749 100644
--- a/src/represent/resultwidget.cpp
+++ b/src/represent/resultwidget.cpp
@@ -1,5 +1,6 @@
 #include "resultwidget.h"
 #include "debug.h"
+#include "manager.h"
 #include "representer.h"
 #include "settings.h"
 #include "task.h"
@@ -11,8 +12,8 @@
 #include <QMenu>
 #include <QMouseEvent>
 
-ResultWidget::ResultWidget(Representer &representer, const Settings &settings,
-                           QWidget *parent)
+ResultWidget::ResultWidget(Manager &manager, Representer &representer,
+                           const Settings &settings, QWidget *parent)
   : QFrame(parent)
   , representer_(representer)
   , settings_(settings)
@@ -50,6 +51,15 @@ ResultWidget::ResultWidget(Representer &representer, const Settings &settings,
     auto edit = contextMenu_->addAction(tr("Edit..."));
     connect(edit, &QAction::triggered,  //
             this, &ResultWidget::edit);
+
+    contextMenu_->addSeparator();
+
+    auto capture = contextMenu_->addAction(tr("New capture"));
+    connect(capture, &QAction::triggered,  //
+            this, [&manager] { manager.capture(); });
+    auto repeatCapture = contextMenu_->addAction(tr("Repeat capture"));
+    connect(repeatCapture, &QAction::triggered,  //
+            this, [&manager] { manager.repeatCapture(); });
   }
 
   installEventFilter(this);
diff --git a/src/represent/resultwidget.h b/src/represent/resultwidget.h
index 6328af0..0e38e9d 100644
--- a/src/represent/resultwidget.h
+++ b/src/represent/resultwidget.h
@@ -11,8 +11,8 @@ class ResultWidget : public QFrame
 {
   Q_OBJECT
 public:
-  ResultWidget(Representer& representer, const Settings& settings,
-               QWidget* parent = nullptr);
+  ResultWidget(Manager& manager, Representer& representer,
+               const Settings& settings, QWidget* parent = nullptr);
 
   const TaskPtr& task() const;
   void show(const TaskPtr& task);