From 51cd59f4364ecda75c2b1dc39104108cc1214c2e Mon Sep 17 00:00:00 2001
From: inoshperera <inoshperera@gmail.com>
Date: Thu, 10 Mar 2022 16:33:09 +0530
Subject: [PATCH] Generate QR with access token for given user

fixes https://gitlab.com/entgra/product-iots/-/issues/1277
---
 .../extension/APIManagementProviderService.java   |  4 +++-
 .../APIManagementProviderServiceImpl.java         | 15 +++++++++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderService.java b/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderService.java
index 7405737237..2507e4f132 100644
--- a/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderService.java
+++ b/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderService.java
@@ -62,11 +62,13 @@ public interface APIManagementProviderService {
      * @param applicationName Application Name
      * @param tokenType Token Type
      * @param validityPeriod Validity Period
+     * @param username Name of the user to create the token. If null, set as carbon context user
      * @return {@link String} Access Token
      * @throws APIManagerException if error occurred while getting the access token for given scopes,
      * validity period etc.
      */
-    AccessTokenInfo getAccessToken(String scopes, String[] tags, String applicationName, String tokenType, String validityPeriod)
+    AccessTokenInfo getAccessToken(String scopes, String[] tags, String applicationName, String
+            tokenType, String validityPeriod, String username)
             throws APIManagerException;
 
 }
diff --git a/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderServiceImpl.java b/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderServiceImpl.java
index 7ba23fc4ce..d6fd929e13 100644
--- a/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderServiceImpl.java
+++ b/components/apimgt-extensions/org.wso2.carbon.apimgt.application.extension/src/main/java/org/wso2/carbon/apimgt/application/extension/APIManagementProviderServiceImpl.java
@@ -233,7 +233,7 @@ public class APIManagementProviderServiceImpl implements APIManagementProviderSe
 
     @Override
     public AccessTokenInfo getAccessToken(String scopes, String[] tags, String applicationName, String tokenType,
-            String validityPeriod) throws APIManagerException {
+            String validityPeriod, String username) throws APIManagerException {
         try {
             String tenantDomain = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantDomain(true);
             ApiApplicationKey clientCredentials = getClientCredentials(tenantDomain, tags, applicationName, tokenType,
@@ -245,15 +245,22 @@ public class APIManagementProviderServiceImpl implements APIManagementProviderSe
                 throw new APIManagerException(msg);
             }
 
-            String user =
-                    PrivilegedCarbonContext.getThreadLocalCarbonContext().getUsername() + "@" + PrivilegedCarbonContext
+            if (username == null || username.isEmpty()) {
+                username =
+                        PrivilegedCarbonContext.getThreadLocalCarbonContext().getUsername() + "@" + PrivilegedCarbonContext
+                                .getThreadLocalCarbonContext().getTenantDomain(true);
+            } else {
+                if (!username.contains("@")) {
+                    username += "@" + PrivilegedCarbonContext
                             .getThreadLocalCarbonContext().getTenantDomain(true);
+                }
+            }
 
             JWTClientManagerService jwtClientManagerService = APIApplicationManagerExtensionDataHolder.getInstance()
                     .getJwtClientManagerService();
             JWTClient jwtClient = jwtClientManagerService.getJWTClient();
             AccessTokenInfo accessTokenForAdmin = jwtClient
-                    .getAccessToken(clientCredentials.getConsumerKey(), clientCredentials.getConsumerSecret(), user,
+                    .getAccessToken(clientCredentials.getConsumerKey(), clientCredentials.getConsumerSecret(), username,
                             scopes);
 
             return accessTokenForAdmin;