From 7167ce41b61d2ba2cdb526777a4233eb84a3b66a Mon Sep 17 00:00:00 2001
From: Unit 193 <unit193@ubuntu.com>
Date: Sat, 6 Dec 2014 17:33:25 -0500
Subject: Imported Upstream version 2.99.6

---
 .../coreSQLiteStudio/impl/dbattacherimpl.h         | 94 ++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 SQLiteStudio3/coreSQLiteStudio/impl/dbattacherimpl.h

(limited to 'SQLiteStudio3/coreSQLiteStudio/impl/dbattacherimpl.h')

diff --git a/SQLiteStudio3/coreSQLiteStudio/impl/dbattacherimpl.h b/SQLiteStudio3/coreSQLiteStudio/impl/dbattacherimpl.h
new file mode 100644
index 0000000..49307c6
--- /dev/null
+++ b/SQLiteStudio3/coreSQLiteStudio/impl/dbattacherimpl.h
@@ -0,0 +1,94 @@
+#ifndef DBATTACHERIMPL_H
+#define DBATTACHERIMPL_H
+
+#include "dbattacher.h"
+
+class DbAttacherImpl : public DbAttacher
+{
+    public:
+        /**
+         * @brief Creates attacher with the given database as the main.
+         * @param db Database that the query will be executed on.
+         */
+        DbAttacherImpl(Db* db);
+
+        bool attachDatabases(const QString& query);
+        bool attachDatabases(const QList<SqliteQueryPtr>& queries);
+        bool attachDatabases(SqliteQueryPtr query);
+        void detachDatabases();
+        BiStrHash getDbNameToAttach() const;
+        QString getQuery() const;
+
+    private:
+        /**
+         * @brief Does the actual job, after having all input queries as parsed objects.
+         * @return true on success, false on failure.
+         */
+        bool attachDatabases();
+
+        /**
+         * @brief Finds tokens representing databases in the query.
+         * @return List of tokens. Some tokens have non-printable value (spaces, etc), others are database names.
+         */
+        TokenList getDbTokens();
+
+        /**
+         * @brief Detaches all databases currently attached by the attacher.
+         *
+         * Also clears names mappings.
+         */
+        void detachAttached();
+
+        /**
+         * @brief Generates mapping of database name to its Db object for all registered databases.
+         */
+        void prepareNameToDbMap();
+
+        /**
+         * @brief Groups tokens by the name of database they refer to.
+         * @param dbTokens Tokens representing databases in the query.
+         *
+         * This method is used to learn if some database is used more than once in the query,
+         * so we attach it only once, then replace all tokens referring to it by the attach name.
+         */
+        QHash<QString,TokenList> groupDbTokens(const TokenList& dbTokens);
+
+        /**
+         * @brief Tries to attach all required databases.
+         * @param groupedDbTokens Database tokens grouped by database name, as returned from groupDbTokens().
+         * @return true on success, false on any problem.
+         *
+         * Major problem that can happen is when "<tt>ATTACH 'path to file'</tt>" fails for any reason. In that case
+         * detachAttached() is called and false is returned.
+         */
+        bool attachAllDbs(const QHash<QString,TokenList>& groupedDbTokens);
+
+        /**
+         * @brief Creates token-to-token replace map to update the query.
+         * @param dbTokens Tokens representing databases in the query.
+         * @return Mapping to be used when replacing tokens in the query.
+         */
+        QHash<TokenPtr, TokenPtr> getTokenMapping(const TokenList& dbTokens);
+
+        /**
+         * @brief Replaces tokens in the query.
+         * @param tokenMapping Map of tokens to replace.
+         *
+         * Replacing takes place in token lists of each query in the queries member.
+         */
+        void replaceTokensInQueries(const QHash<TokenPtr, TokenPtr>& tokenMapping);
+
+        QList<SqliteQueryPtr> queries;
+        Db* db = nullptr;
+        Dialect dialect;
+        BiStrHash dbNameToAttach;
+        StrHash<Db*> nameToDbMap;
+};
+
+class DbAttacherDefaultFactory : public DbAttacherFactory
+{
+    public:
+        DbAttacher* create(Db* db);
+};
+
+#endif // DBATTACHERIMPL_H
-- 
cgit v1.2.3