Refactor ConnectionConfigurator out of PostgresConn.

qgis · Jul 25, 2023 · f1cd812 · f1cd812
1 parent f217cbe
commit f1cd812
Show file tree

Hide file tree

Showing 14 changed files with 358 additions and 331 deletions.
diff --git a/src/core/CMakeLists.txt b/src/core/CMakeLists.txt
@@ -529,6 +529,7 @@ set(QGIS_CORE_SRCS
   qgsspatialindex.cpp
   qgsspatialindexkdbush.cpp
   qgsspatialindexutils.cpp
+  qgssqlconnectionconfigurator.cpp
   qgssqlexpressioncompiler.cpp
   qgssqliteexpressioncompiler.cpp
   qgssqlstatement.cpp

diff --git a/src/core/qgssqlconnectionconfigurator.cpp b/src/core/qgssqlconnectionconfigurator.cpp
@@ -0,0 +1,43 @@
+/***************************************************************************
+  qgspostgresconnectionconfigurator.cpp  -  connection configurator class for
+  PostgeSQL based connectors.
+                             -------------------
+    begin                : 2023/06/06
+    copyright            : (C) 2023 by Alexey Karandashev
+    email                : reflectored at pm dot me
+ ***************************************************************************/
+
+/***************************************************************************
+ *                                                                         *
+ *   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 "qgssqlconnectionconfigurator.h"
+
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sService = new QgsSettingsEntryString( QStringLiteral( "service" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sHost = new QgsSettingsEntryString( QStringLiteral( "host" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sPort = new QgsSettingsEntryString( QStringLiteral( "port" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sDatabase = new QgsSettingsEntryString( QStringLiteral( "database" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sSessionRole = new QgsSettingsEntryString( QStringLiteral( "session_role" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sUsername = new QgsSettingsEntryString( QStringLiteral( "username" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sPassword = new QgsSettingsEntryString( QStringLiteral( "password" ), sTreeConnections );
+const QgsSettingsEntryEnumFlag<QgsDataSourceUri::SslMode> *QgsPostgreSqlConnectionSettings::sSslMode = new QgsSettingsEntryEnumFlag<QgsDataSourceUri::SslMode>( QStringLiteral( "sslmode" ), sTreeConnections, QgsDataSourceUri::SslPrefer );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sPublicOnly = new QgsSettingsEntryBool( QStringLiteral( "publicOnly" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sGeometryColumnsOnly = new QgsSettingsEntryBool( QStringLiteral( "geometryColumnsOnly" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sDontResolveType = new QgsSettingsEntryBool( QStringLiteral( "dontResolveType" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sAllowGeometrylessTables = new QgsSettingsEntryBool( QStringLiteral( "allowGeometrylessTables" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sEstimatedMetadata = new QgsSettingsEntryBool( QStringLiteral( "estimatedMetadata" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sSaveUsername = new QgsSettingsEntryBool( QStringLiteral( "saveUsername" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sSavePassword = new QgsSettingsEntryBool( QStringLiteral( "savePassword" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sOldSave = new QgsSettingsEntryBool( QStringLiteral( "save" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sProjectsInDatabase = new QgsSettingsEntryBool( QStringLiteral( "projectsInDatabase" ), sTreeConnections );
+const QgsSettingsEntryString *QgsPostgreSqlConnectionSettings::sAuthCfg = new QgsSettingsEntryString( QStringLiteral( "authcfg" ), sTreeConnections );
+const QgsSettingsEntryStringList *QgsPostgreSqlConnectionSettings::sKeys = new QgsSettingsEntryStringList( QStringLiteral( "keys" ), sTreeConnections );
+const QgsSettingsEntryBool *QgsPostgreSqlConnectionSettings::sMetadataInDatabase = new QgsSettingsEntryBool( QStringLiteral( "metadataInDatabase" ), sTreeConnections );
+const QgsSettingsEntryInteger *QgsPostgreSqlConnectionSettings::sDefaultTimeout = new QgsSettingsEntryInteger( QStringLiteral( "default_timeout" ), sTreeConnections, 30 );
+const QString QgsPostgreSqlConnectionSettings::mConnectionTypeName = "PostgreSQL";
+const QString QgsPostgreSqlConnectionSettings::mConnectionTypePort = "5432";
diff --git a/src/core/qgssqlconnectionconfigurator.h b/src/core/qgssqlconnectionconfigurator.h
@@ -0,0 +1,173 @@
+/***************************************************************************
+  qgspostgresconnectionconfigurator.h  -  connection configurator class for
+  PostgeSQL based connectors.
+                             -------------------
+    begin                : 2023/06/06
+    copyright            : (C) 2023 by Alexey Karandashev
+    email                : reflectored at pm dot me
+ ***************************************************************************/
+
+/***************************************************************************
+ *                                                                         *
+ *   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 QGSSQLCONNECTIONCONFIGURATOR_H
+#define QGSSQLCONNECTIONCONFIGURATOR_H
+
+#include <QString>
+#include <QtCore/qstring.h>
+#include <QtCore/qstringliteral.h>
+
+#include "qgsdatasourceuri.h"
+#include "qgssettingsentryenumflag.h"
+#include "qgssettingsentryimpl.h"
+#include "qgssettingstree.h"
+#include "qgssettings.h"
+
+class CORE_EXPORT QgsPostgreSqlConnectionSettings SIP_SKIP
+{
+  public:
+    static inline QgsSettingsTreeNamedListNode *sTreeConnections = QgsSettingsTree::sTreeConnections->createNamedListNode( "PostgreSQL", Qgis::SettingsTreeNodeOption::NamedListSelectedItemSetting );
+    static const QgsSettingsEntryString *sService;
+    static const QgsSettingsEntryString *sHost;
+    static const QgsSettingsEntryString *sPort;
+    static const QgsSettingsEntryString *sDatabase;
+    static const QgsSettingsEntryString *sSessionRole;
+    static const QgsSettingsEntryString *sUsername;
+    static const QgsSettingsEntryString *sPassword;
+    static const QgsSettingsEntryEnumFlag<QgsDataSourceUri::SslMode> *sSslMode;
+    static const QgsSettingsEntryBool *sPublicOnly;
+    static const QgsSettingsEntryBool *sGeometryColumnsOnly;
+    static const QgsSettingsEntryBool *sDontResolveType;
+    static const QgsSettingsEntryBool *sAllowGeometrylessTables;
+    static const QgsSettingsEntryBool *sEstimatedMetadata;
+    static const QgsSettingsEntryBool *sSaveUsername;
+    static const QgsSettingsEntryBool *sSavePassword;
+    static const QgsSettingsEntryBool *sOldSave;
+    static const QgsSettingsEntryBool *sProjectsInDatabase;
+    static const QgsSettingsEntryString *sAuthCfg;
+    static const QgsSettingsEntryStringList *sKeys;
+    static const QgsSettingsEntryBool *sMetadataInDatabase;
+    static const QgsSettingsEntryInteger *sDefaultTimeout;
+    static const QString mConnectionTypeName;
+    static const QString mConnectionTypePort;
+};
+
+template <typename T>
+class CORE_EXPORT QgsSqlConnectionConfigurator SIP_SKIP
+{
+  public:
+    static QStringList connectionList()
+    {
+      return T::sTreeConnections->items();
+    }
+
+    static QString selectedConnection()
+    {
+      return T::sTreeConnections->selectedItem();
+    }
+    static void setSelectedConnection( const QString &name )
+    {
+      T::sTreeConnections->setSelectedItem( name );
+    }
+    static QgsDataSourceUri connUri( const QString &connectionName )
+    {
+      QgsDebugMsgLevel( "theConnName = " + connectionName, 2 );
+
+      const QString service = T::sService->value( connectionName );
+      const QString host = T::sHost->value( connectionName );
+      QString port = T::sPort->value( connectionName );
+      if ( port.length() == 0 )
+      {
+        port = T::mConnectionTypePort;
+      }
+      const QString database = T::sDatabase->value( connectionName );
+
+      const bool estimatedMetadata = useEstimatedMetadata( connectionName );
+      QgsDataSourceUri::SslMode sslmode = T::sSslMode->valueWithDefaultOverride( QgsDataSourceUri::SslPrefer, connectionName );
+
+      QString username;
+      QString password;
+      if ( T::sSaveUsername->value( connectionName ) )
+      {
+        username = T::sUsername->value( connectionName );
+      }
+
+      if ( T::sSavePassword->value( connectionName ) )
+      {
+        password = T::sPassword->value( connectionName );
+      }
+
+      // Old save setting
+      if ( T::sOldSave->exists( connectionName ) )
+      {
+        username = T::sUsername->value( connectionName );
+
+        if ( T::sOldSave->value( connectionName ) )
+        {
+          password = T::sPassword->value( connectionName );
+        }
+      }
+
+      const QString authcfg = T::sAuthCfg->value( connectionName );
+
+      QgsDataSourceUri uri;
+      if ( !service.isEmpty() )
+      {
+        uri.setConnection( service, database, username, password, sslmode, authcfg );
+      }
+      else
+      {
+        uri.setConnection( host, port, database, username, password, sslmode, authcfg );
+      }
+      uri.setUseEstimatedMetadata( estimatedMetadata );
+
+      return uri;
+    }
+
+    static bool publicSchemaOnly( const QString &connectionName )
+    {
+      return T::sPublicOnly->valueWithDefaultOverride( false, connectionName );
+    }
+    static bool geometryColumnsOnly( const QString &connectionName )
+    {
+
+      return T::sGeometryColumnsOnly->valueWithDefaultOverride( false, connectionName );
+    }
+
+    static bool dontResolveType( const QString &connectionName )
+    {
+
+      return T::sDontResolveType->valueWithDefaultOverride( false, connectionName );
+    }
+
+    static bool useEstimatedMetadata( const QString &connectionName )
+    {
+
+      return T::sEstimatedMetadata->valueWithDefaultOverride( false, connectionName );
+    }
+
+    static bool allowGeometrylessTables( const QString &connectionName )
+    {
+      return T::sAllowGeometrylessTables->valueWithDefaultOverride( false, connectionName );
+    }
+    static bool allowProjectsInDatabase( const QString &connectionName )
+    {
+      return T::sProjectsInDatabase->valueWithDefaultOverride( false, connectionName );
+    }
+    static void deleteConnection( const QString &connectionName )
+    {
+      T::sTreeConnections->deleteItem( connectionName );
+    }
+    static bool allowMetadataInDatabase( const QString &connectionName )
+    {
+      return T::sMetadataInDatabase->valueWithDefaultOverride( false, connectionName );
+    }
+};
+
+#endif