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