From b00865510faed1fd6c2e30142b01dcac8aa9de4b Mon Sep 17 00:00:00 2001
From: Bruno Windels <brunow@matrix.org>
Date: Fri, 11 Sep 2020 14:40:05 +0200
Subject: [PATCH] add operation store

---
 src/matrix/storage/common.js                  |  1 +
 src/matrix/storage/idb/Transaction.js         |  5 ++
 src/matrix/storage/idb/schema.js              |  2 +
 .../storage/idb/stores/OperationStore.js      | 55 +++++++++++++++++++
 4 files changed, 63 insertions(+)
 create mode 100644 src/matrix/storage/idb/stores/OperationStore.js

diff --git a/src/matrix/storage/common.js b/src/matrix/storage/common.js
index 4a4060b2..f74dafdc 100644
--- a/src/matrix/storage/common.js
+++ b/src/matrix/storage/common.js
@@ -28,6 +28,7 @@ export const STORE_NAMES = Object.freeze([
     "inboundGroupSessions",
     "outboundGroupSessions",
     "groupSessionDecryptions",
+    "operations"
 ]);
 
 export const STORE_MAP = Object.freeze(STORE_NAMES.reduce((nameMap, name) => {
diff --git a/src/matrix/storage/idb/Transaction.js b/src/matrix/storage/idb/Transaction.js
index e0982c54..af6d49ca 100644
--- a/src/matrix/storage/idb/Transaction.js
+++ b/src/matrix/storage/idb/Transaction.js
@@ -30,6 +30,7 @@ import {OlmSessionStore} from "./stores/OlmSessionStore.js";
 import {InboundGroupSessionStore} from "./stores/InboundGroupSessionStore.js";
 import {OutboundGroupSessionStore} from "./stores/OutboundGroupSessionStore.js";
 import {GroupSessionDecryptionStore} from "./stores/GroupSessionDecryptionStore.js";
+import {OperationStore} from "./stores/OperationStore.js";
 
 export class Transaction {
     constructor(txn, allowedStoreNames) {
@@ -111,6 +112,10 @@ export class Transaction {
         return this._store("groupSessionDecryptions", idbStore => new GroupSessionDecryptionStore(idbStore));
     }
 
+    get operations() {
+        return this._store("operations", idbStore => new OperationStore(idbStore));
+    }
+
     complete() {
         return txnAsPromise(this._txn);
     }
diff --git a/src/matrix/storage/idb/schema.js b/src/matrix/storage/idb/schema.js
index 1ed9cadb..809f6729 100644
--- a/src/matrix/storage/idb/schema.js
+++ b/src/matrix/storage/idb/schema.js
@@ -74,4 +74,6 @@ function createE2EEStores(db) {
     db.createObjectStore("inboundGroupSessions", {keyPath: "key"});
     db.createObjectStore("outboundGroupSessions", {keyPath: "roomId"});
     db.createObjectStore("groupSessionDecryptions", {keyPath: "key"});
+    const operations = db.createObjectStore("operations", {keyPath: "id"});
+    operations.createIndex("byTypeAndScope", "typeScopeKey", {unique: false});
 }
diff --git a/src/matrix/storage/idb/stores/OperationStore.js b/src/matrix/storage/idb/stores/OperationStore.js
new file mode 100644
index 00000000..598f80e7
--- /dev/null
+++ b/src/matrix/storage/idb/stores/OperationStore.js
@@ -0,0 +1,55 @@
+/*
+Copyright 2020 The Matrix.org Foundation C.I.C.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+function encodeTypeScopeKey(type, scope) {
+    return `${type}|${scope}`;
+}
+
+export class OperationStore {
+    constructor(store) {
+        this._store = store;
+    }
+
+    getAll() {
+        return this._store.selectAll();
+    }
+
+    async getAllByTypeAndScope(type, scope) {
+        const key = encodeTypeScopeKey(type, scope);
+        const results = [];
+        await this._store.index("byTypeAndScope").iterateWhile(key, value => {
+            if (value.typeScopeKey !== key) {
+                return false;
+            }
+            results.push(value);
+            return true;
+        });
+        return results;
+    }
+
+    add(operation) {
+        operation.typeScopeKey = encodeTypeScopeKey(operation.type, operation.scope);
+        this._store.add(operation);
+    }
+
+    update(operation) {
+        this._store.set(operation);
+    }
+
+    remove(id) {
+        this._store.delete(id);
+    }
+}