From 62dbabc34163b90dccbc5ea684d14a5428d72315 Mon Sep 17 00:00:00 2001
From: teryanik <github@4kt.ru>
Date: Tue, 1 Oct 2019 20:58:29 +0300
Subject: [PATCH] Add option to auto copy URL after upload (#553)

---
 src/config/geneneralconf.cpp      | 15 +++++++++++++++
 src/config/geneneralconf.h        |  2 ++
 src/tools/imgur/imguruploader.cpp |  8 +++++++-
 src/utils/confighandler.cpp       |  8 ++++++++
 src/utils/confighandler.h         |  3 +++
 5 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/src/config/geneneralconf.cpp b/src/config/geneneralconf.cpp
index 7ddce1a8..5ed73ea5 100644
--- a/src/config/geneneralconf.cpp
+++ b/src/config/geneneralconf.cpp
@@ -37,6 +37,7 @@ GeneneralConf::GeneneralConf(QWidget *parent) : QWidget(parent) {
     initShowTrayIcon();
     initAutostart();
     initCloseAfterCapture();
+    initCopyAndCloseAfterUpload();
 
     // this has to be at the end
     initConfingButtons();
@@ -49,6 +50,7 @@ void GeneneralConf::updateComponents() {
     m_sysNotifications->setChecked(config.desktopNotificationValue());
     m_autostart->setChecked(config.startupLaunchValue());
     m_closeAfterCapture->setChecked(config.closeAfterScreenshotValue());
+    m_copyAndCloseAfterUpload->setChecked(config.copyAndCloseAfterUploadEnabled());
 
 #if defined(Q_OS_LINUX) || defined(Q_OS_UNIX)
     m_showTray->setChecked(!config.disabledTrayIconValue());
@@ -222,3 +224,16 @@ void GeneneralConf::initCloseAfterCapture() {
     connect(m_closeAfterCapture, &QCheckBox::clicked, this,
             &GeneneralConf::closeAfterCaptureChanged);
 }
+
+void GeneneralConf::initCopyAndCloseAfterUpload()
+{
+    m_copyAndCloseAfterUpload = new QCheckBox(tr("Copy URL after upload"), this);
+    ConfigHandler config;
+    m_copyAndCloseAfterUpload->setChecked(config.copyAndCloseAfterUploadEnabled());
+    m_copyAndCloseAfterUpload->setToolTip(tr("Copy URL and close window after upload"));
+    m_layout->addWidget(m_copyAndCloseAfterUpload);
+
+    connect(m_copyAndCloseAfterUpload, &QCheckBox::clicked, [](bool checked) {
+        ConfigHandler().setCopyAndCloseAfterUploadEnabled(checked);
+    });
+}
\ No newline at end of file
diff --git a/src/config/geneneralconf.h b/src/config/geneneralconf.h
index 96991ecd..ae40ef76 100644
--- a/src/config/geneneralconf.h
+++ b/src/config/geneneralconf.h
@@ -48,6 +48,7 @@ private:
     QCheckBox *m_helpMessage;
     QCheckBox *m_autostart;
     QCheckBox *m_closeAfterCapture;
+    QCheckBox *m_copyAndCloseAfterUpload;
     QPushButton *m_importButton;
     QPushButton *m_exportButton;
     QPushButton *m_resetButton;
@@ -58,4 +59,5 @@ private:
     void initConfingButtons();
     void initAutostart();
     void initCloseAfterCapture();
+    void initCopyAndCloseAfterUpload();
 };
diff --git a/src/tools/imgur/imguruploader.cpp b/src/tools/imgur/imguruploader.cpp
index db1fb5cd..cd0dfc63 100644
--- a/src/tools/imgur/imguruploader.cpp
+++ b/src/tools/imgur/imguruploader.cpp
@@ -77,7 +77,13 @@ void ImgurUploader::handleReply(QNetworkReply *reply) {
         m_imageURL.setUrl(data[QStringLiteral("link")].toString());
         m_deleteImageURL.setUrl(QStringLiteral("https://imgur.com/delete/%1").arg(
                                     data[QStringLiteral("deletehash")].toString()));
-        onUploadOk();
+        if (ConfigHandler().copyAndCloseAfterUploadEnabled()) {
+            QApplication::clipboard()->setText(m_imageURL.toString());
+            SystemNotification().sendMessage(QObject::tr("URL copied to clipboard."));
+            close();
+        } else {
+            onUploadOk();
+        }
     } else {
         m_infoLabel->setText(reply->errorString());
     }
diff --git a/src/utils/confighandler.cpp b/src/utils/confighandler.cpp
index 5aed0bf6..14f3e55f 100644
--- a/src/utils/confighandler.cpp
+++ b/src/utils/confighandler.cpp
@@ -325,6 +325,14 @@ void ConfigHandler::setCloseAfterScreenshot(const bool close) {
     m_settings.setValue(QStringLiteral("closeAfterScreenshot"), close);
 }
 
+bool ConfigHandler::copyAndCloseAfterUploadEnabled() {
+    return m_settings.value(QStringLiteral("copyAndCloseAfterUpload")).toBool();
+}
+
+void ConfigHandler::setCopyAndCloseAfterUploadEnabled(const bool value) {
+    m_settings.setValue(QStringLiteral("copyAndCloseAfterUpload"), value);
+}
+
 void ConfigHandler::setDefaults() {
     m_settings.clear();
 }
diff --git a/src/utils/confighandler.h b/src/utils/confighandler.h
index 85d93612..1886882c 100644
--- a/src/utils/confighandler.h
+++ b/src/utils/confighandler.h
@@ -71,6 +71,9 @@ public:
     bool closeAfterScreenshotValue();
     void setCloseAfterScreenshot(const bool);
 
+    bool copyAndCloseAfterUploadEnabled();
+    void setCopyAndCloseAfterUploadEnabled(const bool);
+
 
     void setDefaults();
     void setAllTheButtons();