Fix multiple providers

Author: BennuFire

src/application/ApplicationActions.ts

@@ -178,6 +178,12 @@ export const setSSOEnabled = (enabled: boolean, discoveryUrl: string) => ({
   payload: { enabled, discoveryUrl },
 });
 
+export const SET_SSO_PROVIDERS = 'APPLICATION/SET_SSO_PROVIDERS';
+export const setSSOProviders = (providers: []) => ({
+  type: SET_SSO_PROVIDERS,
+  payload: { providers },
+});
+
 export const SET_WAIT_FOR_SSO = 'APPLICATION/SET_WAIT_FOR_SSO';
 export const setWaitForSSO = (wait: boolean) => ({
   type: SET_WAIT_FOR_SSO,

src/application/ApplicationReducer.ts

@@ -21,6 +21,7 @@ import {
   SET_SESSION_PARAMETERS,
   SET_SHARE_DETAILS_FROM_URL,
   SET_SSO_ENABLED,
+  SET_SSO_PROVIDERS,
   SET_STANDALONE_DASHBOARD_DATEBASE,
   SET_STANDALONE_ENABLED,
   SET_STANDALONE_MODE,
@@ -112,6 +113,11 @@ export const applicationReducer = (state = initialState, action: { type: any; pa
       state = update(state, { ssoEnabled: enabled, ssoDiscoveryUrl: discoveryUrl });
       return state;
     }
+    case SET_SSO_PROVIDERS: {
+      const { providers } = payload;
+      state = update(state, { ssoProviders: providers });
+      return state;
+    }
     case SET_WAIT_FOR_SSO: {
       const { wait } = payload;
       state = update(state, { waitForSSO: wait });

src/application/ApplicationSelectors.ts

@@ -64,6 +64,7 @@ export const applicationHasReportHelpModalOpen = (state: any) => {
 export const applicationGetSsoSettings = (state: any) => {
   return {
     ssoEnabled: state.application.ssoEnabled,
+    ssoProviders: state.application.ssoProviders,
     ssoDiscoveryUrl: state.application.ssoDiscoveryUrl,
     cachedSSODiscoveryUrl: state.application.cachedSSODiscoveryUrl,
   };

src/application/ApplicationThunks.ts

@@ -30,6 +30,7 @@ import {
   clearDesktopConnectionProperties,
   clearNotification,
   setSSOEnabled,
+  setSSOProviders,
   setStandaloneEnabled,
   setAboutModalOpen,
   setStandaloneMode,
@@ -340,6 +341,7 @@ export const onConfirmLoadSharedDashboardThunk = () => (dispatch: any, getState:
 export const loadApplicationConfigThunk = () => async (dispatch: any, getState: any) => {
   let config = {
     ssoEnabled: false,
+    ssoProviders: [],
     ssoDiscoveryUrl: 'http://example.com',
     standalone: false,
     standaloneProtocol: 'neo4j',
@@ -377,6 +379,7 @@ export const loadApplicationConfigThunk = () => async (dispatch: any, getState:
     }
     const state = getState();
     dispatch(setSSOEnabled(config.ssoEnabled, state.application.cachedSSODiscoveryUrl));
+    dispatch(setSSOProviders(config.ssoProviders));
 
     const { standalone } = config;
     dispatch(

src/component/sso/SSOLoginButton.tsx

@@ -4,14 +4,23 @@ import { getDiscoveryDataInfo } from './SSOUtils';
 import { ShieldCheckIconOutline } from '@neo4j-ndl/react/icons';
 import { Button, IconButton } from '@neo4j-ndl/react';
 
-export const SSOLoginButton = ({ discoveryAPIUrl, hostname, port, onSSOAttempt, onClick }) => {
+export const SSOLoginButton = ({ discoveryAPIUrl, hostname, port, onSSOAttempt, onClick, providers }) => {
   const [savedSSOProviders, setSSOProviders] = useState([]);
   const [discoveryUrlValidated, setDiscoveryUrlValidated] = useState(undefined);
+
+  const filterByProvidersList = (discoveredProviders, validProviders) => {
+    if (validProviders.length == 0) {
+      return discoveredProviders;
+    }
+    return validProviders.length == 0
+      ? discoveredProviders
+      : discoveredProviders.filter((p) => validProviders.includes(p.id));
+  };
   const attemptManualSSOProviderRetrieval = () => {
     // Do an extra check to see if the hostname provides some SSO provider configuration.
     getDiscoveryDataInfo(`https://${hostname}:${port}`)
       .then((mergedSSOProviders) => {
-        setSSOProviders(mergedSSOProviders);
+        setSSOProviders(filterByProvidersList(mergedSSOProviders, providers));
         if (mergedSSOProviders.length == 0) {
           setDiscoveryUrlValidated(undefined);
         } else {
@@ -26,7 +35,7 @@ export const SSOLoginButton = ({ discoveryAPIUrl, hostname, port, onSSOAttempt,
     // First, try to get the SSO discovery URL from the config.json configuration file and see if it contains anything.
     getDiscoveryDataInfo(discoveryAPIUrl)
       .then((mergedSSOProviders) => {
-        setSSOProviders(mergedSSOProviders);
+        setSSOProviders(filterByProvidersList(mergedSSOProviders, providers));
         if (mergedSSOProviders.length == 0) {
           attemptManualSSOProviderRetrieval();
         } else {

src/modal/ConnectionModal.tsx

@@ -186,6 +186,7 @@ export default function NeoConnectionModal({
                   // Remember credentials on click
                   setConnectionProperties(protocol, url, port, database, '', '');
                 }}
+                providers={ssoSettings.ssoProviders}
               />
             ) : (
               <Button