1 files changed, 31 insertions, 0 deletions
diff --git a/Plugins/DbSqliteWx/dbsqlitewxinstance.cpp b/Plugins/DbSqliteWx/dbsqlitewxinstance.cpp
new file mode 100644
index 0000000..ff5488e
--- /dev/null
+++ b/Plugins/DbSqliteWx/dbsqlitewxinstance.cpp
@@ -0,0 +1,31 @@
+#include "dbsqlitewxinstance.h"
+#include "dbsqlitewx.h"
+
+DbSqliteWxInstance::DbSqliteWxInstance(const QString& name, const QString& path, const QHash<QString, QVariant>& connOptions) :
+    AbstractDb3<WxSQLite>(name, path, connOptions)
+{
+}
+
+void DbSqliteWxInstance::initAfterOpen()
+{
+    SqlQueryPtr res;
+
+    QString key = connOptions[DbSqliteWx::PASSWORD_OPT].toString();
+    if (!key.isEmpty())
+    {
+        res = exec(QString("PRAGMA key = '%1';").arg(key), Flag::NO_LOCK);
+        if (res->isError())
+            qWarning() << "Error while defining WxSqlite3 key:" << res->getErrorText();
+    }
+
+    AbstractDb3<WxSQLite>::initAfterOpen();
+}
+
+QString DbSqliteWxInstance::getAttachSql(Db *otherDb, const QString &generatedAttachName)
+{
+    QString pass = "";
+    if (otherDb->getConnectionOptions().contains(DbSqliteWx::PASSWORD_OPT))
+        pass = otherDb->getConnectionOptions()[DbSqliteWx::PASSWORD_OPT].toString().replace("'", "''");
+
+    return QString("ATTACH '%1' AS %2 KEY '%3';").arg(otherDb->getPath(), generatedAttachName, pass);
+}