From d5929d9ebe6dd561637df57e7c450d58133747e9 Mon Sep 17 00:00:00 2001
From: Bruno Windels <274386+bwindels@users.noreply.github.com>
Date: Thu, 9 Feb 2023 11:53:39 +0100
Subject: [PATCH] make features available in Client and Session

---
 src/domain/login/LoginViewModel.ts | 2 +-
 src/matrix/Client.js               | 5 ++++-
 src/matrix/Session.js              | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/domain/login/LoginViewModel.ts b/src/domain/login/LoginViewModel.ts
index 75d57880..f43361d0 100644
--- a/src/domain/login/LoginViewModel.ts
+++ b/src/domain/login/LoginViewModel.ts
@@ -55,7 +55,7 @@ export class LoginViewModel extends ViewModel<SegmentType, Options> {
         const {ready, defaultHomeserver, loginToken} = options;
         this._ready = ready;
         this._loginToken = loginToken;
-        this._client = new Client(this.platform);
+        this._client = new Client(this.platform, this.features);
         this._homeserver = defaultHomeserver;
         this._initViewModels();
     }
diff --git a/src/matrix/Client.js b/src/matrix/Client.js
index 5b53958b..fabb489b 100644
--- a/src/matrix/Client.js
+++ b/src/matrix/Client.js
@@ -31,6 +31,7 @@ import {TokenLoginMethod} from "./login/TokenLoginMethod";
 import {SSOLoginHelper} from "./login/SSOLoginHelper";
 import {getDehydratedDevice} from "./e2ee/Dehydration.js";
 import {Registration} from "./registration/Registration";
+import {FeatureSet} from "../features";
 
 export const LoadStatus = createEnum(
     "NotLoading",
@@ -53,7 +54,7 @@ export const LoginFailure = createEnum(
 );
 
 export class Client {
-    constructor(platform) {
+    constructor(platform, features = new FeatureSet(0)) {
         this._platform = platform;
         this._sessionStartedByReconnector = false;
         this._status = new ObservableValue(LoadStatus.NotLoading);
@@ -68,6 +69,7 @@ export class Client {
         this._olmPromise = platform.loadOlm();
         this._workerPromise = platform.loadOlmWorker();
         this._accountSetup = undefined;
+        this._features = features;
     }
 
     createNewSessionId() {
@@ -278,6 +280,7 @@ export class Client {
             olmWorker,
             mediaRepository,
             platform: this._platform,
+            features: this._features
         });
         await this._session.load(log);
         if (dehydratedDevice) {
diff --git a/src/matrix/Session.js b/src/matrix/Session.js
index f1b273ff..3bcb0ff7 100644
--- a/src/matrix/Session.js
+++ b/src/matrix/Session.js
@@ -54,7 +54,7 @@ const PUSHER_KEY = "pusher";
 
 export class Session {
     // sessionInfo contains deviceId, userId and homeserver
-    constructor({storage, hsApi, sessionInfo, olm, olmWorker, platform, mediaRepository}) {
+    constructor({storage, hsApi, sessionInfo, olm, olmWorker, platform, mediaRepository, features}) {
         this._platform = platform;
         this._storage = storage;
         this._hsApi = hsApi;