From 2817f40b82ecb043f681d63bee8d54afbf487d43 Mon Sep 17 00:00:00 2001
From: Denis Bykhov <bykhov.denis@gmail.com>
Date: Mon, 20 Mar 2023 20:04:56 +0600
Subject: [PATCH] Fix redirect url (#2775)

Signed-off-by: Denis Bykhov <bykhov.denis@gmail.com>
---
 .../src/components/LoginForm.svelte           | 24 +++++++++++++++++--
 plugins/login-resources/src/utils.ts          |  8 +++----
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/plugins/login-resources/src/components/LoginForm.svelte b/plugins/login-resources/src/components/LoginForm.svelte
index b327aee8f0..6093dfff8f 100644
--- a/plugins/login-resources/src/components/LoginForm.svelte
+++ b/plugins/login-resources/src/components/LoginForm.svelte
@@ -15,9 +15,9 @@
 -->
 <script lang="ts">
   import { OK, setMetadata, Severity, Status } from '@hcengineering/platform'
-  import { getCurrentLocation, navigate, setMetadataLocalStorage } from '@hcengineering/ui'
+  import { getCurrentLocation, navigate, Location, setMetadataLocalStorage } from '@hcengineering/ui'
 
-  import { doLogin } from '../utils'
+  import { doLogin, getWorkspaces, navigateToWorkspace, selectWorkspace } from '../utils'
   import Form from './Form.svelte'
 
   import login from '../plugin'
@@ -53,6 +53,26 @@
         setMetadata(login.metadata.LoginToken, result.token)
         setMetadataLocalStorage(login.metadata.LoginEndpoint, result.endpoint)
         setMetadataLocalStorage(login.metadata.LoginEmail, result.email)
+
+        if (navigateUrl !== undefined) {
+          try {
+            const loc = JSON.parse(decodeURIComponent(navigateUrl)) as Location
+            const workspace = loc.path[1]
+            if (workspace !== undefined) {
+              const workspaces = await getWorkspaces()
+              if (workspaces.find((p) => p.workspace === workspace) !== undefined) {
+                status = new Status(Severity.INFO, login.status.ConnectingToServer, {})
+
+                const [loginStatus, result] = await selectWorkspace(workspace)
+                status = loginStatus
+                navigateToWorkspace(workspace, result, navigateUrl)
+                return
+              }
+            }
+          } catch (err: any) {
+            // Json parse error could be ignored
+          }
+        }
         const loc = getCurrentLocation()
         loc.path[1] = 'selectWorkspace'
         loc.path.length = 2
diff --git a/plugins/login-resources/src/utils.ts b/plugins/login-resources/src/utils.ts
index fb3d82d4bf..fe046d2b55 100644
--- a/plugins/login-resources/src/utils.ts
+++ b/plugins/login-resources/src/utils.ts
@@ -292,12 +292,10 @@ export function navigateToWorkspace (workspace: string, loginInfo?: WorkspaceLog
   setLoginInfo(loginInfo)
 
   if (navigateUrl !== undefined) {
-    const loc = JSON.parse(decodeURIComponent(navigateUrl)) as Location
     try {
-      const url = JSON.parse(decodeURIComponent(loc.query?.navigateUrl ?? '{}')) as Location
-      if (url.path[1] === workspace) {
-        navigate(url)
-
+      const loc = JSON.parse(decodeURIComponent(navigateUrl)) as Location
+      if (loc.path[1] === workspace) {
+        navigate(loc)
         return
       }
     } catch (err: any) {