you can now rename query buffers
[quassel.git] / 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 IdentityId SqliteStorage::createIdentity(UserId user, CoreIdentity &identity) {
195   IdentityId identityId;
196
197   QSqlQuery query(logDb());
198   query.prepare(queryString("insert_identity"));
199   query.bindValue(":userid", user.toInt());
200   query.bindValue(":identityname", identity.identityName());
201   query.bindValue(":realname", identity.realName());
202   query.bindValue(":awaynick", identity.awayNick());
203   query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
204   query.bindValue(":awayreason", identity.awayReason());
205   query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
206   query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
207   query.bindValue(":autoawaytime", identity.autoAwayTime());
208   query.bindValue(":autoawayreason", identity.autoAwayReason());
209   query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
210   query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
211   query.bindValue(":detachawayreason", identity.detachAwayReason());
212   query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
213   query.bindValue(":ident", identity.ident());
214   query.bindValue(":kickreason", identity.kickReason());
215   query.bindValue(":partreason", identity.partReason());
216   query.bindValue(":quitreason", identity.quitReason());
217 #ifdef HAVE_SSL
218   query.bindValue(":sslcert", identity.sslCert().toPem());
219   query.bindValue(":sslkey", identity.sslKey().toPem());
220 #else
221   query.bindValue(":sslcert", QByteArray());
222   query.bindValue(":sslkey", QByteArray());
223 #endif
224
225   safeExec(query);
226
227   identityId = query.lastInsertId().toInt();
228   if(!identityId.isValid()) {
229     watchQuery(query);
230   } else {
231     QSqlQuery deleteNickQuery(logDb());
232     deleteNickQuery.prepare(queryString("delete_nicks"));
233     deleteNickQuery.bindValue(":identityid", identityId.toInt());
234     safeExec(deleteNickQuery);
235
236     QSqlQuery insertNickQuery(logDb());
237     insertNickQuery.prepare(queryString("insert_nick"));
238     foreach(QString nick, identity.nicks()) {
239       insertNickQuery.bindValue(":identityid", identityId.toInt());
240       insertNickQuery.bindValue(":nick", nick);
241       safeExec(insertNickQuery);
242     }
243   }
244   identity.setId(identityId);
245   return identityId;
246 }
247
248 bool SqliteStorage::updateIdentity(UserId user, const CoreIdentity &identity) {
249   QSqlQuery checkQuery(logDb());
250   checkQuery.prepare(queryString("select_checkidentity"));
251   checkQuery.bindValue(":identityid", identity.id().toInt());
252   checkQuery.bindValue(":userid", user.toInt());
253   safeExec(checkQuery);
254
255   if(!checkQuery.first() || checkQuery.value(0).toInt() != 1) // there should be exactly one identity for the given id and user
256     return false;
257
258   QSqlQuery query(logDb());
259   query.prepare(queryString("update_identity"));
260   query.bindValue(":identityname", identity.identityName());
261   query.bindValue(":realname", identity.realName());
262   query.bindValue(":awaynick", identity.awayNick());
263   query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
264   query.bindValue(":awayreason", identity.awayReason());
265   query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
266   query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
267   query.bindValue(":autoawaytime", identity.autoAwayTime());
268   query.bindValue(":autoawayreason", identity.autoAwayReason());
269   query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
270   query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
271   query.bindValue(":detachawayreason", identity.detachAwayReason());
272   query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
273   query.bindValue(":ident", identity.ident());
274   query.bindValue(":kickreason", identity.kickReason());
275   query.bindValue(":partreason", identity.partReason());
276   query.bindValue(":quitreason", identity.quitReason());
277 #ifdef HAVE_SSL
278   query.bindValue(":sslcert", identity.sslCert().toPem());
279   query.bindValue(":sslkey", identity.sslKey().toPem());
280 #else
281   query.bindValue(":sslcert", QByteArray());
282   query.bindValue(":sslkey", QByteArray());
283 #endif
284   query.bindValue(":identityid", identity.id().toInt());
285   safeExec(query);
286
287   QSqlQuery deleteNickQuery(logDb());
288   deleteNickQuery.prepare(queryString("delete_nicks"));
289   deleteNickQuery.bindValue(":identityid", identity.id().toInt());
290   safeExec(deleteNickQuery);
291
292   QSqlQuery insertNickQuery(logDb());
293   insertNickQuery.prepare(queryString("insert_nick"));
294   foreach(QString nick, identity.nicks()) {
295     insertNickQuery.bindValue(":identityid", identity.id().toInt());
296     insertNickQuery.bindValue(":nick", nick);
297     safeExec(insertNickQuery);
298   }
299
300   return true;
301 }
302
303 void SqliteStorage::removeIdentity(UserId user, IdentityId identityId) {
304   QSqlQuery checkQuery(logDb());
305   checkQuery.prepare(queryString("select_checkidentity"));
306   checkQuery.bindValue(":identityid", identityId.toInt());
307   checkQuery.bindValue(":userid", user.toInt());
308   safeExec(checkQuery);
309
310   if(!checkQuery.first() || checkQuery.value(0).toInt() != 1) // there should be exactly one identity for the given id and user
311     return;
312
313   QSqlQuery deleteNickQuery(logDb());
314   deleteNickQuery.prepare(queryString("delete_nicks"));
315   deleteNickQuery.bindValue(":identityid", identityId.toInt());
316   safeExec(deleteNickQuery);
317
318   QSqlQuery deleteIdentityQuery(logDb());
319   deleteIdentityQuery.prepare(queryString("delete_identity"));
320   deleteIdentityQuery.bindValue(":identityid", identityId.toInt());
321   deleteIdentityQuery.bindValue(":userid", user.toInt());
322   safeExec(deleteIdentityQuery);
323 }
324
325 QList<CoreIdentity> SqliteStorage::identities(UserId user) {
326   QList<CoreIdentity> identities;
327
328   QSqlQuery query(logDb());
329   query.prepare(queryString("select_identities"));
330   query.bindValue(":userid", user.toInt());
331
332   QSqlQuery nickQuery(logDb());
333   nickQuery.prepare(queryString("select_nicks"));
334
335   safeExec(query);
336
337   while(query.next()) {
338     CoreIdentity identity(IdentityId(query.value(0).toInt()));
339
340     identity.setIdentityName(query.value(1).toString());
341     identity.setRealName(query.value(2).toString());
342     identity.setAwayNick(query.value(3).toString());
343     identity.setAwayNickEnabled(!!query.value(4).toInt());
344     identity.setAwayReason(query.value(5).toString());
345     identity.setAwayReasonEnabled(!!query.value(6).toInt());
346     identity.setAutoAwayEnabled(!!query.value(7).toInt());
347     identity.setAutoAwayTime(query.value(8).toInt());
348     identity.setAutoAwayReason(query.value(9).toString());
349     identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
350     identity.setDetachAwayEnabled(!!query.value(11).toInt());
351     identity.setDetachAwayReason(query.value(12).toString());
352     identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
353     identity.setIdent(query.value(14).toString());
354     identity.setKickReason(query.value(15).toString());
355     identity.setPartReason(query.value(16).toString());
356     identity.setQuitReason(query.value(17).toString());
357 #ifdef HAVE_SSL
358     identity.setSslCert(query.value(18).toByteArray());
359     identity.setSslKey(query.value(19).toByteArray());
360 #endif
361
362     nickQuery.bindValue(":identityid", identity.id().toInt());
363     QList<QString> nicks;
364     safeExec(nickQuery);
365     watchQuery(nickQuery);
366     while(nickQuery.next()) {
367       nicks << nickQuery.value(0).toString();
368     }
369     identity.setNicks(nicks);
370     identities << identity;
371   }
372   return identities;
373 }
374
375 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info) {
376   NetworkId networkId;
377   QSqlQuery query(logDb());
378   query.prepare(queryString("insert_network"));
379   query.bindValue(":userid", user.toInt());
380   query.bindValue(":networkname", info.networkName);
381   safeExec(query);
382
383   networkId = getNetworkId(user, info.networkName);
384   if(!networkId.isValid()) {
385     watchQuery(query);
386   } else {
387     NetworkInfo newNetworkInfo = info;
388     newNetworkInfo.networkId = networkId;
389     updateNetwork(user, newNetworkInfo);
390   }
391   return networkId;
392 }
393
394 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info) {
395   if(!isValidNetwork(user, info.networkId))
396      return false;
397
398   QSqlQuery updateQuery(logDb());
399   updateQuery.prepare(queryString("update_network"));
400   updateQuery.bindValue(":networkname", info.networkName);
401   updateQuery.bindValue(":identityid", info.identity.toInt());
402   updateQuery.bindValue(":usecustomencoding", info.useCustomEncodings ? 1 : 0);
403   updateQuery.bindValue(":encodingcodec", QString(info.codecForEncoding));
404   updateQuery.bindValue(":decodingcodec", QString(info.codecForDecoding));
405   updateQuery.bindValue(":servercodec", QString(info.codecForServer));
406   updateQuery.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
407   updateQuery.bindValue(":perform", info.perform.join("\n"));
408   updateQuery.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
409   updateQuery.bindValue(":autoidentifyservice", info.autoIdentifyService);
410   updateQuery.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
411   updateQuery.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
412   updateQuery.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
413   updateQuery.bindValue(":autoreconnectretries", info.autoReconnectRetries);
414   updateQuery.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0);
415   updateQuery.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
416   updateQuery.bindValue(":networkid", info.networkId.toInt());
417   safeExec(updateQuery);
418   if(!watchQuery(updateQuery))
419     return false;
420
421   QSqlQuery dropServersQuery(logDb());
422   dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
423   dropServersQuery.bindValue(":networkid", info.networkId.toInt());
424   safeExec(dropServersQuery);
425   if(!watchQuery(dropServersQuery))
426     return false;
427
428   QSqlQuery insertServersQuery(logDb());
429   insertServersQuery.prepare(queryString("insert_server"));
430   foreach(Network::Server server, info.serverList) {
431     insertServersQuery.bindValue(":userid", user.toInt());
432     insertServersQuery.bindValue(":networkid", info.networkId.toInt());
433     insertServersQuery.bindValue(":hostname", server.host);
434     insertServersQuery.bindValue(":port", server.port);
435     insertServersQuery.bindValue(":password", server.password);
436     insertServersQuery.bindValue(":ssl", server.useSsl ? 1 : 0);
437     insertServersQuery.bindValue(":sslversion", server.sslVersion);
438     insertServersQuery.bindValue(":useproxy", server.useProxy ? 1 : 0);
439     insertServersQuery.bindValue(":proxytype", server.proxyType);
440     insertServersQuery.bindValue(":proxyhost", server.proxyHost);
441     insertServersQuery.bindValue(":proxyport", server.proxyPort);
442     insertServersQuery.bindValue(":proxyuser", server.proxyUser);
443     insertServersQuery.bindValue(":proxypass", server.proxyPass);
444
445     safeExec(insertServersQuery);
446     if(!watchQuery(insertServersQuery))
447       return false;
448   }
449
450   return true;
451 }
452
453 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId) {
454   if(!isValidNetwork(user, networkId))
455      return false;
456
457   bool withTransaction = logDb().driver()->hasFeature(QSqlDriver::Transactions);
458   if(withTransaction) {
459     sync();
460     if(!logDb().transaction()) {
461       qWarning() << "SqliteStorage::removeNetwork(): cannot start transaction. continuing with out rollback support!";
462       withTransaction = false;
463     }
464   }
465
466   QSqlQuery deleteBacklogQuery(logDb());
467   deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
468   deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
469   safeExec(deleteBacklogQuery);
470   if(!watchQuery(deleteBacklogQuery)) {
471     if(withTransaction)
472       logDb().rollback();
473     return false;
474   }
475
476   QSqlQuery deleteBuffersQuery(logDb());
477   deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
478   deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
479   safeExec(deleteBuffersQuery);
480   if(!watchQuery(deleteBuffersQuery)) {
481     if(withTransaction)
482       logDb().rollback();
483     return false;
484   }
485
486   QSqlQuery deleteServersQuery(logDb());
487   deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
488   deleteServersQuery.bindValue(":networkid", networkId.toInt());
489   safeExec(deleteServersQuery);
490   if(!watchQuery(deleteServersQuery)) {
491     if(withTransaction)
492       logDb().rollback();
493     return false;
494   }
495
496   QSqlQuery deleteNetworkQuery(logDb());
497   deleteNetworkQuery.prepare(queryString("delete_network"));
498   deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
499   safeExec(deleteNetworkQuery);
500   if(!watchQuery(deleteNetworkQuery)) {
501     if(withTransaction)
502       logDb().rollback();
503     return false;
504   }
505
506   logDb().commit();
507   return true;
508 }
509
510 QList<NetworkInfo> SqliteStorage::networks(UserId user) {
511   QList<NetworkInfo> nets;
512
513   QSqlQuery networksQuery(logDb());
514   networksQuery.prepare(queryString("select_networks_for_user"));
515   networksQuery.bindValue(":userid", user.toInt());
516
517   QSqlQuery serversQuery(logDb());
518   serversQuery.prepare(queryString("select_servers_for_network"));
519
520   safeExec(networksQuery);
521   if(!watchQuery(networksQuery))
522     return nets;
523
524   while(networksQuery.next()) {
525     NetworkInfo net;
526     net.networkId = networksQuery.value(0).toInt();
527     net.networkName = networksQuery.value(1).toString();
528     net.identity = networksQuery.value(2).toInt();
529     net.codecForServer = networksQuery.value(3).toString().toAscii();
530     net.codecForEncoding = networksQuery.value(4).toString().toAscii();
531     net.codecForDecoding = networksQuery.value(5).toString().toAscii();
532     net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
533     net.perform = networksQuery.value(7).toString().split("\n");
534     net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
535     net.autoIdentifyService = networksQuery.value(9).toString();
536     net.autoIdentifyPassword = networksQuery.value(10).toString();
537     net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
538     net.autoReconnectInterval = networksQuery.value(12).toUInt();
539     net.autoReconnectRetries = networksQuery.value(13).toInt();
540     net.unlimitedReconnectRetries = networksQuery.value(14).toInt() == 1 ? true : false;
541     net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false;
542
543     serversQuery.bindValue(":networkid", net.networkId.toInt());
544     safeExec(serversQuery);
545     if(!watchQuery(serversQuery))
546       return nets;
547
548     Network::ServerList servers;
549     while(serversQuery.next()) {
550       Network::Server server;
551       server.host = serversQuery.value(0).toString();
552       server.port = serversQuery.value(1).toUInt();
553       server.password = serversQuery.value(2).toString();
554       server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false;
555       server.sslVersion = serversQuery.value(4).toInt();
556       server.useProxy = serversQuery.value(5).toInt() == 1 ? true : false;
557       server.proxyType = serversQuery.value(6).toInt();
558       server.proxyHost = serversQuery.value(7).toString();
559       server.proxyPort = serversQuery.value(8).toUInt();
560       server.proxyUser = serversQuery.value(9).toString();
561       server.proxyPass = serversQuery.value(10).toString();
562       servers << server;
563     }
564     net.serverList = servers;
565     nets << net;
566   }
567   return nets;
568 }
569
570 bool SqliteStorage::isValidNetwork(UserId user, const NetworkId &networkId) {
571   QSqlQuery query(logDb());
572   query.prepare(queryString("select_networkExists"));
573   query.bindValue(":userid", user.toInt());
574   query.bindValue(":networkid", networkId.toInt());
575   safeExec(query);
576
577   watchQuery(query); // there should not occur any errors
578   if(!query.first())
579     return false;
580
581   Q_ASSERT(!query.next());
582   return true;
583 }
584
585 bool SqliteStorage::isValidBuffer(const UserId &user, const BufferId &bufferId) {
586   QSqlQuery query(logDb());
587   query.prepare(queryString("select_bufferExists"));
588   query.bindValue(":userid", user.toInt());
589   query.bindValue(":bufferid", bufferId.toInt());
590   safeExec(query);
591
592   watchQuery(query);
593   if(!query.first())
594     return false;
595
596   Q_ASSERT(!query.next());
597   return true;
598 }
599
600 NetworkId SqliteStorage::getNetworkId(UserId user, const QString &network) {
601   QSqlQuery query(logDb());
602   query.prepare("SELECT networkid FROM network "
603                 "WHERE userid = :userid AND networkname = :networkname");
604   query.bindValue(":userid", user.toInt());
605   query.bindValue(":networkname", network);
606   safeExec(query);
607
608   if(query.first())
609     return query.value(0).toInt();
610   else
611     return NetworkId();
612 }
613
614 QList<NetworkId> SqliteStorage::connectedNetworks(UserId user) {
615   QList<NetworkId> connectedNets;
616   QSqlQuery query(logDb());
617   query.prepare(queryString("select_connected_networks"));
618   query.bindValue(":userid", user.toInt());
619   safeExec(query);
620   watchQuery(query);
621
622   while(query.next()) {
623     connectedNets << query.value(0).toInt();
624   }
625
626   return connectedNets;
627 }
628
629 void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected) {
630   QSqlQuery query(logDb());
631   query.prepare(queryString("update_network_connected"));
632   query.bindValue(":userid", user.toInt());
633   query.bindValue(":networkid", networkId.toInt());
634   query.bindValue(":connected", isConnected ? 1 : 0);
635   safeExec(query);
636   watchQuery(query);
637 }
638
639 QHash<QString, QString> SqliteStorage::persistentChannels(UserId user, const NetworkId &networkId) {
640   QHash<QString, QString> persistentChans;
641   QSqlQuery query(logDb());
642   query.prepare(queryString("select_persistent_channels"));
643   query.bindValue(":userid", user.toInt());
644   query.bindValue(":networkid", networkId.toInt());
645   safeExec(query);
646   watchQuery(query);
647
648   while(query.next()) {
649     persistentChans[query.value(0).toString()] = query.value(1).toString();
650   }
651
652   return persistentChans;
653 }
654
655 void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined) {
656   QSqlQuery query(logDb());
657   query.prepare(queryString("update_buffer_persistent_channel"));
658   query.bindValue(":userid", user.toInt());
659   query.bindValue(":networkId", networkId.toInt());
660   query.bindValue(":buffercname", channel.toLower());
661   query.bindValue(":joined", isJoined ? 1 : 0);
662   safeExec(query);
663   watchQuery(query);
664 }
665
666 void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key) {
667   QSqlQuery query(logDb());
668   query.prepare(queryString("update_buffer_set_channel_key"));
669   query.bindValue(":userid", user.toInt());
670   query.bindValue(":networkId", networkId.toInt());
671   query.bindValue(":buffercname", channel.toLower());
672   query.bindValue(":key", key);
673   safeExec(query);
674   watchQuery(query);
675 }
676
677
678 void SqliteStorage::createBuffer(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) {
679   QSqlQuery query(logDb());
680   query.prepare(queryString("insert_buffer"));
681   query.bindValue(":userid", user.toInt());
682   query.bindValue(":networkid", networkId.toInt());
683   query.bindValue(":buffertype", (int)type);
684   query.bindValue(":buffername", buffer);
685   query.bindValue(":buffercname", buffer.toLower());
686   safeExec(query);
687
688   watchQuery(query);
689 }
690
691 BufferInfo SqliteStorage::getBufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) {
692   QSqlQuery query(logDb());
693   query.prepare(queryString("select_bufferByName"));
694   query.bindValue(":networkid", networkId.toInt());
695   query.bindValue(":userid", user.toInt());
696   query.bindValue(":buffercname", buffer.toLower());
697   safeExec(query);
698
699   if(!query.first()) {
700     createBuffer(user, networkId, type, buffer);
701     safeExec(query);
702     if(!query.first()) {
703       watchQuery(query);
704       qWarning() << "unable to create BufferInfo for:" << user << networkId << buffer;
705       return BufferInfo();
706     }
707   }
708
709   BufferInfo bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
710   if(query.next()) {
711     qCritical() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
712     qCritical() << "         Query:" << query.lastQuery();
713     qCritical() << "  bound Values:";
714     QList<QVariant> list = query.boundValues().values();
715     for (int i = 0; i < list.size(); ++i)
716       qCritical() << i << ":" << list.at(i).toString().toAscii().data();
717     Q_ASSERT(false);
718   }
719
720   return bufferInfo;
721 }
722
723 BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId) {
724   QSqlQuery query(logDb());
725   query.prepare(queryString("select_buffer_by_id"));
726   query.bindValue(":userid", user.toInt());
727   query.bindValue(":bufferid", bufferId.toInt());
728   safeExec(query);
729   if(!watchQuery(query))
730     return BufferInfo();
731
732   if(!query.first())
733     return BufferInfo();
734
735   BufferInfo bufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
736   Q_ASSERT(!query.next());
737
738   return bufferInfo;
739 }
740
741 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user) {
742   QList<BufferInfo> bufferlist;
743   QSqlQuery query(logDb());
744   query.prepare(queryString("select_buffers"));
745   query.bindValue(":userid", user.toInt());
746
747   safeExec(query);
748   watchQuery(query);
749   while(query.next()) {
750     bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
751   }
752   return bufferlist;
753 }
754
755 QList<BufferId> SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId) {
756   QList<BufferId> bufferList;
757   QSqlQuery query(logDb());
758   query.prepare(queryString("select_buffers_for_network"));
759   query.bindValue(":networkid", networkId.toInt());
760   query.bindValue(":userid", user.toInt());
761
762   safeExec(query);
763   watchQuery(query);
764   while(query.next()) {
765     bufferList << BufferId(query.value(0).toInt());
766   }
767   return bufferList;
768 }
769
770 bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId) {
771   if(!isValidBuffer(user, bufferId))
772     return false;
773
774   QSqlQuery delBacklogQuery(logDb());
775   delBacklogQuery.prepare(queryString("delete_backlog_for_buffer"));
776   delBacklogQuery.bindValue(":bufferid", bufferId.toInt());
777   safeExec(delBacklogQuery);
778   if(!watchQuery(delBacklogQuery))
779     return false;
780
781   QSqlQuery delBufferQuery(logDb());
782   delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
783   delBufferQuery.bindValue(":bufferid", bufferId.toInt());
784   safeExec(delBufferQuery);
785   if(!watchQuery(delBufferQuery))
786     return false;
787
788   return true;
789 }
790
791 bool SqliteStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName) {
792   if(!isValidBuffer(user, bufferId))
793     return false;
794
795   QSqlQuery query(logDb());
796   query.prepare(queryString("update_buffer_name"));
797   query.bindValue(":buffername", newName);
798   query.bindValue(":buffercname", newName.toLower());
799   query.bindValue(":bufferid", bufferId.toInt());
800   safeExec(query);
801   if(query.lastError().isValid()) {
802     // unexepcted error occured (19 == constraint violation)
803     if(query.lastError().number() != 19)
804       watchQuery(query);
805     return false;
806   }
807
808   return true;
809 }
810
811 void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId) {
812   QSqlQuery query(logDb());
813   query.prepare(queryString("update_buffer_lastseen"));
814
815   query.bindValue(":userid", user.toInt());
816   query.bindValue(":bufferid", bufferId.toInt());
817   query.bindValue(":lastseenmsgid", msgId.toInt());
818   safeExec(query);
819   watchQuery(query);
820 }
821
822 QHash<BufferId, MsgId> SqliteStorage::bufferLastSeenMsgIds(UserId user) {
823   QHash<BufferId, MsgId> lastSeenHash;
824   QSqlQuery query(logDb());
825   query.prepare(queryString("select_buffer_lastseen_messages"));
826   query.bindValue(":userid", user.toInt());
827   safeExec(query);
828   if(!watchQuery(query))
829     return lastSeenHash;
830
831   while(query.next()) {
832     lastSeenHash[query.value(0).toInt()] = query.value(1).toInt();
833   }
834   return lastSeenHash;
835 }
836
837 MsgId SqliteStorage::logMessage(Message msg) {
838   QSqlQuery logMessageQuery(logDb());
839   logMessageQuery.prepare(queryString("insert_message"));
840
841   logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
842   logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
843   logMessageQuery.bindValue(":type", msg.type());
844   logMessageQuery.bindValue(":flags", (int)msg.flags());
845   logMessageQuery.bindValue(":sender", msg.sender());
846   logMessageQuery.bindValue(":message", msg.contents());
847   safeExec(logMessageQuery);
848
849   if(logMessageQuery.lastError().isValid()) {
850     // constraint violation - must be NOT NULL constraint - probably the sender is missing...
851     if(logMessageQuery.lastError().number() == 19) {
852       QSqlQuery addSenderQuery(logDb());
853       addSenderQuery.prepare(queryString("insert_sender"));
854       addSenderQuery.bindValue(":sender", msg.sender());
855       safeExec(addSenderQuery);
856       safeExec(logMessageQuery);
857       if(!watchQuery(logMessageQuery))
858         return 0;
859     } else {
860       watchQuery(logMessageQuery);
861     }
862   }
863
864   MsgId msgId = logMessageQuery.lastInsertId().toInt();
865   Q_ASSERT(msgId.isValid());
866   return msgId;
867 }
868
869 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit) {
870   QList<Message> messagelist;
871
872   BufferInfo bufferInfo = getBufferInfo(user, bufferId);
873   if(!bufferInfo.isValid())
874     return messagelist;
875
876   QSqlQuery query(logDb());
877   if(last == -1) {
878     query.prepare(queryString("select_messagesNew"));
879   } else {
880     query.prepare(queryString("select_messages"));
881     query.bindValue(":lastmsg", last.toInt());
882   }
883
884   query.bindValue(":bufferid", bufferId.toInt());
885   query.bindValue(":firstmsg", first.toInt());
886   query.bindValue(":limit", limit);
887   safeExec(query);
888
889   watchQuery(query);
890
891   while(query.next()) {
892     Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
893                 bufferInfo,
894                 (Message::Type)query.value(2).toUInt(),
895                 query.value(5).toString(),
896                 query.value(4).toString(),
897                 (Message::Flags)query.value(3).toUInt());
898     msg.setMsgId(query.value(0).toInt());
899     messagelist << msg;
900   }
901   return messagelist;
902 }
903
904 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit) {
905   QList<Message> messagelist;
906
907   QHash<BufferId, BufferInfo> bufferInfoHash;
908   foreach(BufferInfo bufferInfo, requestBuffers(user)) {
909     bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
910   }
911
912   QSqlQuery query(logDb());
913   if(last == -1) {
914     query.prepare(queryString("select_messagesAllNew"));
915   } else {
916     query.prepare(queryString("select_messagesAll"));
917     query.bindValue(":lastmsg", last.toInt());
918   }
919   query.bindValue(":userid", user.toInt());
920   query.bindValue(":firstmsg", first.toInt());
921   query.bindValue(":limit", limit);
922   safeExec(query);
923
924   watchQuery(query);
925
926   while(query.next()) {
927     Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
928                 bufferInfoHash[query.value(1).toInt()],
929                 (Message::Type)query.value(3).toUInt(),
930                 query.value(6).toString(),
931                 query.value(5).toString(),
932                 (Message::Flags)query.value(4).toUInt());
933     msg.setMsgId(query.value(0).toInt());
934     messagelist << msg;
935   }
936
937   return messagelist;
938 }
939
940 QString SqliteStorage::backlogFile() {
941   return quasselDir().absolutePath() + "/quassel-storage.sqlite";
942 }
943
944 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount) {
945   query.exec();
946
947   if(!query.lastError().isValid())
948     return true;
949
950   switch(query.lastError().number()) {
951   case 5: // SQLITE_BUSY         5   /* The database file is locked */
952   case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
953     if(retryCount < _maxRetryCount)
954       return safeExec(query, retryCount + 1);
955   default:
956     return false;
957   }
958 }