From e424293293692e781ec724b211dcb9ec7dd0cc64 Mon Sep 17 00:00:00 2001
From: RMidhunSuresh <rmidhunsuresh@gmail.com>
Date: Wed, 11 Aug 2021 17:13:49 +0530
Subject: [PATCH] Save homeserver before redirecting

Signed-off-by: RMidhunSuresh <rmidhunsuresh@gmail.com>
---
 src/domain/LoginViewModel.js | 132 +++++++++++++++++++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 src/domain/LoginViewModel.js

diff --git a/src/domain/LoginViewModel.js b/src/domain/LoginViewModel.js
new file mode 100644
index 00000000..a33b4c46
--- /dev/null
+++ b/src/domain/LoginViewModel.js
@@ -0,0 +1,132 @@
+/*
+Copyright 2020 Bruno Windels <bruno@windels.cloud>
+
+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.
+*/
+
+import {ViewModel} from "./ViewModel.js";
+import {SessionLoadViewModel} from "./SessionLoadViewModel.js";
+
+function normalizeHomeserver(homeServer) {
+    try {
+        return new URL(homeServer).origin;
+    } catch (err) {
+        return new URL(`https://${homeServer}`).origin;
+    }
+}
+
+export class LoginViewModel extends ViewModel {
+    constructor(options) {
+        super(options);
+        const {ready, defaultHomeServer, createSessionContainer} = options;
+        this._createSessionContainer = createSessionContainer;
+        this._ready = ready;
+        this._defaultHomeServer = defaultHomeServer;
+        this._sessionContainer = null;
+        this._loadViewModel = null;
+        this._loadViewModelSubscription = null;
+        this._supportsSSOLogin = false;
+        this.queryLogin();
+    }
+
+    get defaultHomeServer() { return this._defaultHomeServer; }
+
+    get loadViewModel() {return this._loadViewModel; }
+
+    get isBusy() {
+        if (!this._loadViewModel) {
+            return false;
+        } else {
+            return this._loadViewModel.loading;
+        }
+    }
+
+    async queryLogin(homeServer = this.defaultHomeServer) {
+        // See if we support SSO, if so shows SSO link
+        /* For this, we'd need to poll queryLogin before we do login()
+        */
+        this._supportsSSOLogin = false;
+        this.emitChange("supportsSSOLogin");
+        if (!this._sessionContainer) {
+            this._sessionContainer = this._createSessionContainer();
+        }
+        const normalizedHS = normalizeHomeserver(homeServer);
+        try {
+            this.loginOptions = await this._sessionContainer.queryLogin(normalizedHS);
+            this._supportsSSOLogin = !!this.loginOptions.sso;
+        }
+        catch (e) {
+            // Something went wrong, assume SSO is not supported
+            console.error("Could not query login methods supported by the homeserver");
+        }
+        this.emitChange("supportsSSOLogin");
+    }
+
+    async login(username, password, homeserver) {
+        homeserver = normalizeHomeserver(homeserver);
+        this._loadViewModelSubscription = this.disposeTracked(this._loadViewModelSubscription);
+        if (this._loadViewModel) {
+            this._loadViewModel = this.disposeTracked(this._loadViewModel);
+        }
+        this._loadViewModel = this.track(new SessionLoadViewModel(this.childOptions({
+            createAndStartSessionContainer: async () => {
+                if (this.loginOptions.password) {
+                    this._sessionContainer.startWithLogin(this.loginOptions.password(username, password));
+                }
+                return this._sessionContainer;
+            },
+            ready: sessionContainer => {
+                // make sure we don't delete the session in dispose when navigating away
+                this._sessionContainer = null;
+                this._ready(sessionContainer);
+            },
+            homeserver,
+        })));
+        this._loadViewModel.start();
+        this.emitChange("loadViewModel");
+        this._loadViewModelSubscription = this.track(this._loadViewModel.disposableOn("change", () => {
+            if (!this._loadViewModel.loading) {
+                this._loadViewModelSubscription = this.disposeTracked(this._loadViewModelSubscription);
+            }
+            this.emitChange("isBusy");
+        }));
+    }
+
+    getSSOLink(homeserver) {
+        const link = `${homeserver}/_matrix/client/r0/login/sso/redirect?redirectUrl=${this.urlCreator.createSSOCallbackURL()}`;
+        return link;
+    }
+
+    async startSSOLogin(homeserver) {
+        const hs = normalizeHomeserver(homeserver);
+        await this.platform.settingsStorage.setString("homeserver", hs);
+        this.platform.openUrl(this.getSSOLink(hs));
+    }
+
+    get cancelUrl() {
+        return this.urlCreator.urlForSegment("session");
+    }
+
+    get supportsSSOLogin() {
+        return this._supportsSSOLogin;
+    }
+
+    dispose() {
+        super.dispose();
+        if (this._sessionContainer) {
+            // if we move away before we're done with initial sync
+            // delete the session
+            this._sessionContainer.deleteSession();
+        }
+    }
+}