src/core/sqlitestorage.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by the Quassel IRC Team                         *
   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 "network.h"
  26
  27 #include "util.h"
  28 #include "logger.h"
  29
  30 int SqliteStorage::_maxRetryCount = 150; // yes this is a large number... only other way to "handle" this is bailing out...
  31
  32 SqliteStorage::SqliteStorage(QObject *parent)
  33   : AbstractSqlStorage(parent)
  34 {
  35 }
  36
  37 SqliteStorage::~SqliteStorage() {
  38 }
  39
  40 bool SqliteStorage::isAvailable() const {
  41   if(!QSqlDatabase::isDriverAvailable("QSQLITE")) return false;
  42   return true;
  43 }
  44
  45 QString SqliteStorage::displayName() const {
  46   return QString("SQLite");
  47 }
  48
  49 QString SqliteStorage::description() const {
  50   return tr("SQLite is a file-based database engine that does not require any setup. It is suitable for small and medium-sized "
  51             "databases that do not require access via network. Use SQLite if your Quassel Core should store its data on the same machine "
  52             "it is running on, and if you only expect a few users to use your core.");
  53 }
  54
  55 int SqliteStorage::installedSchemaVersion() {
  56   QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
  57   if(query.first())
  58     return query.value(0).toInt();
  59
  60   // maybe it's really old... (schema version 0)
  61   query = logDb().exec("SELECT MAX(version) FROM coreinfo");
  62   if(query.first())
  63     return query.value(0).toInt();
  64
  65   return AbstractSqlStorage::installedSchemaVersion();
  66 }
  67
  68 UserId SqliteStorage::addUser(const QString &user, const QString &password) {
  69   QSqlQuery query(logDb());
  70   query.prepare(queryString("insert_quasseluser"));
  71   query.bindValue(":username", user);
  72   query.bindValue(":password", cryptedPassword(password));
  73   safeExec(query);
  74   if(query.lastError().isValid() && query.lastError().number() == 19) { // user already exists - sadly 19 seems to be the general constraint violation error...
  75     return 0;
  76   }
  77
  78   query.prepare(queryString("select_userid"));
  79   query.bindValue(":username", user);
  80   safeExec(query);
  81   query.first();
  82   UserId uid = query.value(0).toInt();
  83   emit userAdded(uid, user);
  84   return uid;
  85 }
  86
  87 void SqliteStorage::updateUser(UserId user, const QString &password) {
  88   QSqlQuery query(logDb());
  89   query.prepare(queryString("update_userpassword"));
  90   query.bindValue(":userid", user.toInt());
  91   query.bindValue(":password", cryptedPassword(password));
  92   safeExec(query);
  93 }
  94
  95 void SqliteStorage::renameUser(UserId user, const QString &newName) {
  96   QSqlQuery query(logDb());
  97   query.prepare(queryString("update_username"));
  98   query.bindValue(":userid", user.toInt());
  99   query.bindValue(":username", newName);
 100   safeExec(query);
 101   emit userRenamed(user, newName);
 102 }
 103
 104 UserId SqliteStorage::validateUser(const QString &user, const QString &password) {
 105   QSqlQuery query(logDb());
 106   query.prepare(queryString("select_authuser"));
 107   query.bindValue(":username", user);
 108   query.bindValue(":password", cryptedPassword(password));
 109   safeExec(query);
 110
 111   if(query.first()) {
 112     return query.value(0).toInt();
 113   } else {
 114     return 0;
 115   }
 116 }
 117
 118 UserId SqliteStorage::internalUser() {
 119   QSqlQuery query(logDb());
 120   query.prepare(queryString("select_internaluser"));
 121   safeExec(query);
 122
 123   if(query.first()) {
 124     return query.value(0).toInt();
 125   } else {
 126     return 0;
 127   }
 128 }
 129
 130 void SqliteStorage::delUser(UserId user) {
 131   QSqlQuery query(logDb());
 132   query.prepare(queryString("delete_backlog_by_uid"));
 133   query.bindValue(":userid", user.toInt());
 134   safeExec(query);
 135
 136   query.prepare(queryString("delete_buffers_by_uid"));
 137   query.bindValue(":userid", user.toInt());
 138   safeExec(query);
 139
 140   query.prepare(queryString("delete_networks_by_uid"));
 141   query.bindValue(":userid", user.toInt());
 142   safeExec(query);
 143
 144   query.prepare(queryString("delete_quasseluser"));
 145   query.bindValue(":userid", user.toInt());
 146   safeExec(query);
 147   // I hate the lack of foreign keys and on delete cascade... :(
 148   emit userRemoved(user);
 149 }
 150
 151 void SqliteStorage::setUserSetting(UserId userId, const QString &settingName, const QVariant &data) {
 152   QByteArray rawData;
 153   QDataStream out(&rawData, QIODevice::WriteOnly);
 154   out.setVersion(QDataStream::Qt_4_2);
 155   out << data;
 156
 157   QSqlQuery query(logDb());
 158   query.prepare(queryString("insert_user_setting"));
 159   query.bindValue(":userid", userId.toInt());
 160   query.bindValue(":settingname", settingName);
 161   query.bindValue(":settingvalue", rawData);
 162   safeExec(query);
 163
 164   if(query.lastError().isValid()) {
 165     QSqlQuery updateQuery(logDb());
 166     updateQuery.prepare(queryString("update_user_setting"));
 167     updateQuery.bindValue(":userid", userId.toInt());
 168     updateQuery.bindValue(":settingname", settingName);
 169     updateQuery.bindValue(":settingvalue", rawData);
 170     safeExec(updateQuery);
 171   }
 172
 173 }
 174
 175 QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData) {
 176   QSqlQuery query(logDb());
 177   query.prepare(queryString("select_user_setting"));
 178   query.bindValue(":userid", userId.toInt());
 179   query.bindValue(":settingname", settingName);
 180   safeExec(query);
 181
 182   if(query.first()) {
 183     QVariant data;
 184     QByteArray rawData = query.value(0).toByteArray();
 185     QDataStream in(&rawData, QIODevice::ReadOnly);
 186     in.setVersion(QDataStream::Qt_4_2);
 187     in >> data;
 188     return data;
 189   } else {
 190     return defaultData;
 191   }
 192 }
 193
 194 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info) {
 195   NetworkId networkId;
 196   QSqlQuery query(logDb());
 197   query.prepare(queryString("insert_network"));
 198   query.bindValue(":userid", user.toInt());
 199   query.bindValue(":networkname", info.networkName);
 200   safeExec(query);
 201
 202   networkId = getNetworkId(user, info.networkName);
 203   if(!networkId.isValid()) {
 204     watchQuery(query);
 205   } else {
 206     NetworkInfo newNetworkInfo = info;
 207     newNetworkInfo.networkId = networkId;
 208     updateNetwork(user, newNetworkInfo);
 209   }
 210   return networkId;
 211 }
 212
 213 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info) {
 214   if(!isValidNetwork(user, info.networkId))
 215      return false;
 216
 217   QSqlQuery updateQuery(logDb());
 218   updateQuery.prepare(queryString("update_network"));
 219   updateQuery.bindValue(":networkname", info.networkName);
 220   updateQuery.bindValue(":identityid", info.identity.toInt());
 221   updateQuery.bindValue(":usecustomencoding", info.useCustomEncodings ? 1 : 0);
 222   updateQuery.bindValue(":encodingcodec", QString(info.codecForEncoding));
 223   updateQuery.bindValue(":decodingcodec", QString(info.codecForDecoding));
 224   updateQuery.bindValue(":servercodec", QString(info.codecForServer));
 225   updateQuery.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
 226   updateQuery.bindValue(":perform", info.perform.join("\n"));
 227   updateQuery.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
 228   updateQuery.bindValue(":autoidentifyservice", info.autoIdentifyService);
 229   updateQuery.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 230   updateQuery.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
 231   updateQuery.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 232   updateQuery.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 233   updateQuery.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0);
 234   updateQuery.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
 235   updateQuery.bindValue(":networkid", info.networkId.toInt());
 236   safeExec(updateQuery);
 237   if(!watchQuery(updateQuery))
 238     return false;
 239
 240   QSqlQuery dropServersQuery(logDb());
 241   dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 242   dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 243   safeExec(dropServersQuery);
 244   if(!watchQuery(dropServersQuery))
 245     return false;
 246
 247   QSqlQuery insertServersQuery(logDb());
 248   insertServersQuery.prepare(queryString("insert_server"));
 249   foreach(Network::Server server, info.serverList) {
 250     insertServersQuery.bindValue(":hostname", server.host);
 251     insertServersQuery.bindValue(":port", server.port);
 252     insertServersQuery.bindValue(":password", server.password);
 253     insertServersQuery.bindValue(":ssl", server.useSsl ? 1 : 0);
 254     insertServersQuery.bindValue(":userid", user.toInt());
 255     insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 256
 257     safeExec(insertServersQuery);
 258     if(!watchQuery(insertServersQuery))
 259       return false;
 260   }
 261
 262   return true;
 263 }
 264
 265 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId) {
 266   if(!isValidNetwork(user, networkId))
 267      return false;
 268
 269   bool withTransaction = logDb().driver()->hasFeature(QSqlDriver::Transactions);
 270   if(withTransaction) {
 271     sync();
 272     if(!logDb().transaction()) {
 273       qWarning() << "SqliteStorage::removeNetwork(): cannot start transaction. continuing with out rollback support!";
 274       withTransaction = false;
 275     }
 276   }
 277
 278   QSqlQuery deleteBacklogQuery(logDb());
 279   deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
 280   deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
 281   safeExec(deleteBacklogQuery);
 282   if(!watchQuery(deleteBacklogQuery)) {
 283     if(withTransaction)
 284       logDb().rollback();
 285     return false;
 286   }
 287
 288   QSqlQuery deleteBuffersQuery(logDb());
 289   deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
 290   deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
 291   safeExec(deleteBuffersQuery);
 292   if(!watchQuery(deleteBuffersQuery)) {
 293     if(withTransaction)
 294       logDb().rollback();
 295     return false;
 296   }
 297
 298   QSqlQuery deleteServersQuery(logDb());
 299   deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
 300   deleteServersQuery.bindValue(":networkid", networkId.toInt());
 301   safeExec(deleteServersQuery);
 302   if(!watchQuery(deleteServersQuery)) {
 303     if(withTransaction)
 304       logDb().rollback();
 305     return false;
 306   }
 307
 308   QSqlQuery deleteNetworkQuery(logDb());
 309   deleteNetworkQuery.prepare(queryString("delete_network"));
 310   deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
 311   safeExec(deleteNetworkQuery);
 312   if(!watchQuery(deleteNetworkQuery)) {
 313     if(withTransaction)
 314       logDb().rollback();
 315     return false;
 316   }
 317
 318   logDb().commit();
 319   return true;
 320 }
 321
 322 QList<NetworkInfo> SqliteStorage::networks(UserId user) {
 323   QList<NetworkInfo> nets;
 324
 325   QSqlQuery networksQuery(logDb());
 326   networksQuery.prepare(queryString("select_networks_for_user"));
 327   networksQuery.bindValue(":userid", user.toInt());
 328
 329   QSqlQuery serversQuery(logDb());
 330   serversQuery.prepare(queryString("select_servers_for_network"));
 331
 332   safeExec(networksQuery);
 333   if(!watchQuery(networksQuery))
 334     return nets;
 335
 336   while(networksQuery.next()) {
 337     NetworkInfo net;
 338     net.networkId = networksQuery.value(0).toInt();
 339     net.networkName = networksQuery.value(1).toString();
 340     net.identity = networksQuery.value(2).toInt();
 341     net.codecForServer = networksQuery.value(3).toString().toAscii();
 342     net.codecForEncoding = networksQuery.value(4).toString().toAscii();
 343     net.codecForDecoding = networksQuery.value(5).toString().toAscii();
 344     net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
 345     net.perform = networksQuery.value(7).toString().split("\n");
 346     net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
 347     net.autoIdentifyService = networksQuery.value(9).toString();
 348     net.autoIdentifyPassword = networksQuery.value(10).toString();
 349     net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
 350     net.autoReconnectInterval = networksQuery.value(12).toUInt();
 351     net.autoReconnectRetries = networksQuery.value(13).toInt();
 352     net.unlimitedReconnectRetries = networksQuery.value(14).toInt() == 1 ? true : false;
 353     net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false;
 354
 355     serversQuery.bindValue(":networkid", net.networkId.toInt());
 356     safeExec(serversQuery);
 357     if(!watchQuery(serversQuery))
 358       return nets;
 359
 360     Network::ServerList servers;
 361     while(serversQuery.next()) {
 362       Network::Server server;
 363       server.host = serversQuery.value(0).toString();
 364       server.port = serversQuery.value(1).toUInt();
 365       server.password = serversQuery.value(2).toString();
 366       server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false;
 367       servers << server;
 368     }
 369     net.serverList = servers;
 370     nets << net;
 371   }
 372   return nets;
 373 }
 374
 375 bool SqliteStorage::isValidNetwork(UserId user, const NetworkId &networkId) {
 376   QSqlQuery query(logDb());
 377   query.prepare(queryString("select_networkExists"));
 378   query.bindValue(":userid", user.toInt());
 379   query.bindValue(":networkid", networkId.toInt());
 380   safeExec(query);
 381
 382   watchQuery(query); // there should not occur any errors
 383   if(!query.first())
 384     return false;
 385
 386   Q_ASSERT(!query.next());
 387   return true;
 388 }
 389
 390 bool SqliteStorage::isValidBuffer(const UserId &user, const BufferId &bufferId) {
 391   QSqlQuery query(logDb());
 392   query.prepare(queryString("select_bufferExists"));
 393   query.bindValue(":userid", user.toInt());
 394   query.bindValue(":bufferid", bufferId.toInt());
 395   safeExec(query);
 396
 397   watchQuery(query);
 398   if(!query.first())
 399     return false;
 400
 401   Q_ASSERT(!query.next());
 402   return true;
 403 }
 404
 405 NetworkId SqliteStorage::getNetworkId(UserId user, const QString &network) {
 406   QSqlQuery query(logDb());
 407   query.prepare("SELECT networkid FROM network "
 408                 "WHERE userid = :userid AND networkname = :networkname");
 409   query.bindValue(":userid", user.toInt());
 410   query.bindValue(":networkname", network);
 411   safeExec(query);
 412
 413   if(query.first())
 414     return query.value(0).toInt();
 415   else
 416     return NetworkId();
 417 }
 418
 419 QList<NetworkId> SqliteStorage::connectedNetworks(UserId user) {
 420   QList<NetworkId> connectedNets;
 421   QSqlQuery query(logDb());
 422   query.prepare(queryString("select_connected_networks"));
 423   query.bindValue(":userid", user.toInt());
 424   safeExec(query);
 425   watchQuery(query);
 426
 427   while(query.next()) {
 428     connectedNets << query.value(0).toInt();
 429   }
 430
 431   return connectedNets;
 432 }
 433
 434 void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected) {
 435   QSqlQuery query(logDb());
 436   query.prepare(queryString("update_network_connected"));
 437   query.bindValue(":userid", user.toInt());
 438   query.bindValue(":networkid", networkId.toInt());
 439   query.bindValue(":connected", isConnected ? 1 : 0);
 440   safeExec(query);
 441   watchQuery(query);
 442 }
 443
 444 QHash<QString, QString> SqliteStorage::persistentChannels(UserId user, const NetworkId &networkId) {
 445   QHash<QString, QString> persistentChans;
 446   QSqlQuery query(logDb());
 447   query.prepare(queryString("select_persistent_channels"));
 448   query.bindValue(":userid", user.toInt());
 449   query.bindValue(":networkid", networkId.toInt());
 450   safeExec(query);
 451   watchQuery(query);
 452
 453   while(query.next()) {
 454     persistentChans[query.value(0).toString()] = query.value(1).toString();
 455   }
 456
 457   return persistentChans;
 458 }
 459
 460 void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined) {
 461   QSqlQuery query(logDb());
 462   query.prepare(queryString("update_buffer_persistent_channel"));
 463   query.bindValue(":userid", user.toInt());
 464   query.bindValue(":networkId", networkId.toInt());
 465   query.bindValue(":buffercname", channel.toLower());
 466   query.bindValue(":joined", isJoined ? 1 : 0);
 467   safeExec(query);
 468   watchQuery(query);
 469 }
 470
 471 void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key) {
 472   QSqlQuery query(logDb());
 473   query.prepare(queryString("update_buffer_set_channel_key"));
 474   query.bindValue(":userid", user.toInt());
 475   query.bindValue(":networkId", networkId.toInt());
 476   query.bindValue(":buffercname", channel.toLower());
 477   query.bindValue(":key", key);
 478   safeExec(query);
 479   watchQuery(query);
 480 }
 481
 482
 483 void SqliteStorage::createBuffer(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) {
 484   QSqlQuery query(logDb());
 485   query.prepare(queryString("insert_buffer"));
 486   query.bindValue(":userid", user.toInt());
 487   query.bindValue(":networkid", networkId.toInt());
 488   query.bindValue(":buffertype", (int)type);
 489   query.bindValue(":buffername", buffer);
 490   query.bindValue(":buffercname", buffer.toLower());
 491   safeExec(query);
 492
 493   watchQuery(query);
 494 }
 495
 496 BufferInfo SqliteStorage::getBufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) {
 497   QSqlQuery query(logDb());
 498   query.prepare(queryString("select_bufferByName"));
 499   query.bindValue(":networkid", networkId.toInt());
 500   query.bindValue(":userid", user.toInt());
 501   query.bindValue(":buffercname", buffer.toLower());
 502   safeExec(query);
 503
 504   if(!query.first()) {
 505     createBuffer(user, networkId, type, buffer);
 506     safeExec(query);
 507     if(!query.first()) {
 508       watchQuery(query);
 509       qWarning() << "unable to create BufferInfo for:" << user << networkId << buffer;
 510       return BufferInfo();
 511     }
 512   }
 513
 514   BufferInfo bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
 515   if(query.next()) {
 516     qCritical() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
 517     qCritical() << "         Query:" << query.lastQuery();
 518     qCritical() << "  bound Values:";
 519     QList<QVariant> list = query.boundValues().values();
 520     for (int i = 0; i < list.size(); ++i)
 521       qCritical() << i << ":" << list.at(i).toString().toAscii().data();
 522     Q_ASSERT(false);
 523   }
 524
 525   return bufferInfo;
 526 }
 527
 528 BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId) {
 529   QSqlQuery query(logDb());
 530   query.prepare(queryString("select_buffer_by_id"));
 531   query.bindValue(":userid", user.toInt());
 532   query.bindValue(":bufferid", bufferId.toInt());
 533   safeExec(query);
 534   if(!watchQuery(query))
 535     return BufferInfo();
 536
 537   if(!query.first())
 538     return BufferInfo();
 539
 540   BufferInfo bufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
 541   Q_ASSERT(!query.next());
 542
 543   return bufferInfo;
 544 }
 545
 546 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user) {
 547   QList<BufferInfo> bufferlist;
 548   QSqlQuery query(logDb());
 549   query.prepare(queryString("select_buffers"));
 550   query.bindValue(":userid", user.toInt());
 551
 552   safeExec(query);
 553   watchQuery(query);
 554   while(query.next()) {
 555     bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
 556   }
 557   return bufferlist;
 558 }
 559
 560 QList<BufferId> SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId) {
 561   QList<BufferId> bufferList;
 562   QSqlQuery query(logDb());
 563   query.prepare(queryString("select_buffers_for_network"));
 564   query.bindValue(":networkid", networkId.toInt());
 565   query.bindValue(":userid", user.toInt());
 566
 567   safeExec(query);
 568   watchQuery(query);
 569   while(query.next()) {
 570     bufferList << BufferId(query.value(0).toInt());
 571   }
 572   return bufferList;
 573 }
 574
 575 bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId) {
 576   if(!isValidBuffer(user, bufferId))
 577     return false;
 578
 579   QSqlQuery delBacklogQuery(logDb());
 580   delBacklogQuery.prepare(queryString("delete_backlog_for_buffer"));
 581   delBacklogQuery.bindValue(":bufferid", bufferId.toInt());
 582   safeExec(delBacklogQuery);
 583   if(!watchQuery(delBacklogQuery))
 584     return false;
 585
 586   QSqlQuery delBufferQuery(logDb());
 587   delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
 588   delBufferQuery.bindValue(":bufferid", bufferId.toInt());
 589   safeExec(delBufferQuery);
 590   if(!watchQuery(delBufferQuery))
 591     return false;
 592
 593   return true;
 594 }
 595
 596 BufferId SqliteStorage::renameBuffer(const UserId &user, const NetworkId &networkId, const QString &newName, const QString &oldName) {
 597   // check if such a buffer exists...
 598   QSqlQuery existsQuery(logDb());
 599   existsQuery.prepare(queryString("select_bufferByName"));
 600   existsQuery.bindValue(":networkid", networkId.toInt());
 601   existsQuery.bindValue(":userid", user.toInt());
 602   existsQuery.bindValue(":buffercname", oldName.toLower());
 603   safeExec(existsQuery);
 604   if(!watchQuery(existsQuery))
 605     return false;
 606
 607   if(!existsQuery.first())
 608     return false;
 609
 610   const int bufferid = existsQuery.value(0).toInt();
 611
 612   Q_ASSERT(!existsQuery.next());
 613
 614   // ... and if the new name is still free.
 615   existsQuery.bindValue(":networkid", networkId.toInt());
 616   existsQuery.bindValue(":userid", user.toInt());
 617   existsQuery.bindValue(":buffercname", newName.toLower());
 618   safeExec(existsQuery);
 619   if(!watchQuery(existsQuery))
 620     return false;
 621
 622   if(existsQuery.first())
 623     return false;
 624
 625   QSqlQuery renameBufferQuery(logDb());
 626   renameBufferQuery.prepare(queryString("update_buffer_name"));
 627   renameBufferQuery.bindValue(":buffername", newName);
 628   renameBufferQuery.bindValue(":buffercname", newName.toLower());
 629   renameBufferQuery.bindValue(":bufferid", bufferid);
 630   safeExec(renameBufferQuery);
 631   if(watchQuery(existsQuery))
 632     return BufferId(bufferid);
 633   else
 634     return BufferId();
 635 }
 636
 637 void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId) {
 638   QSqlQuery query(logDb());
 639   query.prepare(queryString("update_buffer_lastseen"));
 640
 641   query.bindValue(":userid", user.toInt());
 642   query.bindValue(":bufferid", bufferId.toInt());
 643   query.bindValue(":lastseenmsgid", msgId.toInt());
 644   safeExec(query);
 645   watchQuery(query);
 646 }
 647
 648 QHash<BufferId, MsgId> SqliteStorage::bufferLastSeenMsgIds(UserId user) {
 649   QHash<BufferId, MsgId> lastSeenHash;
 650   QSqlQuery query(logDb());
 651   query.prepare(queryString("select_buffer_lastseen_messages"));
 652   query.bindValue(":userid", user.toInt());
 653   safeExec(query);
 654   if(!watchQuery(query))
 655     return lastSeenHash;
 656
 657   while(query.next()) {
 658     lastSeenHash[query.value(0).toInt()] = query.value(1).toInt();
 659   }
 660   return lastSeenHash;
 661 }
 662
 663 MsgId SqliteStorage::logMessage(Message msg) {
 664   QSqlQuery logMessageQuery(logDb());
 665   logMessageQuery.prepare(queryString("insert_message"));
 666
 667   logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
 668   logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
 669   logMessageQuery.bindValue(":type", msg.type());
 670   logMessageQuery.bindValue(":flags", (int)msg.flags());
 671   logMessageQuery.bindValue(":sender", msg.sender());
 672   logMessageQuery.bindValue(":message", msg.contents());
 673   safeExec(logMessageQuery);
 674
 675   if(logMessageQuery.lastError().isValid()) {
 676     // constraint violation - must be NOT NULL constraint - probably the sender is missing...
 677     if(logMessageQuery.lastError().number() == 19) {
 678       QSqlQuery addSenderQuery(logDb());
 679       addSenderQuery.prepare(queryString("insert_sender"));
 680       addSenderQuery.bindValue(":sender", msg.sender());
 681       safeExec(addSenderQuery);
 682       safeExec(logMessageQuery);
 683       if(!watchQuery(logMessageQuery))
 684         return 0;
 685     } else {
 686       watchQuery(logMessageQuery);
 687     }
 688   }
 689
 690   MsgId msgId = logMessageQuery.lastInsertId().toInt();
 691   Q_ASSERT(msgId.isValid());
 692   return msgId;
 693 }
 694
 695 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit) {
 696   QList<Message> messagelist;
 697
 698   BufferInfo bufferInfo = getBufferInfo(user, bufferId);
 699   if(!bufferInfo.isValid())
 700     return messagelist;
 701
 702   QSqlQuery query(logDb());
 703   if(last == -1) {
 704     query.prepare(queryString("select_messagesNew"));
 705   } else {
 706     query.prepare(queryString("select_messages"));
 707     query.bindValue(":lastmsg", last.toInt());
 708   }
 709
 710   query.bindValue(":bufferid", bufferId.toInt());
 711   query.bindValue(":firstmsg", first.toInt());
 712   query.bindValue(":limit", limit);
 713   safeExec(query);
 714
 715   watchQuery(query);
 716
 717   while(query.next()) {
 718     Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
 719                 bufferInfo,
 720                 (Message::Type)query.value(2).toUInt(),
 721                 query.value(5).toString(),
 722                 query.value(4).toString(),
 723                 (Message::Flags)query.value(3).toUInt());
 724     msg.setMsgId(query.value(0).toInt());
 725     messagelist << msg;
 726   }
 727   return messagelist;
 728 }
 729
 730 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit) {
 731   QList<Message> messagelist;
 732
 733   QHash<BufferId, BufferInfo> bufferInfoHash;
 734   foreach(BufferInfo bufferInfo, requestBuffers(user)) {
 735     bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
 736   }
 737
 738   QSqlQuery query(logDb());
 739   if(last == -1) {
 740     query.prepare(queryString("select_messagesAllNew"));
 741   } else {
 742     query.prepare(queryString("select_messagesAll"));
 743     query.bindValue(":lastmsg", last.toInt());
 744   }
 745   query.bindValue(":userid", user.toInt());
 746   query.bindValue(":firstmsg", first.toInt());
 747   query.bindValue(":limit", limit);
 748   safeExec(query);
 749
 750   watchQuery(query);
 751
 752   while(query.next()) {
 753     Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
 754                 bufferInfoHash[query.value(1).toInt()],
 755                 (Message::Type)query.value(3).toUInt(),
 756                 query.value(6).toString(),
 757                 query.value(5).toString(),
 758                 (Message::Flags)query.value(4).toUInt());
 759     msg.setMsgId(query.value(0).toInt());
 760     messagelist << msg;
 761   }
 762
 763   return messagelist;
 764 }
 765
 766 QString SqliteStorage::backlogFile() {
 767   return quasselDir().absolutePath() + "/quassel-storage.sqlite";
 768 }
 769
 770 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount) {
 771   query.exec();
 772
 773   if(!query.lastError().isValid())
 774     return true;
 775
 776   switch(query.lastError().number()) {
 777   case 5: // SQLITE_BUSY         5   /* The database file is locked */
 778   case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
 779     if(retryCount < _maxRetryCount)
 780       return safeExec(query, retryCount + 1);
 781   default:
 782     return false;
 783   }
 784 }