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