From fdbf59cd78b614e22202c57d99a9955c6c6620b8 Mon Sep 17 00:00:00 2001
From: Tuan Dang <dangtony98@gmail.com>
Date: Fri, 9 Feb 2024 08:31:19 -0800
Subject: [PATCH] Redirect users to SAML portal on change org to SAML org

---
 frontend/src/layouts/AppLayout/AppLayout.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/frontend/src/layouts/AppLayout/AppLayout.tsx b/frontend/src/layouts/AppLayout/AppLayout.tsx
index 796ea04e6..d175afe48 100644
--- a/frontend/src/layouts/AppLayout/AppLayout.tsx
+++ b/frontend/src/layouts/AppLayout/AppLayout.tsx
@@ -314,13 +314,16 @@ export const AppLayout = ({ children }: LayoutProps) => {
                           return (
                           <DropdownMenuItem key={org.id}>
                             <Button
-                              onClick={() => {
+                              onClick={async () => {
                                 if (currentOrg?.id === org.id) return;
                                 
                                 if (org.authEnforced) {
                                   // org has an org-level auth method enabled (e.g. SAML)
                                   // -> logout + redirect to SAML SSO
-                                  logOutUser();
+
+                                  await logout.mutateAsync();
+                                  window.open(`/api/v1/sso/redirect/saml2/organizations/${org.slug}`);
+                                  window.close();
                                   return;
                                 }