insert user improvement for Google EMM

components/mobile-plugins/android-plugin/org.wso2.carbon.device.mgt.mobile.android.api/src/main/java/org/wso2/carbon/mdm/services/android/services/impl/AndroidEnterpriseServiceImpl.java

@@ -99,43 +99,9 @@ public class AndroidEnterpriseServiceImpl implements AndroidEnterpriseService {
                     .build();
         }
 
-        EnterpriseConfigs enterpriseConfigs = AndroidEnterpriseUtils.getEnterpriseConfigs();
         String token;
-        boolean deviceIdExist = false;
-
         try {
-            String googleUserId;
+            token = insertUser(enterpriseUser);
-            List<AndroidEnterpriseUser> androidEnterpriseUsers = AndroidAPIUtils.getAndroidPluginService()
-                    .getEnterpriseUser(CarbonContext.getThreadLocalCarbonContext().getUsername());
-            GoogleAPIInvoker googleAPIInvoker = new GoogleAPIInvoker(enterpriseConfigs.getEsa());
-            if (androidEnterpriseUsers != null && androidEnterpriseUsers.size() > 0) {
-                googleUserId = androidEnterpriseUsers.get(0).getGoogleUserId();
-                // If this device is also present, only need to provide a token for this request.
-                for (AndroidEnterpriseUser enterprise : androidEnterpriseUsers) {
-                    if (enterprise.getEmmDeviceId() != null
-                            && enterprise.getEmmDeviceId().equals(enterpriseUser.getAndroidPlayDeviceId())) {
-                        deviceIdExist = true;
-                    }
-                }
-            } else {
-                googleUserId = googleAPIInvoker.insertUser(enterpriseConfigs.getEnterpriseId(), CarbonContext
-                        .getThreadLocalCarbonContext()
-                        .getUsername());
-            }
-            // Fetching an auth token from Google EMM API
-            token = googleAPIInvoker.getToken(enterpriseConfigs.getEnterpriseId(), googleUserId);
-
-            if (!deviceIdExist) {
-                AndroidEnterpriseUser androidEnterpriseUser = new AndroidEnterpriseUser();
-                androidEnterpriseUser.setEmmUsername(CarbonContext.getThreadLocalCarbonContext().getUsername());
-                androidEnterpriseUser.setTenantId(CarbonContext.getThreadLocalCarbonContext().getTenantId());
-                androidEnterpriseUser.setAndroidPlayDeviceId(enterpriseUser.getAndroidPlayDeviceId());
-                androidEnterpriseUser.setEnterpriseId(enterpriseConfigs.getEnterpriseId());
-                androidEnterpriseUser.setEmmDeviceId(enterpriseUser.getEmmDeviceIdentifier());
-                androidEnterpriseUser.setGoogleUserId(googleUserId);
-
-                AndroidAPIUtils.getAndroidPluginService().addEnterpriseUser(androidEnterpriseUser);
-            }
             if (token == null) {
                 return Response.serverError().entity(
                         new ErrorResponse.ErrorResponseBuilder().setMessage("Error when fetching token").build())
@@ -149,6 +115,48 @@ public class AndroidEnterpriseServiceImpl implements AndroidEnterpriseService {
         return Response.status(Response.Status.OK).entity(token).build();
     }
 
+    public String insertUser(EnterpriseUser enterpriseUser) throws EnterpriseServiceException {
+        EnterpriseConfigs enterpriseConfigs = AndroidEnterpriseUtils.getEnterpriseConfigs();
+        String token;
+        boolean deviceIdExist = false;
+
+        String googleUserId;
+        List<AndroidEnterpriseUser> androidEnterpriseUsers = AndroidAPIUtils.getAndroidPluginService()
+                .getEnterpriseUser(CarbonContext.getThreadLocalCarbonContext().getUsername());
+        GoogleAPIInvoker googleAPIInvoker = new GoogleAPIInvoker(enterpriseConfigs.getEsa());
+        if (androidEnterpriseUsers != null && androidEnterpriseUsers.size() > 0) {
+            googleUserId = androidEnterpriseUsers.get(0).getGoogleUserId();
+            // If this device is also present, only need to provide a token for this request.
+            for (AndroidEnterpriseUser enterprise : androidEnterpriseUsers) {
+                if (enterprise.getEmmDeviceId() != null
+                        && enterprise.getEmmDeviceId().equals(enterpriseUser.getAndroidPlayDeviceId())) {
+                    deviceIdExist = true;
+                }
+            }
+        } else {
+            googleUserId = googleAPIInvoker.insertUser(enterpriseConfigs.getEnterpriseId(), CarbonContext
+                    .getThreadLocalCarbonContext()
+                    .getUsername());
+        }
+        // Fetching an auth token from Google EMM API
+        token = googleAPIInvoker.getToken(enterpriseConfigs.getEnterpriseId(), googleUserId);
+
+        if (!deviceIdExist) {
+            AndroidEnterpriseUser androidEnterpriseUser = new AndroidEnterpriseUser();
+            androidEnterpriseUser.setEmmUsername(CarbonContext.getThreadLocalCarbonContext().getUsername());
+            androidEnterpriseUser.setTenantId(CarbonContext.getThreadLocalCarbonContext().getTenantId());
+            androidEnterpriseUser.setAndroidPlayDeviceId(enterpriseUser.getAndroidPlayDeviceId());
+            androidEnterpriseUser.setEnterpriseId(enterpriseConfigs.getEnterpriseId());
+            androidEnterpriseUser.setEmmDeviceId(enterpriseUser.getEmmDeviceIdentifier());
+            androidEnterpriseUser.setGoogleUserId(googleUserId);
+
+            AndroidAPIUtils.getAndroidPluginService().addEnterpriseUser(androidEnterpriseUser);
+        }
+
+        return token;
+
+    }
+
     @Override
     @Produces(MediaType.APPLICATION_JSON)
     @Consumes(MediaType.APPLICATION_JSON)

components/mobile-plugins/android-plugin/org.wso2.carbon.device.mgt.mobile.android.api/src/main/java/org/wso2/carbon/mdm/services/android/services/impl/DeviceManagementServiceImpl.java

@@ -54,9 +54,11 @@ import org.wso2.carbon.device.mgt.common.policy.mgt.monitor.PolicyComplianceExce
 import org.wso2.carbon.device.mgt.core.device.details.mgt.DeviceDetailsMgtException;
 import org.wso2.carbon.device.mgt.core.device.details.mgt.DeviceInformationManager;
 import org.wso2.carbon.device.mgt.core.operation.mgt.CommandOperation;
+import org.wso2.carbon.device.mgt.mobile.android.impl.EnterpriseServiceException;
 import org.wso2.carbon.mdm.services.android.bean.ErrorResponse;
 import org.wso2.carbon.mdm.services.android.bean.wrapper.AndroidApplication;
 import org.wso2.carbon.mdm.services.android.bean.wrapper.AndroidDevice;
+import org.wso2.carbon.mdm.services.android.bean.wrapper.EnterpriseUser;
 import org.wso2.carbon.mdm.services.android.exception.UnexpectedServerErrorException;
 import org.wso2.carbon.mdm.services.android.services.DeviceManagementService;
 import org.wso2.carbon.mdm.services.android.util.AndroidAPIUtils;
@@ -92,6 +94,8 @@ public class DeviceManagementServiceImpl implements DeviceManagementService {
 
     private static final String OPERATION_ERROR_STATUS = "ERROR";
     private static final Log log = LogFactory.getLog(DeviceManagementServiceImpl.class);
+    public static final String GOOGLE_AFW_EMM_ANDROID_ID = "googleEMMAndroidId";
+    public static final String GOOGLE_AFW_DEVICE_ID = "googleEMMDeviceId";
 
     @PUT
     @Path("/{id}/applications")
@@ -233,6 +237,7 @@ public class DeviceManagementServiceImpl implements DeviceManagementService {
                     new ErrorResponse.ErrorResponseBuilder().setCode(400l).setMessage(errorMessage).build());
         }
         try {
+            String token = null;
             Device device = new Device();
             device.setType(DeviceManagementConstants.MobileDeviceTypes.MOBILE_DEVICE_TYPE_ANDROID);
             device.setEnrolmentInfo(androidDevice.getEnrolmentInfo());
@@ -244,6 +249,27 @@ public class DeviceManagementServiceImpl implements DeviceManagementService {
             device.setFeatures(androidDevice.getFeatures());
             device.setProperties(androidDevice.getProperties());
 
+            String googleEMMAndroidId = null;
+            String googleEMMDeviceId = null;
+            if (androidDevice.getProperties() != null) {
+                for (Device.Property property : androidDevice.getProperties()) {
+                    if (property.getName().equals(GOOGLE_AFW_EMM_ANDROID_ID)) {
+                        googleEMMAndroidId = property.getValue();
+                    } else if (property.getName().equals(GOOGLE_AFW_DEVICE_ID)) {
+                        googleEMMDeviceId = property.getValue();
+                    }
+                }
+
+                if (googleEMMAndroidId != null && googleEMMDeviceId != null) {
+                    EnterpriseUser user = new EnterpriseUser();
+                    user.setAndroidPlayDeviceId(googleEMMAndroidId);
+                    user.setEmmDeviceIdentifier(googleEMMDeviceId);
+                    AndroidEnterpriseServiceImpl enterpriseService = new AndroidEnterpriseServiceImpl();
+                    token = enterpriseService.insertUser(user);
+                }
+            }
+
+
             boolean status = AndroidAPIUtils.getDeviceManagementService().enrollDevice(device);
             if (status) {
                 DeviceIdentifier deviceIdentifier = new DeviceIdentifier(androidDevice.getDeviceIdentifier(),
@@ -301,8 +327,12 @@ public class DeviceManagementServiceImpl implements DeviceManagementService {
 
                 Message responseMessage = new Message();
                 responseMessage.setResponseCode(Response.Status.OK.toString());
-                responseMessage.setResponseMessage("Android device, which carries the id '" +
+                if (token == null) {
-                        androidDevice.getDeviceIdentifier() + "' has successfully been enrolled");
+                    responseMessage.setResponseMessage("Android device, which carries the id '" +
+                            androidDevice.getDeviceIdentifier() + "' has successfully been enrolled");
+                } else {
+                    responseMessage.setResponseMessage("Google response token" + token);
+                }
                 return Response.status(Response.Status.OK).entity(responseMessage).build();
             } else {
                 Message responseMessage = new Message();
@@ -339,6 +369,12 @@ public class DeviceManagementServiceImpl implements DeviceManagementService {
             log.error(msg, e);
             throw new UnexpectedServerErrorException(
                     new ErrorResponse.ErrorResponseBuilder().setCode(500l).setMessage(msg).build());
+        } catch (EnterpriseServiceException e) {
+            String msg = "Error occurred while adding user via Google Apis '" +
+                    androidDevice.getDeviceIdentifier() + "'";
+            log.error(msg, e);
+            throw new UnexpectedServerErrorException(
+                    new ErrorResponse.ErrorResponseBuilder().setCode(500l).setMessage(msg).build());
         }
     }