/***************************************************************************
- * Copyright (C) 2005-07 by The Quassel Team *
+ * Copyright (C) 2005-07 by the Quassel IRC Team *
* devel@quassel-irc.org *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
- * (at your option) any later version. *
+ * (at your option) version 3. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
#include <QtSql>
SqliteStorage::SqliteStorage() {
- // TODO I don't think that this path is failsafe for windows users :)
- QString backlogFile = Global::quasselDir + "/quassel-storage.sqlite";
- logDb = QSqlDatabase::addDatabase("QSQLITE");
- logDb.setDatabaseName(backlogFile);
- bool ok = logDb.open();
-
- if(!ok) {
- qWarning(tr("Could not open backlog database: %1").arg(logDb.lastError().text()).toAscii());
- qWarning(tr("Disabling logging...").toAscii());
- Q_ASSERT(ok);
- return;
+ logMessageQuery = NULL;
+ addSenderQuery = NULL;
+ getLastMessageIdQuery = NULL;
+ requestMsgsQuery = NULL;
+ requestMsgsOffsetQuery = NULL;
+ requestMsgsSinceQuery = NULL;
+ requestMsgsSinceOffsetQuery = NULL;
+ requestMsgRangeQuery = NULL;
+ createNetworkQuery = NULL;
+ createBufferQuery = NULL;
+ getBufferInfoQuery = NULL;
+}
+
+SqliteStorage::~SqliteStorage() {
+ if (logMessageQuery) delete logMessageQuery;
+ if (addSenderQuery) delete addSenderQuery;
+ if (getLastMessageIdQuery) delete getLastMessageIdQuery;
+ if (requestMsgsQuery) delete requestMsgsQuery;
+ if (requestMsgsOffsetQuery) delete requestMsgsOffsetQuery;
+ if (requestMsgsSinceQuery) delete requestMsgsSinceQuery;
+ if (requestMsgsSinceOffsetQuery) delete requestMsgsSinceOffsetQuery;
+ if (requestMsgRangeQuery) delete requestMsgRangeQuery;
+ if (createNetworkQuery) delete createNetworkQuery;
+ if (createBufferQuery) delete createBufferQuery;
+ if (getBufferInfoQuery) delete getBufferInfoQuery;
+
+ logDb.close();
+}
+
+bool SqliteStorage::isAvailable() {
+ if(!QSqlDatabase::isDriverAvailable("QSQLITE")) return false;
+ return true;
+}
+
+QString SqliteStorage::displayName() {
+ return QString("SQlite");
+}
+
+bool SqliteStorage::setup(const QVariantMap &settings) {
+ bool ok;
+ // this extra scope is needed to be able to remove the database connection later
+ {
+ logDb = QSqlDatabase::addDatabase("QSQLITE", "quassel_setup");
+ logDb.setDatabaseName(SqliteStorage::backlogFile(true));
+ ok = logDb.open();
+
+ if (!ok) {
+ qWarning(tr("Could not open backlog database: %1").arg(logDb.lastError().text()).toAscii());
+ } else {
+ logDb.exec("CREATE TABLE quasseluser ("
+ "userid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "username TEXT UNIQUE NOT NULL,"
+ "password BLOB NOT NULL)");
+
+ logDb.exec("CREATE TABLE sender ("
+ "senderid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "sender TEXT UNIQUE NOT NULL)");
+
+ logDb.exec("CREATE TABLE network ("
+ "networkid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "userid INTEGER NOT NULL,"
+ "networkname TEXT NOT NULL,"
+ "UNIQUE (userid, networkname))");
+
+ logDb.exec("CREATE TABLE buffergroup ("
+ "groupid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "userid INTEGER NOT NULL,"
+ "displayname TEXT)");
+
+ logDb.exec("CREATE TABLE buffer ("
+ "bufferid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "userid INTEGER NOT NULL,"
+ "groupid INTEGER,"
+ "networkid INTEGER NOT NULL,"
+ "buffername TEXT NOT NULL)");
+
+ logDb.exec("CREATE UNIQUE INDEX buffer_idx "
+ "ON buffer(userid, networkid, buffername)");
+
+ logDb.exec("CREATE TABLE backlog ("
+ "messageid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "time INTEGER NOT NULL,"
+ "bufferid INTEGER NOT NULL,"
+ "type INTEGER NOT NULL,"
+ "flags INTEGER NOT NULL,"
+ "senderid INTEGER NOT NULL,"
+ "message TEXT)");
+
+ logDb.exec("CREATE TABLE coreinfo ("
+ "updateid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
+ "version INTEGER NOT NULL)");
+
+ logDb.exec("INSERT INTO coreinfo (version) VALUES (0)");
+
+ // something fucked up -> no logging possible
+ // FIXME logDb.lastError is reset whenever exec is called
+ if(logDb.lastError().isValid()) {
+ qWarning(tr("Could not create backlog table: %1").arg(logDb.lastError().text()).toAscii());
+ qWarning(tr("Disabling logging...").toAscii());
+ Q_ASSERT(false); // quassel does require logging
+ ok = false;
+ }
+
+ logDb.close();
+ }
+ }
+
+ QSqlDatabase::removeDatabase("quassel_setup");
+ return ok;
+}
+
+bool SqliteStorage::init(const QVariantMap &settings) {
+ bool ok;
+ // i need the extra scope to be able to remove the database connection
+ {
+ logDb = QSqlDatabase::database("quassel_connection", false);
+ if (!logDb.isValid()) {
+ logDb = QSqlDatabase::addDatabase("QSQLITE", "quassel_connection");
+ }
+ logDb.setDatabaseName(SqliteStorage::backlogFile());
+ ok = logDb.open();
+ if (!ok) {
+ qWarning(tr("Could not open backlog database: %1").arg(logDb.lastError().text()).toAscii());
+ }
+ }
+
+ if (!ok) {
+ //QSqlDatabase::removeDatabase("quassel_connection");
+ return false;
}
// check if the db schema is up to date
//checkVersion(query.value(0));
qDebug() << "Sqlite is ready. Quassel Schema Version:" << query.value(0).toUInt();
} else {
- initDb();
+ qWarning("Sqlite is not ready!");
+ return false;
}
// we will need those pretty often... so let's speed things up:
createBufferQuery = new QSqlQuery(logDb);
- createBufferQuery->prepare("INSERT INTO buffer (userid, networkid, buffername) VALUES (:userid, (SELECT networkid FROM network WHERE networkname = :networkname), :buffername)");
+ createBufferQuery->prepare("INSERT INTO buffer (userid, networkid, buffername) VALUES (:userid, (SELECT networkid FROM network WHERE networkname = :networkname AND userid = :userid2), :buffername)");
createNetworkQuery = new QSqlQuery(logDb);
createNetworkQuery->prepare("INSERT INTO network (userid, networkname) VALUES (:userid, :networkname)");
- getBufferIdQuery = new QSqlQuery(logDb);
- getBufferIdQuery->prepare("SELECT bufferid FROM buffer "
+ getBufferInfoQuery = new QSqlQuery(logDb);
+ getBufferInfoQuery->prepare("SELECT bufferid FROM buffer "
"JOIN network ON buffer.networkid = network.networkid "
- "WHERE network.networkname = :networkname AND buffer.userid = :userid AND buffer.buffername = :buffername ");
+ "WHERE network.networkname = :networkname AND network.userid = :userid AND buffer.userid = :userid2 AND lower(buffer.buffername) = lower(:buffername)");
logMessageQuery = new QSqlQuery(logDb);
logMessageQuery->prepare("INSERT INTO backlog (time, bufferid, type, flags, senderid, message) "
"WHERE (buffer.bufferid = :bufferid OR buffer.groupid = (SELECT groupid FROM buffer WHERE bufferid = :bufferid2)) AND "
"backlog.messageid >= :firstmsg AND backlog.messageid <= :lastmsg "
"ORDER BY messageid DESC ");
-
-}
-
-SqliteStorage::~SqliteStorage() {
- //logDb.close();
- delete logMessageQuery;
- delete addSenderQuery;
- delete getLastMessageIdQuery;
- delete requestMsgsQuery;
- delete requestMsgsOffsetQuery;
- delete requestMsgsSinceQuery;
- delete requestMsgsSinceOffsetQuery;
- delete requestMsgRangeQuery;
- delete createNetworkQuery;
- delete createBufferQuery;
- delete getBufferIdQuery;
- logDb.close();
-}
-
-
-void SqliteStorage::initDb() {
- logDb.exec("CREATE TABLE quasseluser ("
- "userid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "username TEXT UNIQUE NOT NULL,"
- "password BLOB NOT NULL)");
-
- logDb.exec("CREATE TABLE sender ("
- "senderid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "sender TEXT UNIQUE NOT NULL)");
-
- logDb.exec("CREATE TABLE network ("
- "networkid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "userid INTEGER NOT NULL,"
- "networkname TEXT NOT NULL,"
- "UNIQUE (userid, networkname))");
-
- logDb.exec("CREATE TABLE buffergroup ("
- "groupid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "userid INTEGER NOT NULL,"
- "displayname TEXT)");
- logDb.exec("CREATE TABLE buffer ("
- "bufferid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "userid INTEGER NOT NULL,"
- "groupid INTEGER,"
- "networkid INTEGER NOT NULL,"
- "buffername TEXT NOT NULL)");
-
- logDb.exec("CREATE UNIQUE INDEX buffer_idx "
- "ON buffer(userid, networkid, buffername)");
-
- logDb.exec("CREATE TABLE backlog ("
- "messageid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "time INTEGER NOT NULL,"
- "bufferid INTEGER NOT NULL,"
- "type INTEGER NOT NULL,"
- "flags INTEGER NOT NULL,"
- "senderid INTEGER NOT NULL,"
- "message TEXT NOT NULL)");
-
- logDb.exec("CREATE TABLE coreinfo ("
- "updateid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,"
- "version INTEGER NOT NULL)");
-
- logDb.exec("INSERT INTO coreinfo (version) VALUES (0)");
-
-
- // something fucked up -> no logging possible
- // FIXME logDb.lastError is reset whenever exec is called
- if(logDb.lastError().isValid()) {
- qWarning(tr("Could not create backlog table: %1").arg(logDb.lastError().text()).toAscii());
- qWarning(tr("Disabling logging...").toAscii());
- Q_ASSERT(false); // quassel does require logging
- }
-}
-
-bool SqliteStorage::isAvailable() {
- if(!QSqlDatabase::isDriverAvailable("QSQLITE")) return false;
return true;
}
-QString SqliteStorage::displayName() {
- // I think the class name is a good start here
- return QString("SqliteStorage");
-}
-
-UserId SqliteStorage::addUser(QString user, QString password) {
+UserId SqliteStorage::addUser(const QString &user, const QString &password) {
QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
cryptopass = cryptopass.toHex();
return uid;
}
-void SqliteStorage::updateUser(UserId user, QString password) {
+void SqliteStorage::updateUser(UserId user, const QString &password) {
QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
cryptopass = cryptopass.toHex();
query.exec();
}
-void SqliteStorage::renameUser(UserId user, QString newName) {
+void SqliteStorage::renameUser(UserId user, const QString &newName) {
QSqlQuery query(logDb);
query.prepare("UPDATE quasseluser SET username = :username WHERE userid = :userid");
query.bindValue(":userid", user);
emit userRenamed(user, newName);
}
-UserId SqliteStorage::validateUser(QString user, QString password) {
+UserId SqliteStorage::validateUser(const QString &user, const QString &password) {
QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
cryptopass = cryptopass.toHex();
emit userRemoved(user);
}
-void SqliteStorage::createBuffer(UserId user, QString network, QString buffer) {
+void SqliteStorage::createBuffer(UserId user, const QString &network, const QString &buffer) {
createBufferQuery->bindValue(":userid", user);
+ createBufferQuery->bindValue(":userid2", user); // Qt can't handle same placeholder twice (maybe sqlites fault)
createBufferQuery->bindValue(":networkname", network);
createBufferQuery->bindValue(":buffername", buffer);
createBufferQuery->exec();
}
}
-BufferId SqliteStorage::getBufferId(UserId user, QString network, QString buffer) {
- BufferId bufferid;
- getBufferIdQuery->bindValue(":networkname", network);
- getBufferIdQuery->bindValue(":userid", user);
- getBufferIdQuery->bindValue(":buffername", buffer);
- getBufferIdQuery->exec();
+uint SqliteStorage::getNetworkId(UserId user, const QString &network) {
+ QSqlQuery query(logDb);
+ query.prepare("SELECT networkid FROM network "
+ "WHERE userid = :userid AND networkname = :networkname");
+ query.bindValue(":userid", user);
+ query.bindValue(":networkname", network);
+ query.exec();
+
+ if(query.first())
+ return query.value(0).toUInt();
+ else {
+ createBuffer(user, network, "");
+ query.exec();
+ if(query.first())
+ return query.value(0).toUInt();
+ else {
+ qWarning() << "NETWORK NOT FOUND:" << network << "for User:" << user;
+ return 0;
+ }
+ }
+}
- if(!getBufferIdQuery->first()) {
+BufferInfo SqliteStorage::getBufferInfo(UserId user, const QString &network, const QString &buffer) {
+ BufferInfo bufferid;
+ // TODO: get rid of this hackaround
+ uint networkId = getNetworkId(user, network);
+ getBufferInfoQuery->bindValue(":networkname", network);
+ getBufferInfoQuery->bindValue(":userid", user);
+ getBufferInfoQuery->bindValue(":userid2", user); // Qt can't handle same placeholder twice... though I guess it's sqlites fault
+ getBufferInfoQuery->bindValue(":buffername", buffer);
+ getBufferInfoQuery->exec();
+
+ if(!getBufferInfoQuery->first()) {
createBuffer(user, network, buffer);
- getBufferIdQuery->exec();
- if(getBufferIdQuery->first()) {
- bufferid = BufferId(getBufferIdQuery->value(0).toUInt(), network, buffer);
- emit bufferIdUpdated(bufferid);
+ getBufferInfoQuery->exec();
+ if(getBufferInfoQuery->first()) {
+ bufferid = BufferInfo(getBufferInfoQuery->value(0).toUInt(), networkId, 0, network, buffer);
+ emit bufferInfoUpdated(bufferid);
}
} else {
- bufferid = BufferId(getBufferIdQuery->value(0).toUInt(), network, buffer);
+ bufferid = BufferInfo(getBufferInfoQuery->value(0).toUInt(), networkId, 0, network, buffer);
}
- Q_ASSERT(!getBufferIdQuery->next());
+ Q_ASSERT(!getBufferInfoQuery->next());
return bufferid;
}
-QList<BufferId> SqliteStorage::requestBuffers(UserId user, QDateTime since) {
- QList<BufferId> bufferlist;
+QList<BufferInfo> SqliteStorage::requestBuffers(UserId user, QDateTime since) {
+ QList<BufferInfo> bufferlist;
QSqlQuery query(logDb);
query.prepare("SELECT DISTINCT buffer.bufferid, networkname, buffername FROM buffer "
"JOIN network ON buffer.networkid = network.networkid "
query.exec();
while(query.next()) {
- bufferlist << BufferId(query.value(0).toUInt(), query.value(1).toString(), query.value(2).toString());
+ bufferlist << BufferInfo(query.value(0).toUInt(), getNetworkId(user, query.value(1).toString()), 0, query.value(1).toString(), query.value(2).toString());
}
return bufferlist;
}
MsgId SqliteStorage::logMessage(Message msg) {
- logMessageQuery->bindValue(":time", msg.timeStamp.toTime_t());
- logMessageQuery->bindValue(":bufferid", msg.buffer.uid());
- logMessageQuery->bindValue(":type", msg.type);
- logMessageQuery->bindValue(":flags", msg.flags);
- logMessageQuery->bindValue(":sender", msg.sender);
- logMessageQuery->bindValue(":message", msg.text);
+ logMessageQuery->bindValue(":time", msg.timestamp().toTime_t());
+ logMessageQuery->bindValue(":bufferid", msg.buffer().uid());
+ logMessageQuery->bindValue(":type", msg.type());
+ logMessageQuery->bindValue(":flags", msg.flags());
+ logMessageQuery->bindValue(":sender", msg.sender());
+ logMessageQuery->bindValue(":message", msg.text());
logMessageQuery->exec();
if(logMessageQuery->lastError().isValid()) {
// constraint violation - must be NOT NULL constraint - probably the sender is missing...
if(logMessageQuery->lastError().number() == 19) {
- addSenderQuery->bindValue(":sender", msg.sender);
+ addSenderQuery->bindValue(":sender", msg.sender());
addSenderQuery->exec();
+ watchQuery(addSenderQuery);
logMessageQuery->exec();
- Q_ASSERT(!logMessageQuery->lastError().isValid());
+ if(!watchQuery(logMessageQuery))
+ return 0;
} else {
- qDebug() << "unhandled DB Error in logMessage(): Number:" << logMessageQuery->lastError().number() << "ErrMsg:" << logMessageQuery->lastError().text();
+ watchQuery(logMessageQuery);
}
}
- getLastMessageIdQuery->bindValue(":time", msg.timeStamp.toTime_t());
- getLastMessageIdQuery->bindValue(":bufferid", msg.buffer.uid());
- getLastMessageIdQuery->bindValue(":type", msg.type);
- getLastMessageIdQuery->bindValue(":sender", msg.sender);
+ getLastMessageIdQuery->bindValue(":time", msg.timestamp().toTime_t());
+ getLastMessageIdQuery->bindValue(":bufferid", msg.buffer().uid());
+ getLastMessageIdQuery->bindValue(":type", msg.type());
+ getLastMessageIdQuery->bindValue(":sender", msg.sender());
getLastMessageIdQuery->exec();
if(getLastMessageIdQuery->first()) {
return getLastMessageIdQuery->value(0).toUInt();
} else { // somethin went wrong... :(
- qDebug() << getLastMessageIdQuery->lastQuery() << "time/bufferid/type/sender:" << msg.timeStamp.toTime_t() << msg.buffer.uid() << msg.type << msg.sender;
+ qDebug() << getLastMessageIdQuery->lastQuery() << "time/bufferid/type/sender:" << msg.timestamp().toTime_t() << msg.buffer().uid() << msg.type() << msg.sender();
Q_ASSERT(false);
return 0;
}
}
-QList<Message> SqliteStorage::requestMsgs(BufferId buffer, int lastmsgs, int offset) {
+QList<Message> SqliteStorage::requestMsgs(BufferInfo buffer, int lastmsgs, int offset) {
QList<Message> messagelist;
// we have to determine the real offset first
requestMsgsOffsetQuery->bindValue(":bufferid", buffer.uid());
requestMsgsQuery->value(5).toString(),
requestMsgsQuery->value(4).toString(),
requestMsgsQuery->value(3).toUInt());
- msg.msgId = requestMsgsQuery->value(0).toUInt();
+ msg.setMsgId(requestMsgsQuery->value(0).toUInt());
messagelist << msg;
}
return messagelist;
}
-QList<Message> SqliteStorage::requestMsgs(BufferId buffer, QDateTime since, int offset) {
+QList<Message> SqliteStorage::requestMsgs(BufferInfo buffer, QDateTime since, int offset) {
QList<Message> messagelist;
// we have to determine the real offset first
requestMsgsSinceOffsetQuery->bindValue(":bufferid", buffer.uid());
requestMsgsSinceQuery->value(5).toString(),
requestMsgsSinceQuery->value(4).toString(),
requestMsgsSinceQuery->value(3).toUInt());
- msg.msgId = requestMsgsSinceQuery->value(0).toUInt();
+ msg.setMsgId(requestMsgsSinceQuery->value(0).toUInt());
messagelist << msg;
}
}
-QList<Message> SqliteStorage::requestMsgRange(BufferId buffer, int first, int last) {
+QList<Message> SqliteStorage::requestMsgRange(BufferInfo buffer, int first, int last) {
QList<Message> messagelist;
requestMsgRangeQuery->bindValue(":bufferid", buffer.uid());
requestMsgRangeQuery->bindValue(":bufferid2", buffer.uid());
requestMsgRangeQuery->value(5).toString(),
requestMsgRangeQuery->value(4).toString(),
requestMsgRangeQuery->value(3).toUInt());
- msg.msgId = requestMsgRangeQuery->value(0).toUInt();
+ msg.setMsgId(requestMsgRangeQuery->value(0).toUInt());
messagelist << msg;
}
return messagelist;
}
-void SqliteStorage::importOldBacklog() {
- QSqlQuery query(logDb);
- int user;
- query.prepare("SELECT MIN(userid) FROM quasseluser");
- query.exec();
- if(!query.first()) {
- qDebug() << "create a user first!";
- } else {
- user = query.value(0).toUInt();
+QString SqliteStorage::backlogFile(bool createPath) {
+ // kinda ugly, but I currently see no other way to do that
+#ifdef Q_OS_WIN32
+ QString quasselDir = QDir::homePath() + qgetenv("APPDATA") + "\\quassel\\";
+#else
+ QString quasselDir = QDir::homePath() + "/.quassel/";
+#endif
+
+ if (createPath) {
+ QDir *qDir = new QDir(quasselDir);
+ if (!qDir->exists(quasselDir)) {
+ qDir->mkpath(quasselDir);
+ }
+ delete qDir;
}
- query.prepare("DELETE FROM backlog WHERE bufferid IN (SELECT DISTINCT bufferid FROM buffer WHERE userid = :userid");
- query.bindValue(":userid", user);
- query.exec();
- query.prepare("DELETE FROM buffer WHERE userid = :userid");
- query.bindValue(":userid", user);
- query.exec();
- query.prepare("DELETE FROM buffergroup WHERE userid = :userid");
- query.bindValue(":userid", user);
- query.exec();
- query.prepare("DELETE FROM network WHERE userid = :userid");
- query.bindValue(":userid", user);
- query.exec();
- logDb.commit();
- qDebug() << "All userdata has been deleted";
- qDebug() << "importing old backlog files...";
- initBackLogOld(user);
- logDb.commit();
- return;
-}
-
+ return quasselDir + "quassel-storage.sqlite";
+}
+bool SqliteStorage::watchQuery(QSqlQuery *query) {
+ if(query->lastError().isValid()) {
+ qWarning() << "unhandled Error in QSqlQuery!";
+ qWarning() << " last Query:" << query->lastQuery();
+ qWarning() << " executed Query:" << query->executedQuery();
+ qWarning() << " bound Values:" << query->boundValues();
+ qWarning() << " Error Number:" << query->lastError().number();
+ qWarning() << " Error Message:" << query->lastError().text();
+ qWarning() << " Driver Message:" << query->lastError().driverText();
+ qWarning() << " DB Message:" << query->lastError().databaseText();
+
+ return false;
+ }
+ return true;
+}