Mark unmark files from database view

databaseview.cpp

@@ -6,6 +6,8 @@
 #include <QHeaderView>
 #include <QSqlError>
 #include <QDebug>
+#include <QMenu>
+#include <QContextMenuEvent>
 #include <iostream>
 
 const QStringList DEFAULT_COLUMNS = {"EXPTIME", "OBJECT", "RA", "DEC"};
@@ -124,16 +126,40 @@ void FITSFileModel::prepareQuery()
         qDebug() << lastError();
 }
 
+DatabaseTableView::DatabaseTableView(QWidget *parent) : QTableView(parent)
+{
+}
+
+void DatabaseTableView::contextMenuEvent(QContextMenuEvent *event)
+{
+    QMenu menu;
+    QAction *mark = menu.addAction(tr("Mark"));
+    QAction *unmark = menu.addAction(tr("Unmark"));
+
+    QAction *a = menu.exec(event->globalPos());
+    if(a == nullptr)
+        return;
+
+    QModelIndexList indexes = selectionModel()->selectedRows();
+
+    if(a == mark)
+        emit filesMarked(indexes);
+    else if(a == unmark)
+        emit filesUnmarked(indexes);
+
+}
+
 DataBaseView::DataBaseView(Database *database, QWidget *parent) : QWidget(parent)
   ,m_database(database)
 {
     QVBoxLayout *layout = new QVBoxLayout(this);
     setLayout(layout);
 
-    m_tableView = new QTableView(this);
+    m_tableView = new DatabaseTableView(this);
     m_tableView->verticalHeader()->setDefaultSectionSize(1);
     m_tableView->setSortingEnabled(true);
     m_tableView->horizontalHeader()->setSortIndicatorShown(true);
+    m_tableView->setSelectionBehavior(QAbstractItemView::SelectRows);
     layout->addWidget(m_tableView);
     connect(m_tableView, &QTableView::activated, this, &DataBaseView::itemActivated);
 
@@ -151,6 +177,19 @@ DataBaseView::DataBaseView(Database *database, QWidget *parent) : QWidget(parent
     hlayout->addWidget(selectColumnsButton);
     connect(selectColumnsButton, &QPushButton::pressed, this, &DataBaseView::selectColumns);
 
+    connect(m_tableView, &DatabaseTableView::filesMarked, [this](QModelIndexList indexes){
+        QStringList files;
+        for(auto &index : indexes)
+            files.append(index.data().toString());
+        m_database->mark(files);
+    });
+    connect(m_tableView, &DatabaseTableView::filesUnmarked, [this](QModelIndexList indexes){
+        QStringList files;
+        for(auto &index : indexes)
+            files.append(index.data().toString());
+        m_database->unmark(files);
+    });
+
     for(int i=0; i<3; i++)
     {
         m_filterKeyword[i] = new QComboBox(this);
@@ -214,4 +253,3 @@ void DataBaseView::applyFilter()
     }
     m_model->setFilter(keys, values);
 }
-