diff options
Diffstat (limited to 'plugins/Dbx_sqlite/src/dbevents.cpp')
-rw-r--r-- | plugins/Dbx_sqlite/src/dbevents.cpp | 1398 |
1 files changed, 699 insertions, 699 deletions
diff --git a/plugins/Dbx_sqlite/src/dbevents.cpp b/plugins/Dbx_sqlite/src/dbevents.cpp index f4a4618c2e..c4af7cf487 100644 --- a/plugins/Dbx_sqlite/src/dbevents.cpp +++ b/plugins/Dbx_sqlite/src/dbevents.cpp @@ -1,699 +1,699 @@ -#include "stdafx.h" - -//TODO: hide it inside cursor class -static const char normal_order_query[] = "SELECT id FROM events_srt WHERE contact_id = ? ORDER BY timestamp, id;"; -static const char normal_order_pos_query[] = "SELECT id FROM events_srt WHERE contact_id = ? AND id >= ? ORDER BY timestamp, id;"; - -static const char reverse_order_query[] = "SELECT id FROM events_srt WHERE contact_id = ? ORDER BY timestamp desc, id DESC;"; -static const char reverse_order_pos_query[] = "SELECT id FROM events_srt WHERE contact_id = ? AND id <= ? ORDER BY timestamp desc, id DESC;"; - -static const char add_event_sort_query[] = "INSERT INTO events_srt(id, contact_id, timestamp) VALUES (?, ?, ?);"; - -void CDbxSQLite::InitEvents() -{ - sqlite3_stmt *stmt = nullptr; - sqlite3_prepare_v2(m_db, "SELECT DISTINCT module FROM events;", -1, &stmt, nullptr); - int rc = 0; - while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) { - const char *module = (char*)sqlite3_column_text(stmt, 0); - if (mir_strlen(module) > 0) - m_modules.insert(mir_strdup(module)); - } - logError(rc, __FILE__, __LINE__); - sqlite3_finalize(stmt); -} - -void CDbxSQLite::UninitEvents() -{ - for (auto &module : m_modules) - mir_free(module); - m_modules.destroy(); -} - -int CDbxSQLite::GetEventCount(MCONTACT hContact) -{ - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - if (cc->HasCount()) - return cc->m_count; - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT COUNT(1) FROM events_srt WHERE contact_id = ? LIMIT 1;", qEvCount); - sqlite3_bind_int64(stmt, 1, hContact); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - cc->m_count = (rc != SQLITE_ROW) ? 0 : sqlite3_column_int64(stmt, 0); - sqlite3_reset(stmt); - return cc->m_count; -} - -MEVENT CDbxSQLite::AddEvent(MCONTACT hContact, const DBEVENTINFO *dbei) -{ - if (dbei == nullptr) - return 0; - - if (dbei->timestamp == 0) - return 0; - - MCONTACT hNotifyContact = hContact; - DBCachedContact *cc, *ccSub = nullptr; - if (hContact != 0) { - if ((cc = m_cache->GetCachedContact(hContact)) == nullptr) - return 0; - - if (cc->IsSub()) { - ccSub = cc; - if ((cc = m_cache->GetCachedContact(cc->parentID)) == nullptr) - return 0; - - // set default sub to the event's source - if (!(dbei->flags & DBEF_SENT)) - db_mc_setDefault(cc->contactID, hContact, false); - if (db_mc_isEnabled()) - hNotifyContact = cc->contactID; // and add an event to a metahistory - } - } - else cc = &m_system; - - if (cc == nullptr) - return 0; - - if (m_safetyMode) - if (NotifyEventHooks(g_hevEventFiltered, hNotifyContact, (LPARAM)dbei)) - return 0; - - DBEVENTINFO tmp = *dbei; - const char *szEventId; - if (tmp.szId != nullptr) { - tmp.flags |= DBEF_HAS_ID; - szEventId = tmp.szId; - } - else szEventId = ""; - - mir_ptr<uint8_t> pCryptBlob; - if (m_bEncrypted) { - size_t len; - uint8_t *pResult = m_crypto->encodeBuffer(tmp.pBlob, tmp.cbBlob, &len); - if (pResult != nullptr) { - pCryptBlob = tmp.pBlob = pResult; - tmp.cbBlob = (uint16_t)len; - tmp.flags |= DBEF_ENCRYPTED; - } - } - - mir_cslockfull lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("INSERT INTO events(contact_id, module, timestamp, type, flags, data, server_id, is_read) VALUES (?, ?, ?, ?, ?, ?, ?, ?);", qEvAdd); - sqlite3_bind_int64(stmt, 1, hContact); - sqlite3_bind_text(stmt, 2, tmp.szModule, (int)mir_strlen(tmp.szModule), nullptr); - sqlite3_bind_int64(stmt, 3, tmp.timestamp); - sqlite3_bind_int(stmt, 4, tmp.eventType); - sqlite3_bind_int64(stmt, 5, tmp.flags); - sqlite3_bind_blob(stmt, 6, tmp.pBlob, tmp.cbBlob, nullptr); - sqlite3_bind_text(stmt, 7, szEventId, (int)mir_strlen(szEventId), nullptr); - sqlite3_bind_int(stmt, 8, tmp.markedRead()); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - - MEVENT hDbEvent = sqlite3_last_insert_rowid(m_db); - - stmt = InitQuery(add_event_sort_query, qEvAddSrt); - sqlite3_bind_int64(stmt, 1, hDbEvent); - sqlite3_bind_int64(stmt, 2, cc->contactID); - sqlite3_bind_int64(stmt, 3, tmp.timestamp); - rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - - if (ccSub != nullptr) { - stmt = InitQuery(add_event_sort_query, qEvAddSrt); - sqlite3_bind_int64(stmt, 1, hDbEvent); - sqlite3_bind_int64(stmt, 2, ccSub->contactID); - sqlite3_bind_int64(stmt, 3, tmp.timestamp); - rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); //is this necessary ? - } - - char *module = m_modules.find((char *)tmp.szModule); - if (module == nullptr) - m_modules.insert(mir_strdup(tmp.szModule)); - - lock.unlock(); - - DBFlush(); - if (m_safetyMode && !(tmp.flags & DBEF_TEMPORARY)) - NotifyEventHooks(g_hevEventAdded, hNotifyContact, (LPARAM)hDbEvent); - - return hDbEvent; -} - -///////////////////////////////////////////////////////////////////////////////////////// - -int CDbxSQLite::DeleteEventMain(MEVENT hDbEvent) -{ - auto *stmt = InitQuery("DELETE FROM events WHERE id = ?;", qEvDel); - sqlite3_bind_int64(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - return rc; -} - -int CDbxSQLite::DeleteEventSrt(MEVENT hDbEvent, MCONTACT hContact, uint32_t ts) -{ - auto *stmt = InitQuery("DELETE FROM events_srt WHERE id = ? AND contact_id = ? AND timestamp = ?;", qEvDelSrt); - sqlite3_bind_int64(stmt, 1, hDbEvent); - sqlite3_bind_int64(stmt, 2, hContact); - sqlite3_bind_int64(stmt, 3, ts); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - return rc; -} - -BOOL CDbxSQLite::DeleteEvent(MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return 1; - - MEVENT hContact; - uint32_t ts; - { - sqlite3_stmt *stmt = InitQuery("SELECT contact_id, timestamp FROM events WHERE id = ? LIMIT 1;", qEvGetContact2); - sqlite3_bind_int64(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return 2; - } - hContact = sqlite3_column_int64(stmt, 0); - ts = sqlite3_column_int64(stmt, 1); - sqlite3_reset(stmt); - } - - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 1; - - mir_cslockfull lock(m_csDbAccess); - int rc = DeleteEventMain(hDbEvent); - if (rc != SQLITE_DONE) - return 1; - - rc = DeleteEventSrt(hDbEvent, hContact, ts); - if (rc != SQLITE_DONE) - return 1; - - lock.unlock(); - - DBFlush(); - NotifyEventHooks(g_hevEventDeleted, hContact, hDbEvent); - return 0; -} - -///////////////////////////////////////////////////////////////////////////////////////// - -BOOL CDbxSQLite::EditEvent(MCONTACT hContact, MEVENT hDbEvent, const DBEVENTINFO *dbei) -{ - if (dbei == nullptr) - return 1; - - if (dbei->timestamp == 0) - return 1; - - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 1; - - DBEVENTINFO tmp = *dbei; - mir_ptr<uint8_t> pCryptBlob; - if (m_bEncrypted) { - size_t len; - uint8_t *pResult = m_crypto->encodeBuffer(tmp.pBlob, tmp.cbBlob, &len); - if (pResult != nullptr) { - pCryptBlob = tmp.pBlob = pResult; - tmp.cbBlob = (uint16_t)len; - tmp.flags |= DBEF_ENCRYPTED; - } - } - - mir_cslockfull lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("UPDATE events SET module = ?, timestamp = ?, type = ?, flags = ?, data = ?, is_read = ? WHERE id = ?;", qEvEdit); - sqlite3_bind_text(stmt, 1, tmp.szModule, (int)mir_strlen(tmp.szModule), nullptr); - sqlite3_bind_int64(stmt, 2, tmp.timestamp); - sqlite3_bind_int(stmt, 3, tmp.eventType); - sqlite3_bind_int64(stmt, 4, tmp.flags); - sqlite3_bind_blob(stmt, 5, tmp.pBlob, tmp.cbBlob, nullptr); - sqlite3_bind_int(stmt, 6, tmp.markedRead()); - sqlite3_bind_int64(stmt, 7, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - - char *module = m_modules.find((char *)tmp.szModule); - if (module == nullptr) - m_modules.insert(mir_strdup(tmp.szModule)); - - lock.unlock(); - - DBFlush(); - NotifyEventHooks(g_hevEventEdited, hContact, (LPARAM)hDbEvent); - return 0; -} - -int CDbxSQLite::GetBlobSize(MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return -1; - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT LENGTH(data) FROM events WHERE id = ? LIMIT 1;", qEvBlobSize); - sqlite3_bind_int(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return -1; - } - - LONG res = sqlite3_column_int64(stmt, 0); - sqlite3_reset(stmt); - return res; -} - -static char g_szId[100]; - -BOOL CDbxSQLite::GetEvent(MEVENT hDbEvent, DBEVENTINFO *dbei) -{ - if (hDbEvent == 0) - return 1; - - if (dbei == nullptr) - return 1; - - if (dbei->cbBlob > 0 && dbei->pBlob == nullptr) { - dbei->cbBlob = 0; - return 1; - } - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT module, timestamp, type, flags, server_id, length(data), data FROM events WHERE id = ? LIMIT 1;", qEvGet); - sqlite3_bind_int64(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return 1; - } - - char *module = (char *)sqlite3_column_text(stmt, 0); - dbei->szModule = m_modules.find(module); - if (dbei->szModule == nullptr) - return 1; - - dbei->timestamp = sqlite3_column_int64(stmt, 1); - dbei->eventType = sqlite3_column_int(stmt, 2); - dbei->flags = sqlite3_column_int64(stmt, 3); - - char *pszId = (char *)sqlite3_column_text(stmt, 4); - if (mir_strlen(pszId)) { - mir_strncpy(g_szId, pszId, sizeof(g_szId)); - dbei->szId = g_szId; - } - else dbei->szId = nullptr; - - int32_t cbBlob = sqlite3_column_int64(stmt, 5); - size_t bytesToCopy = cbBlob; - if (dbei->cbBlob == -1) - dbei->pBlob = (uint8_t*)mir_calloc(cbBlob + 2); - else if (dbei->cbBlob < cbBlob) - bytesToCopy = dbei->cbBlob; - - dbei->cbBlob = cbBlob; - if (bytesToCopy && dbei->pBlob) { - uint8_t *data = (uint8_t *)sqlite3_column_blob(stmt, 6); - - if (dbei->flags & DBEF_ENCRYPTED) { - dbei->flags &= ~DBEF_ENCRYPTED; - size_t len; - uint8_t* pBlob = (uint8_t*)m_crypto->decodeBuffer(data, cbBlob, &len); - if (pBlob == nullptr) - return 1; - - memcpy(dbei->pBlob, pBlob, bytesToCopy); - if (bytesToCopy > len) - memset(dbei->pBlob + len, 0, bytesToCopy - len); - - mir_free(pBlob); - } - else memcpy(dbei->pBlob, data, bytesToCopy); - } - sqlite3_reset(stmt); - return 0; -} - -BOOL CDbxSQLite::MarkEventRead(MCONTACT hContact, MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return -1; - - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return -1; - - int rows; - { - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("UPDATE events SET flags = flags | 4, is_read = 1 WHERE id = ? AND is_read = 0;", qEvSetFlags); - sqlite3_bind_int64(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - rows = sqlite3_changes(m_db); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - if (rc != SQLITE_DONE) - return -1; - } - - DBFlush(); - NotifyEventHooks(g_hevMarkedRead, hContact, (LPARAM)hDbEvent); - return (rows == 0 ? 0 : DBEF_READ); -} - -MCONTACT CDbxSQLite::GetEventContact(MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return INVALID_CONTACT_ID; - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT contact_id FROM events WHERE id = ? LIMIT 1;", qEvGetContact); - sqlite3_bind_int64(stmt, 1, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return INVALID_CONTACT_ID; - } - MCONTACT hContact = sqlite3_column_int64(stmt, 0); - sqlite3_reset(stmt); - return hContact; -} - -MEVENT CDbxSQLite::FindFirstUnreadEvent(MCONTACT hContact) -{ - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - mir_cslock lock(m_csDbAccess); - - if (cc->IsMeta()) { - if (cc->nSubs == 0) - return 0; - - CMStringA query("SELECT id FROM events WHERE is_read = 0 AND contact_id IN ("); - for (int k = 0; k < cc->nSubs; k++) - query.AppendFormat("%lu, ", cc->pSubs[k]); - query.Delete(query.GetLength() - 2, 2); - query.Append(") ORDER BY timestamp LIMIT 1;"); - - sqlite3_stmt *stmt; - sqlite3_prepare_v2(m_db, query, -1, &stmt, nullptr); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_finalize(stmt); - return 0; - } - - MEVENT ret = sqlite3_column_int64(stmt, 0); - sqlite3_finalize(stmt); - return ret; - } - - sqlite3_stmt *stmt = InitQuery("SELECT id FROM events WHERE contact_id = ? AND is_read = 0 ORDER BY timestamp LIMIT 1;", qEvFindUnread); - sqlite3_bind_int64(stmt, 1, hContact); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return 0; - } - - MEVENT ret = sqlite3_column_int64(stmt, 0); - sqlite3_reset(stmt); - return ret; -} - -///////////////////////////////////////////////////////////////////////////////////////// -// First/next event - -MEVENT CDbxSQLite::FindFirstEvent(MCONTACT hContact) -{ - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - mir_cslock lock(m_csDbAccess); - - if (fwd.cur) - sqlite3_reset(fwd.cur); - - fwd.hContact = hContact; - fwd.cur = InitQuery(normal_order_query, qEvFindFirst); - sqlite3_bind_int64(fwd.cur, 1, hContact); - - int rc = sqlite3_step(fwd.cur); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - fwd.clear(); - return 0; - } - return fwd.hEvent = sqlite3_column_int64(fwd.cur, 0); -} - -MEVENT CDbxSQLite::FindNextEvent(MCONTACT hContact, MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return 0; - - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - if (hContact != fwd.hContact || hDbEvent != fwd.hEvent) { - if (fwd.cur) - sqlite3_reset(fwd.cur); - - fwd.hContact = hContact; - fwd.cur = InitQuery("SELECT id FROM events_srt WHERE contact_id = ? AND id > ? ORDER BY timestamp, id;", qEvFindNext); - sqlite3_bind_int64(fwd.cur, 1, hContact); - sqlite3_bind_int64(fwd.cur, 2, hDbEvent); - } - - int rc = sqlite3_step(fwd.cur); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - fwd.clear(); - return 0; - } - - return fwd.hEvent = sqlite3_column_int64(fwd.cur, 0); -} - -///////////////////////////////////////////////////////////////////////////////////////// -// Last/prev event - -MEVENT CDbxSQLite::FindLastEvent(MCONTACT hContact) -{ - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - mir_cslock lock(m_csDbAccess); - - if (back.cur) - sqlite3_reset(back.cur); - - back.hContact = hContact; - back.cur = InitQuery(reverse_order_query, qEvFindLast); - sqlite3_bind_int64(back.cur, 1, hContact); - int rc = sqlite3_step(back.cur); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - back.clear(); - return 0; - } - - return back.hEvent = sqlite3_column_int64(back.cur, 0); -} - -MEVENT CDbxSQLite::FindPrevEvent(MCONTACT hContact, MEVENT hDbEvent) -{ - if (hDbEvent == 0) - return 0; - - DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system; - if (cc == nullptr) - return 0; - - if (hContact != back.hContact || hDbEvent != back.hEvent) { - if (back.cur) - sqlite3_reset(back.cur); - - back.hContact = hContact; - back.cur = InitQuery("SELECT id FROM events_srt WHERE contact_id = ? AND id < ? ORDER BY timestamp desc, id DESC;", qEvFindPrev); - sqlite3_bind_int64(back.cur, 1, hContact); - sqlite3_bind_int64(back.cur, 2, hDbEvent); - } - - int rc = sqlite3_step(back.cur); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - back.clear(); - return 0; - } - - return back.hEvent = sqlite3_column_int64(back.cur, 0); -} - -///////////////////////////////////////////////////////////////////////////////////////// -// Metacontacts - -BOOL CDbxSQLite::MetaMergeHistory(DBCachedContact *ccMeta, DBCachedContact *ccSub) -{ - //TODO: test this - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT id, timestamp FROM events WHERE contact_id = ?;", qEvMetaMerge); - sqlite3_bind_int64(stmt, 1, ccSub->contactID); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - while (rc == SQLITE_ROW) { - sqlite3_stmt *stmt2 = InitQuery(add_event_sort_query, qEvAddSrt); - sqlite3_bind_int64(stmt2, 1, sqlite3_column_int64(stmt, 0)); - sqlite3_bind_int64(stmt2, 2, ccMeta->contactID); - sqlite3_bind_int64(stmt2, 3, sqlite3_column_int64(stmt, 1)); - int rc2 = sqlite3_step(stmt2); - logError(rc2, __FILE__, __LINE__); - sqlite3_reset(stmt2); - rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - } - - sqlite3_reset(stmt); - DBFlush(); - return TRUE; -} - -BOOL CDbxSQLite::MetaSplitHistory(DBCachedContact *ccMeta, DBCachedContact *) -{ - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("DELETE FROM events_srt WHERE contact_id = ?;", qEvMetaSplit); - sqlite3_bind_int64(stmt, 1, ccMeta->contactID); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - sqlite3_reset(stmt); - if (rc != SQLITE_DONE) - return 1; - - DBFlush(); - return TRUE; -} - -///////////////////////////////////////////////////////////////////////////////////////// -// Server ids - -MEVENT CDbxSQLite::GetEventById(LPCSTR szModule, LPCSTR szId) -{ - if (szModule == nullptr || szId == nullptr) - return 0; - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("SELECT id, timestamp FROM events WHERE module = ? AND server_id = ? LIMIT 1;", qEvGetById); - sqlite3_bind_text(stmt, 1, szModule, (int)mir_strlen(szModule), nullptr); - sqlite3_bind_text(stmt, 2, szId, (int)mir_strlen(szId), nullptr); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - sqlite3_reset(stmt); - return 0; - } - MEVENT hDbEvent = sqlite3_column_int64(stmt, 0); - sqlite3_reset(stmt); - return hDbEvent; -} - -int CDbxSQLite::UpdateEventId(MEVENT hDbEvent, LPCSTR szId) -{ - if (hDbEvent == 0 || mir_strlen(szId) == 0) - return 1; - - mir_cslock lock(m_csDbAccess); - sqlite3_stmt *stmt = InitQuery("UPDATE events SET server_id = ? WHERE id = ?;", qEvUpdateId); - sqlite3_bind_text(stmt, 1, szId, (int)mir_strlen(szId), nullptr); - sqlite3_bind_int64(stmt, 2, hDbEvent); - int rc = sqlite3_step(stmt); - logError(rc, __FILE__, __LINE__); - int rows = sqlite3_changes(m_db); - sqlite3_reset(stmt); - return (rows == 0) ? 2 : 0; -} - -///////////////////////////////////////////////////////////////////////////////////////// -// Event cursors - -STDMETHODIMP_(DB::EventCursor *) CDbxSQLite::EventCursor(MCONTACT hContact, MEVENT hDbEvent) -{ - return new CDbxSQLiteEventCursor(hContact, m_db, hDbEvent); -} - -STDMETHODIMP_(DB::EventCursor *) CDbxSQLite::EventCursorRev(MCONTACT hContact, MEVENT hDbEvent) -{ - return new CDbxSQLiteEventCursor(hContact, m_db, hDbEvent, true); -} - -CDbxSQLiteEventCursor::CDbxSQLiteEventCursor(MCONTACT _1, sqlite3 *_db, MEVENT hDbEvent, bool reverse) : - EventCursor(_1), m_db(_db) -{ - if (reverse) { - if (!hDbEvent) - sqlite3_prepare_v2(m_db, reverse_order_query, -1, &cursor, nullptr); - else - sqlite3_prepare_v2(m_db, reverse_order_pos_query, -1, &cursor, nullptr); - } - else { - if (!hDbEvent) - sqlite3_prepare_v2(m_db, normal_order_query, -1, &cursor, nullptr); - else - sqlite3_prepare_v2(m_db, normal_order_pos_query, -1, &cursor, nullptr); - } - sqlite3_bind_int64(cursor, 1, hContact); - if (hDbEvent) - sqlite3_bind_int64(cursor, 2, hDbEvent); -} - -CDbxSQLiteEventCursor::~CDbxSQLiteEventCursor() -{ - if (cursor) - sqlite3_reset(cursor); -} - -MEVENT CDbxSQLiteEventCursor::FetchNext() -{ - if (!cursor) - return 0; - - int rc = sqlite3_step(cursor); - logError(rc, __FILE__, __LINE__); - if (rc != SQLITE_ROW) { - //empty response - //reset sql cursor - sqlite3_reset(cursor); - cursor = nullptr; - return 0; - } - return sqlite3_column_int64(cursor, 0); -} +#include "stdafx.h"
+
+//TODO: hide it inside cursor class
+static const char normal_order_query[] = "SELECT id FROM events_srt WHERE contact_id = ? ORDER BY timestamp, id;";
+static const char normal_order_pos_query[] = "SELECT id FROM events_srt WHERE contact_id = ? AND id >= ? ORDER BY timestamp, id;";
+
+static const char reverse_order_query[] = "SELECT id FROM events_srt WHERE contact_id = ? ORDER BY timestamp desc, id DESC;";
+static const char reverse_order_pos_query[] = "SELECT id FROM events_srt WHERE contact_id = ? AND id <= ? ORDER BY timestamp desc, id DESC;";
+
+static const char add_event_sort_query[] = "INSERT INTO events_srt(id, contact_id, timestamp) VALUES (?, ?, ?);";
+
+void CDbxSQLite::InitEvents()
+{
+ sqlite3_stmt *stmt = nullptr;
+ sqlite3_prepare_v2(m_db, "SELECT DISTINCT module FROM events;", -1, &stmt, nullptr);
+ int rc = 0;
+ while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) {
+ const char *module = (char*)sqlite3_column_text(stmt, 0);
+ if (mir_strlen(module) > 0)
+ m_modules.insert(mir_strdup(module));
+ }
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_finalize(stmt);
+}
+
+void CDbxSQLite::UninitEvents()
+{
+ for (auto &module : m_modules)
+ mir_free(module);
+ m_modules.destroy();
+}
+
+int CDbxSQLite::GetEventCount(MCONTACT hContact)
+{
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ if (cc->HasCount())
+ return cc->m_count;
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT COUNT(1) FROM events_srt WHERE contact_id = ? LIMIT 1;", qEvCount);
+ sqlite3_bind_int64(stmt, 1, hContact);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ cc->m_count = (rc != SQLITE_ROW) ? 0 : sqlite3_column_int64(stmt, 0);
+ sqlite3_reset(stmt);
+ return cc->m_count;
+}
+
+MEVENT CDbxSQLite::AddEvent(MCONTACT hContact, const DBEVENTINFO *dbei)
+{
+ if (dbei == nullptr)
+ return 0;
+
+ if (dbei->timestamp == 0)
+ return 0;
+
+ MCONTACT hNotifyContact = hContact;
+ DBCachedContact *cc, *ccSub = nullptr;
+ if (hContact != 0) {
+ if ((cc = m_cache->GetCachedContact(hContact)) == nullptr)
+ return 0;
+
+ if (cc->IsSub()) {
+ ccSub = cc;
+ if ((cc = m_cache->GetCachedContact(cc->parentID)) == nullptr)
+ return 0;
+
+ // set default sub to the event's source
+ if (!(dbei->flags & DBEF_SENT))
+ db_mc_setDefault(cc->contactID, hContact, false);
+ if (db_mc_isEnabled())
+ hNotifyContact = cc->contactID; // and add an event to a metahistory
+ }
+ }
+ else cc = &m_system;
+
+ if (cc == nullptr)
+ return 0;
+
+ if (m_safetyMode)
+ if (NotifyEventHooks(g_hevEventFiltered, hNotifyContact, (LPARAM)dbei))
+ return 0;
+
+ DBEVENTINFO tmp = *dbei;
+ const char *szEventId;
+ if (tmp.szId != nullptr) {
+ tmp.flags |= DBEF_HAS_ID;
+ szEventId = tmp.szId;
+ }
+ else szEventId = "";
+
+ mir_ptr<uint8_t> pCryptBlob;
+ if (m_bEncrypted) {
+ size_t len;
+ uint8_t *pResult = m_crypto->encodeBuffer(tmp.pBlob, tmp.cbBlob, &len);
+ if (pResult != nullptr) {
+ pCryptBlob = tmp.pBlob = pResult;
+ tmp.cbBlob = (uint16_t)len;
+ tmp.flags |= DBEF_ENCRYPTED;
+ }
+ }
+
+ mir_cslockfull lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("INSERT INTO events(contact_id, module, timestamp, type, flags, data, server_id, is_read) VALUES (?, ?, ?, ?, ?, ?, ?, ?);", qEvAdd);
+ sqlite3_bind_int64(stmt, 1, hContact);
+ sqlite3_bind_text(stmt, 2, tmp.szModule, (int)mir_strlen(tmp.szModule), nullptr);
+ sqlite3_bind_int64(stmt, 3, tmp.timestamp);
+ sqlite3_bind_int(stmt, 4, tmp.eventType);
+ sqlite3_bind_int64(stmt, 5, tmp.flags);
+ sqlite3_bind_blob(stmt, 6, tmp.pBlob, tmp.cbBlob, nullptr);
+ sqlite3_bind_text(stmt, 7, szEventId, (int)mir_strlen(szEventId), nullptr);
+ sqlite3_bind_int(stmt, 8, tmp.markedRead());
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+
+ MEVENT hDbEvent = sqlite3_last_insert_rowid(m_db);
+
+ stmt = InitQuery(add_event_sort_query, qEvAddSrt);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ sqlite3_bind_int64(stmt, 2, cc->contactID);
+ sqlite3_bind_int64(stmt, 3, tmp.timestamp);
+ rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+
+ if (ccSub != nullptr) {
+ stmt = InitQuery(add_event_sort_query, qEvAddSrt);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ sqlite3_bind_int64(stmt, 2, ccSub->contactID);
+ sqlite3_bind_int64(stmt, 3, tmp.timestamp);
+ rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt); //is this necessary ?
+ }
+
+ char *module = m_modules.find((char *)tmp.szModule);
+ if (module == nullptr)
+ m_modules.insert(mir_strdup(tmp.szModule));
+
+ lock.unlock();
+
+ DBFlush();
+ if (m_safetyMode && !(tmp.flags & DBEF_TEMPORARY))
+ NotifyEventHooks(g_hevEventAdded, hNotifyContact, (LPARAM)hDbEvent);
+
+ return hDbEvent;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+
+int CDbxSQLite::DeleteEventMain(MEVENT hDbEvent)
+{
+ auto *stmt = InitQuery("DELETE FROM events WHERE id = ?;", qEvDel);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+ return rc;
+}
+
+int CDbxSQLite::DeleteEventSrt(MEVENT hDbEvent, MCONTACT hContact, uint32_t ts)
+{
+ auto *stmt = InitQuery("DELETE FROM events_srt WHERE id = ? AND contact_id = ? AND timestamp = ?;", qEvDelSrt);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ sqlite3_bind_int64(stmt, 2, hContact);
+ sqlite3_bind_int64(stmt, 3, ts);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+ return rc;
+}
+
+BOOL CDbxSQLite::DeleteEvent(MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return 1;
+
+ MEVENT hContact;
+ uint32_t ts;
+ {
+ sqlite3_stmt *stmt = InitQuery("SELECT contact_id, timestamp FROM events WHERE id = ? LIMIT 1;", qEvGetContact2);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return 2;
+ }
+ hContact = sqlite3_column_int64(stmt, 0);
+ ts = sqlite3_column_int64(stmt, 1);
+ sqlite3_reset(stmt);
+ }
+
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 1;
+
+ mir_cslockfull lock(m_csDbAccess);
+ int rc = DeleteEventMain(hDbEvent);
+ if (rc != SQLITE_DONE)
+ return 1;
+
+ rc = DeleteEventSrt(hDbEvent, hContact, ts);
+ if (rc != SQLITE_DONE)
+ return 1;
+
+ lock.unlock();
+
+ DBFlush();
+ NotifyEventHooks(g_hevEventDeleted, hContact, hDbEvent);
+ return 0;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+
+BOOL CDbxSQLite::EditEvent(MCONTACT hContact, MEVENT hDbEvent, const DBEVENTINFO *dbei)
+{
+ if (dbei == nullptr)
+ return 1;
+
+ if (dbei->timestamp == 0)
+ return 1;
+
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 1;
+
+ DBEVENTINFO tmp = *dbei;
+ mir_ptr<uint8_t> pCryptBlob;
+ if (m_bEncrypted) {
+ size_t len;
+ uint8_t *pResult = m_crypto->encodeBuffer(tmp.pBlob, tmp.cbBlob, &len);
+ if (pResult != nullptr) {
+ pCryptBlob = tmp.pBlob = pResult;
+ tmp.cbBlob = (uint16_t)len;
+ tmp.flags |= DBEF_ENCRYPTED;
+ }
+ }
+
+ mir_cslockfull lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("UPDATE events SET module = ?, timestamp = ?, type = ?, flags = ?, data = ?, is_read = ? WHERE id = ?;", qEvEdit);
+ sqlite3_bind_text(stmt, 1, tmp.szModule, (int)mir_strlen(tmp.szModule), nullptr);
+ sqlite3_bind_int64(stmt, 2, tmp.timestamp);
+ sqlite3_bind_int(stmt, 3, tmp.eventType);
+ sqlite3_bind_int64(stmt, 4, tmp.flags);
+ sqlite3_bind_blob(stmt, 5, tmp.pBlob, tmp.cbBlob, nullptr);
+ sqlite3_bind_int(stmt, 6, tmp.markedRead());
+ sqlite3_bind_int64(stmt, 7, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+
+ char *module = m_modules.find((char *)tmp.szModule);
+ if (module == nullptr)
+ m_modules.insert(mir_strdup(tmp.szModule));
+
+ lock.unlock();
+
+ DBFlush();
+ NotifyEventHooks(g_hevEventEdited, hContact, (LPARAM)hDbEvent);
+ return 0;
+}
+
+int CDbxSQLite::GetBlobSize(MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return -1;
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT LENGTH(data) FROM events WHERE id = ? LIMIT 1;", qEvBlobSize);
+ sqlite3_bind_int(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return -1;
+ }
+
+ LONG res = sqlite3_column_int64(stmt, 0);
+ sqlite3_reset(stmt);
+ return res;
+}
+
+static char g_szId[100];
+
+BOOL CDbxSQLite::GetEvent(MEVENT hDbEvent, DBEVENTINFO *dbei)
+{
+ if (hDbEvent == 0)
+ return 1;
+
+ if (dbei == nullptr)
+ return 1;
+
+ if (dbei->cbBlob > 0 && dbei->pBlob == nullptr) {
+ dbei->cbBlob = 0;
+ return 1;
+ }
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT module, timestamp, type, flags, server_id, length(data), data FROM events WHERE id = ? LIMIT 1;", qEvGet);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return 1;
+ }
+
+ char *module = (char *)sqlite3_column_text(stmt, 0);
+ dbei->szModule = m_modules.find(module);
+ if (dbei->szModule == nullptr)
+ return 1;
+
+ dbei->timestamp = sqlite3_column_int64(stmt, 1);
+ dbei->eventType = sqlite3_column_int(stmt, 2);
+ dbei->flags = sqlite3_column_int64(stmt, 3);
+
+ char *pszId = (char *)sqlite3_column_text(stmt, 4);
+ if (mir_strlen(pszId)) {
+ mir_strncpy(g_szId, pszId, sizeof(g_szId));
+ dbei->szId = g_szId;
+ }
+ else dbei->szId = nullptr;
+
+ int32_t cbBlob = sqlite3_column_int64(stmt, 5);
+ size_t bytesToCopy = cbBlob;
+ if (dbei->cbBlob == -1)
+ dbei->pBlob = (uint8_t*)mir_calloc(cbBlob + 2);
+ else if (dbei->cbBlob < cbBlob)
+ bytesToCopy = dbei->cbBlob;
+
+ dbei->cbBlob = cbBlob;
+ if (bytesToCopy && dbei->pBlob) {
+ uint8_t *data = (uint8_t *)sqlite3_column_blob(stmt, 6);
+
+ if (dbei->flags & DBEF_ENCRYPTED) {
+ dbei->flags &= ~DBEF_ENCRYPTED;
+ size_t len;
+ uint8_t* pBlob = (uint8_t*)m_crypto->decodeBuffer(data, cbBlob, &len);
+ if (pBlob == nullptr)
+ return 1;
+
+ memcpy(dbei->pBlob, pBlob, bytesToCopy);
+ if (bytesToCopy > len)
+ memset(dbei->pBlob + len, 0, bytesToCopy - len);
+
+ mir_free(pBlob);
+ }
+ else memcpy(dbei->pBlob, data, bytesToCopy);
+ }
+ sqlite3_reset(stmt);
+ return 0;
+}
+
+BOOL CDbxSQLite::MarkEventRead(MCONTACT hContact, MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return -1;
+
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return -1;
+
+ int rows;
+ {
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("UPDATE events SET flags = flags | 4, is_read = 1 WHERE id = ? AND is_read = 0;", qEvSetFlags);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ rows = sqlite3_changes(m_db);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+ if (rc != SQLITE_DONE)
+ return -1;
+ }
+
+ DBFlush();
+ NotifyEventHooks(g_hevMarkedRead, hContact, (LPARAM)hDbEvent);
+ return (rows == 0 ? 0 : DBEF_READ);
+}
+
+MCONTACT CDbxSQLite::GetEventContact(MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return INVALID_CONTACT_ID;
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT contact_id FROM events WHERE id = ? LIMIT 1;", qEvGetContact);
+ sqlite3_bind_int64(stmt, 1, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return INVALID_CONTACT_ID;
+ }
+ MCONTACT hContact = sqlite3_column_int64(stmt, 0);
+ sqlite3_reset(stmt);
+ return hContact;
+}
+
+MEVENT CDbxSQLite::FindFirstUnreadEvent(MCONTACT hContact)
+{
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ mir_cslock lock(m_csDbAccess);
+
+ if (cc->IsMeta()) {
+ if (cc->nSubs == 0)
+ return 0;
+
+ CMStringA query("SELECT id FROM events WHERE is_read = 0 AND contact_id IN (");
+ for (int k = 0; k < cc->nSubs; k++)
+ query.AppendFormat("%lu, ", cc->pSubs[k]);
+ query.Delete(query.GetLength() - 2, 2);
+ query.Append(") ORDER BY timestamp LIMIT 1;");
+
+ sqlite3_stmt *stmt;
+ sqlite3_prepare_v2(m_db, query, -1, &stmt, nullptr);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_finalize(stmt);
+ return 0;
+ }
+
+ MEVENT ret = sqlite3_column_int64(stmt, 0);
+ sqlite3_finalize(stmt);
+ return ret;
+ }
+
+ sqlite3_stmt *stmt = InitQuery("SELECT id FROM events WHERE contact_id = ? AND is_read = 0 ORDER BY timestamp LIMIT 1;", qEvFindUnread);
+ sqlite3_bind_int64(stmt, 1, hContact);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return 0;
+ }
+
+ MEVENT ret = sqlite3_column_int64(stmt, 0);
+ sqlite3_reset(stmt);
+ return ret;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// First/next event
+
+MEVENT CDbxSQLite::FindFirstEvent(MCONTACT hContact)
+{
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ mir_cslock lock(m_csDbAccess);
+
+ if (fwd.cur)
+ sqlite3_reset(fwd.cur);
+
+ fwd.hContact = hContact;
+ fwd.cur = InitQuery(normal_order_query, qEvFindFirst);
+ sqlite3_bind_int64(fwd.cur, 1, hContact);
+
+ int rc = sqlite3_step(fwd.cur);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ fwd.clear();
+ return 0;
+ }
+ return fwd.hEvent = sqlite3_column_int64(fwd.cur, 0);
+}
+
+MEVENT CDbxSQLite::FindNextEvent(MCONTACT hContact, MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return 0;
+
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ if (hContact != fwd.hContact || hDbEvent != fwd.hEvent) {
+ if (fwd.cur)
+ sqlite3_reset(fwd.cur);
+
+ fwd.hContact = hContact;
+ fwd.cur = InitQuery("SELECT id FROM events_srt WHERE contact_id = ? AND id > ? ORDER BY timestamp, id;", qEvFindNext);
+ sqlite3_bind_int64(fwd.cur, 1, hContact);
+ sqlite3_bind_int64(fwd.cur, 2, hDbEvent);
+ }
+
+ int rc = sqlite3_step(fwd.cur);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ fwd.clear();
+ return 0;
+ }
+
+ return fwd.hEvent = sqlite3_column_int64(fwd.cur, 0);
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// Last/prev event
+
+MEVENT CDbxSQLite::FindLastEvent(MCONTACT hContact)
+{
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ mir_cslock lock(m_csDbAccess);
+
+ if (back.cur)
+ sqlite3_reset(back.cur);
+
+ back.hContact = hContact;
+ back.cur = InitQuery(reverse_order_query, qEvFindLast);
+ sqlite3_bind_int64(back.cur, 1, hContact);
+ int rc = sqlite3_step(back.cur);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ back.clear();
+ return 0;
+ }
+
+ return back.hEvent = sqlite3_column_int64(back.cur, 0);
+}
+
+MEVENT CDbxSQLite::FindPrevEvent(MCONTACT hContact, MEVENT hDbEvent)
+{
+ if (hDbEvent == 0)
+ return 0;
+
+ DBCachedContact *cc = (hContact) ? m_cache->GetCachedContact(hContact) : &m_system;
+ if (cc == nullptr)
+ return 0;
+
+ if (hContact != back.hContact || hDbEvent != back.hEvent) {
+ if (back.cur)
+ sqlite3_reset(back.cur);
+
+ back.hContact = hContact;
+ back.cur = InitQuery("SELECT id FROM events_srt WHERE contact_id = ? AND id < ? ORDER BY timestamp desc, id DESC;", qEvFindPrev);
+ sqlite3_bind_int64(back.cur, 1, hContact);
+ sqlite3_bind_int64(back.cur, 2, hDbEvent);
+ }
+
+ int rc = sqlite3_step(back.cur);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ back.clear();
+ return 0;
+ }
+
+ return back.hEvent = sqlite3_column_int64(back.cur, 0);
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// Metacontacts
+
+BOOL CDbxSQLite::MetaMergeHistory(DBCachedContact *ccMeta, DBCachedContact *ccSub)
+{
+ //TODO: test this
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT id, timestamp FROM events WHERE contact_id = ?;", qEvMetaMerge);
+ sqlite3_bind_int64(stmt, 1, ccSub->contactID);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ while (rc == SQLITE_ROW) {
+ sqlite3_stmt *stmt2 = InitQuery(add_event_sort_query, qEvAddSrt);
+ sqlite3_bind_int64(stmt2, 1, sqlite3_column_int64(stmt, 0));
+ sqlite3_bind_int64(stmt2, 2, ccMeta->contactID);
+ sqlite3_bind_int64(stmt2, 3, sqlite3_column_int64(stmt, 1));
+ int rc2 = sqlite3_step(stmt2);
+ logError(rc2, __FILE__, __LINE__);
+ sqlite3_reset(stmt2);
+ rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ }
+
+ sqlite3_reset(stmt);
+ DBFlush();
+ return TRUE;
+}
+
+BOOL CDbxSQLite::MetaSplitHistory(DBCachedContact *ccMeta, DBCachedContact *)
+{
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("DELETE FROM events_srt WHERE contact_id = ?;", qEvMetaSplit);
+ sqlite3_bind_int64(stmt, 1, ccMeta->contactID);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ sqlite3_reset(stmt);
+ if (rc != SQLITE_DONE)
+ return 1;
+
+ DBFlush();
+ return TRUE;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// Server ids
+
+MEVENT CDbxSQLite::GetEventById(LPCSTR szModule, LPCSTR szId)
+{
+ if (szModule == nullptr || szId == nullptr)
+ return 0;
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("SELECT id, timestamp FROM events WHERE module = ? AND server_id = ? LIMIT 1;", qEvGetById);
+ sqlite3_bind_text(stmt, 1, szModule, (int)mir_strlen(szModule), nullptr);
+ sqlite3_bind_text(stmt, 2, szId, (int)mir_strlen(szId), nullptr);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ sqlite3_reset(stmt);
+ return 0;
+ }
+ MEVENT hDbEvent = sqlite3_column_int64(stmt, 0);
+ sqlite3_reset(stmt);
+ return hDbEvent;
+}
+
+int CDbxSQLite::UpdateEventId(MEVENT hDbEvent, LPCSTR szId)
+{
+ if (hDbEvent == 0 || mir_strlen(szId) == 0)
+ return 1;
+
+ mir_cslock lock(m_csDbAccess);
+ sqlite3_stmt *stmt = InitQuery("UPDATE events SET server_id = ? WHERE id = ?;", qEvUpdateId);
+ sqlite3_bind_text(stmt, 1, szId, (int)mir_strlen(szId), nullptr);
+ sqlite3_bind_int64(stmt, 2, hDbEvent);
+ int rc = sqlite3_step(stmt);
+ logError(rc, __FILE__, __LINE__);
+ int rows = sqlite3_changes(m_db);
+ sqlite3_reset(stmt);
+ return (rows == 0) ? 2 : 0;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// Event cursors
+
+STDMETHODIMP_(DB::EventCursor *) CDbxSQLite::EventCursor(MCONTACT hContact, MEVENT hDbEvent)
+{
+ return new CDbxSQLiteEventCursor(hContact, m_db, hDbEvent);
+}
+
+STDMETHODIMP_(DB::EventCursor *) CDbxSQLite::EventCursorRev(MCONTACT hContact, MEVENT hDbEvent)
+{
+ return new CDbxSQLiteEventCursor(hContact, m_db, hDbEvent, true);
+}
+
+CDbxSQLiteEventCursor::CDbxSQLiteEventCursor(MCONTACT _1, sqlite3 *_db, MEVENT hDbEvent, bool reverse) :
+ EventCursor(_1), m_db(_db)
+{
+ if (reverse) {
+ if (!hDbEvent)
+ sqlite3_prepare_v2(m_db, reverse_order_query, -1, &cursor, nullptr);
+ else
+ sqlite3_prepare_v2(m_db, reverse_order_pos_query, -1, &cursor, nullptr);
+ }
+ else {
+ if (!hDbEvent)
+ sqlite3_prepare_v2(m_db, normal_order_query, -1, &cursor, nullptr);
+ else
+ sqlite3_prepare_v2(m_db, normal_order_pos_query, -1, &cursor, nullptr);
+ }
+ sqlite3_bind_int64(cursor, 1, hContact);
+ if (hDbEvent)
+ sqlite3_bind_int64(cursor, 2, hDbEvent);
+}
+
+CDbxSQLiteEventCursor::~CDbxSQLiteEventCursor()
+{
+ if (cursor)
+ sqlite3_reset(cursor);
+}
+
+MEVENT CDbxSQLiteEventCursor::FetchNext()
+{
+ if (!cursor)
+ return 0;
+
+ int rc = sqlite3_step(cursor);
+ logError(rc, __FILE__, __LINE__);
+ if (rc != SQLITE_ROW) {
+ //empty response
+ //reset sql cursor
+ sqlite3_reset(cursor);
+ cursor = nullptr;
+ return 0;
+ }
+ return sqlite3_column_int64(cursor, 0);
+}
|