From 929a94dc73c82796dec9f70bac4c5d11427430bb Mon Sep 17 00:00:00 2001
From: Vigneshan <vigneshan@entgra.io>
Date: Wed, 16 Dec 2020 11:55:37 +0530
Subject: [PATCH] Add jwt bearer grant type for sso service providers

---
 .../entgra/ui/request/interceptor/SsoLoginHandler.java | 10 ++++++++--
 .../ui/request/interceptor/util/HandlerConstants.java  |  1 +
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/SsoLoginHandler.java b/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/SsoLoginHandler.java
index 1ea7d030e13..620040c0d7c 100644
--- a/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/SsoLoginHandler.java
+++ b/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/SsoLoginHandler.java
@@ -228,8 +228,14 @@ public class SsoLoginHandler extends HttpServlet {
      */
     private StringEntity constructAppGrantTypeUpdatePayload() {
         JSONObject jsonObject = new JSONObject();
-        jsonObject.put("supportedGrantTypes", new JSONArray(new Object[]{HandlerConstants.CODE_GRANT_TYPE,
-                HandlerConstants.REFRESH_TOKEN_GRANT_TYPE, HandlerConstants.PASSWORD_GRANT_TYPE}));
+        jsonObject.put("supportedGrantTypes",
+                new JSONArray(new Object[]{
+                        HandlerConstants.CODE_GRANT_TYPE,
+                        HandlerConstants.REFRESH_TOKEN_GRANT_TYPE,
+                        HandlerConstants.PASSWORD_GRANT_TYPE,
+                        HandlerConstants.JWT_BEARER_GRANT_TYPE
+                })
+        );
         jsonObject.put(HandlerConstants.CALLBACK_URL_KEY, iotsCoreUrl + baseContextPath + HandlerConstants.SSO_LOGIN_CALLBACK);
         String payload = jsonObject.toString();
         return new StringEntity(payload, ContentType.APPLICATION_JSON);
diff --git a/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/util/HandlerConstants.java b/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/util/HandlerConstants.java
index 8c0e79c2a6d..1efcb84f955 100644
--- a/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/util/HandlerConstants.java
+++ b/components/ui-request-interceptor/io.entgra.ui.request.interceptor/src/main/java/io/entgra/ui/request/interceptor/util/HandlerConstants.java
@@ -52,6 +52,7 @@ public class HandlerConstants {
     public static final String CODE_GRANT_TYPE = "authorization_code";
     public static final String REFRESH_TOKEN_GRANT_TYPE = "refresh_token";
     public static final String PASSWORD_GRANT_TYPE = "password";
+    public static final String JWT_BEARER_GRANT_TYPE = "urn:ietf:params:oauth:grant-type:jwt-bearer";
 
     public static final String SCHEME_SEPARATOR = "://";
     public static final String COLON = ":";