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