1/***************************************************************************
2 * Copyright (C) 2005-2014 by the Quassel Project *
3 * devel@quassel-irc.org *
4 * *
5 * This program is free software; you can redistribute it and/or modify *
6 * it under the terms of the GNU General Public License as published by *
7 * the Free Software Foundation; either version 2 of the License, or *
8 * (at your option) version 3. *
9 * *
10 * This program is distributed in the hope that it will be useful, *
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13 * GNU General Public License for more details. *
14 * *
15 * You should have received a copy of the GNU General Public License *
16 * along with this program; if not, write to the *
17 * Free Software Foundation, Inc., *
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
19 ***************************************************************************/
20
21#include "postgresqlstorage.h"
22
23#include <QtSql>
24
25#include "logger.h"
26#include "network.h"
27#include "quassel.h"
28
29PostgreSqlStorage::PostgreSqlStorage(QObject *parent)
30 : AbstractSqlStorage(parent),
31 _port(-1)
32{
33}
34
35
36PostgreSqlStorage::~PostgreSqlStorage()
37{
38}
39
40
41AbstractSqlMigrationWriter *PostgreSqlStorage::createMigrationWriter()
42{
43 PostgreSqlMigrationWriter *writer = new PostgreSqlMigrationWriter();
44 QVariantMap properties;
45 properties["Username"] = _userName;
46 properties["Password"] = _password;
47 properties["Hostname"] = _hostName;
48 properties["Port"] = _port;
49 properties["Database"] = _databaseName;
50 writer->setConnectionProperties(properties);
51 return writer;
52}
53
54
55bool PostgreSqlStorage::isAvailable() const
56{
57 qDebug() << QSqlDatabase::drivers();
58 if (!QSqlDatabase::isDriverAvailable("QPSQL")) return false;
59 return true;
60}
61
62
63QString PostgreSqlStorage::displayName() const
64{
65 return QString("PostgreSQL");
66}
67
68
69QString PostgreSqlStorage::description() const
70{
71 // FIXME: proper description
72 return tr("PostgreSQL Turbo Bomber HD!");
73}
74
75
76QStringList PostgreSqlStorage::setupKeys() const
77{
78 QStringList keys;
79 keys << "Username"
80 << "Password"
81 << "Hostname"
82 << "Port"
83 << "Database";
84 return keys;
85}
86
87
88QVariantMap PostgreSqlStorage::setupDefaults() const
89{
90 QVariantMap map;
91 map["Username"] = QVariant(QString("quassel"));
92 map["Hostname"] = QVariant(QString("localhost"));
93 map["Port"] = QVariant(5432);
94 map["Database"] = QVariant(QString("quassel"));
95 return map;
96}
97
98
99bool PostgreSqlStorage::initDbSession(QSqlDatabase &db)
100{
101 // check whether the Qt driver performs string escaping or not.
102 // i.e. test if it doubles slashes.
103 QSqlField testField;
104 testField.setType(QVariant::String);
105 testField.setValue("\\");
106 QString formattedString = db.driver()->formatValue(testField);
107 switch(formattedString.count('\\')) {
108 case 2:
109 // yes it does... and we cannot do anything to change the behavior of Qt.
110 // If this is a legacy DB (Postgres < 8.2), then everything is already ok,
111 // as this is the expected behavior.
112 // If it is a newer version, switch to legacy mode.
113
114 quWarning() << "Switching Postgres to legacy mode. (set standard conforming strings to off)";
115 // If the following calls fail, it is a legacy DB anyways, so it doesn't matter
116 // and no need to check the outcome.
117 db.exec("set standard_conforming_strings = off");
118 db.exec("set escape_string_warning = off");
119 break;
120 case 1:
121 // ok, so Qt does not escape...
122 // That means we have to ensure that postgres uses standard conforming strings...
123 {
124 QSqlQuery query = db.exec("set standard_conforming_strings = on");
125 if (query.lastError().isValid()) {
126 // We cannot enable standard conforming strings...
127 // since Quassel does no escaping by itself, this would yield a major vulnerability.
128 quError() << "Failed to enable standard_conforming_strings for the Postgres db!";
129 return false;
130 }
131 }
132 break;
133 default:
134 // The slash got replaced with 0 or more than 2 slashes! o_O
135 quError() << "Your version of Qt does something _VERY_ strange to slashes in QSqlQueries! You should consult your trusted doctor!";
136 return false;
137 break;
138 }
139 return true;
140}
141
142
143void PostgreSqlStorage::setConnectionProperties(const QVariantMap &properties)
144{
145 _userName = properties["Username"].toString();
146 _password = properties["Password"].toString();
147 _hostName = properties["Hostname"].toString();
148 _port = properties["Port"].toInt();
149 _databaseName = properties["Database"].toString();
150}
151
152
153int PostgreSqlStorage::installedSchemaVersion()
154{
155 QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
156 if (query.first())
157 return query.value(0).toInt();
158
159 // maybe it's really old... (schema version 0)
160 query = logDb().exec("SELECT MAX(version) FROM coreinfo");
161 if (query.first())
162 return query.value(0).toInt();
163
164 return AbstractSqlStorage::installedSchemaVersion();
165}
166
167
168bool PostgreSqlStorage::updateSchemaVersion(int newVersion)
169{
170 QSqlQuery query(logDb());
171 query.prepare("UPDATE coreinfo SET value = :version WHERE key = 'schemaversion'");
172 query.bindValue(":version", newVersion);
173 query.exec();
174
175 bool success = true;
176 if (query.lastError().isValid()) {
177 qCritical() << "PostgreSqlStorage::updateSchemaVersion(int): Updating schema version failed!";
178 success = false;
179 }
180 return success;
181}
182
183
184bool PostgreSqlStorage::setupSchemaVersion(int version)
185{
186 QSqlQuery query(logDb());
187 query.prepare("INSERT INTO coreinfo (key, value) VALUES ('schemaversion', :version)");
188 query.bindValue(":version", version);
189 query.exec();
190
191 bool success = true;
192 if (query.lastError().isValid()) {
193 qCritical() << "PostgreSqlStorage::setupSchemaVersion(int): Updating schema version failed!";
194 success = false;
195 }
196 return success;
197}
198
199
200UserId PostgreSqlStorage::addUser(const QString &user, const QString &password)
201{
202 QSqlQuery query(logDb());
203 query.prepare(queryString("insert_quasseluser"));
204 query.bindValue(":username", user);
205 query.bindValue(":password", cryptedPassword(password));
206 safeExec(query);
207 if (!watchQuery(query))
208 return 0;
209
210 query.first();
211 UserId uid = query.value(0).toInt();
212 emit userAdded(uid, user);
213 return uid;
214}
215
216
217bool PostgreSqlStorage::updateUser(UserId user, const QString &password)
218{
219 QSqlQuery query(logDb());
220 query.prepare(queryString("update_userpassword"));
221 query.bindValue(":userid", user.toInt());
222 query.bindValue(":password", cryptedPassword(password));
223 safeExec(query);
224 return query.numRowsAffected() != 0;
225}
226
227
228void PostgreSqlStorage::renameUser(UserId user, const QString &newName)
229{
230 QSqlQuery query(logDb());
231 query.prepare(queryString("update_username"));
232 query.bindValue(":userid", user.toInt());
233 query.bindValue(":username", newName);
234 safeExec(query);
235 emit userRenamed(user, newName);
236}
237
238
239UserId PostgreSqlStorage::validateUser(const QString &user, const QString &password)
240{
241 QSqlQuery query(logDb());
242 query.prepare(queryString("select_authuser"));
243 query.bindValue(":username", user);
244 query.bindValue(":password", cryptedPassword(password));
245 safeExec(query);
246
247 if (query.first()) {
248 return query.value(0).toInt();
249 }
250 else {
251 return 0;
252 }
253}
254
255
256UserId PostgreSqlStorage::getUserId(const QString &user)
257{
258 QSqlQuery query(logDb());
259 query.prepare(queryString("select_userid"));
260 query.bindValue(":username", user);
261 safeExec(query);
262
263 if (query.first()) {
264 return query.value(0).toInt();
265 }
266 else {
267 return 0;
268 }
269}
270
271
272UserId PostgreSqlStorage::internalUser()
273{
274 QSqlQuery query(logDb());
275 query.prepare(queryString("select_internaluser"));
276 safeExec(query);
277
278 if (query.first()) {
279 return query.value(0).toInt();
280 }
281 else {
282 return 0;
283 }
284}
285
286
287void PostgreSqlStorage::delUser(UserId user)
288{
289 QSqlDatabase db = logDb();
290 if (!db.transaction()) {
291 qWarning() << "PostgreSqlStorage::delUser(): cannot start transaction!";
292 return;
293 }
294
295 QSqlQuery query(db);
296 query.prepare(queryString("delete_quasseluser"));
297 query.bindValue(":userid", user.toInt());
298 safeExec(query);
299 if (!watchQuery(query)) {
300 db.rollback();
301 return;
302 }
303 else {
304 db.commit();
305 emit userRemoved(user);
306 }
307}
308
309
310void PostgreSqlStorage::setUserSetting(UserId userId, const QString &settingName, const QVariant &data)
311{
312 QByteArray rawData;
313 QDataStream out(&rawData, QIODevice::WriteOnly);
314 out.setVersion(QDataStream::Qt_4_2);
315 out << data;
316
317 QSqlDatabase db = logDb();
318 QSqlQuery selectQuery(db);
319 selectQuery.prepare(queryString("select_user_setting"));
320 selectQuery.bindValue(":userid", userId.toInt());
321 selectQuery.bindValue(":settingname", settingName);
322 safeExec(selectQuery);
323
324 QString setQueryString;
325 if (!selectQuery.first()) {
326 setQueryString = queryString("insert_user_setting");
327 }
328 else {
329 setQueryString = queryString("update_user_setting");
330 }
331
332 QSqlQuery setQuery(db);
333 setQuery.prepare(setQueryString);
334 setQuery.bindValue(":userid", userId.toInt());
335 setQuery.bindValue(":settingname", settingName);
336 setQuery.bindValue(":settingvalue", rawData);
337 safeExec(setQuery);
338}
339
340
341QVariant PostgreSqlStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData)
342{
343 QSqlQuery query(logDb());
344 query.prepare(queryString("select_user_setting"));
345 query.bindValue(":userid", userId.toInt());
346 query.bindValue(":settingname", settingName);
347 safeExec(query);
348
349 if (query.first()) {
350 QVariant data;
351 QByteArray rawData = query.value(0).toByteArray();
352 QDataStream in(&rawData, QIODevice::ReadOnly);
353 in.setVersion(QDataStream::Qt_4_2);
354 in >> data;
355 return data;
356 }
357 else {
358 return defaultData;
359 }
360}
361
362
363IdentityId PostgreSqlStorage::createIdentity(UserId user, CoreIdentity &identity)
364{
365 IdentityId identityId;
366
367 QSqlDatabase db = logDb();
368 if (!db.transaction()) {
369 qWarning() << "PostgreSqlStorage::createIdentity(): Unable to start Transaction!";
370 qWarning() << " -" << qPrintable(db.lastError().text());
371 return identityId;
372 }
373
374 QSqlQuery query(db);
375 query.prepare(queryString("insert_identity"));
376 query.bindValue(":userid", user.toInt());
377 query.bindValue(":identityname", identity.identityName());
378 query.bindValue(":realname", identity.realName());
379 query.bindValue(":awaynick", identity.awayNick());
380 query.bindValue(":awaynickenabled", identity.awayNickEnabled());
381 query.bindValue(":awayreason", identity.awayReason());
382 query.bindValue(":awayreasonenabled", identity.awayReasonEnabled());
383 query.bindValue(":autoawayenabled", identity.awayReasonEnabled());
384 query.bindValue(":autoawaytime", identity.autoAwayTime());
385 query.bindValue(":autoawayreason", identity.autoAwayReason());
386 query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled());
387 query.bindValue(":detachawayenabled", identity.detachAwayEnabled());
388 query.bindValue(":detachawayreason", identity.detachAwayReason());
389 query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled());
390 query.bindValue(":ident", identity.ident());
391 query.bindValue(":kickreason", identity.kickReason());
392 query.bindValue(":partreason", identity.partReason());
393 query.bindValue(":quitreason", identity.quitReason());
394#ifdef HAVE_SSL
395 query.bindValue(":sslcert", identity.sslCert().toPem());
396 query.bindValue(":sslkey", identity.sslKey().toPem());
397#else
398 query.bindValue(":sslcert", QByteArray());
399 query.bindValue(":sslkey", QByteArray());
400#endif
401 safeExec(query);
402 if (query.lastError().isValid()) {
403 watchQuery(query);
404 db.rollback();
405 return IdentityId();
406 }
407
408 query.first();
409 identityId = query.value(0).toInt();
410 identity.setId(identityId);
411
412 if (!identityId.isValid()) {
413 watchQuery(query);
414 db.rollback();
415 return IdentityId();
416 }
417
418 QSqlQuery insertNickQuery(db);
419 insertNickQuery.prepare(queryString("insert_nick"));
420 foreach(QString nick, identity.nicks()) {
421 insertNickQuery.bindValue(":identityid", identityId.toInt());
422 insertNickQuery.bindValue(":nick", nick);
423 safeExec(insertNickQuery);
424 if (!watchQuery(insertNickQuery)) {
425 db.rollback();
426 return IdentityId();
427 }
428 }
429
430 if (!db.commit()) {
431 qWarning() << "PostgreSqlStorage::createIdentity(): committing data failed!";
432 qWarning() << " -" << qPrintable(db.lastError().text());
433 return IdentityId();
434 }
435 return identityId;
436}
437
438
439bool PostgreSqlStorage::updateIdentity(UserId user, const CoreIdentity &identity)
440{
441 QSqlDatabase db = logDb();
442 if (!db.transaction()) {
443 qWarning() << "PostgreSqlStorage::updateIdentity(): Unable to start Transaction!";
444 qWarning() << " -" << qPrintable(db.lastError().text());
445 return false;
446 }
447
448 QSqlQuery checkQuery(db);
449 checkQuery.prepare(queryString("select_checkidentity"));
450 checkQuery.bindValue(":identityid", identity.id().toInt());
451 checkQuery.bindValue(":userid", user.toInt());
452 safeExec(checkQuery);
453
454 // there should be exactly one identity for the given id and user
455 if (!checkQuery.first() || checkQuery.value(0).toInt() != 1) {
456 db.rollback();
457 return false;
458 }
459
460 QSqlQuery query(db);
461 query.prepare(queryString("update_identity"));
462 query.bindValue(":identityname", identity.identityName());
463 query.bindValue(":realname", identity.realName());
464 query.bindValue(":awaynick", identity.awayNick());
465 query.bindValue(":awaynickenabled", identity.awayNickEnabled());
466 query.bindValue(":awayreason", identity.awayReason());
467 query.bindValue(":awayreasonenabled", identity.awayReasonEnabled());
468 query.bindValue(":autoawayenabled", identity.awayReasonEnabled());
469 query.bindValue(":autoawaytime", identity.autoAwayTime());
470 query.bindValue(":autoawayreason", identity.autoAwayReason());
471 query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled());
472 query.bindValue(":detachawayenabled", identity.detachAwayEnabled());
473 query.bindValue(":detachawayreason", identity.detachAwayReason());
474 query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled());
475 query.bindValue(":ident", identity.ident());
476 query.bindValue(":kickreason", identity.kickReason());
477 query.bindValue(":partreason", identity.partReason());
478 query.bindValue(":quitreason", identity.quitReason());
479#ifdef HAVE_SSL
480 query.bindValue(":sslcert", identity.sslCert().toPem());
481 query.bindValue(":sslkey", identity.sslKey().toPem());
482#else
483 query.bindValue(":sslcert", QByteArray());
484 query.bindValue(":sslkey", QByteArray());
485#endif
486 query.bindValue(":identityid", identity.id().toInt());
487
488 safeExec(query);
489 if (!watchQuery(query)) {
490 db.rollback();
491 return false;
492 }
493
494 QSqlQuery deleteNickQuery(db);
495 deleteNickQuery.prepare(queryString("delete_nicks"));
496 deleteNickQuery.bindValue(":identityid", identity.id().toInt());
497 safeExec(deleteNickQuery);
498 if (!watchQuery(deleteNickQuery)) {
499 db.rollback();
500 return false;
501 }
502
503 QSqlQuery insertNickQuery(db);
504 insertNickQuery.prepare(queryString("insert_nick"));
505 foreach(QString nick, identity.nicks()) {
506 insertNickQuery.bindValue(":identityid", identity.id().toInt());
507 insertNickQuery.bindValue(":nick", nick);
508 safeExec(insertNickQuery);
509 if (!watchQuery(insertNickQuery)) {
510 db.rollback();
511 return false;
512 }
513 }
514
515 if (!db.commit()) {
516 qWarning() << "PostgreSqlStorage::updateIdentity(): committing data failed!";
517 qWarning() << " -" << qPrintable(db.lastError().text());
518 return false;
519 }
520 return true;
521}
522
523
524void PostgreSqlStorage::removeIdentity(UserId user, IdentityId identityId)
525{
526 QSqlDatabase db = logDb();
527 if (!db.transaction()) {
528 qWarning() << "PostgreSqlStorage::removeIdentity(): Unable to start Transaction!";
529 qWarning() << " -" << qPrintable(db.lastError().text());
530 return;
531 }
532
533 QSqlQuery query(db);
534 query.prepare(queryString("delete_identity"));
535 query.bindValue(":identityid", identityId.toInt());
536 query.bindValue(":userid", user.toInt());
537 safeExec(query);
538 if (!watchQuery(query)) {
539 db.rollback();
540 }
541 else {
542 db.commit();
543 }
544}
545
546
547QList<CoreIdentity> PostgreSqlStorage::identities(UserId user)
548{
549 QList<CoreIdentity> identities;
550
551 QSqlDatabase db = logDb();
552 if (!beginReadOnlyTransaction(db)) {
553 qWarning() << "PostgreSqlStorage::identites(): cannot start read only transaction!";
554 qWarning() << " -" << qPrintable(db.lastError().text());
555 return identities;
556 }
557
558 QSqlQuery query(db);
559 query.prepare(queryString("select_identities"));
560 query.bindValue(":userid", user.toInt());
561
562 QSqlQuery nickQuery(db);
563 nickQuery.prepare(queryString("select_nicks"));
564
565 safeExec(query);
566
567 while (query.next()) {
568 CoreIdentity identity(IdentityId(query.value(0).toInt()));
569
570 identity.setIdentityName(query.value(1).toString());
571 identity.setRealName(query.value(2).toString());
572 identity.setAwayNick(query.value(3).toString());
573 identity.setAwayNickEnabled(!!query.value(4).toInt());
574 identity.setAwayReason(query.value(5).toString());
575 identity.setAwayReasonEnabled(!!query.value(6).toInt());
576 identity.setAutoAwayEnabled(!!query.value(7).toInt());
577 identity.setAutoAwayTime(query.value(8).toInt());
578 identity.setAutoAwayReason(query.value(9).toString());
579 identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
580 identity.setDetachAwayEnabled(!!query.value(11).toInt());
581 identity.setDetachAwayReason(query.value(12).toString());
582 identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
583 identity.setIdent(query.value(14).toString());
584 identity.setKickReason(query.value(15).toString());
585 identity.setPartReason(query.value(16).toString());
586 identity.setQuitReason(query.value(17).toString());
587#ifdef HAVE_SSL
588 identity.setSslCert(query.value(18).toByteArray());
589 identity.setSslKey(query.value(19).toByteArray());
590#endif
591
592 nickQuery.bindValue(":identityid", identity.id().toInt());
593 QList<QString> nicks;
594 safeExec(nickQuery);
595 watchQuery(nickQuery);
596 while (nickQuery.next()) {
597 nicks << nickQuery.value(0).toString();
598 }
599 identity.setNicks(nicks);
600 identities << identity;
601 }
602 db.commit();
603 return identities;
604}
605
606
607NetworkId PostgreSqlStorage::createNetwork(UserId user, const NetworkInfo &info)
608{
609 NetworkId networkId;
610
611 QSqlDatabase db = logDb();
612 if (!db.transaction()) {
613 qWarning() << "PostgreSqlStorage::createNetwork(): failed to begin transaction!";
614 qWarning() << " -" << qPrintable(db.lastError().text());
615 return false;
616 }
617
618 QSqlQuery query(db);
619 query.prepare(queryString("insert_network"));
620 query.bindValue(":userid", user.toInt());
621 bindNetworkInfo(query, info);
622 safeExec(query);
623 if (query.lastError().isValid()) {
624 watchQuery(query);
625 db.rollback();
626 return NetworkId();
627 }
628
629 query.first();
630 networkId = query.value(0).toInt();
631
632 if (!networkId.isValid()) {
633 watchQuery(query);
634 db.rollback();
635 return NetworkId();
636 }
637
638 QSqlQuery insertServersQuery(db);
639 insertServersQuery.prepare(queryString("insert_server"));
640 foreach(Network::Server server, info.serverList) {
641 insertServersQuery.bindValue(":userid", user.toInt());
642 insertServersQuery.bindValue(":networkid", networkId.toInt());
643 bindServerInfo(insertServersQuery, server);
644 safeExec(insertServersQuery);
645 if (!watchQuery(insertServersQuery)) {
646 db.rollback();
647 return NetworkId();
648 }
649 }
650
651 if (!db.commit()) {
652 qWarning() << "PostgreSqlStorage::createNetwork(): committing data failed!";
653 qWarning() << " -" << qPrintable(db.lastError().text());
654 return NetworkId();
655 }
656 return networkId;
657}
658
659
660void PostgreSqlStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info)
661{
662 query.bindValue(":networkname", info.networkName);
663 query.bindValue(":identityid", info.identity.isValid() ? info.identity.toInt() : QVariant());
664 query.bindValue(":encodingcodec", QString(info.codecForEncoding));
665 query.bindValue(":decodingcodec", QString(info.codecForDecoding));
666 query.bindValue(":servercodec", QString(info.codecForServer));
667 query.bindValue(":userandomserver", info.useRandomServer);
668 query.bindValue(":perform", info.perform.join("\n"));
669 query.bindValue(":useautoidentify", info.useAutoIdentify);
670 query.bindValue(":autoidentifyservice", info.autoIdentifyService);
671 query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
672 query.bindValue(":usesasl", info.useSasl);
673 query.bindValue(":saslaccount", info.saslAccount);
674 query.bindValue(":saslpassword", info.saslPassword);
675 query.bindValue(":useautoreconnect", info.useAutoReconnect);
676 query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
677 query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
678 query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries);
679 query.bindValue(":rejoinchannels", info.rejoinChannels);
680 if (info.networkId.isValid())
681 query.bindValue(":networkid", info.networkId.toInt());
682}
683
684
685void PostgreSqlStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server)
686{
687 query.bindValue(":hostname", server.host);
688 query.bindValue(":port", server.port);
689 query.bindValue(":password", server.password);
690 query.bindValue(":ssl", server.useSsl);
691 query.bindValue(":sslversion", server.sslVersion);
692 query.bindValue(":useproxy", server.useProxy);
693 query.bindValue(":proxytype", server.proxyType);
694 query.bindValue(":proxyhost", server.proxyHost);
695 query.bindValue(":proxyport", server.proxyPort);
696 query.bindValue(":proxyuser", server.proxyUser);
697 query.bindValue(":proxypass", server.proxyPass);
698}
699
700
701bool PostgreSqlStorage::updateNetwork(UserId user, const NetworkInfo &info)
702{
703 QSqlDatabase db = logDb();
704 if (!db.transaction()) {
705 qWarning() << "PostgreSqlStorage::updateNetwork(): failed to begin transaction!";
706 qWarning() << " -" << qPrintable(db.lastError().text());
707 return false;
708 }
709
710 QSqlQuery updateQuery(db);
711 updateQuery.prepare(queryString("update_network"));
712 updateQuery.bindValue(":userid", user.toInt());
713 bindNetworkInfo(updateQuery, info);
714 safeExec(updateQuery);
715 if (!watchQuery(updateQuery)) {
716 db.rollback();
717 return false;
718 }
719 if (updateQuery.numRowsAffected() != 1) {
720 // seems this is not our network...
721 db.rollback();
722 return false;
723 }
724
725 QSqlQuery dropServersQuery(db);
726 dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
727 dropServersQuery.bindValue(":networkid", info.networkId.toInt());
728 safeExec(dropServersQuery);
729 if (!watchQuery(dropServersQuery)) {
730 db.rollback();
731 return false;
732 }
733
734 QSqlQuery insertServersQuery(db);
735 insertServersQuery.prepare(queryString("insert_server"));
736 foreach(Network::Server server, info.serverList) {
737 insertServersQuery.bindValue(":userid", user.toInt());
738 insertServersQuery.bindValue(":networkid", info.networkId.toInt());
739 bindServerInfo(insertServersQuery, server);
740 safeExec(insertServersQuery);
741 if (!watchQuery(insertServersQuery)) {
742 db.rollback();
743 return false;
744 }
745 }
746
747 if (!db.commit()) {
748 qWarning() << "PostgreSqlStorage::updateNetwork(): committing data failed!";
749 qWarning() << " -" << qPrintable(db.lastError().text());
750 return false;
751 }
752 return true;
753}
754
755
756bool PostgreSqlStorage::removeNetwork(UserId user, const NetworkId &networkId)
757{
758 QSqlDatabase db = logDb();
759 if (!db.transaction()) {
760 qWarning() << "PostgreSqlStorage::removeNetwork(): cannot start transaction!";
761 qWarning() << " -" << qPrintable(db.lastError().text());
762 return false;
763 }
764
765 QSqlQuery query(db);
766 query.prepare(queryString("delete_network"));
767 query.bindValue(":userid", user.toInt());
768 query.bindValue(":networkid", networkId.toInt());
769 safeExec(query);
770 if (!watchQuery(query)) {
771 db.rollback();
772 return false;
773 }
774
775 db.commit();
776 return true;
777}
778
779
780QList<NetworkInfo> PostgreSqlStorage::networks(UserId user)
781{
782 QList<NetworkInfo> nets;
783
784 QSqlDatabase db = logDb();
785 if (!beginReadOnlyTransaction(db)) {
786 qWarning() << "PostgreSqlStorage::networks(): cannot start read only transaction!";
787 qWarning() << " -" << qPrintable(db.lastError().text());
788 return nets;
789 }
790
791 QSqlQuery networksQuery(db);
792 networksQuery.prepare(queryString("select_networks_for_user"));
793 networksQuery.bindValue(":userid", user.toInt());
794
795 QSqlQuery serversQuery(db);
796 serversQuery.prepare(queryString("select_servers_for_network"));
797
798 safeExec(networksQuery);
799 if (!watchQuery(networksQuery)) {
800 db.rollback();
801 return nets;
802 }
803
804 while (networksQuery.next()) {
805 NetworkInfo net;
806 net.networkId = networksQuery.value(0).toInt();
807 net.networkName = networksQuery.value(1).toString();
808 net.identity = networksQuery.value(2).toInt();
809 net.codecForServer = networksQuery.value(3).toString().toLatin1();
810 net.codecForEncoding = networksQuery.value(4).toString().toLatin1();
811 net.codecForDecoding = networksQuery.value(5).toString().toLatin1();
812 net.useRandomServer = networksQuery.value(6).toBool();
813 net.perform = networksQuery.value(7).toString().split("\n");
814 net.useAutoIdentify = networksQuery.value(8).toBool();
815 net.autoIdentifyService = networksQuery.value(9).toString();
816 net.autoIdentifyPassword = networksQuery.value(10).toString();
817 net.useAutoReconnect = networksQuery.value(11).toBool();
818 net.autoReconnectInterval = networksQuery.value(12).toUInt();
819 net.autoReconnectRetries = networksQuery.value(13).toInt();
820 net.unlimitedReconnectRetries = networksQuery.value(14).toBool();
821 net.rejoinChannels = networksQuery.value(15).toBool();
822 net.useSasl = networksQuery.value(16).toBool();
823 net.saslAccount = networksQuery.value(17).toString();
824 net.saslPassword = networksQuery.value(18).toString();
825
826 serversQuery.bindValue(":networkid", net.networkId.toInt());
827 safeExec(serversQuery);
828 if (!watchQuery(serversQuery)) {
829 db.rollback();
830 return nets;
831 }
832
833 Network::ServerList servers;
834 while (serversQuery.next()) {
835 Network::Server server;
836 server.host = serversQuery.value(0).toString();
837 server.port = serversQuery.value(1).toUInt();
838 server.password = serversQuery.value(2).toString();
839 server.useSsl = serversQuery.value(3).toBool();
840 server.sslVersion = serversQuery.value(4).toInt();
841 server.useProxy = serversQuery.value(5).toBool();
842 server.proxyType = serversQuery.value(6).toInt();
843 server.proxyHost = serversQuery.value(7).toString();
844 server.proxyPort = serversQuery.value(8).toUInt();
845 server.proxyUser = serversQuery.value(9).toString();
846 server.proxyPass = serversQuery.value(10).toString();
847 servers << server;
848 }
849 net.serverList = servers;
850 nets << net;
851 }
852 db.commit();
853 return nets;
854}
855
856
857QList<NetworkId> PostgreSqlStorage::connectedNetworks(UserId user)
858{
859 QList<NetworkId> connectedNets;
860
861 QSqlDatabase db = logDb();
862 if (!beginReadOnlyTransaction(db)) {
863 qWarning() << "PostgreSqlStorage::connectedNetworks(): cannot start read only transaction!";
864 qWarning() << " -" << qPrintable(db.lastError().text());
865 return connectedNets;
866 }
867
868 QSqlQuery query(db);
869 query.prepare(queryString("select_connected_networks"));
870 query.bindValue(":userid", user.toInt());
871 safeExec(query);
872 watchQuery(query);
873
874 while (query.next()) {
875 connectedNets << query.value(0).toInt();
876 }
877
878 db.commit();
879 return connectedNets;
880}
881
882
883void PostgreSqlStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected)
884{
885 QSqlQuery query(logDb());
886 query.prepare(queryString("update_network_connected"));
887 query.bindValue(":userid", user.toInt());
888 query.bindValue(":networkid", networkId.toInt());
889 query.bindValue(":connected", isConnected);
890 safeExec(query);
891 watchQuery(query);
892}
893
894
895QHash<QString, QString> PostgreSqlStorage::persistentChannels(UserId user, const NetworkId &networkId)
896{
897 QHash<QString, QString> persistentChans;
898
899 QSqlDatabase db = logDb();
900 if (!beginReadOnlyTransaction(db)) {
901 qWarning() << "PostgreSqlStorage::persistentChannels(): cannot start read only transaction!";
902 qWarning() << " -" << qPrintable(db.lastError().text());
903 return persistentChans;
904 }
905
906 QSqlQuery query(db);
907 query.prepare(queryString("select_persistent_channels"));
908 query.bindValue(":userid", user.toInt());
909 query.bindValue(":networkid", networkId.toInt());
910 safeExec(query);
911 watchQuery(query);
912
913 while (query.next()) {
914 persistentChans[query.value(0).toString()] = query.value(1).toString();
915 }
916
917 db.commit();
918 return persistentChans;
919}
920
921
922void PostgreSqlStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined)
923{
924 QSqlQuery query(logDb());
925 query.prepare(queryString("update_buffer_persistent_channel"));
926 query.bindValue(":userid", user.toInt());
927 query.bindValue(":networkId", networkId.toInt());
928 query.bindValue(":buffercname", channel.toLower());
929 query.bindValue(":joined", isJoined);
930 safeExec(query);
931 watchQuery(query);
932}
933
934
935void PostgreSqlStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key)
936{
937 QSqlQuery query(logDb());
938 query.prepare(queryString("update_buffer_set_channel_key"));
939 query.bindValue(":userid", user.toInt());
940 query.bindValue(":networkId", networkId.toInt());
941 query.bindValue(":buffercname", channel.toLower());
942 query.bindValue(":key", key);
943 safeExec(query);
944 watchQuery(query);
945}
946
947
948QString PostgreSqlStorage::awayMessage(UserId user, NetworkId networkId)
949{
950 QSqlQuery query(logDb());
951 query.prepare(queryString("select_network_awaymsg"));
952 query.bindValue(":userid", user.toInt());
953 query.bindValue(":networkid", networkId.toInt());
954 safeExec(query);
955 watchQuery(query);
956 QString awayMsg;
957 if (query.first())
958 awayMsg = query.value(0).toString();
959 return awayMsg;
960}
961
962
963void PostgreSqlStorage::setAwayMessage(UserId user, NetworkId networkId, const QString &awayMsg)
964{
965 QSqlQuery query(logDb());
966 query.prepare(queryString("update_network_set_awaymsg"));
967 query.bindValue(":userid", user.toInt());
968 query.bindValue(":networkid", networkId.toInt());
969 query.bindValue(":awaymsg", awayMsg);
970 safeExec(query);
971 watchQuery(query);
972}
973
974
975QString PostgreSqlStorage::userModes(UserId user, NetworkId networkId)
976{
977 QSqlQuery query(logDb());
978 query.prepare(queryString("select_network_usermode"));
979 query.bindValue(":userid", user.toInt());
980 query.bindValue(":networkid", networkId.toInt());
981 safeExec(query);
982 watchQuery(query);
983 QString modes;
984 if (query.first())
985 modes = query.value(0).toString();
986 return modes;
987}
988
989
990void PostgreSqlStorage::setUserModes(UserId user, NetworkId networkId, const QString &userModes)
991{
992 QSqlQuery query(logDb());
993 query.prepare(queryString("update_network_set_usermode"));
994 query.bindValue(":userid", user.toInt());
995 query.bindValue(":networkid", networkId.toInt());
996 query.bindValue(":usermode", userModes);
997 safeExec(query);
998 watchQuery(query);
999}
1000
1001
1002BufferInfo PostgreSqlStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create)
1003{
1004 QSqlDatabase db = logDb();
1005 if (!db.transaction()) {
1006 qWarning() << "PostgreSqlStorage::bufferInfo(): cannot start read only transaction!";
1007 qWarning() << " -" << qPrintable(db.lastError().text());
1008 return BufferInfo();
1009 }
1010
1011 QSqlQuery query(db);
1012 query.prepare(queryString("select_bufferByName"));
1013 query.bindValue(":networkid", networkId.toInt());
1014 query.bindValue(":userid", user.toInt());
1015 query.bindValue(":buffercname", buffer.toLower());
1016 safeExec(query);
1017
1018 if (query.first()) {
1019 BufferInfo bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
1020 if (query.next()) {
1021 qCritical() << "PostgreSqlStorage::bufferInfo(): received more then one Buffer!";
1022 qCritical() << " Query:" << query.lastQuery();
1023 qCritical() << " bound Values:";
1024 QList<QVariant> list = query.boundValues().values();
1025 for (int i = 0; i < list.size(); ++i)
1026 qCritical() << i << ":" << list.at(i).toString().toLatin1().data();
1027 Q_ASSERT(false);
1028 }
1029 db.commit();
1030 return bufferInfo;
1031 }
1032
1033 if (!create) {
1034 db.rollback();
1035 return BufferInfo();
1036 }
1037
1038 QSqlQuery createQuery(db);
1039 createQuery.prepare(queryString("insert_buffer"));
1040 createQuery.bindValue(":userid", user.toInt());
1041 createQuery.bindValue(":networkid", networkId.toInt());
1042 createQuery.bindValue(":buffertype", (int)type);
1043 createQuery.bindValue(":buffername", buffer);
1044 createQuery.bindValue(":buffercname", buffer.toLower());
1045 createQuery.bindValue(":joined", type & BufferInfo::ChannelBuffer ? true : false);
1046
1047 safeExec(createQuery);
1048
1049 if (createQuery.lastError().isValid()) {
1050 qWarning() << "PostgreSqlStorage::bufferInfo(): unable to create buffer";
1051 watchQuery(createQuery);
1052 db.rollback();
1053 return BufferInfo();
1054 }
1055
1056 createQuery.first();
1057
1058 BufferInfo bufferInfo = BufferInfo(createQuery.value(0).toInt(), networkId, type, 0, buffer);
1059 db.commit();
1060 return bufferInfo;
1061}
1062
1063
1064BufferInfo PostgreSqlStorage::getBufferInfo(UserId user, const BufferId &bufferId)
1065{
1066 QSqlQuery query(logDb());
1067 query.prepare(queryString("select_buffer_by_id"));
1068 query.bindValue(":userid", user.toInt());
1069 query.bindValue(":bufferid", bufferId.toInt());
1070 safeExec(query);
1071 if (!watchQuery(query))
1072 return BufferInfo();
1073
1074 if (!query.first())
1075 return BufferInfo();
1076
1077 BufferInfo bufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
1078 Q_ASSERT(!query.next());
1079
1080 return bufferInfo;
1081}
1082
1083
1084QList<BufferInfo> PostgreSqlStorage::requestBuffers(UserId user)
1085{
1086 QList<BufferInfo> bufferlist;
1087
1088 QSqlDatabase db = logDb();
1089 if (!beginReadOnlyTransaction(db)) {
1090 qWarning() << "PostgreSqlStorage::requestBuffers(): cannot start read only transaction!";
1091 qWarning() << " -" << qPrintable(db.lastError().text());
1092 return bufferlist;
1093 }
1094
1095 QSqlQuery query(db);
1096 query.prepare(queryString("select_buffers"));
1097 query.bindValue(":userid", user.toInt());
1098
1099 safeExec(query);
1100 watchQuery(query);
1101 while (query.next()) {
1102 bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
1103 }
1104 db.commit();
1105 return bufferlist;
1106}
1107
1108
1109QList<BufferId> PostgreSqlStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId)
1110{
1111 QList<BufferId> bufferList;
1112
1113 QSqlDatabase db = logDb();
1114 if (!beginReadOnlyTransaction(db)) {
1115 qWarning() << "PostgreSqlStorage::requestBufferIdsForNetwork(): cannot start read only transaction!";
1116 qWarning() << " -" << qPrintable(db.lastError().text());
1117 return bufferList;
1118 }
1119
1120 QSqlQuery query(db);
1121 query.prepare(queryString("select_buffers_for_network"));
1122 query.bindValue(":networkid", networkId.toInt());
1123 query.bindValue(":userid", user.toInt());
1124
1125 safeExec(query);
1126 watchQuery(query);
1127 while (query.next()) {
1128 bufferList << BufferId(query.value(0).toInt());
1129 }
1130 db.commit();
1131 return bufferList;
1132}
1133
1134
1135bool PostgreSqlStorage::removeBuffer(const UserId &user, const BufferId &bufferId)
1136{
1137 QSqlDatabase db = logDb();
1138 if (!db.transaction()) {
1139 qWarning() << "PostgreSqlStorage::removeBuffer(): cannot start transaction!";
1140 return false;
1141 }
1142
1143 QSqlQuery query(db);
1144 query.prepare(queryString("delete_buffer_for_bufferid"));
1145 query.bindValue(":userid", user.toInt());
1146 query.bindValue(":bufferid", bufferId.toInt());
1147 safeExec(query);
1148 if (!watchQuery(query)) {
1149 db.rollback();
1150 return false;
1151 }
1152
1153 int numRows = query.numRowsAffected();
1154 switch (numRows) {
1155 case 0:
1156 db.commit();
1157 return false;
1158 case 1:
1159 db.commit();
1160 return true;
1161 default:
1162 // there was more then one buffer deleted...
1163 qWarning() << "PostgreSqlStorage::removeBuffer(): Userid" << user << "BufferId" << "caused deletion of" << numRows << "Buffers! Rolling back transaction...";
1164 db.rollback();
1165 return false;
1166 }
1167}
1168
1169
1170bool PostgreSqlStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName)
1171{
1172 QSqlDatabase db = logDb();
1173 if (!db.transaction()) {
1174 qWarning() << "PostgreSqlStorage::renameBuffer(): cannot start transaction!";
1175 return false;
1176 }
1177
1178 QSqlQuery query(db);
1179 query.prepare(queryString("update_buffer_name"));
1180 query.bindValue(":buffername", newName);
1181 query.bindValue(":buffercname", newName.toLower());
1182 query.bindValue(":userid", user.toInt());
1183 query.bindValue(":bufferid", bufferId.toInt());
1184 safeExec(query);
1185 if (query.lastError().isValid()) {
1186 watchQuery(query);
1187 db.rollback();
1188 return false;
1189 }
1190
1191 int numRows = query.numRowsAffected();
1192 switch (numRows) {
1193 case 0:
1194 db.commit();
1195 return false;
1196 case 1:
1197 db.commit();
1198 return true;
1199 default:
1200 // there was more then one buffer deleted...
1201 qWarning() << "PostgreSqlStorage::renameBuffer(): Userid" << user << "BufferId" << "affected" << numRows << "Buffers! Rolling back transaction...";
1202 db.rollback();
1203 return false;
1204 }
1205}
1206
1207
1208bool PostgreSqlStorage::mergeBuffersPermanently(const UserId &user, const BufferId &bufferId1, const BufferId &bufferId2)
1209{
1210 QSqlDatabase db = logDb();
1211 if (!db.transaction()) {
1212 qWarning() << "PostgreSqlStorage::mergeBuffersPermanently(): cannot start transaction!";
1213 qWarning() << " -" << qPrintable(db.lastError().text());
1214 return false;
1215 }
1216
1217 QSqlQuery checkQuery(db);
1218 checkQuery.prepare("SELECT count(*) FROM buffer "
1219 "WHERE userid = :userid AND bufferid IN (:buffer1, :buffer2)");
1220 checkQuery.bindValue(":userid", user.toInt());
1221 checkQuery.bindValue(":buffer1", bufferId1.toInt());
1222 checkQuery.bindValue(":buffer2", bufferId2.toInt());
1223 safeExec(checkQuery);
1224 if (!watchQuery(checkQuery)) {
1225 db.rollback();
1226 return false;
1227 }
1228 checkQuery.first();
1229 if (checkQuery.value(0).toInt() != 2) {
1230 db.rollback();
1231 return false;
1232 }
1233
1234 QSqlQuery query(db);
1235 query.prepare(queryString("update_backlog_bufferid"));
1236 query.bindValue(":oldbufferid", bufferId2.toInt());
1237 query.bindValue(":newbufferid", bufferId1.toInt());
1238 safeExec(query);
1239 if (!watchQuery(query)) {
1240 db.rollback();
1241 return false;
1242 }
1243
1244 QSqlQuery delBufferQuery(logDb());
1245 delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1246 delBufferQuery.bindValue(":userid", user.toInt());
1247 delBufferQuery.bindValue(":bufferid", bufferId2.toInt());
1248 safeExec(delBufferQuery);
1249 if (!watchQuery(delBufferQuery)) {
1250 db.rollback();
1251 return false;
1252 }
1253
1254 db.commit();
1255 return true;
1256}
1257
1258
1259void PostgreSqlStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1260{
1261 QSqlQuery query(logDb());
1262 query.prepare(queryString("update_buffer_lastseen"));
1263
1264 query.bindValue(":userid", user.toInt());
1265 query.bindValue(":bufferid", bufferId.toInt());
1266 query.bindValue(":lastseenmsgid", msgId.toInt());
1267 safeExec(query);
1268 watchQuery(query);
1269}
1270
1271
1272QHash<BufferId, MsgId> PostgreSqlStorage::bufferLastSeenMsgIds(UserId user)
1273{
1274 QHash<BufferId, MsgId> lastSeenHash;
1275
1276 QSqlDatabase db = logDb();
1277 if (!beginReadOnlyTransaction(db)) {
1278 qWarning() << "PostgreSqlStorage::bufferLastSeenMsgIds(): cannot start read only transaction!";
1279 qWarning() << " -" << qPrintable(db.lastError().text());
1280 return lastSeenHash;
1281 }
1282
1283 QSqlQuery query(db);
1284 query.prepare(queryString("select_buffer_lastseen_messages"));
1285 query.bindValue(":userid", user.toInt());
1286 safeExec(query);
1287 if (!watchQuery(query)) {
1288 db.rollback();
1289 return lastSeenHash;
1290 }
1291
1292 while (query.next()) {
1293 lastSeenHash[query.value(0).toInt()] = query.value(1).toInt();
1294 }
1295
1296 db.commit();
1297 return lastSeenHash;
1298}
1299
1300
1301void PostgreSqlStorage::setBufferMarkerLineMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1302{
1303 QSqlQuery query(logDb());
1304 query.prepare(queryString("update_buffer_markerlinemsgid"));
1305
1306 query.bindValue(":userid", user.toInt());
1307 query.bindValue(":bufferid", bufferId.toInt());
1308 query.bindValue(":markerlinemsgid", msgId.toInt());
1309 safeExec(query);
1310 watchQuery(query);
1311}
1312
1313
1314QHash<BufferId, MsgId> PostgreSqlStorage::bufferMarkerLineMsgIds(UserId user)
1315{
1316 QHash<BufferId, MsgId> markerLineHash;
1317
1318 QSqlDatabase db = logDb();
1319 if (!beginReadOnlyTransaction(db)) {
1320 qWarning() << "PostgreSqlStorage::bufferMarkerLineMsgIds(): cannot start read only transaction!";
1321 qWarning() << " -" << qPrintable(db.lastError().text());
1322 return markerLineHash;
1323 }
1324
1325 QSqlQuery query(db);
1326 query.prepare(queryString("select_buffer_markerlinemsgids"));
1327 query.bindValue(":userid", user.toInt());
1328 safeExec(query);
1329 if (!watchQuery(query)) {
1330 db.rollback();
1331 return markerLineHash;
1332 }
1333
1334 while (query.next()) {
1335 markerLineHash[query.value(0).toInt()] = query.value(1).toInt();
1336 }
1337
1338 db.commit();
1339 return markerLineHash;
1340}
1341
1342
1343bool PostgreSqlStorage::logMessage(Message &msg)
1344{
1345 QSqlDatabase db = logDb();
1346 if (!db.transaction()) {
1347 qWarning() << "PostgreSqlStorage::logMessage(): cannot start transaction!";
1348 qWarning() << " -" << qPrintable(db.lastError().text());
1349 return false;
1350 }
1351
1352 QSqlQuery getSenderIdQuery = executePreparedQuery("select_senderid", msg.sender(), db);
1353 int senderId;
1354 if (getSenderIdQuery.first()) {
1355 senderId = getSenderIdQuery.value(0).toInt();
1356 }
1357 else {
1358 // it's possible that the sender was already added by another thread
1359 // since the insert might fail we're setting a savepoint
1360 savePoint("sender_sp1", db);
1361 QSqlQuery addSenderQuery = executePreparedQuery("insert_sender", msg.sender(), db);
1362
1363 if (addSenderQuery.lastError().isValid()) {
1364 rollbackSavePoint("sender_sp1", db);
1365 getSenderIdQuery = db.exec(getSenderIdQuery.lastQuery());
1366 getSenderIdQuery.first();
1367 senderId = getSenderIdQuery.value(0).toInt();
1368 }
1369 else {
1370 releaseSavePoint("sender_sp1", db);
1371 addSenderQuery.first();
1372 senderId = addSenderQuery.value(0).toInt();
1373 }
1374 }
1375
1376 QVariantList params;
1377 params << msg.timestamp()
1378 << msg.bufferInfo().bufferId().toInt()
1379 << msg.type()
1380 << (int)msg.flags()
1381 << senderId
1382 << msg.contents();
1383 QSqlQuery logMessageQuery = executePreparedQuery("insert_message", params, db);
1384
1385 if (!watchQuery(logMessageQuery)) {
1386 db.rollback();
1387 return false;
1388 }
1389
1390 logMessageQuery.first();
1391 MsgId msgId = logMessageQuery.value(0).toInt();
1392 db.commit();
1393 if (msgId.isValid()) {
1394 msg.setMsgId(msgId);
1395 return true;
1396 }
1397 else {
1398 return false;
1399 }
1400}
1401
1402
1403bool PostgreSqlStorage::logMessages(MessageList &msgs)
1404{
1405 QSqlDatabase db = logDb();
1406 if (!db.transaction()) {
1407 qWarning() << "PostgreSqlStorage::logMessage(): cannot start transaction!";
1408 qWarning() << " -" << qPrintable(db.lastError().text());
1409 return false;
1410 }
1411
1412 QList<int> senderIdList;
1413 QHash<QString, int> senderIds;
1414 QSqlQuery addSenderQuery;
1415 QSqlQuery selectSenderQuery;;
1416 for (int i = 0; i < msgs.count(); i++) {
1417 const QString &sender = msgs.at(i).sender();
1418 if (senderIds.contains(sender)) {
1419 senderIdList << senderIds[sender];
1420 continue;
1421 }
1422
1423 selectSenderQuery = executePreparedQuery("select_senderid", sender, db);
1424 if (selectSenderQuery.first()) {
1425 senderIdList << selectSenderQuery.value(0).toInt();
1426 senderIds[sender] = selectSenderQuery.value(0).toInt();
1427 }
1428 else {
1429 savePoint("sender_sp", db);
1430 addSenderQuery = executePreparedQuery("insert_sender", sender, db);
1431 if (addSenderQuery.lastError().isValid()) {
1432 // seems it was inserted meanwhile... by a different thread
1433 rollbackSavePoint("sender_sp", db);
1434 selectSenderQuery = db.exec(selectSenderQuery.lastQuery());
1435 selectSenderQuery.first();
1436 senderIdList << selectSenderQuery.value(0).toInt();
1437 senderIds[sender] = selectSenderQuery.value(0).toInt();
1438 }
1439 else {
1440 releaseSavePoint("sender_sp", db);
1441 addSenderQuery.first();
1442 senderIdList << addSenderQuery.value(0).toInt();
1443 senderIds[sender] = addSenderQuery.value(0).toInt();
1444 }
1445 }
1446 }
1447
1448 // yes we loop twice over the same list. This avoids alternating queries.
1449 bool error = false;
1450 for (int i = 0; i < msgs.count(); i++) {
1451 Message &msg = msgs[i];
1452 QVariantList params;
1453 params << msg.timestamp()
1454 << msg.bufferInfo().bufferId().toInt()
1455 << msg.type()
1456 << (int)msg.flags()
1457 << senderIdList.at(i)
1458 << msg.contents();
1459 QSqlQuery logMessageQuery = executePreparedQuery("insert_message", params, db);
1460 if (!watchQuery(logMessageQuery)) {
1461 db.rollback();
1462 error = true;
1463 break;
1464 }
1465 else {
1466 logMessageQuery.first();
1467 msg.setMsgId(logMessageQuery.value(0).toInt());
1468 }
1469 }
1470
1471 if (error) {
1472 // we had a rollback in the db so we need to reset all msgIds
1473 for (int i = 0; i < msgs.count(); i++) {
1474 msgs[i].setMsgId(MsgId());
1475 }
1476 return false;
1477 }
1478
1479 db.commit();
1480 return true;
1481}
1482
1483
1484QList<Message> PostgreSqlStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit)
1485{
1486 QList<Message> messagelist;
1487
1488 QSqlDatabase db = logDb();
1489 if (!beginReadOnlyTransaction(db)) {
1490 qWarning() << "PostgreSqlStorage::requestMsgs(): cannot start read only transaction!";
1491 qWarning() << " -" << qPrintable(db.lastError().text());
1492 return messagelist;
1493 }
1494
1495 BufferInfo bufferInfo = getBufferInfo(user, bufferId);
1496 if (!bufferInfo.isValid()) {
1497 db.rollback();
1498 return messagelist;
1499 }
1500
1501 QString queryName;
1502 QVariantList params;
1503 if (last == -1 && first == -1) {
1504 queryName = "select_messages";
1505 }
1506 else if (last == -1) {
1507 queryName = "select_messagesNewerThan";
1508 params << first.toInt();
1509 }
1510 else {
1511 queryName = "select_messagesRange";
1512 params << first.toInt();
1513 params << last.toInt();
1514 }
1515 params << bufferId.toInt();
1516 if (limit != -1)
1517 params << limit;
1518 else
1519 params << "ALL";
1520
1521 QSqlQuery query = executePreparedQuery(queryName, params, db);
1522
1523 if (!watchQuery(query)) {
1524 qDebug() << "select_messages failed";
1525 db.rollback();
1526 return messagelist;
1527 }
1528
1529 QDateTime timestamp;
1530 while (query.next()) {
1531 timestamp = query.value(1).toDateTime();
1532 timestamp.setTimeSpec(Qt::UTC);
1533 Message msg(timestamp,
1534 bufferInfo,
1535 (Message::Type)query.value(2).toUInt(),
1536 query.value(5).toString(),
1537 query.value(4).toString(),
1538 (Message::Flags)query.value(3).toUInt());
1539 msg.setMsgId(query.value(0).toInt());
1540 messagelist << msg;
1541 }
1542
1543 db.commit();
1544 return messagelist;
1545}
1546
1547
1548QList<Message> PostgreSqlStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit)
1549{
1550 QList<Message> messagelist;
1551
1552 // requestBuffers uses it's own transaction.
1553 QHash<BufferId, BufferInfo> bufferInfoHash;
1554 foreach(BufferInfo bufferInfo, requestBuffers(user)) {
1555 bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
1556 }
1557
1558 QSqlDatabase db = logDb();
1559 if (!beginReadOnlyTransaction(db)) {
1560 qWarning() << "PostgreSqlStorage::requestAllMsgs(): cannot start read only transaction!";
1561 qWarning() << " -" << qPrintable(db.lastError().text());
1562 return messagelist;
1563 }
1564
1565 QSqlQuery query(db);
1566 if (last == -1) {
1567 query.prepare(queryString("select_messagesAllNew"));
1568 }
1569 else {
1570 query.prepare(queryString("select_messagesAll"));
1571 query.bindValue(":lastmsg", last.toInt());
1572 }
1573 query.bindValue(":userid", user.toInt());
1574 query.bindValue(":firstmsg", first.toInt());
1575 safeExec(query);
1576 if (!watchQuery(query)) {
1577 db.rollback();
1578 return messagelist;
1579 }
1580
1581 QDateTime timestamp;
1582 for (int i = 0; i < limit && query.next(); i++) {
1583 timestamp = query.value(1).toDateTime();
1584 timestamp.setTimeSpec(Qt::UTC);
1585 Message msg(timestamp,
1586 bufferInfoHash[query.value(1).toInt()],
1587 (Message::Type)query.value(3).toUInt(),
1588 query.value(6).toString(),
1589 query.value(5).toString(),
1590 (Message::Flags)query.value(4).toUInt());
1591 msg.setMsgId(query.value(0).toInt());
1592 messagelist << msg;
1593 }
1594
1595 db.commit();
1596 return messagelist;
1597}
1598
1599
1600// void PostgreSqlStorage::safeExec(QSqlQuery &query) {
1601// qDebug() << "PostgreSqlStorage::safeExec";
1602// qDebug() << " executing:\n" << query.executedQuery();
1603// qDebug() << " bound Values:";
1604// QList<QVariant> list = query.boundValues().values();
1605// for (int i = 0; i < list.size(); ++i)
1606// qCritical() << i << ": " << list.at(i).toString().toLatin1().data();
1607
1608// query.exec();
1609
1610// qDebug() << "Success:" << !query.lastError().isValid();
1611// qDebug();
1612
1613// if(!query.lastError().isValid())
1614// return;
1615
1616// qDebug() << "==================== ERROR ====================";
1617// watchQuery(query);
1618// qDebug() << "===============================================";
1619// qDebug();
1620// return;
1621// }
1622
1623bool PostgreSqlStorage::beginReadOnlyTransaction(QSqlDatabase &db)
1624{
1625 QSqlQuery query = db.exec("BEGIN TRANSACTION READ ONLY");
1626 return !query.lastError().isValid();
1627}
1628
1629
1630QSqlQuery PostgreSqlStorage::prepareAndExecuteQuery(const QString &queryname, const QString &paramstring, const QSqlDatabase &db)
1631{
1632 // Query preparing is done lazily. That means that instead of always checking if the query is already prepared
1633 // we just EXECUTE and catch the error
1634 QSqlQuery query;
1635
1636 db.exec("SAVEPOINT quassel_prepare_query");
1637 if (paramstring.isNull()) {
1638 query = db.exec(QString("EXECUTE quassel_%1").arg(queryname));
1639 }
1640 else {
1641 query = db.exec(QString("EXECUTE quassel_%1 (%2)").arg(queryname).arg(paramstring));
1642 }
1643
1644 if (db.lastError().isValid()) {
1645 // and once again: Qt leaves us without error codes so we either parse (language dependant(!)) strings
1646 // or we just guess the error. As we're only interested in unprepared queries, this will be our guess. :)
1647 db.exec("ROLLBACK TO SAVEPOINT quassel_prepare_query");
1648 QSqlQuery checkQuery = db.exec(QString("SELECT count(name) FROM pg_prepared_statements WHERE name = 'quassel_%1' AND from_sql = TRUE").arg(queryname.toLower()));
1649 checkQuery.first();
1650 if (checkQuery.value(0).toInt() == 0) {
1651 db.exec(QString("PREPARE quassel_%1 AS %2").arg(queryname).arg(queryString(queryname)));
1652 if (db.lastError().isValid()) {
1653 qWarning() << "PostgreSqlStorage::prepareQuery(): unable to prepare query:" << queryname << "AS" << queryString(queryname);
1654 qWarning() << " Error:" << db.lastError().text();
1655 return QSqlQuery(db);
1656 }
1657 }
1658 // we alwas execute the query again, even if the query was already prepared.
1659 // this ensures, that the error is properly propagated to the calling function
1660 // (otherwise the last call would be the testing select to pg_prepared_statements
1661 // which always gives a proper result and the error would be lost)
1662 if (paramstring.isNull()) {
1663 query = db.exec(QString("EXECUTE quassel_%1").arg(queryname));
1664 }
1665 else {
1666 query = db.exec(QString("EXECUTE quassel_%1 (%2)").arg(queryname).arg(paramstring));
1667 }
1668 }
1669 else {
1670 // only release the SAVEPOINT
1671 db.exec("RELEASE SAVEPOINT quassel_prepare_query");
1672 }
1673 return query;
1674}
1675
1676
1677QSqlQuery PostgreSqlStorage::executePreparedQuery(const QString &queryname, const QVariantList &params, const QSqlDatabase &db)
1678{
1679 QSqlDriver *driver = db.driver();
1680
1681 QStringList paramStrings;
1682 QSqlField field;
1683 for (int i = 0; i < params.count(); i++) {
1684 const QVariant &value = params.at(i);
1685 field.setType(value.type());
1686 if (value.isNull())
1687 field.clear();
1688 else
1689 field.setValue(value);
1690
1691 paramStrings << driver->formatValue(field);
1692 }
1693
1694 if (params.isEmpty()) {
1695 return prepareAndExecuteQuery(queryname, db);
1696 }
1697 else {
1698 return prepareAndExecuteQuery(queryname, paramStrings.join(", "), db);
1699 }
1700}
1701
1702
1703QSqlQuery PostgreSqlStorage::executePreparedQuery(const QString &queryname, const QVariant &param, const QSqlDatabase &db)
1704{
1705 QSqlField field;
1706 field.setType(param.type());
1707 if (param.isNull())
1708 field.clear();
1709 else
1710 field.setValue(param);
1711
1712 QString paramString = db.driver()->formatValue(field);
1713 return prepareAndExecuteQuery(queryname, paramString, db);
1714}
1715
1716
1717void PostgreSqlStorage::deallocateQuery(const QString &queryname, const QSqlDatabase &db)
1718{
1719 db.exec(QString("DEALLOCATE quassel_%1").arg(queryname));
1720}
1721
1722
1723// ========================================
1724// PostgreSqlMigrationWriter
1725// ========================================
1726PostgreSqlMigrationWriter::PostgreSqlMigrationWriter()
1727 : PostgreSqlStorage()
1728{
1729}
1730
1731
1732bool PostgreSqlMigrationWriter::prepareQuery(MigrationObject mo)
1733{
1734 QString query;
1735 switch (mo) {
1736 case QuasselUser:
1737 query = queryString("migrate_write_quasseluser");
1738 break;
1739 case Sender:
1740 query = queryString("migrate_write_sender");
1741 break;
1742 case Identity:
1743 _validIdentities.clear();
1744 query = queryString("migrate_write_identity");
1745 break;
1746 case IdentityNick:
1747 query = queryString("migrate_write_identity_nick");
1748 break;
1749 case Network:
1750 query = queryString("migrate_write_network");
1751 break;
1752 case Buffer:
1753 query = queryString("migrate_write_buffer");
1754 break;
1755 case Backlog:
1756 query = queryString("migrate_write_backlog");
1757 break;
1758 case IrcServer:
1759 query = queryString("migrate_write_ircserver");
1760 break;
1761 case UserSetting:
1762 query = queryString("migrate_write_usersetting");
1763 break;
1764 }
1765 newQuery(query, logDb());
1766 return true;
1767}
1768
1769
1770//bool PostgreSqlMigrationWriter::writeUser(const QuasselUserMO &user) {
1771bool PostgreSqlMigrationWriter::writeMo(const QuasselUserMO &user)
1772{
1773 bindValue(0, user.id.toInt());
1774 bindValue(1, user.username);
1775 bindValue(2, user.password);
1776 return exec();
1777}
1778
1779
1780//bool PostgreSqlMigrationWriter::writeSender(const SenderMO &sender) {
1781bool PostgreSqlMigrationWriter::writeMo(const SenderMO &sender)
1782{
1783 bindValue(0, sender.senderId);
1784 bindValue(1, sender.sender);
1785 return exec();
1786}
1787
1788
1789//bool PostgreSqlMigrationWriter::writeIdentity(const IdentityMO &identity) {
1790bool PostgreSqlMigrationWriter::writeMo(const IdentityMO &identity)
1791{
1792 _validIdentities << identity.id.toInt();
1793 bindValue(0, identity.id.toInt());
1794 bindValue(1, identity.userid.toInt());
1795 bindValue(2, identity.identityname);
1796 bindValue(3, identity.realname);
1797 bindValue(4, identity.awayNick);
1798 bindValue(5, identity.awayNickEnabled);
1799 bindValue(6, identity.awayReason);
1800 bindValue(7, identity.awayReasonEnabled);
1801 bindValue(8, identity.autoAwayEnabled);
1802 bindValue(9, identity.autoAwayTime);
1803 bindValue(10, identity.autoAwayReason);
1804 bindValue(11, identity.autoAwayReasonEnabled);
1805 bindValue(12, identity.detachAwayEnabled);
1806 bindValue(13, identity.detachAwayReason);
1807 bindValue(14, identity.detchAwayReasonEnabled);
1808 bindValue(15, identity.ident);
1809 bindValue(16, identity.kickReason);
1810 bindValue(17, identity.partReason);
1811 bindValue(18, identity.quitReason);
1812 bindValue(19, identity.sslCert);
1813 bindValue(20, identity.sslKey);
1814 return exec();
1815}
1816
1817
1818//bool PostgreSqlMigrationWriter::writeIdentityNick(const IdentityNickMO &identityNick) {
1819bool PostgreSqlMigrationWriter::writeMo(const IdentityNickMO &identityNick)
1820{
1821 bindValue(0, identityNick.nickid);
1822 bindValue(1, identityNick.identityId.toInt());
1823 bindValue(2, identityNick.nick);
1824 return exec();
1825}
1826
1827
1828//bool PostgreSqlMigrationWriter::writeNetwork(const NetworkMO &network) {
1829bool PostgreSqlMigrationWriter::writeMo(const NetworkMO &network)
1830{
1831 bindValue(0, network.networkid.toInt());
1832 bindValue(1, network.userid.toInt());
1833 bindValue(2, network.networkname);
1834 if (_validIdentities.contains(network.identityid.toInt()))
1835 bindValue(3, network.identityid.toInt());
1836 else
1837 bindValue(3, QVariant());
1838 bindValue(4, network.encodingcodec);
1839 bindValue(5, network.decodingcodec);
1840 bindValue(6, network.servercodec);
1841 bindValue(7, network.userandomserver);
1842 bindValue(8, network.perform);
1843 bindValue(9, network.useautoidentify);
1844 bindValue(10, network.autoidentifyservice);
1845 bindValue(11, network.autoidentifypassword);
1846 bindValue(12, network.useautoreconnect);
1847 bindValue(13, network.autoreconnectinterval);
1848 bindValue(14, network.autoreconnectretries);
1849 bindValue(15, network.unlimitedconnectretries);
1850 bindValue(16, network.rejoinchannels);
1851 bindValue(17, network.connected);
1852 bindValue(18, network.usermode);
1853 bindValue(19, network.awaymessage);
1854 bindValue(20, network.attachperform);
1855 bindValue(21, network.detachperform);
1856 bindValue(22, network.usesasl);
1857 bindValue(23, network.saslaccount);
1858 bindValue(24, network.saslpassword);
1859 return exec();
1860}
1861
1862
1863//bool PostgreSqlMigrationWriter::writeBuffer(const BufferMO &buffer) {
1864bool PostgreSqlMigrationWriter::writeMo(const BufferMO &buffer)
1865{
1866 bindValue(0, buffer.bufferid.toInt());
1867 bindValue(1, buffer.userid.toInt());
1868 bindValue(2, buffer.groupid);
1869 bindValue(3, buffer.networkid.toInt());
1870 bindValue(4, buffer.buffername);
1871 bindValue(5, buffer.buffercname);
1872 bindValue(6, (int)buffer.buffertype);
1873 bindValue(7, buffer.lastseenmsgid);
1874 bindValue(8, buffer.markerlinemsgid);
1875 bindValue(9, buffer.key);
1876 bindValue(10, buffer.joined);
1877 return exec();
1878}
1879
1880
1881//bool PostgreSqlMigrationWriter::writeBacklog(const BacklogMO &backlog) {
1882bool PostgreSqlMigrationWriter::writeMo(const BacklogMO &backlog)
1883{
1884 bindValue(0, backlog.messageid.toInt());
1885 bindValue(1, backlog.time);
1886 bindValue(2, backlog.bufferid.toInt());
1887 bindValue(3, backlog.type);
1888 bindValue(4, (int)backlog.flags);
1889 bindValue(5, backlog.senderid);
1890 bindValue(6, backlog.message);
1891 return exec();
1892}
1893
1894
1895//bool PostgreSqlMigrationWriter::writeIrcServer(const IrcServerMO &ircserver) {
1896bool PostgreSqlMigrationWriter::writeMo(const IrcServerMO &ircserver)
1897{
1898 bindValue(0, ircserver.serverid);
1899 bindValue(1, ircserver.userid.toInt());
1900 bindValue(2, ircserver.networkid.toInt());
1901 bindValue(3, ircserver.hostname);
1902 bindValue(4, ircserver.port);
1903 bindValue(5, ircserver.password);
1904 bindValue(6, ircserver.ssl);
1905 bindValue(7, ircserver.sslversion);
1906 bindValue(8, ircserver.useproxy);
1907 bindValue(9, ircserver.proxytype);
1908 bindValue(10, ircserver.proxyhost);
1909 bindValue(11, ircserver.proxyport);
1910 bindValue(12, ircserver.proxyuser);
1911 bindValue(13, ircserver.proxypass);
1912 return exec();
1913}
1914
1915
1916//bool PostgreSqlMigrationWriter::writeUserSetting(const UserSettingMO &userSetting) {
1917bool PostgreSqlMigrationWriter::writeMo(const UserSettingMO &userSetting)
1918{
1919 bindValue(0, userSetting.userid.toInt());
1920 bindValue(1, userSetting.settingname);
1921 bindValue(2, userSetting.settingvalue);
1922 return exec();
1923}
1924
1925
1926bool PostgreSqlMigrationWriter::postProcess()
1927{
1928 QSqlDatabase db = logDb();
1929 QList<Sequence> sequences;
1930 sequences << Sequence("backlog", "messageid")
1931 << Sequence("buffer", "bufferid")
1932 << Sequence("identity", "identityid")
1933 << Sequence("identity_nick", "nickid")
1934 << Sequence("ircserver", "serverid")
1935 << Sequence("network", "networkid")
1936 << Sequence("quasseluser", "userid")
1937 << Sequence("sender", "senderid");
1938 QList<Sequence>::const_iterator iter;
1939 for (iter = sequences.constBegin(); iter != sequences.constEnd(); iter++) {
1940 resetQuery();
1941 newQuery(QString("SELECT setval('%1_%2_seq', max(%2)) FROM %1").arg(iter->table, iter->field), db);
1942 if (!exec())
1943 return false;
1944 }
1945 return true;
1946}
1947