summaryrefslogtreecommitdiffstats
path: root/Plugins/DbSqliteWx/dbsqlitewxinstance.cpp
blob: 0335976de2d9021d2f2ca28aac1af41e64fdf554 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#include "dbsqlitewxinstance.h"
#include "dbsqlitewx.h"

DbSqliteWxInstance::DbSqliteWxInstance(const QString& name, const QString& path, const QHash<QString, QVariant>& connOptions) :
    AbstractDb3<WxSQLite>(name, path, connOptions)
{
}

Db* DbSqliteWxInstance::clone() const
{
    return new DbSqliteWxInstance(name, path, connOptions);
}

QString DbSqliteWxInstance::getTypeClassName() const
{
    return "DbSqliteWxInstance";
}

void DbSqliteWxInstance::initAfterOpen()
{
    SqlQueryPtr res;

    QString cipher = connOptions[DbSqliteWx::CIPHER_OPT].toString();
    if (!cipher.isEmpty())
    {
        res = exec(QString("PRAGMA cipher = '%1';").arg(cipher), Flag::NO_LOCK);
        if (res->isError())
            qWarning() << "Error while defining WxSqlite3 cipher:" << res->getErrorText();
    }

    QString pragmas = connOptions[DbSqliteWx::PRAGMAS_OPT].toString();
    QStringList pragmaList = quickSplitQueries(pragmas);
    for (const QString& pragma : pragmaList)
    {
        res = exec(pragma, Flag::NO_LOCK);
        if (res->isError())
            qWarning() << "Error while defining WxSqlite3 pragma" << pragma << ":" << res->getErrorText();
    }

    QString key = connOptions[DbSqliteWx::PASSWORD_OPT].toString();
    if (!key.isEmpty())
    {
        res = exec(QString("PRAGMA key = '%1';").arg(escapeString(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);
}