[mssql] Move GUI code to new files + remove legacy data item calls

qgis · Jul 1, 2019 · 2a16f3e · 2a16f3e
1 parent 04630d5
commit 2a16f3e
Show file tree

Hide file tree

Showing 8 changed files with 352 additions and 293 deletions.
diff --git a/src/providers/mssql/CMakeLists.txt b/src/providers/mssql/CMakeLists.txt
@@ -18,10 +18,13 @@ SET(MSSQL_MOC_HDRS
 
 IF (WITH_GUI)
   SET(MSSQL_SRCS ${MSSQL_SRCS}
+    qgsmssqlprovidergui.cpp
+    qgsmssqldataitemguiprovider.cpp
     qgsmssqlsourceselect.cpp
     qgsmssqlnewconnection.cpp
   )
   SET(MSSQL_MOC_HDRS ${MSSQL_MOC_HDRS}
+    qgsmssqldataitemguiprovider.h
     qgsmssqlsourceselect.h
     qgsmssqlnewconnection.h
   )

diff --git a/src/providers/mssql/qgsmssqldataitemguiprovider.cpp b/src/providers/mssql/qgsmssqldataitemguiprovider.cpp
@@ -0,0 +1,213 @@
+/***************************************************************************
+  qgsmssqldataitemguiprovider.cpp
+  --------------------------------------
+  Date                 : June 2019
+  Copyright            : (C) 2019 by Martin Dobias
+  Email                : wonder dot sk at gmail dot com
+ ***************************************************************************
+ *                                                                         *
+ *   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 "qgsmssqlconnection.h"
+#include "qgsmssqldataitemguiprovider.h"
+#include "qgsmssqldataitems.h"
+#include "qgsmssqlnewconnection.h"
+#include "qgsmssqlsourceselect.h"
+
+#include <QInputDialog>
+#include <QMessageBox>
+
+
+void QgsMssqlDataItemGuiProvider::populateContextMenu( QgsDataItem *item, QMenu *menu, const QList<QgsDataItem *> &, QgsDataItemGuiContext )
+{
+  if ( QgsMssqlRootItem *rootItem = qobject_cast< QgsMssqlRootItem * >( item ) )
+  {
+    QAction *actionNew = new QAction( tr( "New Connection…" ), menu );
+    connect( actionNew, &QAction::triggered, this, [rootItem] { newConnection( rootItem ); } );
+    menu->addAction( actionNew );
+  }
+  else if ( QgsMssqlConnectionItem *connItem = qobject_cast< QgsMssqlConnectionItem * >( item ) )
+  {
+    QAction *actionRefresh = new QAction( tr( "Refresh" ), menu );
+    connect( actionRefresh, &QAction::triggered, this, [connItem]
+    {
+      connItem->refresh();
+      if ( connItem->parent() )
+        connItem->parent()->refreshConnections();
+    } );
+    menu->addAction( actionRefresh );
+
+    menu->addSeparator();
+
+    QAction *actionEdit = new QAction( tr( "Edit Connection…" ), menu );
+    connect( actionEdit, &QAction::triggered, this, [connItem] { editConnection( connItem ); } );
+    menu->addAction( actionEdit );
+
+    QAction *actionDelete = new QAction( tr( "Delete Connection" ), menu );
+    connect( actionDelete, &QAction::triggered, this, [connItem] { deleteConnection( connItem ); } );
+    menu->addAction( actionDelete );
+
+    menu->addSeparator();
+
+    QAction *actionShowNoGeom = new QAction( tr( "Show Non-spatial Tables" ), menu );
+    actionShowNoGeom->setCheckable( true );
+    actionShowNoGeom->setChecked( connItem->allowGeometrylessTables() );
+    connect( actionShowNoGeom, &QAction::toggled, connItem, &QgsMssqlConnectionItem::setAllowGeometrylessTables );
+    menu->addAction( actionShowNoGeom );
+
+    QAction *actionCreateSchema = new QAction( tr( "Create Schema…" ), menu );
+    connect( actionCreateSchema, &QAction::triggered, this, [connItem] { createSchema( connItem ); } );
+    menu->addAction( actionCreateSchema );
+  }
+  else if ( QgsMssqlSchemaItem *schemaItem = qobject_cast< QgsMssqlSchemaItem * >( item ) )
+  {
+    QAction *actionRefresh = new QAction( tr( "Refresh" ), menu );
+    connect( actionRefresh, &QAction::triggered, this, [schemaItem]
+    {
+      if ( schemaItem->parent() )
+        schemaItem->parent()->refresh();
+    } );
+    menu->addAction( actionRefresh );
+  }
+  else if ( QgsMssqlLayerItem *layerItem = qobject_cast< QgsMssqlLayerItem * >( item ) )
+  {
+    // truncate
+    QAction *actionTruncateLayer = new QAction( tr( "Truncate Table" ), menu );
+    connect( actionTruncateLayer, &QAction::triggered, this, [layerItem] { truncateTable( layerItem ); } );
+    menu->addAction( actionTruncateLayer );
+  }
+}
+
+bool QgsMssqlDataItemGuiProvider::deleteLayer( QgsLayerItem *item, QgsDataItemGuiContext )
+{
+  if ( QgsMssqlLayerItem *layerItem = qobject_cast< QgsMssqlLayerItem * >( item ) )
+  {
+    QgsMssqlConnectionItem *connItem = qobject_cast<QgsMssqlConnectionItem *>( layerItem->parent() ? layerItem->parent()->parent() : nullptr );
+    const QgsMssqlLayerProperty &layerInfo = layerItem->layerInfo();
+
+    if ( QMessageBox::question( nullptr, QObject::tr( "Delete Table" ),
+                                QObject::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;
+
+    QString errCause;
+    bool res = QgsMssqlConnection::dropTable( layerItem->uri(), &errCause );
+    if ( !res )
+    {
+      QMessageBox::warning( nullptr, tr( "Delete Table" ), errCause );
+    }
+    else
+    {
+      QMessageBox::information( nullptr, tr( "Delete Table" ), tr( "Table deleted successfully." ) );
+      if ( connItem )
+        connItem->refresh();
+    }
+    return true;
+  }
+  return false;
+}
+
+bool QgsMssqlDataItemGuiProvider::acceptDrop( QgsDataItem *item, QgsDataItemGuiContext )
+{
+  if ( qobject_cast< QgsMssqlConnectionItem * >( item ) )
+    return true;
+  if ( qobject_cast< QgsMssqlSchemaItem * >( item ) )
+    return true;
+
+  return false;
+}
+
+bool QgsMssqlDataItemGuiProvider::handleDrop( QgsDataItem *item, QgsDataItemGuiContext, const QMimeData *data, Qt::DropAction )
+{
+  if ( QgsMssqlConnectionItem *connItem = qobject_cast< QgsMssqlConnectionItem * >( item ) )
+  {
+    return connItem->handleDrop( data, QString() );
+  }
+  else if ( QgsMssqlSchemaItem *schemaItem = qobject_cast< QgsMssqlSchemaItem * >( item ) )
+  {
+    QgsMssqlConnectionItem *connItem = qobject_cast<QgsMssqlConnectionItem *>( schemaItem->parent() );
+    if ( !connItem )
+      return false;
+
+    return connItem->handleDrop( data, schemaItem->name() );
+  }
+  return false;
+}
+
+void QgsMssqlDataItemGuiProvider::newConnection( QgsDataItem *item )
+{
+  QgsMssqlNewConnection nc( nullptr );
+  if ( nc.exec() )
+  {
+    item->refreshConnections();
+  }
+}
+
+void QgsMssqlDataItemGuiProvider::editConnection( QgsDataItem *item )
+{
+  QgsMssqlNewConnection nc( nullptr, item->name() );
+  if ( nc.exec() )
+  {
+    // the parent should be updated
+    item->parent()->refreshConnections();
+  }
+}
+
+void QgsMssqlDataItemGuiProvider::deleteConnection( QgsDataItem *item )
+{
+  if ( QMessageBox::question( nullptr, QObject::tr( "Delete Connection" ),
+                              QObject::tr( "Are you sure you want to delete the connection to %1?" ).arg( item->name() ),
+                              QMessageBox::Yes | QMessageBox::No, QMessageBox::No ) != QMessageBox::Yes )
+    return;
+
+  QgsMssqlSourceSelect::deleteConnection( item->name() );
+  // the parent should be updated
+  item->parent()->refreshConnections();
+}
+
+void QgsMssqlDataItemGuiProvider::createSchema( QgsMssqlConnectionItem *connItem )
+{
+  QString schemaName = QInputDialog::getText( nullptr, tr( "Create Schema" ), tr( "Schema name:" ) );
+  if ( schemaName.isEmpty() )
+    return;
+
+  QString uri = connItem->connInfo();
+  QString error;
+  if ( !QgsMssqlConnection::createSchema( uri, schemaName, &error ) )
+  {
+    QMessageBox::warning( nullptr, tr( "Create Schema" ), tr( "Unable to create schema %1\n%2" ).arg( schemaName,
+                          error ) );
+    return;
+  }
+
+  connItem->refresh();
+  // the parent should be updated
+  if ( connItem->parent() )
+    connItem->parent()->refreshConnections();
+}
+
+
+void QgsMssqlDataItemGuiProvider::truncateTable( QgsMssqlLayerItem *layerItem )
+{
+  const QgsMssqlLayerProperty &layerInfo = layerItem->layerInfo();
+  if ( QMessageBox::question( nullptr, QObject::tr( "Truncate Table" ),
+                              QObject::tr( "Are you sure you want to truncate [%1].[%2]?\n\nThis will delete all data within the table." ).arg( layerInfo.schemaName, layerInfo.tableName ),
+                              QMessageBox::Yes | QMessageBox::No, QMessageBox::No ) != QMessageBox::Yes )
+    return;
+
+  QString errCause;
+  bool res = QgsMssqlConnection::truncateTable( layerItem->uri(), &errCause );
+  if ( !res )
+  {
+    QMessageBox::warning( nullptr, tr( "Truncate Table" ), errCause );
+  }
+  else
+  {
+    QMessageBox::information( nullptr, tr( "Truncate Table" ), tr( "Table truncated successfully." ) );
+  }
+}
diff --git a/src/providers/mssql/qgsmssqldataitemguiprovider.h b/src/providers/mssql/qgsmssqldataitemguiprovider.h
@@ -0,0 +1,47 @@
+/***************************************************************************
+  qgsmssqldataitemguiprovider.h
+  --------------------------------------
+  Date                 : June 2019
+  Copyright            : (C) 2019 by Martin Dobias
+  Email                : wonder dot sk at gmail dot com
+ ***************************************************************************
+ *                                                                         *
+ *   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 QGSMSSQLDATAITEMGUIPROVIDER_H
+#define QGSMSSQLDATAITEMGUIPROVIDER_H
+
+#include "qgsdataitemguiprovider.h"
+
+class QgsMssqlConnectionItem;
+class QgsMssqlLayerItem;
+
+class QgsMssqlDataItemGuiProvider : public QObject, public QgsDataItemGuiProvider
+{
+    Q_OBJECT
+  public:
+
+    QString name() override { return QStringLiteral( "MSSQL" ); }
+
+    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 createSchema( QgsMssqlConnectionItem *connItem );
+    static void truncateTable( QgsMssqlLayerItem *layerItem );
+};
+
+#endif // QGSMSSQLDATAITEMGUIPROVIDER_H