Update APIs to support custom authentication management.

wso2 · Dec 2, 2024 · 9097241 · 9097241
1 parent 660f2fd
commit 9097241
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/...rbon/identity/api/server/authenticators/v1/core/ServerAuthenticatorManagementService.java b/...rbon/identity/api/server/authenticators/v1/core/ServerAuthenticatorManagementService.java
@@ -115,7 +115,13 @@ public List<Authenticator> getAuthenticators(String filter, Integer limit, Integ
             LocalAuthenticatorConfig[] localAuthenticatorConfigs = AuthenticatorsServiceHolder.getInstance()
                     .getApplicationManagementService().getAllLocalAuthenticators(ContextLoader
                             .getTenantDomainFromContext());
-
+            List<UserDefinedLocalAuthenticatorConfig> userDefinedLocalAuthConfigs =
+                    AuthenticatorsServiceHolder.getInstance().getApplicationAuthenticatorService()
+                    .getUserDefinedLocalAuthenticators(ContextLoader.getTenantDomainFromContext());
+            if (CollectionUtils.isNotEmpty(userDefinedLocalAuthConfigs)) {
+                localAuthenticatorConfigs = (LocalAuthenticatorConfig[]) ArrayUtils.addAll(localAuthenticatorConfigs,
+                        userDefinedLocalAuthConfigs.toArray(new LocalAuthenticatorConfig[0]));
+            }
             int localAuthenticatorsCount = localAuthenticatorConfigs.length;
             RequestPathAuthenticatorConfig[] requestPathAuthenticatorConfigs = new RequestPathAuthenticatorConfig[0];
 
@@ -155,6 +161,8 @@ public List<Authenticator> getAuthenticators(String filter, Integer limit, Integ
                     null);
         } catch (IdentityProviderManagementException e) {
             throw handleIdPException(e, Constants.ErrorMessage.ERROR_CODE_ERROR_LISTING_IDPS, null);
+        } catch (AuthenticatorMgtException e) {
+            throw new RuntimeException(e);
         }
     }
 
@@ -1002,7 +1010,7 @@ private APIError handleAuthenticatorException(AuthenticatorMgtException e) {
                                 errorCode : Constants.AUTHENTICATOR_ERROR_PREFIX + errorCode;
                 errorResponse.setCode(errorCode);
             }
-            errorResponse.setDescription(e.getMessage());
+            errorResponse.setDescription(e.getDescription());
             status = Response.Status.BAD_REQUEST;
         } else if (e instanceof AuthenticatorMgtServerException) {
             if (e.getErrorCode() != null) {
@@ -1012,7 +1020,7 @@ private APIError handleAuthenticatorException(AuthenticatorMgtException e) {
                                 errorCode : Constants.AUTHENTICATOR_ERROR_PREFIX + errorCode;
                 errorResponse.setCode(errorCode);
             }
-            errorResponse.setDescription(e.getMessage());
+            errorResponse.setDescription(e.getDescription());
             status = Response.Status.INTERNAL_SERVER_ERROR;
         } else {
             status = Response.Status.INTERNAL_SERVER_ERROR;

diff --git a/...on/identity/api/server/authenticators/v1/impl/LocalAuthenticatorConfigBuilderFactory.java b/...on/identity/api/server/authenticators/v1/impl/LocalAuthenticatorConfigBuilderFactory.java
@@ -123,9 +123,9 @@ private static UserDefinedAuthenticatorEndpointConfig buildEndpointConfig(Endpoi
                     .entrySet().stream().collect(Collectors.toMap(
                             Map.Entry::getKey, entry -> entry.getValue().toString())));
             return endpointConfigBuilder.build();
-        } catch (NoSuchElementException e) {
+        } catch (NoSuchElementException | IllegalArgumentException e) {
             Constants.ErrorMessage error = ERROR_CODE_INVALID_ENDPOINT_CONFIG;
-            throw new AuthenticatorMgtClientException(error.getCode(), error.getMessage(), error.getMessage());
+            throw new AuthenticatorMgtClientException(error.getCode(), error.getMessage(), e.getMessage());
         }
     }