1 files changed, 81 insertions, 0 deletions
diff --git a/lib/database.h b/lib/database.h
new file mode 100644
index 00000000..8a133f8e
--- /dev/null
+++ b/lib/database.h
@@ -0,0 +1,81 @@
+// SPDX-FileCopyrightText: 2021 Tobias Fella <fella@posteo.de>
+// SPDX-License-Identifier: LGPL-2.1-or-later
+
+#pragma once
+
+#include <QtCore/QObject>
+#include <QtSql/QSqlQuery>
+#include <QtCore/QVector>
+
+#include <QtCore/QHash>
+
+#include "e2ee/e2ee.h"
+
+namespace Quotient {
+
+class QUOTIENT_API Database : public QObject
+{
+    Q_OBJECT
+public:
+    Database(const QString& matrixId, const QString& deviceId, QObject* parent);
+
+    int version();
+    void transaction();
+    void commit();
+    QSqlQuery execute(const QString &queryString);
+    QSqlQuery execute(QSqlQuery &query);
+    QSqlDatabase database();
+    QSqlQuery prepareQuery(const QString& quaryString);
+
+    QByteArray accountPickle();
+    void setAccountPickle(const QByteArray &pickle);
+    void clear();
+    void saveOlmSession(const QString& senderKey, const QString& sessionId,
+                        const QByteArray& pickle, const QDateTime& timestamp);
+    UnorderedMap<QString, std::vector<QOlmSessionPtr>> loadOlmSessions(
+        const PicklingMode& picklingMode);
+    UnorderedMap<QString, QOlmInboundGroupSessionPtr> loadMegolmSessions(
+        const QString& roomId, const PicklingMode& picklingMode);
+    void saveMegolmSession(const QString& roomId, const QString& sessionId,
+                           const QByteArray& pickle, const QString& senderId,
+                           const QString& olmSessionId);
+    void addGroupSessionIndexRecord(const QString& roomId,
+                                    const QString& sessionId, uint32_t index,
+                                    const QString& eventId, qint64 ts);
+    std::pair<QString, qint64> groupSessionIndexRecord(const QString& roomId,
+                                                       const QString& sessionId,
+                                                       qint64 index);
+    void clearRoomData(const QString& roomId);
+    void setOlmSessionLastReceived(const QString& sessionId,
+                                   const QDateTime& timestamp);
+    QOlmOutboundGroupSessionPtr loadCurrentOutboundMegolmSession(
+        const QString& roomId, const PicklingMode& picklingMode);
+    void saveCurrentOutboundMegolmSession(
+        const QString& roomId, const PicklingMode& picklingMode,
+        const QOlmOutboundGroupSession& session);
+    void updateOlmSession(const QString& senderKey, const QString& sessionId,
+                          const QByteArray& pickle);
+
+    // Returns a map UserId -> [DeviceId] that have not received key yet
+    QMultiHash<QString, QString> devicesWithoutKey(
+        const QString& roomId, QMultiHash<QString, QString> devices,
+        const QString& sessionId);
+    // 'devices' contains tuples {userId, deviceId, curveKey}
+    void setDevicesReceivedKey(
+        const QString& roomId,
+        const QVector<std::tuple<QString, QString, QString>>& devices,
+        const QString& sessionId, int index);
+
+    bool isSessionVerified(const QString& edKey);
+    void setSessionVerified(const QString& edKeyId);
+
+private:
+    void migrateTo1();
+    void migrateTo2();
+    void migrateTo3();
+    void migrateTo4();
+    void migrateTo5();
+
+    QString m_matrixId;
+};
+} // namespace Quotient