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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
#include "collationmanagerimpl.h"
#include "services/pluginmanager.h"
#include "plugins/scriptingplugin.h"
#include "services/notifymanager.h"
#include "services/dbmanager.h"
#include "common/utils.h"
#include <QDebug>
class CollationFunctionInfoImpl : public ScriptingPlugin::FunctionInfo
{
public:
QString getName() const {return QString();}
QStringList getArguments() const {return {"first", "second"};}
bool getUndefinedArgs() const {return false;}
};
CollationFunctionInfoImpl collationFunctionInfo;
CollationManagerImpl::CollationManagerImpl()
{
init();
}
void CollationManagerImpl::setCollations(const QList<CollationManager::CollationPtr>& newCollations)
{
collations = newCollations;
refreshCollationsByKey();
storeInConfig();
emit collationListChanged();
}
QList<CollationManager::CollationPtr> CollationManagerImpl::getAllCollations() const
{
return collations;
}
CollationManager::CollationPtr CollationManagerImpl::getCollation(const QString &name) const
{
if (!collationsByKey.contains(name))
{
qCritical() << "Could not find requested collation" << name << ".";
return nullptr;
}
return collationsByKey[name];
}
QList<CollationManager::CollationPtr> CollationManagerImpl::getCollationsForDatabase(const QString& dbName) const
{
QList<CollationPtr> results;
for (const CollationPtr& coll : collations)
{
if (coll->allDatabases || coll->databases.contains(dbName, Qt::CaseInsensitive))
results << coll;
}
return results;
}
int CollationManagerImpl::evaluate(const QString& name, const QString& value1, const QString& value2)
{
if (!collationsByKey.contains(name))
{
qWarning() << "Could not find requested collation" << name << ", so using default collation.";
return evaluateDefault(value1, value2);
}
ScriptingPlugin* plugin = PLUGINS->getScriptingPlugin(collationsByKey[name]->lang);
if (!plugin)
{
qWarning() << "Plugin for collation" << name << ", not loaded, so using default collation.";
return evaluateDefault(value1, value2);
}
QString err;
QVariant result = plugin->evaluate(collationsByKey[name]->code, collationFunctionInfo, {value1, value2}, &err);
if (!err.isNull())
{
qWarning() << "Error while evaluating collation:" << err;
return evaluateDefault(value1, value2);
}
bool ok;
int intResult = result.toInt(&ok);
if (!ok)
{
qWarning() << "Not integer result from collation:" << result.toString();
return evaluateDefault(value1, value2);
}
return intResult;
}
int CollationManagerImpl::evaluateDefault(const QString& value1, const QString& value2)
{
return value1.compare(value2, Qt::CaseInsensitive);
}
void CollationManagerImpl::init()
{
loadFromConfig();
refreshCollationsByKey();
}
void CollationManagerImpl::storeInConfig()
{
QVariantList list;
QHash<QString,QVariant> collHash;
for (CollationPtr coll : collations)
{
collHash["name"] = coll->name;
collHash["type"] = coll->type;
collHash["lang"] = coll->lang;
collHash["code"] = coll->code;
collHash["allDatabases"] = coll->allDatabases;
collHash["databases"] =common(DBLIST->getDbNames(), coll->databases);
list << collHash;
}
CFG_CORE.Internal.Collations.set(list);
}
void CollationManagerImpl::loadFromConfig()
{
collations.clear();
QVariantList list = CFG_CORE.Internal.Collations.get();
QHash<QString,QVariant> collHash;
CollationPtr coll;
for (const QVariant& var : list)
{
collHash = var.toHash();
coll = CollationPtr::create();
coll->name = collHash["name"].toString();
if (collHash.contains("type") && collHash["type"].toInt() == CollationType::EXTENSION_BASED)
coll->type = CollationType::EXTENSION_BASED;
else
coll->type = CollationType::FUNCTION_BASED;
coll->lang = updateScriptingQtLang(collHash["lang"].toString());
coll->code = collHash["code"].toString();
coll->databases = collHash["databases"].toStringList();
coll->allDatabases = collHash["allDatabases"].toBool();
collations << coll;
}
}
void CollationManagerImpl::refreshCollationsByKey()
{
collationsByKey.clear();
for (CollationPtr collation : collations)
collationsByKey[collation->name] = collation;
}
QString CollationManagerImpl::updateScriptingQtLang(const QString& lang) const
{
if (lang == "QtScript")
return QStringLiteral("JavaScript");
return lang;
}
|
