1 /***************************************************************************
2 * Copyright (C) 2005-07 by The Quassel IRC Development Team *
3 * devel@quassel-irc.org *
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) any later version. *
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. *
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 ***************************************************************************/
25 #include "coreproxy.h"
26 #include "sqlitestorage.h"
31 Core *Core::instanceptr = 0;
33 Core * Core::instance() {
34 if(instanceptr) return instanceptr;
35 instanceptr = new Core();
40 void Core::destroy() {
50 if(!SqliteStorage::isAvailable()) {
51 qFatal("Sqlite is currently required! Please make sure your Qt library has sqlite support enabled.");
53 //SqliteStorage::init();
54 storage = new SqliteStorage();
55 connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(updateGlobalData(UserId, QString)));
56 connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
57 //startListening(); // FIXME
58 if(Global::runMode == Global::Monolithic) { // TODO Make GUI user configurable
60 guiUser = storage->validateUser("Default", "password");
61 } catch(Storage::AuthError) {
62 guiUser = storage->addUser("Default", "password");
65 Global::setGuiUser(guiUser);
66 createSession(guiUser);
69 // Read global settings from config file
71 s.beginGroup("Global");
72 foreach(QString unum, s.childGroups()) {
73 UserId uid = unum.toUInt();
75 foreach(QString key, s.childKeys()) {
76 Global::updateData(uid, key, s.value(key));
84 foreach(QTcpSocket *sock, validClients.keys()) {
91 CoreSession *Core::session(UserId uid) {
92 Core *core = instance();
93 if(core->sessions.contains(uid)) return core->sessions[uid];
97 CoreSession *Core::localSession() {
98 Core *core = instance();
99 if(core->guiUser && core->sessions.contains(core->guiUser)) return core->sessions[core->guiUser];
103 CoreSession *Core::createSession(UserId uid) {
104 Core *core = instance();
105 Q_ASSERT(!core->sessions.contains(uid));
106 CoreSession *sess = new CoreSession(uid, core->storage);
107 core->sessions[uid] = sess;
108 connect(sess, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)), core, SLOT(recvProxySignal(CoreSignal, QVariant, QVariant, QVariant)));
113 bool Core::startListening(uint port) {
114 if(!server.listen(QHostAddress::Any, port)) {
115 qWarning(QString(QString("Could not open GUI client port %1: %2").arg(port).arg(server.errorString())).toAscii());
118 qDebug() << "Listening for GUI clients on port" << server.serverPort();
122 void Core::stopListening() {
124 qDebug() << "No longer listening for GUI clients.";
127 void Core::incomingConnection() {
128 // TODO implement SSL
129 QTcpSocket *socket = server.nextPendingConnection();
130 connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
131 connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
132 blockSizes.insert(socket, (quint32)0);
133 qDebug() << "Client connected from " << socket->peerAddress().toString();
136 void Core::clientHasData() {
137 QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
138 Q_ASSERT(socket && blockSizes.contains(socket));
139 quint32 bsize = blockSizes.value(socket);
141 while(readDataFromDevice(socket, bsize, item)) {
142 if(validClients.contains(socket)) {
143 QList<QVariant> sigdata = item.toList();
144 if((ClientSignal)sigdata[0].toInt() == GS_UPDATE_GLOBAL_DATA) {
145 processClientUpdate(socket, sigdata[1].toString(), sigdata[2]);
147 sessions[validClients[socket]]->processSignal((ClientSignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]);
150 // we need to auth the client
152 processClientInit(socket, item);
153 } catch(Storage::AuthError) {
154 qWarning() << "Authentification error!"; // FIXME
157 } catch(Exception e) {
158 qWarning() << "Client init error:" << e.msg();
163 blockSizes[socket] = bsize = 0;
165 blockSizes[socket] = bsize;
168 void Core::clientDisconnected() {
169 QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
170 blockSizes.remove(socket);
171 validClients.remove(socket);
172 qDebug() << "Client disconnected.";
173 // TODO remove unneeded sessions - if necessary/possible...
176 void Core::processClientInit(QTcpSocket *socket, const QVariant &v) {
177 VarMap msg = v.toMap();
178 if(msg["GUIProtocol"].toUInt() != GUI_PROTOCOL) {
179 //qWarning() << "Client version mismatch.";
180 throw Exception("GUI client version mismatch");
183 UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString()); // throws exception if this failed
185 // Find or create session for validated user
187 if(sessions.contains(uid)) sess = sessions[uid];
189 sess = createSession(uid);
190 validClients[socket] = uid;
195 QStringList dataKeys = Global::keys(uid);
197 foreach(key, dataKeys) {
198 coreData[key] = Global::data(key);
200 reply["CoreData"] = coreData;
201 reply["SessionState"] = sess->sessionState();
202 QList<QVariant> sigdata;
203 sigdata.append(CS_CORE_STATE); sigdata.append(QVariant(reply)); sigdata.append(QVariant()); sigdata.append(QVariant());
204 writeDataToDevice(socket, QVariant(sigdata));
205 sess->sendServerStates();
208 void Core::processClientUpdate(QTcpSocket *socket, QString key, const QVariant &data) {
209 UserId uid = validClients[socket];
210 Global::updateData(uid, key, data);
211 QList<QVariant> sigdata;
212 sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
213 foreach(QTcpSocket *s, validClients.keys()) {
214 if(validClients[s] == uid && s != socket) writeDataToDevice(s, QVariant(sigdata));
218 void Core::updateGlobalData(UserId uid, QString key) {
219 QVariant data = Global::data(uid, key);
220 QList<QVariant> sigdata;
221 sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
222 foreach(QTcpSocket *socket, validClients.keys()) {
223 if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
227 void Core::recvProxySignal(CoreSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
228 CoreSession *sess = qobject_cast<CoreSession*>(sender());
230 UserId uid = sess->userId();
231 QList<QVariant> sigdata;
232 sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
233 //qDebug() << "Sending signal: " << sigdata;
234 foreach(QTcpSocket *socket, validClients.keys()) {
235 if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
240 // Read global settings from config file
242 s.beginGroup("Global");
244 foreach(key, s.childKeys()) {
245 global->updateData(key, s.value(key));
248 global->updateData("CoreReady", true);
249 // Now that we are in sync, we can connect signals to automatically store further updates.
250 // I don't think we care if global data changed locally or if it was updated by a client.
251 connect(global, SIGNAL(dataUpdatedRemotely(QString)), SLOT(globalDataUpdated(QString)));
252 connect(global, SIGNAL(dataPutLocally(QString)), SLOT(globalDataUpdated(QString)));
257 CoreSession::CoreSession(UserId uid, Storage *_storage) : user(uid), storage(_storage) {
258 coreProxy = new CoreProxy();
260 connect(coreProxy, SIGNAL(send(CoreSignal, QVariant, QVariant, QVariant)), this, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)));
261 connect(coreProxy, SIGNAL(requestServerStates()), this, SIGNAL(serverStateRequested()));
262 connect(coreProxy, SIGNAL(gsRequestConnect(QStringList)), this, SLOT(connectToIrc(QStringList)));
263 connect(coreProxy, SIGNAL(gsUserInput(BufferId, QString)), this, SLOT(msgFromGui(BufferId, QString)));
264 connect(coreProxy, SIGNAL(gsImportBacklog()), storage, SLOT(importOldBacklog()));
265 connect(coreProxy, SIGNAL(gsRequestBacklog(BufferId, QVariant, QVariant)), this, SLOT(sendBacklog(BufferId, QVariant, QVariant)));
266 connect(this, SIGNAL(displayMsg(Message)), coreProxy, SLOT(csDisplayMsg(Message)));
267 connect(this, SIGNAL(displayStatusMsg(QString, QString)), coreProxy, SLOT(csDisplayStatusMsg(QString, QString)));
268 connect(this, SIGNAL(backlogData(BufferId, QList<QVariant>, bool)), coreProxy, SLOT(csBacklogData(BufferId, QList<QVariant>, bool)));
269 connect(this, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
270 connect(storage, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
271 connect(Global::instance(), SIGNAL(dataUpdatedRemotely(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
272 connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
275 CoreSession::~CoreSession() {
279 UserId CoreSession::userId() {
283 void CoreSession::processSignal(ClientSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
284 coreProxy->recv(sig, arg1, arg2, arg3);
287 void CoreSession::globalDataUpdated(UserId uid, QString key) {
288 Q_ASSERT(uid == userId());
289 QVariant data = Global::data(userId(), key);
291 s.setValue(QString("Global/%1/").arg(userId())+key, data);
294 void CoreSession::connectToIrc(QStringList networks) {
295 foreach(QString net, networks) {
296 if(servers.contains(net)) {
299 Server *server = new Server(userId(), net);
300 connect(this, SIGNAL(serverStateRequested()), server, SLOT(sendState()));
301 connect(this, SIGNAL(connectToIrc(QString)), server, SLOT(connectToIrc(QString)));
302 connect(this, SIGNAL(disconnectFromIrc(QString)), server, SLOT(disconnectFromIrc(QString)));
303 connect(this, SIGNAL(msgFromGui(QString, QString, QString)), server, SLOT(userInput(QString, QString, QString)));
305 connect(server, SIGNAL(connected(QString)), this, SLOT(serverConnected(QString)));
306 connect(server, SIGNAL(disconnected(QString)), this, SLOT(serverDisconnected(QString)));
308 connect(server, SIGNAL(serverState(QString, VarMap)), coreProxy, SLOT(csServerState(QString, VarMap)));
309 //connect(server, SIGNAL(displayMsg(Message)), this, SLOT(recvMessageFromServer(Message)));
310 connect(server, SIGNAL(displayMsg(Message::Type, QString, QString, QString, quint8)), this, SLOT(recvMessageFromServer(Message::Type, QString, QString, QString, quint8)));
311 connect(server, SIGNAL(displayStatusMsg(QString)), this, SLOT(recvStatusMsgFromServer(QString)));
312 connect(server, SIGNAL(modeSet(QString, QString, QString)), coreProxy, SLOT(csModeSet(QString, QString, QString)));
313 connect(server, SIGNAL(topicSet(QString, QString, QString)), coreProxy, SLOT(csTopicSet(QString, QString, QString)));
314 connect(server, SIGNAL(nickAdded(QString, QString, VarMap)), coreProxy, SLOT(csNickAdded(QString, QString, VarMap)));
315 connect(server, SIGNAL(nickRenamed(QString, QString, QString)), coreProxy, SLOT(csNickRenamed(QString, QString, QString)));
316 connect(server, SIGNAL(nickRemoved(QString, QString)), coreProxy, SLOT(csNickRemoved(QString, QString)));
317 connect(server, SIGNAL(nickUpdated(QString, QString, VarMap)), coreProxy, SLOT(csNickUpdated(QString, QString, VarMap)));
318 connect(server, SIGNAL(ownNickSet(QString, QString)), coreProxy, SLOT(csOwnNickSet(QString, QString)));
319 connect(server, SIGNAL(queryRequested(QString, QString)), coreProxy, SLOT(csQueryRequested(QString, QString)));
320 // TODO add error handling
321 connect(server, SIGNAL(connected(QString)), coreProxy, SLOT(csServerConnected(QString)));
322 connect(server, SIGNAL(disconnected(QString)), coreProxy, SLOT(csServerDisconnected(QString)));
325 servers[net] = server;
327 emit connectToIrc(net);
331 void CoreSession::serverConnected(QString net) {
332 storage->getBufferId(userId(), net); // create status buffer
335 void CoreSession::serverDisconnected(QString net) {
338 coreProxy->csServerDisconnected(net);
341 void CoreSession::msgFromGui(BufferId bufid, QString msg) {
342 emit msgFromGui(bufid.network(), bufid.buffer(), msg);
345 // ALL messages coming pass through these functions before going to the GUI.
346 // So this is the perfect place for storing the backlog and log stuff.
348 void CoreSession::recvMessageFromServer(Message::Type type, QString target, QString text, QString sender, quint8 flags) {
349 Server *s = qobject_cast<Server*>(this->sender());
352 if((flags & Message::PrivMsg) && !(flags & Message::Self)) {
353 buf = storage->getBufferId(user, s->getNetwork(), nickFromMask(sender));
355 buf = storage->getBufferId(user, s->getNetwork(), target);
357 Message msg(buf, type, text, sender, flags);
358 msg.msgId = storage->logMessage(msg); //qDebug() << msg.msgId;
360 emit displayMsg(msg);
363 void CoreSession::recvStatusMsgFromServer(QString msg) {
364 Server *s = qobject_cast<Server*>(sender());
366 emit displayStatusMsg(s->getNetwork(), msg);
370 QList<BufferId> CoreSession::buffers() const {
371 return storage->requestBuffers(user);
375 QVariant CoreSession::sessionState() {
377 QList<QVariant> bufs;
378 foreach(BufferId id, storage->requestBuffers(user)) { bufs.append(QVariant::fromValue(id)); }
384 void CoreSession::sendServerStates() {
385 emit serverStateRequested();
388 void CoreSession::sendBacklog(BufferId id, QVariant v1, QVariant v2) {
390 QList<Message> msglist;
391 if(v1.type() == QVariant::DateTime) {
395 msglist = storage->requestMsgs(id, v1.toInt(), v2.toInt());
398 // Send messages out in smaller packages - we don't want to make the signal data too large!
399 for(int i = 0; i < msglist.count(); i++) {
400 log.append(QVariant::fromValue(msglist[i]));
401 if(log.count() >= 5) {
402 emit backlogData(id, log, i >= msglist.count() - 1);
406 if(log.count() > 0) emit backlogData(id, log, true);