Implement QgsHanaDataItemGuiProvider interface

qgis · Dec 7, 2020 · 5b109ae · 5b109ae
1 parent 473157d
commit 5b109ae
Show file tree

Hide file tree

Showing 9 changed files with 301 additions and 236 deletions.
diff --git a/src/providers/hana/CMakeLists.txt b/src/providers/hana/CMakeLists.txt
@@ -29,11 +29,14 @@ SET (HANA_MOC_HDRS
 IF (WITH_GUI)
   SET(HANA_SRCS ${HANA_SRCS}
     qgshanadataitems.cpp
+    qgshanadataitemguiprovider.cpp
     qgshananewconnection.cpp
+    qgshanaprovidergui.cpp
     qgshanasourceselect.cpp
   )
   SET(HANA_MOC_HDRS ${HANA_MOC_HDRS}
     qgshananewconnection.h
+    qgshanadataitemguiprovider.h
     qgshanasourceselect.h
   )
 ENDIF ()

diff --git a/src/providers/hana/qgshanadataitemguiprovider.cpp b/src/providers/hana/qgshanadataitemguiprovider.cpp
@@ -0,0 +1,164 @@
+/***************************************************************************
+   qgshanadataitemguiprovider.cpp
+   --------------------------------------
+   Date      : 31-05-2019
+   Copyright : (C) SAP SE
+   Author    : Maksim Rylov
+/***************************************************************************
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ ***************************************************************************/
+#include "qgshanaconnection.h"
+#include "qgshanadataitems.h"
+#include "qgshanadataitemguiprovider.h"
+#include "qgshananewconnection.h"
+#include "qgshanaprovider.h"
+#include "qgsnewnamedialog.h"
+
+#include <QInputDialog>
+#include <QMessageBox>
+
+void QgsHanaDataItemGuiProvider::populateContextMenu(
+  QgsDataItem *item, QMenu *menu, const QList<QgsDataItem *> &, QgsDataItemGuiContext)
+{
+  if (QgsHanaRootItem *rootItem = qobject_cast<QgsHanaRootItem *>(item))
+  {
+    QAction *actionNew = new QAction(tr("New Connection…"), this);
+    connect(actionNew, &QAction::triggered, this, [rootItem] { newConnection(rootItem); });
+    menu->addAction(actionNew);
+  }
+
+  if (QgsHanaConnectionItem *connItem = qobject_cast<QgsHanaConnectionItem *>(item))
+  {
+    QAction *actionRefresh = new QAction(tr("Refresh"), this);
+    connect(actionRefresh, &QAction::triggered, this, [connItem] { refreshConnection(connItem); });
+    menu->addAction(actionRefresh);
+
+    menu->addSeparator();
+
+    QAction *actionEdit = new QAction(tr("Edit Connection…"), this);
+    connect(actionEdit, &QAction::triggered, this, [connItem] { editConnection(connItem); });
+    menu->addAction(actionEdit);
+
+    QAction *actionDelete = new QAction(tr("Delete Connection"), this);
+    connect(actionDelete, &QAction::triggered, this, [connItem] { deleteConnection(connItem); });
+    menu->addAction(actionDelete);
+  }
+
+  if (QgsHanaSchemaItem *schemaItem = qobject_cast<QgsHanaSchemaItem *>(item))
+  {
+    QAction *actionRefresh = new QAction(tr("Refresh"), this);
+    connect(actionRefresh, &QAction::triggered, this, [schemaItem] { schemaItem->refresh(); });
+    menu->addAction(actionRefresh);
+  }
+}
+
+bool QgsHanaDataItemGuiProvider::deleteLayer(QgsLayerItem *item, QgsDataItemGuiContext context)
+{
+  if (QgsHanaLayerItem *layerItem = qobject_cast<QgsHanaLayerItem *>(item))
+  {
+    const QgsHanaLayerProperty &layerInfo = layerItem->layerInfo();
+    QString typeName = layerInfo.isView ? tr("View") : tr("Table");
+
+    if (QMessageBox::question(nullptr, tr("Delete %1").arg(typeName),
+      tr("Are you sure you want to delete %1.%2?").arg(layerInfo.schemaName, layerInfo.tableName),
+      QMessageBox::Yes | QMessageBox::No, QMessageBox::No) != QMessageBox::Yes)
+      return true;
+
+    QgsHanaConnectionItem* connItem = qobject_cast<QgsHanaConnectionItem*>(layerItem->parent()->parent());
+
+    QgsHanaConnectionRef conn(connItem->name());
+    if (conn.isNull())
+      return false;
+
+    QString errMessage;
+    bool res = conn->dropTable(layerInfo.schemaName, layerInfo.tableName, &errMessage);
+    if (!res)
+    {
+      QMessageBox::warning(nullptr, tr("Delete %1").arg(typeName), errMessage);
+    }
+    else
+    {
+      QMessageBox::information(nullptr,
+        tr("Delete %1").arg(typeName),
+        tr("%1 deleted successfully.").arg(typeName));
+      if ( layerItem->parent() )
+        layerItem->parent()->refresh();
+    }
+    return res;
+  }
+  return false;
+}
+
+bool QgsHanaDataItemGuiProvider::acceptDrop(QgsDataItem *item, QgsDataItemGuiContext)
+{
+  if (qobject_cast<QgsHanaConnectionItem *>(item))
+    return true;
+  if (qobject_cast<QgsHanaSchemaItem *>(item))
+    return true;
+
+  return false;
+}
+
+bool QgsHanaDataItemGuiProvider::handleDrop(
+  QgsDataItem *item, QgsDataItemGuiContext, const QMimeData *data, Qt::DropAction)
+{
+  if (QgsHanaConnectionItem *connItem = qobject_cast<QgsHanaConnectionItem *>(item))
+  {
+    return connItem->handleDrop(data, QString());
+  }
+  else if (QgsHanaSchemaItem *schemaItem = qobject_cast<QgsHanaSchemaItem *>(item))
+  {
+    QgsHanaConnectionItem *connItem = qobject_cast<QgsHanaConnectionItem *>(schemaItem->parent());
+    if (!connItem)
+      return false;
+
+    return connItem->handleDrop(data, schemaItem->name());
+  }
+  return false;
+}
+
+void QgsHanaDataItemGuiProvider::newConnection(QgsDataItem *item)
+{
+  QgsHanaNewConnection nc(nullptr);
+  if (nc.exec())
+  {
+    item->refresh();
+  }
+}
+
+void QgsHanaDataItemGuiProvider::editConnection(QgsDataItem *item)
+{
+  QgsHanaNewConnection nc(nullptr, item->name());
+  if ( nc.exec() )
+  {
+    // the parent should be updated
+    if (item->parent())
+      item->parent()->refreshConnections();
+  }
+}
+
+void QgsHanaDataItemGuiProvider::deleteConnection(QgsDataItem *item)
+{
+  if (QMessageBox::question(nullptr, QObject::tr("Delete Connection"),
+    tr("Are you sure you want to delete the connection to %1?").arg(item->name()),
+    QMessageBox::Yes | QMessageBox::No, QMessageBox::No) != QMessageBox::Yes)
+    return;
+
+  QgsHanaSettings::removeConnection(item->name());
+  // the parent should be updated
+  if (item->parent())
+    item->parent()->refreshConnections();
+}
+
+void QgsHanaDataItemGuiProvider::refreshConnection(QgsDataItem *item)
+{
+  item->refresh();
+  // the parent should be updated
+  if (item->parent())
+    item->parent()->refreshConnections();
+}
diff --git a/src/providers/hana/qgshanadataitemguiprovider.h b/src/providers/hana/qgshanadataitemguiprovider.h
@@ -0,0 +1,45 @@
+/***************************************************************************
+   qgshanadataitemguiprovider.h
+   --------------------------------------
+   Date      : 31-05-2019
+   Copyright : (C) SAP SE
+   Author    : Maksim Rylov
+/***************************************************************************
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ ***************************************************************************/
+#ifndef QGSHANADATAITEMGUIPROVIDER_H
+#define QGSHANADATAITEMGUIPROVIDER_H
+
+#include "qgsdataitemguiprovider.h"
+
+class QgsHanaSchemaItem;
+class QgsHanaLayerItem;
+
+class QgsHanaDataItemGuiProvider : public QObject, public QgsDataItemGuiProvider
+{
+  Q_OBJECT
+public:
+
+  QString name() override { return QStringLiteral("HANA"); }
+
+  void populateContextMenu(QgsDataItem *item, QMenu *menu,
+    const QList<QgsDataItem *> &selectedItems, QgsDataItemGuiContext context) override;
+
+  bool deleteLayer(QgsLayerItem *item, QgsDataItemGuiContext context) override;
+
+  bool acceptDrop(QgsDataItem *item, QgsDataItemGuiContext context) override;
+  bool handleDrop(QgsDataItem *item, QgsDataItemGuiContext context, const QMimeData *data, Qt::DropAction action) override;
+
+private:
+  static void newConnection(QgsDataItem *item);
+  static void editConnection(QgsDataItem *item);
+  static void deleteConnection(QgsDataItem *item);
+  static void refreshConnection(QgsDataItem *item);
+};
+
+#endif // QGSHANADATAITEMGUIPROVIDER_H