main-db.cpp 74.7 KB
Newer Older
Ronan's avatar
Ronan committed
1
/*
2
 * main-db.cpp
Ghislain MARY's avatar
Ghislain MARY committed
3
 * Copyright (C) 2010-2017 Belledonne Communications SARL
Ronan's avatar
Ronan committed
4
 *
Ghislain MARY's avatar
Ghislain MARY committed
5 6 7 8
 * 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.
Ronan's avatar
Ronan committed
9 10 11 12 13 14 15
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
Ghislain MARY's avatar
Ghislain MARY committed
16 17
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
Ronan's avatar
Ronan committed
18 19
 */

20
#include <algorithm>
21
#include <ctime>
22

Ronan's avatar
Ronan committed
23 24 25 26
#ifdef SOCI_ENABLED
	#include <soci/soci.h>
#endif // ifdef SOCI_ENABLED

27 28
#include "linphone/utils/utils.h"

Ronan's avatar
Ronan committed
29
#include "chat/chat-message/chat-message-p.h"
30
#include "chat/chat-room/chat-room-p.h"
31
#include "chat/chat-room/client-group-chat-room.h"
32
#include "chat/chat-room/server-group-chat-room.h"
33
#include "conference/participant-p.h"
34
#include "content/content-type.h"
Ronan's avatar
Ronan committed
35
#include "content/content.h"
Ronan's avatar
Ronan committed
36
#include "core/core-p.h"
37
#include "db/session/db-session-provider.h"
Ronan's avatar
Ronan committed
38
#include "event-log/event-log-p.h"
39
#include "event-log/events.h"
Ronan's avatar
Ronan committed
40
#include "logger/logger.h"
41
#include "main-db-key-p.h"
Ronan's avatar
Ronan committed
42
#include "main-db-p.h"
Ronan's avatar
Ronan committed
43 44 45

// =============================================================================

46 47 48 49 50
// See: http://soci.sourceforge.net/doc/3.2/exchange.html
// Part: Object lifetime and immutability

// -----------------------------------------------------------------------------

Ronan's avatar
Ronan committed
51 52
using namespace std;

Ronan's avatar
Ronan committed
53 54 55 56
LINPHONE_BEGIN_NAMESPACE

// -----------------------------------------------------------------------------

57
MainDb::MainDb (const shared_ptr<Core> &core) : AbstractDb(*new MainDbPrivate), CoreAccessor(core) {}
Ronan's avatar
Ronan committed
58

59 60
#ifdef SOCI_ENABLED

Ronan's avatar
Ronan committed
61
// -----------------------------------------------------------------------------
62
// Soci backend.
Ronan's avatar
Ronan committed
63 64
// -----------------------------------------------------------------------------

65 66 67 68 69
	template<typename T>
	struct EnumToSql {
		T first;
		const char *second;
	};
70

71 72 73 74 75 76 77
	template<typename T>
	static constexpr const char *mapEnumToSql (const EnumToSql<T> enumToSql[], size_t n, T key) {
		return n == 0 ? "" : (
			enumToSql[n - 1].first == key ? enumToSql[n - 1].second : mapEnumToSql(enumToSql, n - 1, key)
		);
	}

78
	static constexpr EnumToSql<MainDb::Filter> eventFilterToSql[] = {
79
		{ MainDb::ConferenceCallFilter, "3, 4" },
Ronan's avatar
Ronan committed
80
		{ MainDb::ConferenceChatMessageFilter, "5" },
81
		{ MainDb::ConferenceInfoFilter, "1, 2, 6, 7, 8, 9, 10, 11, 12" }
82
	};
83

84
	static constexpr const char *mapEventFilterToSql (MainDb::Filter filter) {
85 86 87
		return mapEnumToSql(
			eventFilterToSql, sizeof eventFilterToSql / sizeof eventFilterToSql[0], filter
		);
88 89
	}

Ronan's avatar
Ronan committed
90 91
// -----------------------------------------------------------------------------

Ronan's avatar
Ronan committed
92 93 94 95 96
	static string buildSqlEventFilter (
		const list<MainDb::Filter> &filters,
		MainDb::FilterMask mask,
		const string &condKeyWord = "WHERE"
	) {
97
		L_ASSERT(
98
			find_if(filters.cbegin(), filters.cend(), [](const MainDb::Filter &filter) {
99 100
				return filter == MainDb::NoFilter;
			}) == filters.cend()
101 102
		);

103
		if (mask == MainDb::NoFilter)
104 105 106 107 108 109 110 111 112 113
			return "";

		bool isStart = true;
		string sql;
		for (const auto &filter : filters) {
			if (!(mask & filter))
				continue;

			if (isStart) {
				isStart = false;
Ronan's avatar
Ronan committed
114
				sql += " " + condKeyWord + " type IN (";
115
			} else
Ronan's avatar
Ronan committed
116
				sql += ", ";
117 118 119
			sql += mapEventFilterToSql(filter);
		}

Ronan's avatar
Ronan committed
120 121 122
		if (!isStart)
			sql += ") ";

123 124 125
		return sql;
	}

126 127
// -----------------------------------------------------------------------------

128 129 130 131 132 133
	long long MainDbPrivate::resolveId (const soci::row &row, int col) const {
		L_Q();
		// See: http://soci.sourceforge.net/doc/master/backends/
		// `row id` is not supported by soci on Sqlite3. It's necessary to cast id to int...
		return q->getBackend() == AbstractDb::Sqlite3
			? static_cast<long long>(row.get<int>(0))
Benjamin REIS's avatar
Benjamin REIS committed
134
			: static_cast<long long>(row.get<unsigned long long>(0));
135 136
	}

137
	long long MainDbPrivate::insertSipAddress (const string &sipAddress) {
138 139 140
		L_Q();
		soci::session *session = dbSession.getBackendSession<soci::session>();

141 142
		long long id = selectSipAddressId(sipAddress);
		if (id >= 0)
143 144
			return id;

Ronan's avatar
Ronan committed
145
		lInfo() << "Insert new sip address in database: `" << sipAddress << "`.";
146 147
		*session << "INSERT INTO sip_address (value) VALUES (:sipAddress)", soci::use(sipAddress);
		return q->getLastInsertId();
148 149
	}

150
	void MainDbPrivate::insertContent (long long eventId, const Content &content) {
151
		L_Q();
152 153
		soci::session *session = dbSession.getBackendSession<soci::session>();

154 155
		const long long &contentTypeId = insertContentType(content.getContentType().asString());
		const string &body = content.getBodyAsString();
156
		*session << "INSERT INTO chat_message_content (event_id, content_type_id, body) VALUES"
Ronan's avatar
Ronan committed
157
			"  (:eventId, :contentTypeId, :body)", soci::use(eventId), soci::use(contentTypeId),
158
			soci::use(body);
159

160
		const long long &messageContentId = q->getLastInsertId();
Sylvain Berfini's avatar
Sylvain Berfini committed
161
		if (content.getContentType().isFile()) {
162
			const FileContent &fileContent = static_cast<const FileContent &>(content);
163 164 165
			const string &fileName = fileContent.getFileName();
			const size_t &fileSize = fileContent.getFileSize();
			const string &filePath = fileContent.getFilePath();
Sylvain Berfini's avatar
Sylvain Berfini committed
166 167
			*session << "INSERT INTO chat_message_file_content (chat_message_content_id, name, size, path) VALUES "
				" (:contentId, :name, :size, :path)",
168
				soci::use(messageContentId), soci::use(fileName), soci::use(fileSize), soci::use(filePath);
Sylvain Berfini's avatar
Sylvain Berfini committed
169 170
		}

171
		for (const auto &appData : content.getAppDataMap())
Ronan's avatar
Ronan committed
172 173
			*session << "INSERT INTO chat_message_content_app_data (chat_message_content_id, name, data) VALUES"
				"  (:messageContentId, :name, :data)",
174
				soci::use(messageContentId), soci::use(appData.first), soci::use(appData.second);
175 176
	}

177 178 179
	void MainDbPrivate::updateContent (long long eventId, long long messageContentId, const Content &content) {
		soci::session *session = dbSession.getBackendSession<soci::session>();

180 181
		const long long &contentTypeId = insertContentType(content.getContentType().asString());
		const string &body = content.getBodyAsString();
182
		*session << "UPDATE chat_message_content SET content_type_id=:contentTypeId, body=:body WHERE event_id=:eventId",
183
			soci::use(contentTypeId), soci::use(body), soci::use(eventId);
184 185 186 187 188 189

		for (const auto &appData : content.getAppDataMap())
			*session << "UPDATE chat_message_content_app_data SET name=:name, data=:data WHERE chat_message_content_id=:messageContentId",
				soci::use(appData.first), soci::use(appData.second), soci::use(messageContentId);
	}

190 191 192 193
	void MainDbPrivate::removeContentsForChatMessageEvent (long long eventId) {
		soci::session *session = dbSession.getBackendSession<soci::session>();

		*session << "DELETE FROM chat_message_content WHERE event_id=:eventId", soci::use(eventId);
Benjamin REIS's avatar
Benjamin REIS committed
194

195 196 197 198 199 200 201
		//TODO: remove file content if exists
		//*session << "DELETE FROM chat_message_file_content WHERE chat_message_content_id=:messageContentId", soci::use(messageContentId);

		//TODO: remove contents' app_data
		//*session << "DELETE FROM chat_message_content_app_data WHERE chat_message_content_id=:messageContentId", soci::use(messageContentId);
	}

202
	long long MainDbPrivate::insertContentType (const string &contentType) {
203 204 205
		L_Q();
		soci::session *session = dbSession.getBackendSession<soci::session>();

206
		long long id;
207 208
		*session << "SELECT id FROM content_type WHERE value = :contentType", soci::use(contentType), soci::into(id);
		if (session->got_data())
209 210
			return id;

Ronan's avatar
Ronan committed
211
		lInfo() << "Insert new content type in database: `" << contentType << "`.";
212 213
		*session << "INSERT INTO content_type (value) VALUES (:contentType)", soci::use(contentType);
		return q->getLastInsertId();
214 215
	}

216
	long long MainDbPrivate::insertBasicChatRoom (
217 218
		long long peerSipAddressId,
		long long localSipAddressId,
219
		const tm &creationTime
220 221 222
	) {
		L_Q();

223 224
		soci::session *session = dbSession.getBackendSession<soci::session>();

225
		long long id = selectChatRoomId(peerSipAddressId, localSipAddressId);
226 227
		if (id >= 0)
			return id;
228

229
		static const int capabilities = static_cast<int>(ChatRoom::Capabilities::Basic);
230 231 232 233 234 235 236 237 238
		lInfo() << "Insert new chat room in database: (peer=" << peerSipAddressId <<
			", local=" << localSipAddressId << ", capabilities=" << capabilities << ").";
		*session << "INSERT INTO chat_room ("
			"  peer_sip_address_id, local_sip_address_id, creation_time, last_update_time, capabilities"
			") VALUES (:peerSipAddressId, :localSipAddressId, :creationTime, :lastUpdateTime, :capabilities)",
			soci::use(peerSipAddressId), soci::use(localSipAddressId), soci::use(creationTime), soci::use(creationTime),
			soci::use(capabilities);

		return q->getLastInsertId();
239 240
	}

241
	long long MainDbPrivate::insertChatRoom (const shared_ptr<ChatRoom> &chatRoom) {
242 243 244 245 246
		L_Q();

		soci::session *session = dbSession.getBackendSession<soci::session>();

		const ChatRoomId &chatRoomId = chatRoom->getChatRoomId();
247 248
		const long long &peerSipAddressId = insertSipAddress(chatRoomId.getPeerAddress().asString());
		const long long &localSipAddressId = insertSipAddress(chatRoomId.getLocalAddress().asString());
249 250 251 252 253 254 255 256 257 258 259

		long long id = selectChatRoomId(peerSipAddressId, localSipAddressId);
		if (id >= 0) {
			lWarning() << "Unable to insert chat room (it already exists): (peer=" << peerSipAddressId <<
				", local=" << localSipAddressId << ").";
			return id;
		}

		lInfo() << "Insert new chat room in database: (peer=" << peerSipAddressId <<
			", local=" << localSipAddressId << ").";

260 261 262
		const tm &creationTime = Utils::getTimeTAsTm(chatRoom->getCreationTime());
		const int &capabilities = static_cast<int>(chatRoom->getCapabilities());
		const string &subject = chatRoom->getSubject();
263
		const int &flags = chatRoom->hasBeenLeft();
264
		*session << "INSERT INTO chat_room ("
265
			"  peer_sip_address_id, local_sip_address_id, creation_time, last_update_time, capabilities, subject, flags"
266
			") VALUES (:peerSipAddressId, :localSipAddressId, :creationTime, :lastUpdateTime, :capabilities, :subject, :flags)",
267
			soci::use(peerSipAddressId), soci::use(localSipAddressId), soci::use(creationTime), soci::use(creationTime),
268
			soci::use(capabilities), soci::use(subject), soci::use(flags);
269 270

		id = q->getLastInsertId();
271 272 273 274

		if (!chatRoom->canHandleParticipants())
			return id;

275 276 277 278 279 280
		// Do not add 'me' when creating a server-group-chat-room.
		if (chatRoomId.getLocalAddress() != chatRoomId.getPeerAddress()) {
			shared_ptr<Participant> me = chatRoom->getMe();
			insertChatRoomParticipant(id, insertSipAddress(me->getAddress().asString()), me->isAdmin());
		}

281
		for (const auto &participant : chatRoom->getParticipants())
282
			insertChatRoomParticipant(id, insertSipAddress(participant->getAddress().asString()), participant->isAdmin());
283 284

		return id;
285 286
	}

287
	void MainDbPrivate::insertChatRoomParticipant (long long chatRoomId, long long sipAddressId, bool isAdmin) {
288 289
		// See: https://stackoverflow.com/a/15299655 (cast to reference)

290
		soci::session *session = dbSession.getBackendSession<soci::session>();
291 292
		soci::statement statement = (
			session->prepare << "UPDATE chat_room_participant SET is_admin = :isAdmin"
293
				"  WHERE chat_room_id = :chatRoomId AND participant_sip_address_id = :sipAddressId",
294
				soci::use(static_cast<const int &>(isAdmin)), soci::use(chatRoomId), soci::use(sipAddressId)
295 296
		);
		statement.execute(true);
Ronan's avatar
Ronan committed
297 298
		if (statement.get_affected_rows() == 0) {
			lInfo() << "Insert new chat room participant in database: `" << sipAddressId << "` (isAdmin=" << isAdmin << ").";
299
			*session << "INSERT INTO chat_room_participant (chat_room_id, participant_sip_address_id, is_admin)"
300
				"  VALUES (:chatRoomId, :sipAddressId, :isAdmin)",
301
				soci::use(chatRoomId), soci::use(sipAddressId), soci::use(static_cast<const int &>(isAdmin));
Ronan's avatar
Ronan committed
302
		}
303 304
	}

305
	void MainDbPrivate::insertChatMessageParticipant (long long eventId, long long sipAddressId, int state) {
306 307
		soci::session *session = dbSession.getBackendSession<soci::session>();
		soci::statement statement = (
308
			session->prepare << "UPDATE chat_message_participant SET state = :state"
309
				"  WHERE event_id = :eventId AND participant_sip_address_id = :sipAddressId",
Ronan's avatar
Ronan committed
310
				soci::use(state), soci::use(eventId), soci::use(sipAddressId)
311 312
		);
		statement.execute(true);
313
		if (statement.get_affected_rows() == 0 && state != static_cast<int>(ChatMessage::State::Displayed))
314
			*session << "INSERT INTO chat_message_participant (event_id, participant_sip_address_id, state)"
Ronan's avatar
Ronan committed
315 316
				"  VALUES (:eventId, :sipAddressId, :state)",
				soci::use(eventId), soci::use(sipAddressId), soci::use(state);
317 318
	}

319
	long long MainDbPrivate::selectSipAddressId (const string &sipAddress) const {
320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348
		soci::session *session = dbSession.getBackendSession<soci::session>();

		long long id;
		*session << "SELECT id FROM sip_address WHERE value = :sipAddress", soci::use(sipAddress), soci::into(id);
		return session->got_data() ? id : -1;
	}

	long long MainDbPrivate::selectChatRoomId (long long peerSipAddressId, long long localSipAddressId) const {
		soci::session *session = dbSession.getBackendSession<soci::session>();

		long long id;
		*session << "SELECT id FROM chat_room"
			"  WHERE peer_sip_address_id = :peerSipAddressId AND local_sip_address_id = :localSipAddressId",
			soci::use(peerSipAddressId), soci::use(localSipAddressId), soci::into(id);
		return session->got_data() ? id : -1;
	}

	long long MainDbPrivate::selectChatRoomId (const ChatRoomId &chatRoomId) const {
		long long peerSipAddressId = selectSipAddressId(chatRoomId.getPeerAddress().asString());
		if (peerSipAddressId < 0)
			return -1;

		long long localSipAddressId = selectSipAddressId(chatRoomId.getLocalAddress().asString());
		if (localSipAddressId < 0)
			return -1;

		return selectChatRoomId(peerSipAddressId, localSipAddressId);
	}

349 350 351 352 353 354 355
	void MainDbPrivate::deleteChatRoomParticipant (long long chatRoomId, long long participantSipAddressId) {
		soci::session *session = dbSession.getBackendSession<soci::session>();
		*session << "DELETE FROM chat_room_participant"
			"  WHERE chat_room_id = :chatRoomId AND participant_sip_address_id = :participantSipAddressId",
			soci::use(chatRoomId), soci::use(participantSipAddressId);
	}

Ronan's avatar
Ronan committed
356 357
// -----------------------------------------------------------------------------

358
	shared_ptr<EventLog> MainDbPrivate::selectGenericConferenceEvent (
359
		long long eventId,
360
		EventLog::Type type,
361
		time_t creationTime,
362
		const ChatRoomId &chatRoomId
363
	) const {
Ronan's avatar
Ronan committed
364 365
		shared_ptr<EventLog> eventLog;

Ronan's avatar
Ronan committed
366 367 368 369 370
		switch (type) {
			case EventLog::Type::None:
				return nullptr;

			case EventLog::Type::ConferenceCreated:
371
			case EventLog::Type::ConferenceTerminated:
Ronan's avatar
Ronan committed
372 373
				eventLog = selectConferenceEvent(eventId, type, creationTime, chatRoomId);
				break;
Ronan's avatar
Ronan committed
374

375 376
			case EventLog::Type::ConferenceCallStart:
			case EventLog::Type::ConferenceCallEnd:
Ronan's avatar
Ronan committed
377 378
				eventLog = selectConferenceCallEvent(eventId, type, creationTime, chatRoomId);
				break;
Ronan's avatar
Ronan committed
379 380

			case EventLog::Type::ConferenceChatMessage:
Ronan's avatar
Ronan committed
381 382
				eventLog = selectConferenceChatMessageEvent(eventId, type, creationTime, chatRoomId);
				break;
Ronan's avatar
Ronan committed
383 384 385 386 387

			case EventLog::Type::ConferenceParticipantAdded:
			case EventLog::Type::ConferenceParticipantRemoved:
			case EventLog::Type::ConferenceParticipantSetAdmin:
			case EventLog::Type::ConferenceParticipantUnsetAdmin:
Ronan's avatar
Ronan committed
388 389
				eventLog = selectConferenceParticipantEvent(eventId, type, creationTime, chatRoomId);
				break;
Ronan's avatar
Ronan committed
390 391 392

			case EventLog::Type::ConferenceParticipantDeviceAdded:
			case EventLog::Type::ConferenceParticipantDeviceRemoved:
Ronan's avatar
Ronan committed
393 394
				eventLog = selectConferenceParticipantDeviceEvent(eventId, type, creationTime, chatRoomId);
				break;
Ronan's avatar
Ronan committed
395 396

			case EventLog::Type::ConferenceSubjectChanged:
Ronan's avatar
Ronan committed
397 398 399 400
				eventLog = selectConferenceSubjectEvent(eventId, type, creationTime, chatRoomId);
				break;
		}

401 402
		if (eventLog)
			cache(eventLog, eventId);
Ronan's avatar
Ronan committed
403

404
		return eventLog;
Ronan's avatar
Ronan committed
405 406
	}

407
	shared_ptr<EventLog> MainDbPrivate::selectConferenceEvent (
408
		long long,
409
		EventLog::Type type,
410
		time_t creationTime,
411
		const ChatRoomId &chatRoomId
412
	) const {
413 414
		return make_shared<ConferenceEvent>(
			type,
415
			creationTime,
416
			chatRoomId
417
		);
Ronan's avatar
Ronan committed
418 419
	}

420
	shared_ptr<EventLog> MainDbPrivate::selectConferenceCallEvent (
421
		long long eventId,
422
		EventLog::Type type,
423
		time_t creationTime,
424
		const ChatRoomId &chatRoomId
425
	) const {
Ronan's avatar
Ronan committed
426 427 428 429
		// TODO.
		return nullptr;
	}

430
	shared_ptr<EventLog> MainDbPrivate::selectConferenceChatMessageEvent (
431
		long long eventId,
432
		EventLog::Type type,
433
		time_t creationTime,
434
		const ChatRoomId &chatRoomId
435
	) const {
Ronan's avatar
Ronan committed
436 437 438
		L_Q();

		shared_ptr<Core> core = q->getCore();
439
		shared_ptr<ChatRoom> chatRoom = core->findChatRoom(chatRoomId);
Ronan's avatar
Ronan committed
440 441 442
		if (!chatRoom)
			return nullptr;

443
		// 1 - Fetch chat message.
444 445 446
		shared_ptr<ChatMessage> chatMessage = getChatMessageFromCache(eventId);
		if (chatMessage)
			goto end;
447
		{
448 449
			string fromSipAddress;
			string toSipAddress;
450 451 452

			tm messageTime;

453
			string imdnMessageId;
454

455 456 457 458 459
			int state;
			int direction;
			int isSecured;

			soci::session *session = dbSession.getBackendSession<soci::session>();
460
			*session << "SELECT from_sip_address.value, to_sip_address.value, time, imdn_message_id, state, direction, is_secured"
461 462
				"  FROM event, conference_chat_message_event, sip_address AS from_sip_address,"
				"  sip_address AS to_sip_address"
463 464
				"  WHERE event_id = :eventId"
				"  AND event_id = event.id"
465 466
				"  AND from_sip_address_id = from_sip_address.id"
				"  AND to_sip_address_id = to_sip_address.id", soci::into(fromSipAddress), soci::into(toSipAddress),
467 468
				soci::into(messageTime), soci::into(imdnMessageId), soci::into(state), soci::into(direction),
				soci::into(isSecured), soci::use(eventId);
469

470 471 472 473
			chatMessage = shared_ptr<ChatMessage>(new ChatMessage(
				chatRoom,
				static_cast<ChatMessage::Direction>(direction)
			));
474 475
			chatMessage->setIsSecured(static_cast<bool>(isSecured));

476 477
			ChatMessagePrivate *dChatMessage = chatMessage->getPrivate();
			dChatMessage->setState(static_cast<ChatMessage::State>(state), true);
478

479 480 481 482
			dChatMessage->forceFromAddress(IdentityAddress(fromSipAddress));
			dChatMessage->forceToAddress(IdentityAddress(toSipAddress));

			dChatMessage->setTime(Utils::getTmAsTimeT(messageTime));
483
		}
Ronan's avatar
Ronan committed
484

485 486 487
		// 2 - Fetch contents.
		{
			soci::session *session = dbSession.getBackendSession<soci::session>();
488 489
			static const string query = "SELECT chat_message_content.id, content_type.id, content_type.value, body"
				"  FROM chat_message_content, content_type"
490 491 492
				"  WHERE event_id = :eventId AND content_type_id = content_type.id";
			soci::rowset<soci::row> rows = (session->prepare << query, soci::use(eventId));
			for (const auto &row : rows) {
493
				ContentType contentType(row.get<string>(2));
494
				const long long &contentId = resolveId(row, 0);
495 496 497 498 499
				Content *content;

				if (contentType == ContentType::FileTransfer)
					content = new FileTransferContent();
				else if (contentType.isFile()) {
500
					// 2.1 - Fetch contents' file informations.
501 502 503 504 505
					string name;
					int size;
					string path;

					*session << "SELECT name, size, path FROM chat_message_file_content"
Ronan's avatar
Ronan committed
506
						"  WHERE chat_message_content_id = :contentId",
Ronan's avatar
Ronan committed
507
						soci::into(name), soci::into(size), soci::into(path), soci::use(contentId);
508 509 510 511 512 513 514 515

					FileContent *fileContent = new FileContent();
					fileContent->setFileName(name);
					fileContent->setFileSize(static_cast<size_t>(size));
					fileContent->setFilePath(path);

					content = fileContent;
				} else
Ronan's avatar
Ronan committed
516
					content = new Content();
517

518
				// 2.2 - Fetch contents' app data.
Ronan's avatar
Ronan committed
519 520 521 522 523
				static const string query = "SELECT name, data FROM chat_message_content_app_data"
					"  WHERE chat_message_content_id = :contentId";
				soci::rowset<soci::row> rows = (session->prepare << query, soci::use(contentId));
				for (const auto &row : rows)
					content->setAppData(row.get<string>(0), row.get<string>(1));
524

525
				content->setContentType(contentType);
Sylvain Berfini's avatar
Sylvain Berfini committed
526
				content->setBody(row.get<string>(3));
527
				chatMessage->addContent(*content);
528
			}
Ronan's avatar
Ronan committed
529
		}
530

531 532
		cache(chatMessage, eventId);

533
	end:
534
		return make_shared<ConferenceChatMessageEvent>(
535
			creationTime,
536 537
			chatMessage
		);
Ronan's avatar
Ronan committed
538 539
	}

540
	shared_ptr<EventLog> MainDbPrivate::selectConferenceParticipantEvent (
541
		long long eventId,
542
		EventLog::Type type,
543
		time_t creationTime,
544
		const ChatRoomId &chatRoomId
545
	) const {
546 547 548 549
		unsigned int notifyId;
		string participantAddress;

		soci::session *session = dbSession.getBackendSession<soci::session>();
550 551
		*session << "SELECT notify_id, participant_address.value"
			"  FROM conference_notified_event, conference_participant_event, sip_address as participant_address"
552
			"  WHERE conference_participant_event.event_id = :eventId"
Ronan's avatar
Ronan committed
553
			"    AND conference_notified_event.event_id = conference_participant_event.event_id"
554
			"    AND participant_address.id = participant_sip_address_id",
555 556 557 558
			soci::into(notifyId), soci::into(participantAddress), soci::use(eventId);

		return make_shared<ConferenceParticipantEvent>(
			type,
559
			creationTime,
560
			chatRoomId,
561
			notifyId,
562
			IdentityAddress(participantAddress)
563
		);
Ronan's avatar
Ronan committed
564 565
	}

566
	shared_ptr<EventLog> MainDbPrivate::selectConferenceParticipantDeviceEvent (
567
		long long eventId,
568
		EventLog::Type type,
569
		time_t creationTime,
570
		const ChatRoomId &chatRoomId
571
	) const {
572 573
		unsigned int notifyId;
		string participantAddress;
574
		string deviceAddress;
575 576

		soci::session *session = dbSession.getBackendSession<soci::session>();
577
		*session << "SELECT notify_id, participant_address.value, device_address.value"
578
			"  FROM conference_notified_event, conference_participant_event, conference_participant_device_event,"
579
			"    sip_address AS participant_address, sip_address AS device_address"
580 581
			"  WHERE conference_participant_device_event.event_id = :eventId"
			"    AND conference_participant_event.event_id = conference_participant_device_event.event_id"
582
			"    AND conference_notified_event.event_id = conference_participant_event.event_id"
583 584
			"    AND participant_address.id = participant_sip_address_id"
			"    AND device_address.id = device_sip_address_id",
585
			soci::into(notifyId), soci::into(participantAddress), soci::into(deviceAddress), soci::use(eventId);
586 587 588

		return make_shared<ConferenceParticipantDeviceEvent>(
			type,
589
			creationTime,
590
			chatRoomId,
591
			notifyId,
592 593
			IdentityAddress(participantAddress),
			IdentityAddress(deviceAddress)
594
		);
Ronan's avatar
Ronan committed
595 596
	}

597
	shared_ptr<EventLog> MainDbPrivate::selectConferenceSubjectEvent (
598
		long long eventId,
599
		EventLog::Type type,
600
		time_t creationTime,
601
		const ChatRoomId &chatRoomId
602 603 604 605
	) const {
		unsigned int notifyId;
		string subject;

606 607 608 609 610 611
		soci::session *session = dbSession.getBackendSession<soci::session>();
		*session << "SELECT notify_id, subject"
			"  FROM conference_notified_event, conference_subject_event"
			"  WHERE conference_subject_event.event_id = :eventId"
			"    AND conference_notified_event.event_id = conference_subject_event.event_id",
			soci::into(notifyId), soci::into(subject), soci::use(eventId);
612 613

		return make_shared<ConferenceSubjectEvent>(
614
			creationTime,
615
			chatRoomId,
616 617 618
			notifyId,
			subject
		);
Ronan's avatar
Ronan committed
619 620
	}

621 622
// -----------------------------------------------------------------------------

Ronan's avatar
Ronan committed
623
	long long MainDbPrivate::insertEvent (const shared_ptr<EventLog> &eventLog) {
624 625 626
		L_Q();
		soci::session *session = dbSession.getBackendSession<soci::session>();

627 628
		const int &type = static_cast<int>(eventLog->getType());
		const tm &creationTime = Utils::getTimeTAsTm(eventLog->getCreationTime());
629
		*session << "INSERT INTO event (type, creation_time) VALUES (:type, :creationTime)",
630 631 632
			soci::use(type),
			soci::use(creationTime);

633
		return q->getLastInsertId();
634 635
	}

Ronan's avatar
Ronan committed
636
	long long MainDbPrivate::insertConferenceEvent (const shared_ptr<EventLog> &eventLog, long long *chatRoomId) {
637 638 639
		shared_ptr<ConferenceEvent> conferenceEvent = static_pointer_cast<ConferenceEvent>(eventLog);

		long long eventId = -1;
640
		const long long &curChatRoomId = selectChatRoomId(conferenceEvent->getChatRoomId());
641 642 643 644 645 646 647 648 649
		if (curChatRoomId < 0) {
			// A conference event can be inserted in database only if chat room exists.
			// Otherwise it's an error.
			const ChatRoomId &chatRoomId = conferenceEvent->getChatRoomId();
			lError() << "Unable to find chat room storage id of (peer=" +
				chatRoomId.getPeerAddress().asString() +
				", local=" + chatRoomId.getLocalAddress().asString() + "`).";
		} else {
			eventId = insertEvent(eventLog);
650

651 652 653 654
			soci::session *session = dbSession.getBackendSession<soci::session>();
			*session << "INSERT INTO conference_event (event_id, chat_room_id)"
				"  VALUES (:eventId, :chatRoomId)", soci::use(eventId), soci::use(curChatRoomId);

655
			const tm &lastUpdateTime = Utils::getTimeTAsTm(eventLog->getCreationTime());
656
			*session << "UPDATE chat_room SET last_update_time = :lastUpdateTime"
657
				"  WHERE id = :chatRoomId", soci::use(lastUpdateTime),
658
				soci::use(curChatRoomId);
659 660

			if (eventLog->getType() == EventLog::Type::ConferenceTerminated)
661
				*session << "UPDATE chat_room SET flags = 1 WHERE id = :chatRoomId", soci::use(curChatRoomId);
662
		}
663 664 665 666 667 668 669

		if (chatRoomId)
			*chatRoomId = curChatRoomId;

		return eventId;
	}

Ronan's avatar
Ronan committed
670
	long long MainDbPrivate::insertConferenceCallEvent (const shared_ptr<EventLog> &eventLog) {
671 672 673 674
		// TODO.
		return 0;
	}

Ronan's avatar
Ronan committed
675 676
	long long MainDbPrivate::insertConferenceChatMessageEvent (const shared_ptr<EventLog> &eventLog) {
		shared_ptr<ChatMessage> chatMessage = static_pointer_cast<ConferenceChatMessageEvent>(eventLog)->getChatMessage();
677 678 679 680 681 682
		shared_ptr<ChatRoom> chatRoom = chatMessage->getChatRoom();
		if (!chatRoom) {
			lError() << "Unable to get a valid chat room. It was removed from database.";
			return -1;
		}

683
		const long long &eventId = insertConferenceEvent(eventLog);
684 685
		if (eventId < 0)
			return -1;
686

687
		soci::session *session = dbSession.getBackendSession<soci::session>();
688 689 690 691 692 693 694 695 696

		const long long &fromSipAddressId = insertSipAddress(chatMessage->getFromAddress().asString());
		const long long &toSipAddressId = insertSipAddress(chatMessage->getToAddress().asString());
		const tm &messageTime = Utils::getTimeTAsTm(chatMessage->getTime());
		const int &state = static_cast<int>(chatMessage->getState());
		const int &direction = static_cast<int>(chatMessage->getDirection());
		const string &imdnMessageId = chatMessage->getImdnMessageId();
		const int &isSecured = chatMessage->isSecured() ? 1 : 0;

697
		*session << "INSERT INTO conference_chat_message_event ("
698
			"  event_id, from_sip_address_id, to_sip_address_id,"
699
			"  time, state, direction, imdn_message_id, is_secured"
700 701
			") VALUES ("
			"  :eventId, :localSipaddressId, :remoteSipaddressId,"
702
			"  :time, :state, :direction, :imdnMessageId, :isSecured"
703
			")", soci::use(eventId), soci::use(fromSipAddressId), soci::use(toSipAddressId),
704 705
			soci::use(messageTime), soci::use(state), soci::use(direction),
			soci::use(imdnMessageId), soci::use(isSecured);
706

707
		for (const Content *content : chatMessage->getContents())
708
			insertContent(eventId, *content);
709

710 711 712
		return eventId;
	}

713
	void MainDbPrivate::updateConferenceChatMessageEvent (const shared_ptr<EventLog> &eventLog) {
714 715 716 717 718 719 720 721
		shared_ptr<ChatMessage> chatMessage = static_pointer_cast<ConferenceChatMessageEvent>(eventLog)->getChatMessage();
		shared_ptr<ChatRoom> chatRoom = chatMessage->getChatRoom();
		if (!chatRoom) {
			lError() << "Unable to get a valid chat room. It was removed from database.";
			return;
		}

		const EventLogPrivate *dEventLog = eventLog->getPrivate();
722
		MainDbKeyPrivate *dEventKey = static_cast<MainDbKey &>(dEventLog->dbKey).getPrivate();
723
		const long long &eventId = dEventKey->storageId;
724 725

		soci::session *session = dbSession.getBackendSession<soci::session>();
726 727 728
		const int &state = static_cast<int>(chatMessage->getState());
		*session << "UPDATE conference_chat_message_event SET state = :state WHERE event_id = :eventId",
			soci::use(state), soci::use(eventId);
729

730 731 732 733
		//TODO: improve
		removeContentsForChatMessageEvent(eventId);
		for (const Content *content : chatMessage->getContents())
			insertContent(eventId, *content);
734 735
	}

736 737
	long long MainDbPrivate::insertConferenceNotifiedEvent (const shared_ptr<EventLog> &eventLog, long long *chatRoomId) {
		long long curChatRoomId;
738
		const long long &eventId = insertConferenceEvent(eventLog, &curChatRoomId);
739 740 741
		if (eventId < 0)
			return -1;

742
		const unsigned int &lastNotifyId = static_pointer_cast<ConferenceNotifiedEvent>(eventLog)->getNotifyId();
743 744 745

		soci::session *session = dbSession.getBackendSession<soci::session>();
		*session << "INSERT INTO conference_notified_event (event_id, notify_id)"
Ronan's avatar
Ronan committed
746
			"  VALUES (:eventId, :notifyId)", soci::use(eventId), soci::use(lastNotifyId);
747
		*session << "UPDATE chat_room SET last_notify_id = :lastNotifyId WHERE id = :chatRoomId",
748 749 750 751
			soci::use(lastNotifyId), soci::use(curChatRoomId);

		if (chatRoomId)
			*chatRoomId = curChatRoomId;
752

753 754 755
		return eventId;
	}

Ronan's avatar
Ronan committed
756
	long long MainDbPrivate::insertConferenceParticipantEvent (const shared_ptr<EventLog> &eventLog) {
757
		long long chatRoomId;
758
		const long long &eventId = insertConferenceNotifiedEvent(eventLog, &chatRoomId);
759 760 761
		if (eventId < 0)
			return -1;

762 763 764
		shared_ptr<ConferenceParticipantEvent> participantEvent =
			static_pointer_cast<ConferenceParticipantEvent>(eventLog);

765
		const long long &participantAddressId = insertSipAddress(
766
			participantEvent->getParticipantAddress().asString()
767 768
		);

769
		soci::session *session = dbSession.getBackendSession<soci::session>();
770
		*session << "INSERT INTO conference_participant_event (event_id, participant_sip_address_id)"
771
			"  VALUES (:eventId, :participantAddressId)", soci::use(eventId), soci::use(participantAddressId);
772

773 774 775 776 777 778 779 780 781
		bool isAdmin = eventLog->getType() == EventLog::Type::ConferenceParticipantSetAdmin;
		switch (eventLog->getType()) {
			case EventLog::Type::ConferenceParticipantAdded:
			case EventLog::Type::ConferenceParticipantSetAdmin:
			case EventLog::Type::ConferenceParticipantUnsetAdmin:
				insertChatRoomParticipant(chatRoomId, participantAddressId, isAdmin);
				break;

			case EventLog::Type::ConferenceParticipantRemoved:
782 783
				deleteChatRoomParticipant(chatRoomId, participantAddressId);
				break;
784 785 786 787 788

			default:
				break;
		}

789 790 791
		return eventId;
	}

Ronan's avatar
Ronan committed