qimageuploader: add copy links to treeview context menu

Author: zenden2k

Source/qimageuploader/CMakeLists.txt

@@ -47,6 +47,8 @@ set(SOURCE_FILES
     QtScriptFunctionsImpl.cpp
     Gui/controls/ServerSelectorWidget.cpp
     Gui/controls/ProgressRingWidget.cpp
+    Gui/controls/UploadTreeView.cpp
+    Gui/controls/UploadTreeView.h
     ../Core/Settings/QtGuiSettings.cpp
     ../Core/Settings/CommonGuiSettings.cpp
     ../Core/AbstractServerIconCache.cpp

Source/qimageuploader/Gui/MainWindow.cpp

@@ -65,7 +65,7 @@ MainWindow::MainWindow(CUploadEngineList* engineList, LogWindow* logWindow, QWid
     ui->treeView->setColumnWidth(0, 150); // Filename column
     ui->treeView->setColumnWidth(1, 150); // Status column
     ui->treeView->setColumnWidth(2, 150); // Progress column
-
+    ui->treeView->setSelectionBehavior(QAbstractItemView::SelectRows);
     connect(ui->treeView, &QTreeView::doubleClicked, this, &MainWindow::itemDoubleClicked);
     connect(ui->treeView, &QTreeView::customContextMenuRequested, this, &MainWindow::onCustomContextMenu);
 
@@ -310,11 +310,33 @@ void MainWindow::onCustomContextMenu(const QPoint& point) {
         contextMenu->setDefaultAction(viewCodeAction);
 
         if (internalItem->task) {
-            auto uploadResult = internalItem->task->uploadResult();
-            QString url = QString::fromUtf8(uploadResult->getDirectUrl().c_str());
-            if (url.isEmpty()) {
-                url = QString::fromUtf8(uploadResult->getDownloadUrl().c_str());
+            auto* uploadResult = internalItem->task->uploadResult();
+            QString directUrl = QString::fromStdString(uploadResult->getDirectUrl());
+            QString viewUrl = QString::fromStdString(uploadResult->getDownloadUrl());
+
+            if (!directUrl.isEmpty()) {
+                QAction* copyDirectLinkAction = new QAction(tr("Copy direct link"), contextMenu);
+                copyDirectLinkAction->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_C));
+                connect(copyDirectLinkAction, &QAction::triggered, [directUrl](bool checked) {
+                    QClipboard* clipboard = QApplication::clipboard();
+                    clipboard->setText(directUrl);
+                });
+                contextMenu->addAction(copyDirectLinkAction);
+            }
+
+            if (!viewUrl.isEmpty()) {
+                QAction* copyViewLinkAction = new QAction(tr("Copy view link"), contextMenu);
+                if (directUrl.isEmpty()) {
+                    copyViewLinkAction->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_C));
+                }
+                connect(copyViewLinkAction, &QAction::triggered, [viewUrl](bool checked) {
+                    QClipboard* clipboard = QApplication::clipboard();
+                    clipboard->setText(viewUrl);
+                });
+                contextMenu->addAction(copyViewLinkAction);
             }
+            contextMenu->addSeparator();
+            QString url = directUrl.isEmpty() ? viewUrl : directUrl;
             if (!url.isEmpty()) {
                 QAction* viewInBrowser = new QAction(tr("Open in browser"), contextMenu);
                 connect(viewInBrowser, &QAction::triggered, [url](bool checked)

Source/qimageuploader/Gui/controls/UploadTreeView.cpp

@@ -0,0 +1,33 @@
+#include "UploadTreeView.h"
+
+#include <algorithm>
+
+#include <QApplication>
+#include <QClipboard>
+
+UploadTreeView::UploadTreeView(QWidget *parent): QTreeView(parent) {
+
+}
+
+void UploadTreeView::keyPressEvent(QKeyEvent * event)
+{
+    if(event->matches(QKeySequence::Copy)) {
+        copy();
+    } else {
+        QTreeView::keyPressEvent(event);
+    }
+}
+
+void UploadTreeView::copy() {
+    QItemSelectionModel * selection = selectionModel();
+    QModelIndexList indexes = selection->selectedIndexes();
+
+    if(indexes.size() < 1) {
+        return;
+    }
+    std::sort(indexes.begin(), indexes.end());
+    QModelIndex index = indexes.first();
+
+    QString url = model()->data(index, Qt::UserRole).toString();
+    QApplication::clipboard()->setText(url);
+}

Source/qimageuploader/Gui/controls/UploadTreeView.h

@@ -0,0 +1,18 @@
+#ifndef UPLOADTREEVIEW_H
+#define UPLOADTREEVIEW_H
+
+#include <QTreeView>
+#include <QWidget>
+#include <QKeyEvent>
+
+class UploadTreeView : public QTreeView
+{
+    Q_OBJECT
+public:
+    UploadTreeView(QWidget *parent = nullptr);
+    void keyPressEvent(QKeyEvent * event) override;
+private:
+    void copy();
+};
+
+#endif // UPLOADTREEVIEW_H

Source/qimageuploader/Gui/models/uploadtreemodel.cpp

@@ -50,6 +50,13 @@ QVariant UploadTreeModel::data(const QModelIndex &index, int role) const
             return QBrush(QColor(Qt::darkGreen));
         }
     }
+
+    if (role == Qt::UserRole && index.column() == 0 && obj->task) {
+        std::string directUrl = obj->task->uploadResult()->getDirectUrl();
+        std::string viewUrl = obj->task->uploadResult()->getDownloadUrl();
+        return directUrl.empty() ? QString::fromStdString(viewUrl) : QString::fromStdString(directUrl);
+    }
+
     if (role != Qt::DisplayRole)
         return {};
 
@@ -58,8 +65,8 @@ QVariant UploadTreeModel::data(const QModelIndex &index, int role) const
     QString result;
     if(index.column() == 0)
     {
-        if ( obj->task ) {
-            result = U2Q(obj->task->title());
+        if (obj->task ) {
+            result = U2Q(obj->task->title());  
         } else if ( obj->session) {
 
             result = QString(tr("Session %1 (%2 files)").arg(obj->index + 1).arg(obj->session->taskCount()));

Source/qimageuploader/Gui/ui/MainWindow.ui

@@ -20,7 +20,7 @@
   <widget class="QWidget" name="centralWidget">
    <layout class="QVBoxLayout" name="verticalLayout">
     <item>
-     <widget class="QTreeView" name="treeView">
+     <widget class="UploadTreeView" name="treeView">
       <property name="expandsOnDoubleClick">
        <bool>false</bool>
       </property>
@@ -38,7 +38,7 @@
       <item>
        <spacer name="horizontalSpacer">
         <property name="orientation">
-         <enum>Qt::Horizontal</enum>
+         <enum>Qt::Orientation::Horizontal</enum>
         </property>
         <property name="sizeHint" stdset="0">
          <size>
@@ -70,7 +70,7 @@
     <bool>false</bool>
    </property>
    <property name="toolButtonStyle">
-    <enum>Qt::ToolButtonTextBesideIcon</enum>
+    <enum>Qt::ToolButtonStyle::ToolButtonTextBesideIcon</enum>
    </property>
    <property name="floatable">
     <bool>false</bool>
@@ -145,6 +145,13 @@
   </action>
  </widget>
  <layoutdefault spacing="6" margin="11"/>
+ <customwidgets>
+  <customwidget>
+   <class>UploadTreeView</class>
+   <extends>QTreeView</extends>
+   <header>Gui/controls/uploadtreeview.h</header>
+  </customwidget>
+ </customwidgets>
  <resources>
   <include location="../../qimageuploader.qrc"/>
  </resources>