From 794b814c470e6c60e2a532720e7ec939b71c1ef7 Mon Sep 17 00:00:00 2001
From: Marcus Huderle <huderlem@gmail.com>
Date: Tue, 13 Feb 2018 21:01:16 -0800
Subject: [PATCH] Add right-click context menu for map groups

---
 mainwindow.cpp | 45 +++++++++++++++++++++++++++++++++++++++++----
 mainwindow.h   |  9 +++++++++
 2 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/mainwindow.cpp b/mainwindow.cpp
index 5a5dcfc4..54cad524 100755
--- a/mainwindow.cpp
+++ b/mainwindow.cpp
@@ -293,13 +293,13 @@ void MainWindow::populateMapList() {
     mapIcon.addFile(QStringLiteral(":/icons/map.ico"), QSize(), QIcon::Normal, QIcon::Off);
     mapIcon.addFile(QStringLiteral(":/icons/image.ico"), QSize(), QIcon::Normal, QIcon::On);
 
-    QStandardItemModel *model = new QStandardItemModel;
+    mapListModel = new QStandardItemModel;
 
     QStandardItem *entry = new QStandardItem;
     entry->setText(project->getProjectTitle());
     entry->setIcon(folderIcon);
     entry->setEditable(false);
-    model->appendRow(entry);
+    mapListModel->appendRow(entry);
 
     QStandardItem *maps = new QStandardItem;
     maps->setText("maps");
@@ -314,6 +314,8 @@ void MainWindow::populateMapList() {
         group->setText(group_name);
         group->setIcon(mapFolderIcon);
         group->setEditable(false);
+        group->setData(group_name, Qt::UserRole);
+        group->setData("map_group", MapListUserRoles::TypeRole);
         maps->appendRow(group);
         QStringList *names = project->groupedMapNames->value(i);
         for (int j = 0; j < names->length(); j++) {
@@ -323,17 +325,52 @@ void MainWindow::populateMapList() {
             map->setIcon(mapIcon);
             map->setEditable(false);
             map->setData(map_name, Qt::UserRole);
+            map->setData("map_name", MapListUserRoles::TypeRole);
             group->appendRow(map);
-            //ui->mapList->setExpanded(model->indexFromItem(map), false); // redundant
         }
     }
 
-    ui->mapList->setModel(model);
+    // Right-clicking on items in the map list tree view brings up a context menu.
+    ui->mapList->setContextMenuPolicy(Qt::CustomContextMenu);
+    connect(ui->mapList, SIGNAL(customContextMenuRequested(const QPoint &)),
+            this, SLOT(onOpenMapListContextMenu(const QPoint &)));
+
+    ui->mapList->setModel(mapListModel);
     ui->mapList->setUpdatesEnabled(true);
     ui->mapList->expandToDepth(2);
     ui->mapList->repaint();
 }
 
+void MainWindow::onOpenMapListContextMenu(const QPoint &point)
+{
+    QModelIndex index = ui->mapList->indexAt(point);
+    if (!index.isValid()) {
+        return;
+    }
+
+    QStandardItem *selectedItem = mapListModel->itemFromIndex(index);
+    QVariant itemType = selectedItem->data(MapListUserRoles::TypeRole);
+    if (!itemType.isValid()) {
+        return;
+    }
+
+    // Build custom context menu depending on which type of item was selected (map group, map name, etc.)
+    if (itemType == "map_group") {
+        QString groupName = selectedItem->data(Qt::UserRole).toString();
+        QMenu* menu = new QMenu();
+        QActionGroup* actions = new QActionGroup(menu);
+        actions->addAction(menu->addAction("Add New Map to Group"))->setData(groupName);
+        connect(actions, SIGNAL(triggered(QAction*)), this, SLOT(addNewMapToGroup(QAction*)));
+        menu->exec(QCursor::pos());
+    }
+}
+
+void MainWindow::addNewMapToGroup(QAction* triggeredAction)
+{
+    QString groupName = triggeredAction->data().toString();
+    qDebug() << "Adding new map " << groupName;
+}
+
 void MainWindow::on_mapList_activated(const QModelIndex &index)
 {
     QVariant data = index.data(Qt::UserRole);
diff --git a/mainwindow.h b/mainwindow.h
index 6e0b5bcf..36a8010f 100755
--- a/mainwindow.h
+++ b/mainwindow.h
@@ -4,6 +4,7 @@
 #include <QString>
 #include <QModelIndex>
 #include <QMainWindow>
+#include <QStandardItemModel>
 #include <QGraphicsPixmapItem>
 #include <QGraphicsItemGroup>
 #include <QGraphicsSceneMouseEvent>
@@ -65,8 +66,12 @@ private slots:
 
     void on_toolButton_Dropper_clicked();
 
+    void onOpenMapListContextMenu(const QPoint &point);
+    void addNewMapToGroup(QAction* triggeredAction);
+
 private:
     Ui::MainWindow *ui;
+    QStandardItemModel *mapListModel;
     Editor *editor = NULL;
     void setMap(QString);
     void populateMapList();
@@ -83,4 +88,8 @@ private:
     void checkToolButtons();
 };
 
+enum MapListUserRoles {
+    TypeRole = Qt::UserRole + 10, // Used to differentiate between the different layers of the map list tree view.
+};
+
 #endif // MAINWINDOW_H