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