From dcba6d1500e13492240c90d35eada669210544d4 Mon Sep 17 00:00:00 2001
From: RMidhunSuresh <hi@midhun.dev>
Date: Tue, 11 Oct 2022 18:05:54 +0530
Subject: [PATCH] Split method

---
 src/matrix/Client.js | 48 ++++++++++++++++++++++++++++++++++----------
 1 file changed, 37 insertions(+), 11 deletions(-)

diff --git a/src/matrix/Client.js b/src/matrix/Client.js
index f81b9085..2670ce07 100644
--- a/src/matrix/Client.js
+++ b/src/matrix/Client.js
@@ -137,7 +137,7 @@ export class Client {
     async startRegistration(homeserver, username, password, initialDeviceDisplayName, flowSelector) {
         const request = this._platform.request;
         const hsApi = new HomeServerApi({homeserver, request});
-        const registration = new Registration(hsApi, {
+        const registration = new Registration(homeserver, hsApi, {
             username,
             password,
             initialDeviceDisplayName,
@@ -146,6 +146,21 @@ export class Client {
         return registration;
     }
 
+    async startWithFinishedRegistration(registration) {
+        this._platform.logger.run("startWithFinishedRegistration", async (log) => {
+            const sessionInfo = registration.sessionInfo;
+            if (!sessionInfo) {
+                throw new Error("Registration.sessionInfo is not available; are you sure that registration is finished?");
+            }
+            await this.startWithSessionInfo({
+                accessToken: sessionInfo.access_token,
+                deviceId: sessionInfo.device_id,
+                userId: sessionInfo.user_id,
+                homeserver: registration.homeserver,
+            }, true, log);
+        });
+    }
+
     async startWithLogin(loginMethod, {inspectAccountSetup} = {}) {
         const currentStatus = this._status.get();
         if (currentStatus !== LoadStatus.LoginFailed &&
@@ -156,23 +171,17 @@ export class Client {
         this._resetStatus();
         await this._platform.logger.run("login", async log => {
             this._status.set(LoadStatus.Login);
-            const clock = this._platform.clock;
             let sessionInfo;
             try {
                 const request = this._platform.request;
                 const hsApi = new HomeServerApi({homeserver: loginMethod.homeserver, request});
                 const loginData = await loginMethod.login(hsApi, "Hydrogen", log);
-                const sessionId = this.createNewSessionId();
                 sessionInfo = {
-                    id: sessionId,
                     deviceId: loginData.device_id,
                     userId: loginData.user_id,
-                    homeServer: loginMethod.homeserver, // deprecate this over time
                     homeserver: loginMethod.homeserver,
                     accessToken: loginData.access_token,
-                    lastUsed: clock.now()
                 };
-                log.set("id", sessionId);
             } catch (err) {
                 this._error = err;
                 if (err.name === "HomeServerError") {
@@ -191,9 +200,26 @@ export class Client {
                 }
                 return;
             }
+            await this.startWithSessionInfo(sessionInfo, inspectAccountSetup, log);
+        });
+    }
+
+    async startWithSessionInfo({deviceId, userId, accessToken, homeserver}, inspectAccountSetup, log) {
+        await log.wrap("startWithSessionInfo", async (l) => {
+            const id = this.createNewSessionId();
+            const lastUsed = this._platform.clock.now();
+            const sessionInfo = {
+                id,
+                deviceId,
+                userId,
+                homeServer: homeserver, // deprecate this over time
+                homeserver,
+                accessToken,
+                lastUsed,
+            };
             let dehydratedDevice;
             if (inspectAccountSetup) {
-                dehydratedDevice = await this._inspectAccountAfterLogin(sessionInfo, log);
+                dehydratedDevice = await this._inspectAccountAfterLogin(sessionInfo, l);
                 if (dehydratedDevice) {
                     sessionInfo.deviceId = dehydratedDevice.deviceId;
                 }
@@ -203,10 +229,10 @@ export class Client {
             // LoadStatus.Error in case of an error,
             // so separate try/catch
             try {
-                await this._loadSessionInfo(sessionInfo, dehydratedDevice, log);
-                log.set("status", this._status.get());
+                await this._loadSessionInfo(sessionInfo, dehydratedDevice, l);
+                l.set("status", this._status.get());
             } catch (err) {
-                log.catch(err);
+                l.catch(err);
                 // free olm Account that might be contained
                 dehydratedDevice?.dispose();
                 this._error = err;