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 bool PostgreSqlStorage::setup(const QVariantMap &settings) {
  83   bool success = AbstractSqlStorage::setup(settings);
  84   if(success) {
  85     logDb().exec(QString("ALTER USER %1 SET standard_conforming_strings TO on").arg(userName()));
  86   }
  87   return success;
  88 }
  89
  90 void PostgreSqlStorage::setConnectionProperties(const QVariantMap &properties) {
  91   _userName = properties["Username"].toString();
  92   _password = properties["Password"].toString();
  93   _hostName = properties["Hostname"].toString();
  94   _port = properties["Port"].toInt();
  95   _databaseName = properties["Database"].toString();
  96 }
  97
  98 int PostgreSqlStorage::installedSchemaVersion() {
  99   QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
 100   if(query.first())
 101     return query.value(0).toInt();
 102
 103   // maybe it's really old... (schema version 0)
 104   query = logDb().exec("SELECT MAX(version) FROM coreinfo");
 105   if(query.first())
 106     return query.value(0).toInt();
 107
 108   return AbstractSqlStorage::installedSchemaVersion();
 109 }
 110
 111 bool PostgreSqlStorage::updateSchemaVersion(int newVersion) {
 112   QSqlQuery query(logDb());
 113   query.prepare("UPDATE coreinfo SET value = :version WHERE key = 'schemaversion'");
 114   query.bindValue(":version", newVersion);
 115   query.exec();
 116
 117   bool success = true;
 118   if(query.lastError().isValid()) {
 119     qCritical() << "PostgreSqlStorage::updateSchemaVersion(int): Updating schema version failed!";
 120     success = false;
 121   }
 122   return success;
 123 }
 124
 125 bool PostgreSqlStorage::setupSchemaVersion(int version) {
 126   QSqlQuery query(logDb());
 127   query.prepare("INSERT INTO coreinfo (key, value) VALUES ('schemaversion', :version)");
 128   query.bindValue(":version", version);
 129   query.exec();
 130
 131   bool success = true;
 132   if(query.lastError().isValid()) {
 133     qCritical() << "PostgreSqlStorage::setupSchemaVersion(int): Updating schema version failed!";
 134     success = false;
 135   }
 136   return success;
 137 }
 138
 139 UserId PostgreSqlStorage::addUser(const QString &user, const QString &password) {
 140   QSqlQuery query(logDb());
 141   query.prepare(queryString("insert_quasseluser"));
 142   query.bindValue(":username", user);
 143   query.bindValue(":password", cryptedPassword(password));
 144   safeExec(query);
 145   if(!watchQuery(query))
 146     return 0;
 147
 148   query.first();
 149   UserId uid = query.value(0).toInt();
 150   emit userAdded(uid, user);
 151   return uid;
 152 }
 153
 154 bool PostgreSqlStorage::updateUser(UserId user, const QString &password) {
 155   QSqlQuery query(logDb());
 156   query.prepare(queryString("update_userpassword"));
 157   query.bindValue(":userid", user.toInt());
 158   query.bindValue(":password", cryptedPassword(password));
 159   safeExec(query);
 160   return query.numRowsAffected() != 0;
 161 }
 162
 163 void PostgreSqlStorage::renameUser(UserId user, const QString &newName) {
 164   QSqlQuery query(logDb());
 165   query.prepare(queryString("update_username"));
 166   query.bindValue(":userid", user.toInt());
 167   query.bindValue(":username", newName);
 168   safeExec(query);
 169   emit userRenamed(user, newName);
 170 }
 171
 172 UserId PostgreSqlStorage::validateUser(const QString &user, const QString &password) {
 173   QSqlQuery query(logDb());
 174   query.prepare(queryString("select_authuser"));
 175   query.bindValue(":username", user);
 176   query.bindValue(":password", cryptedPassword(password));
 177   safeExec(query);
 178
 179   if(query.first()) {
 180     return query.value(0).toInt();
 181   } else {
 182     return 0;
 183   }
 184 }
 185
 186 UserId PostgreSqlStorage::getUserId(const QString &user) {
 187   QSqlQuery query(logDb());
 188   query.prepare(queryString("select_userid"));
 189   query.bindValue(":username", user);
 190   safeExec(query);
 191
 192   if(query.first()) {
 193     return query.value(0).toInt();
 194   } else {
 195     return 0;
 196   }
 197 }
 198
 199 UserId PostgreSqlStorage::internalUser() {
 200   QSqlQuery query(logDb());
 201   query.prepare(queryString("select_internaluser"));
 202   safeExec(query);
 203
 204   if(query.first()) {
 205     return query.value(0).toInt();
 206   } else {
 207     return 0;
 208   }
 209 }
 210
 211 void PostgreSqlStorage::delUser(UserId user) {
 212   QSqlDatabase db = logDb();
 213   if(!db.transaction()) {
 214     qWarning() << "PostgreSqlStorage::delUser(): cannot start transaction!";
 215     return;
 216   }
 217
 218   QSqlQuery query(db);
 219   query.prepare(queryString("delete_quasseluser"));
 220   query.bindValue(":userid", user.toInt());
 221   safeExec(query);
 222   if(!watchQuery(query)) {
 223     db.rollback();
 224     return;
 225   } else {
 226     db.commit();
 227     emit userRemoved(user);
 228   }
 229 }
 230
 231 void PostgreSqlStorage::setUserSetting(UserId userId, const QString &settingName, const QVariant &data) {
 232   QByteArray rawData;
 233   QDataStream out(&rawData, QIODevice::WriteOnly);
 234   out.setVersion(QDataStream::Qt_4_2);
 235   out << data;
 236
 237   QSqlDatabase db = logDb();
 238   QSqlQuery query(db);
 239   query.prepare(queryString("insert_user_setting"));
 240   query.bindValue(":userid", userId.toInt());
 241   query.bindValue(":settingname", settingName);
 242   query.bindValue(":settingvalue", rawData);
 243   safeExec(query);
 244
 245   if(query.lastError().isValid()) {
 246     QSqlQuery updateQuery(db);
 247     updateQuery.prepare(queryString("update_user_setting"));
 248     updateQuery.bindValue(":userid", userId.toInt());
 249     updateQuery.bindValue(":settingname", settingName);
 250     updateQuery.bindValue(":settingvalue", rawData);
 251     safeExec(updateQuery);
 252   }
 253
 254 }
 255
 256 QVariant PostgreSqlStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData) {
 257   QSqlQuery query(logDb());
 258   query.prepare(queryString("select_user_setting"));
 259   query.bindValue(":userid", userId.toInt());
 260   query.bindValue(":settingname", settingName);
 261   safeExec(query);
 262
 263   if(query.first()) {
 264     QVariant data;
 265     QByteArray rawData = query.value(0).toByteArray();
 266     QDataStream in(&rawData, QIODevice::ReadOnly);
 267     in.setVersion(QDataStream::Qt_4_2);
 268     in >> data;
 269     return data;
 270   } else {
 271     return defaultData;
 272   }
 273 }
 274
 275 IdentityId PostgreSqlStorage::createIdentity(UserId user, CoreIdentity &identity) {
 276   IdentityId identityId;
 277
 278   QSqlDatabase db = logDb();
 279   if(!db.transaction()) {
 280     qWarning() << "PostgreSqlStorage::createIdentity(): Unable to start Transaction!";
 281     qWarning() << " -" << qPrintable(db.lastError().text());
 282     return identityId;
 283   }
 284
 285   QSqlQuery query(db);
 286   query.prepare(queryString("insert_identity"));
 287   query.bindValue(":userid", user.toInt());
 288   query.bindValue(":identityname", identity.identityName());
 289   query.bindValue(":realname", identity.realName());
 290   query.bindValue(":awaynick", identity.awayNick());
 291   query.bindValue(":awaynickenabled", identity.awayNickEnabled());
 292   query.bindValue(":awayreason", identity.awayReason());
 293   query.bindValue(":awayreasonenabled", identity.awayReasonEnabled());
 294   query.bindValue(":autoawayenabled", identity.awayReasonEnabled());
 295   query.bindValue(":autoawaytime", identity.autoAwayTime());
 296   query.bindValue(":autoawayreason", identity.autoAwayReason());
 297   query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled());
 298   query.bindValue(":detachawayenabled", identity.detachAwayEnabled());
 299   query.bindValue(":detachawayreason", identity.detachAwayReason());
 300   query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled());
 301   query.bindValue(":ident", identity.ident());
 302   query.bindValue(":kickreason", identity.kickReason());
 303   query.bindValue(":partreason", identity.partReason());
 304   query.bindValue(":quitreason", identity.quitReason());
 305 #ifdef HAVE_SSL
 306   query.bindValue(":sslcert", identity.sslCert().toPem());
 307   query.bindValue(":sslkey", identity.sslKey().toPem());
 308 #else
 309   query.bindValue(":sslcert", QByteArray());
 310   query.bindValue(":sslkey", QByteArray());
 311 #endif
 312   safeExec(query);
 313   if(query.lastError().isValid()) {
 314     watchQuery(query);
 315     db.rollback();
 316     return IdentityId();
 317   }
 318
 319   query.first();
 320   identityId = query.value(0).toInt();
 321   identity.setId(identityId);
 322
 323   if(!identityId.isValid()) {
 324     watchQuery(query);
 325     db.rollback();
 326     return IdentityId();
 327   }
 328
 329   QSqlQuery insertNickQuery(db);
 330   insertNickQuery.prepare(queryString("insert_nick"));
 331   foreach(QString nick, identity.nicks()) {
 332     insertNickQuery.bindValue(":identityid", identityId.toInt());
 333     insertNickQuery.bindValue(":nick", nick);
 334     safeExec(insertNickQuery);
 335     if(!watchQuery(insertNickQuery)) {
 336       db.rollback();
 337       return IdentityId();
 338     }
 339   }
 340
 341   if(!db.commit()) {
 342     qWarning() << "PostgreSqlStorage::createIdentity(): commiting data failed!";
 343     qWarning() << " -" << qPrintable(db.lastError().text());
 344     return IdentityId();
 345   }
 346   return identityId;
 347 }
 348
 349 bool PostgreSqlStorage::updateIdentity(UserId user, const CoreIdentity &identity) {
 350   QSqlDatabase db = logDb();
 351   if(!db.transaction()) {
 352     qWarning() << "PostgreSqlStorage::updateIdentity(): Unable to start Transaction!";
 353     qWarning() << " -" << qPrintable(db.lastError().text());
 354     return false;
 355   }
 356
 357   QSqlQuery checkQuery(db);
 358   checkQuery.prepare(queryString("select_checkidentity"));
 359   checkQuery.bindValue(":identityid", identity.id().toInt());
 360   checkQuery.bindValue(":userid", user.toInt());
 361   safeExec(checkQuery);
 362
 363   // there should be exactly one identity for the given id and user
 364   if(!checkQuery.first() || checkQuery.value(0).toInt() != 1) {
 365     db.rollback();
 366     return false;
 367   }
 368
 369   QSqlQuery query(db);
 370   query.prepare(queryString("update_identity"));
 371   query.bindValue(":identityname", identity.identityName());
 372   query.bindValue(":realname", identity.realName());
 373   query.bindValue(":awaynick", identity.awayNick());
 374   query.bindValue(":awaynickenabled", identity.awayNickEnabled());
 375   query.bindValue(":awayreason", identity.awayReason());
 376   query.bindValue(":awayreasonenabled", identity.awayReasonEnabled());
 377   query.bindValue(":autoawayenabled", identity.awayReasonEnabled());
 378   query.bindValue(":autoawaytime", identity.autoAwayTime());
 379   query.bindValue(":autoawayreason", identity.autoAwayReason());
 380   query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled());
 381   query.bindValue(":detachawayenabled", identity.detachAwayEnabled());
 382   query.bindValue(":detachawayreason", identity.detachAwayReason());
 383   query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled());
 384   query.bindValue(":ident", identity.ident());
 385   query.bindValue(":kickreason", identity.kickReason());
 386   query.bindValue(":partreason", identity.partReason());
 387   query.bindValue(":quitreason", identity.quitReason());
 388 #ifdef HAVE_SSL
 389   query.bindValue(":sslcert", identity.sslCert().toPem());
 390   query.bindValue(":sslkey", identity.sslKey().toPem());
 391 #else
 392   query.bindValue(":sslcert", QByteArray());
 393   query.bindValue(":sslkey", QByteArray());
 394 #endif
 395   query.bindValue(":identityid", identity.id().toInt());
 396
 397   safeExec(query);
 398   if(!watchQuery(query)) {
 399     db.rollback();
 400     return false;
 401   }
 402
 403   QSqlQuery deleteNickQuery(db);
 404   deleteNickQuery.prepare(queryString("delete_nicks"));
 405   deleteNickQuery.bindValue(":identityid", identity.id().toInt());
 406   safeExec(deleteNickQuery);
 407   if(!watchQuery(deleteNickQuery)) {
 408     db.rollback();
 409     return false;
 410   }
 411
 412   QSqlQuery insertNickQuery(db);
 413   insertNickQuery.prepare(queryString("insert_nick"));
 414   foreach(QString nick, identity.nicks()) {
 415     insertNickQuery.bindValue(":identityid", identity.id().toInt());
 416     insertNickQuery.bindValue(":nick", nick);
 417     safeExec(insertNickQuery);
 418     if(!watchQuery(insertNickQuery)) {
 419       db.rollback();
 420       return false;
 421     }
 422   }
 423
 424   if(!db.commit()) {
 425     qWarning() << "PostgreSqlStorage::updateIdentity(): commiting data failed!";
 426     qWarning() << " -" << qPrintable(db.lastError().text());
 427     return false;
 428   }
 429   return true;
 430 }
 431
 432 void PostgreSqlStorage::removeIdentity(UserId user, IdentityId identityId) {
 433   QSqlDatabase db = logDb();
 434   if(!db.transaction()) {
 435     qWarning() << "PostgreSqlStorage::removeIdentity(): Unable to start Transaction!";
 436     qWarning() << " -" << qPrintable(db.lastError().text());
 437     return;
 438   }
 439
 440   QSqlQuery query(db);
 441   query.prepare(queryString("delete_identity"));
 442   query.bindValue(":identityid", identityId.toInt());
 443   query.bindValue(":userid", user.toInt());
 444   safeExec(query);
 445   if(!watchQuery(query)) {
 446     db.rollback();
 447   } else {
 448     db.commit();
 449   }
 450 }
 451
 452 QList<CoreIdentity> PostgreSqlStorage::identities(UserId user) {
 453   QList<CoreIdentity> identities;
 454
 455   QSqlDatabase db = logDb();
 456   if(!beginReadOnlyTransaction(db)) {
 457     qWarning() << "PostgreSqlStorage::identites(): cannot start read only transaction!";
 458     qWarning() << " -" << qPrintable(db.lastError().text());
 459     return identities;
 460   }
 461
 462   QSqlQuery query(db);
 463   query.prepare(queryString("select_identities"));
 464   query.bindValue(":userid", user.toInt());
 465
 466   QSqlQuery nickQuery(db);
 467   nickQuery.prepare(queryString("select_nicks"));
 468
 469   safeExec(query);
 470
 471   while(query.next()) {
 472     CoreIdentity identity(IdentityId(query.value(0).toInt()));
 473
 474     identity.setIdentityName(query.value(1).toString());
 475     identity.setRealName(query.value(2).toString());
 476     identity.setAwayNick(query.value(3).toString());
 477     identity.setAwayNickEnabled(!!query.value(4).toInt());
 478     identity.setAwayReason(query.value(5).toString());
 479     identity.setAwayReasonEnabled(!!query.value(6).toInt());
 480     identity.setAutoAwayEnabled(!!query.value(7).toInt());
 481     identity.setAutoAwayTime(query.value(8).toInt());
 482     identity.setAutoAwayReason(query.value(9).toString());
 483     identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
 484     identity.setDetachAwayEnabled(!!query.value(11).toInt());
 485     identity.setDetachAwayReason(query.value(12).toString());
 486     identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
 487     identity.setIdent(query.value(14).toString());
 488     identity.setKickReason(query.value(15).toString());
 489     identity.setPartReason(query.value(16).toString());
 490     identity.setQuitReason(query.value(17).toString());
 491 #ifdef HAVE_SSL
 492     identity.setSslCert(query.value(18).toByteArray());
 493     identity.setSslKey(query.value(19).toByteArray());
 494 #endif
 495
 496     nickQuery.bindValue(":identityid", identity.id().toInt());
 497     QList<QString> nicks;
 498     safeExec(nickQuery);
 499     watchQuery(nickQuery);
 500     while(nickQuery.next()) {
 501       nicks << nickQuery.value(0).toString();
 502     }
 503     identity.setNicks(nicks);
 504     identities << identity;
 505   }
 506   db.commit();
 507   return identities;
 508 }
 509
 510 NetworkId PostgreSqlStorage::createNetwork(UserId user, const NetworkInfo &info) {
 511   NetworkId networkId;
 512
 513   QSqlDatabase db = logDb();
 514   if(!db.transaction()) {
 515     qWarning() << "PostgreSqlStorage::createNetwork(): failed to begin transaction!";
 516     qWarning() << " -" << qPrintable(db.lastError().text());
 517     return false;
 518   }
 519
 520   QSqlQuery query(db);
 521   query.prepare(queryString("insert_network"));
 522   query.bindValue(":userid", user.toInt());
 523   bindNetworkInfo(query, info);
 524   safeExec(query);
 525   if(query.lastError().isValid()) {
 526     watchQuery(query);
 527     db.rollback();
 528     return NetworkId();
 529   }
 530
 531   query.first();
 532   networkId = query.value(0).toInt();
 533
 534   if(!networkId.isValid()) {
 535     watchQuery(query);
 536     db.rollback();
 537     return NetworkId();
 538   }
 539
 540   QSqlQuery insertServersQuery(db);
 541   insertServersQuery.prepare(queryString("insert_server"));
 542   foreach(Network::Server server, info.serverList) {
 543     insertServersQuery.bindValue(":userid", user.toInt());
 544     insertServersQuery.bindValue(":networkid", networkId.toInt());
 545     bindServerInfo(insertServersQuery, server);
 546     safeExec(insertServersQuery);
 547     if(!watchQuery(insertServersQuery)) {
 548       db.rollback();
 549       return NetworkId();
 550     }
 551   }
 552
 553   if(!db.commit()) {
 554     qWarning() << "PostgreSqlStorage::createNetwork(): commiting data failed!";
 555     qWarning() << " -" << qPrintable(db.lastError().text());
 556     return NetworkId();
 557   }
 558   return networkId;
 559 }
 560
 561 void PostgreSqlStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info) {
 562   query.bindValue(":networkname", info.networkName);
 563   query.bindValue(":identityid", info.identity.isValid() ? info.identity.toInt() : QVariant());
 564   query.bindValue(":encodingcodec", QString(info.codecForEncoding));
 565   query.bindValue(":decodingcodec", QString(info.codecForDecoding));
 566   query.bindValue(":servercodec", QString(info.codecForServer));
 567   query.bindValue(":userandomserver", info.useRandomServer);
 568   query.bindValue(":perform", info.perform.join("\n"));
 569   query.bindValue(":useautoidentify", info.useAutoIdentify);
 570   query.bindValue(":autoidentifyservice", info.autoIdentifyService);
 571   query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 572   query.bindValue(":useautoreconnect", info.useAutoReconnect);
 573   query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 574   query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 575   query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries);
 576   query.bindValue(":rejoinchannels", info.rejoinChannels);
 577   if(info.networkId.isValid())
 578     query.bindValue(":networkid", info.networkId.toInt());
 579 }
 580
 581 void PostgreSqlStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server) {
 582   query.bindValue(":hostname", server.host);
 583   query.bindValue(":port", server.port);
 584   query.bindValue(":password", server.password);
 585   query.bindValue(":ssl", server.useSsl);
 586   query.bindValue(":sslversion", server.sslVersion);
 587   query.bindValue(":useproxy", server.useProxy);
 588   query.bindValue(":proxytype", server.proxyType);
 589   query.bindValue(":proxyhost", server.proxyHost);
 590   query.bindValue(":proxyport", server.proxyPort);
 591   query.bindValue(":proxyuser", server.proxyUser);
 592   query.bindValue(":proxypass", server.proxyPass);
 593 }
 594
 595 bool PostgreSqlStorage::updateNetwork(UserId user, const NetworkInfo &info) {
 596   QSqlDatabase db = logDb();
 597   if(!db.transaction()) {
 598     qWarning() << "PostgreSqlStorage::updateNetwork(): failed to begin transaction!";
 599     qWarning() << " -" << qPrintable(db.lastError().text());
 600     return false;
 601   }
 602
 603   QSqlQuery updateQuery(db);
 604   updateQuery.prepare(queryString("update_network"));
 605   updateQuery.bindValue(":userid", user.toInt());
 606   bindNetworkInfo(updateQuery, info);
 607   safeExec(updateQuery);
 608   if(!watchQuery(updateQuery)) {
 609     db.rollback();
 610     return false;
 611   }
 612   if(updateQuery.numRowsAffected() != 1) {
 613     // seems this is not our network...
 614     db.rollback();
 615     return false;
 616   }
 617
 618   QSqlQuery dropServersQuery(db);
 619   dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 620   dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 621   safeExec(dropServersQuery);
 622   if(!watchQuery(dropServersQuery)) {
 623     db.rollback();
 624     return false;
 625   }
 626
 627   QSqlQuery insertServersQuery(db);
 628   insertServersQuery.prepare(queryString("insert_server"));
 629   foreach(Network::Server server, info.serverList) {
 630     insertServersQuery.bindValue(":userid", user.toInt());
 631     insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 632     bindServerInfo(insertServersQuery, server);
 633     safeExec(insertServersQuery);
 634     if(!watchQuery(insertServersQuery)) {
 635       db.rollback();
 636       return false;
 637     }
 638   }
 639
 640   if(!db.commit()) {
 641     qWarning() << "PostgreSqlStorage::updateNetwork(): commiting data failed!";
 642     qWarning() << " -" << qPrintable(db.lastError().text());
 643     return false;
 644   }
 645   return true;
 646 }
 647
 648 bool PostgreSqlStorage::removeNetwork(UserId user, const NetworkId &networkId) {
 649   QSqlDatabase db = logDb();
 650   if(!db.transaction()) {
 651     qWarning() << "PostgreSqlStorage::removeNetwork(): cannot start transaction!";
 652     qWarning() << " -" << qPrintable(db.lastError().text());
 653     return false;
 654   }
 655
 656   QSqlQuery query(db);
 657   query.prepare(queryString("delete_network"));
 658   query.bindValue(":userid", user.toInt());
 659   query.bindValue(":networkid", networkId.toInt());
 660   safeExec(query);
 661   if(!watchQuery(query)) {
 662     db.rollback();
 663     return false;
 664   }
 665
 666   db.commit();
 667   return true;
 668 }
 669
 670 QList<NetworkInfo> PostgreSqlStorage::networks(UserId user) {
 671   QList<NetworkInfo> nets;
 672
 673   QSqlDatabase db = logDb();
 674   if(!beginReadOnlyTransaction(db)) {
 675     qWarning() << "PostgreSqlStorage::networks(): cannot start read only transaction!";
 676     qWarning() << " -" << qPrintable(db.lastError().text());
 677     return nets;
 678   }
 679
 680   QSqlQuery networksQuery(db);
 681   networksQuery.prepare(queryString("select_networks_for_user"));
 682   networksQuery.bindValue(":userid", user.toInt());
 683
 684   QSqlQuery serversQuery(db);
 685   serversQuery.prepare(queryString("select_servers_for_network"));
 686
 687   safeExec(networksQuery);
 688   if(!watchQuery(networksQuery)) {
 689     db.rollback();
 690     return nets;
 691   }
 692
 693   while(networksQuery.next()) {
 694     NetworkInfo net;
 695     net.networkId = networksQuery.value(0).toInt();
 696     net.networkName = networksQuery.value(1).toString();
 697     net.identity = networksQuery.value(2).toInt();
 698     net.codecForServer = networksQuery.value(3).toString().toAscii();
 699     net.codecForEncoding = networksQuery.value(4).toString().toAscii();
 700     net.codecForDecoding = networksQuery.value(5).toString().toAscii();
 701     net.useRandomServer = networksQuery.value(6).toBool();
 702     net.perform = networksQuery.value(7).toString().split("\n");
 703     net.useAutoIdentify = networksQuery.value(8).toBool();
 704     net.autoIdentifyService = networksQuery.value(9).toString();
 705     net.autoIdentifyPassword = networksQuery.value(10).toString();
 706     net.useAutoReconnect = networksQuery.value(11).toBool();
 707     net.autoReconnectInterval = networksQuery.value(12).toUInt();
 708     net.autoReconnectRetries = networksQuery.value(13).toInt();
 709     net.unlimitedReconnectRetries = networksQuery.value(14).toBool();
 710     net.rejoinChannels = networksQuery.value(15).toBool();
 711
 712     serversQuery.bindValue(":networkid", net.networkId.toInt());
 713     safeExec(serversQuery);
 714     if(!watchQuery(serversQuery)) {
 715       db.rollback();
 716       return nets;
 717     }
 718
 719     Network::ServerList servers;
 720     while(serversQuery.next()) {
 721       Network::Server server;
 722       server.host = serversQuery.value(0).toString();
 723       server.port = serversQuery.value(1).toUInt();
 724       server.password = serversQuery.value(2).toString();
 725       server.useSsl = serversQuery.value(3).toBool();
 726       server.sslVersion = serversQuery.value(4).toInt();
 727       server.useProxy = serversQuery.value(5).toBool();
 728       server.proxyType = serversQuery.value(6).toInt();
 729       server.proxyHost = serversQuery.value(7).toString();
 730       server.proxyPort = serversQuery.value(8).toUInt();
 731       server.proxyUser = serversQuery.value(9).toString();
 732       server.proxyPass = serversQuery.value(10).toString();
 733       servers << server;
 734     }
 735     net.serverList = servers;
 736     nets << net;
 737   }
 738   db.commit();
 739   return nets;
 740 }
 741
 742 QList<NetworkId> PostgreSqlStorage::connectedNetworks(UserId user) {
 743   QList<NetworkId> connectedNets;
 744
 745   QSqlDatabase db = logDb();
 746   if(!beginReadOnlyTransaction(db)) {
 747     qWarning() << "PostgreSqlStorage::connectedNetworks(): cannot start read only transaction!";
 748     qWarning() << " -" << qPrintable(db.lastError().text());
 749     return connectedNets;
 750   }
 751
 752   QSqlQuery query(db);
 753   query.prepare(queryString("select_connected_networks"));
 754   query.bindValue(":userid", user.toInt());
 755   safeExec(query);
 756   watchQuery(query);
 757
 758   while(query.next()) {
 759     connectedNets << query.value(0).toInt();
 760   }
 761
 762   db.commit();
 763   return connectedNets;
 764 }
 765
 766 void PostgreSqlStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected) {
 767   QSqlQuery query(logDb());
 768   query.prepare(queryString("update_network_connected"));
 769   query.bindValue(":userid", user.toInt());
 770   query.bindValue(":networkid", networkId.toInt());
 771   query.bindValue(":connected", isConnected);
 772   safeExec(query);
 773   watchQuery(query);
 774 }
 775
 776 QHash<QString, QString> PostgreSqlStorage::persistentChannels(UserId user, const NetworkId &networkId) {
 777   QHash<QString, QString> persistentChans;
 778
 779   QSqlDatabase db = logDb();
 780   if(!beginReadOnlyTransaction(db)) {
 781     qWarning() << "PostgreSqlStorage::persistentChannels(): cannot start read only transaction!";
 782     qWarning() << " -" << qPrintable(db.lastError().text());
 783     return persistentChans;
 784   }
 785
 786   QSqlQuery query(db);
 787   query.prepare(queryString("select_persistent_channels"));
 788   query.bindValue(":userid", user.toInt());
 789   query.bindValue(":networkid", networkId.toInt());
 790   safeExec(query);
 791   watchQuery(query);
 792
 793   while(query.next()) {
 794     persistentChans[query.value(0).toString()] = query.value(1).toString();
 795   }
 796
 797   db.commit();
 798   return persistentChans;
 799 }
 800
 801 void PostgreSqlStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined) {
 802   QSqlQuery query(logDb());
 803   query.prepare(queryString("update_buffer_persistent_channel"));
 804   query.bindValue(":userid", user.toInt());
 805   query.bindValue(":networkId", networkId.toInt());
 806   query.bindValue(":buffercname", channel.toLower());
 807   query.bindValue(":joined", isJoined);
 808   safeExec(query);
 809   watchQuery(query);
 810 }
 811
 812 void PostgreSqlStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key) {
 813   QSqlQuery query(logDb());
 814   query.prepare(queryString("update_buffer_set_channel_key"));
 815   query.bindValue(":userid", user.toInt());
 816   query.bindValue(":networkId", networkId.toInt());
 817   query.bindValue(":buffercname", channel.toLower());
 818   query.bindValue(":key", key);
 819   safeExec(query);
 820   watchQuery(query);
 821 }
 822
 823 QString PostgreSqlStorage::awayMessage(UserId user, NetworkId networkId) {
 824   QSqlQuery query(logDb());
 825   query.prepare(queryString("select_network_awaymsg"));
 826   query.bindValue(":userid", user.toInt());
 827   query.bindValue(":networkid", networkId.toInt());
 828   safeExec(query);
 829   watchQuery(query);
 830   QString awayMsg;
 831   if(query.first())
 832     awayMsg = query.value(0).toString();
 833   return awayMsg;
 834 }
 835
 836 void PostgreSqlStorage::setAwayMessage(UserId user, NetworkId networkId, const QString &awayMsg) {
 837   QSqlQuery query(logDb());
 838   query.prepare(queryString("update_network_set_awaymsg"));
 839   query.bindValue(":userid", user.toInt());
 840   query.bindValue(":networkid", networkId.toInt());
 841   query.bindValue(":awaymsg", awayMsg);
 842   safeExec(query);
 843   watchQuery(query);
 844 }
 845
 846 QString PostgreSqlStorage::userModes(UserId user, NetworkId networkId) {
 847   QSqlQuery query(logDb());
 848   query.prepare(queryString("select_network_usermode"));
 849   query.bindValue(":userid", user.toInt());
 850   query.bindValue(":networkid", networkId.toInt());
 851   safeExec(query);
 852   watchQuery(query);
 853   QString modes;
 854   if(query.first())
 855     modes = query.value(0).toString();
 856   return modes;
 857 }
 858
 859 void PostgreSqlStorage::setUserModes(UserId user, NetworkId networkId, const QString &userModes) {
 860   QSqlQuery query(logDb());
 861   query.prepare(queryString("update_network_set_usermode"));
 862   query.bindValue(":userid", user.toInt());
 863   query.bindValue(":networkid", networkId.toInt());
 864   query.bindValue(":usermode", userModes);
 865   safeExec(query);
 866   watchQuery(query);
 867 }
 868
 869 BufferInfo PostgreSqlStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create) {
 870   QSqlDatabase db = logDb();
 871   if(!db.transaction()) {
 872     qWarning() << "PostgreSqlStorage::bufferInfo(): cannot start read only transaction!";
 873     qWarning() << " -" << qPrintable(db.lastError().text());
 874     return BufferInfo();
 875   }
 876
 877   QSqlQuery query(db);
 878   query.prepare(queryString("select_bufferByName"));
 879   query.bindValue(":networkid", networkId.toInt());
 880   query.bindValue(":userid", user.toInt());
 881   query.bindValue(":buffercname", buffer.toLower());
 882   safeExec(query);
 883
 884   if(query.first()) {
 885     BufferInfo bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
 886     if(query.next()) {
 887       qCritical() << "PostgreSqlStorage::bufferInfo(): received more then one Buffer!";
 888       qCritical() << "         Query:" << query.lastQuery();
 889       qCritical() << "  bound Values:";
 890       QList<QVariant> list = query.boundValues().values();
 891       for (int i = 0; i < list.size(); ++i)
 892         qCritical() << i << ":" << list.at(i).toString().toAscii().data();
 893       Q_ASSERT(false);
 894     }
 895     db.commit();
 896     return bufferInfo;
 897   }
 898
 899   if(!create) {
 900     db.rollback();
 901     return BufferInfo();
 902   }
 903
 904   QSqlQuery createQuery(db);
 905   createQuery.prepare(queryString("insert_buffer"));
 906   createQuery.bindValue(":userid", user.toInt());
 907   createQuery.bindValue(":networkid", networkId.toInt());
 908   createQuery.bindValue(":buffertype", (int)type);
 909   createQuery.bindValue(":buffername", buffer);
 910   createQuery.bindValue(":buffercname", buffer.toLower());
 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 }