1 files changed, 43 insertions, 0 deletions
diff --git a/lib/database.h b/lib/database.h
new file mode 100644
index 00000000..96256a55
--- /dev/null
+++ b/lib/database.h
@@ -0,0 +1,43 @@
+// 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 "e2ee/e2ee.h"
+
+namespace Quotient {
+class Database : public QObject
+{
+    Q_OBJECT
+
+public:
+    Database(const QString& matrixId, 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);
+    UnorderedMap<QString, std::vector<QOlmSessionPtr>> loadOlmSessions(const PicklingMode& picklingMode);
+    UnorderedMap<std::pair<QString, QString>, QOlmInboundGroupSessionPtr> loadMegolmSessions(const QString& roomId, const PicklingMode& picklingMode);
+    void saveMegolmSession(const QString& roomId, const QString& senderKey, const QString& sessionKey, const QByteArray& pickle);
+    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);
+
+private:
+    void migrateTo1();
+    QString m_matrixId;
+};
+}