From bb7fca059242e209414c470a1d6352c0a357fdc9 Mon Sep 17 00:00:00 2001
From: Bruno Windels <brunow@matrix.org>
Date: Mon, 20 Apr 2020 22:26:04 +0200
Subject: [PATCH] support deleting the session from the container

---
 src/matrix/SessionContainer.js    | 24 +++++++++++++++++++++---
 src/matrix/storage/idb/Storage.js |  4 ++++
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/src/matrix/SessionContainer.js b/src/matrix/SessionContainer.js
index b907cc7e..504d1034 100644
--- a/src/matrix/SessionContainer.js
+++ b/src/matrix/SessionContainer.js
@@ -39,6 +39,8 @@ export class SessionContainer {
         this._reconnector = null;
         this._session = null;
         this._sync = null;
+        this._sessionId = null;
+        this._storage = null;
     }
 
     _createNewSessionId() {
@@ -119,17 +121,18 @@ export class SessionContainer {
             request: this._request,
             reconnector: this._reconnector,
         });
-        const storage = await this._storageFactory.create(sessionInfo.id);
+        this._sessionId = sessionInfo.id;
+        this._storage = await this._storageFactory.create(sessionInfo.id);
         // no need to pass access token to session
         const filteredSessionInfo = {
             deviceId: sessionInfo.deviceId,
             userId: sessionInfo.userId,
             homeServer: sessionInfo.homeServer,
         };
-        this._session = new Session({storage, sessionInfo: filteredSessionInfo, hsApi});
+        this._session = new Session({storage: this._storage, sessionInfo: filteredSessionInfo, hsApi});
         await this._session.load();
         
-        this._sync = new Sync({hsApi, storage, session: this._session});
+        this._sync = new Sync({hsApi, storage: this._storage, session: this._session});
         // notify sync and session when back online
         this._reconnectSubscription = this._reconnector.connectionStatus.subscribe(state => {
             if (state === ConnectionStatus.Online) {
@@ -206,6 +209,21 @@ export class SessionContainer {
             this._waitForFirstSyncHandle.dispose();
             this._waitForFirstSyncHandle = null;
         }
+        if (this._storage) {
+            this._storage.close();
+        }
+    }
+
+    async deleteSession() {
+        if (this._sessionId) {
+            // if one fails, don't block the other from trying
+            // also, run in parallel
+            await Promise.all([
+                this._storageFactory.delete(this._sessionId),
+                this._sessionInfoStorage.delete(this._sessionId),
+            ]);
+            this._sessionId = null;
+        }
     }
 }
 
diff --git a/src/matrix/storage/idb/Storage.js b/src/matrix/storage/idb/Storage.js
index 6db4b1c1..812e1cc8 100644
--- a/src/matrix/storage/idb/Storage.js
+++ b/src/matrix/storage/idb/Storage.js
@@ -37,4 +37,8 @@ export class Storage {
             throw new StorageError("readWriteTxn failed", err);
         }
     }
+
+    close() {
+        this._db.close();
+    }
 }