@ -35,8 +35,12 @@ import org.wso2.carbon.device.mgt.common.configuration.mgt.AmbiguousConfiguratio
import org.wso2.carbon.device.mgt.common.configuration.mgt.DeviceConfiguration ;
import org.wso2.carbon.device.mgt.common.exceptions.DeviceManagementException ;
import org.wso2.carbon.device.mgt.common.exceptions.DeviceNotFoundException ;
import org.wso2.carbon.device.mgt.common.exceptions.OTPManagementException ;
import org.wso2.carbon.device.mgt.common.general.TenantDetail ;
import org.wso2.carbon.device.mgt.common.otp.mgt.OTPEmailTypes ;
import org.wso2.carbon.device.mgt.common.otp.mgt.dto.OneTimePinDTO ;
import org.wso2.carbon.device.mgt.common.permission.mgt.PermissionManagementException ;
import org.wso2.carbon.device.mgt.common.spi.OTPManagementService ;
import org.wso2.carbon.device.mgt.core.DeviceManagementConstants ;
import org.wso2.carbon.device.mgt.core.config.DeviceConfigurationManager ;
import org.wso2.carbon.device.mgt.core.config.DeviceManagementConfig ;
@ -77,7 +81,8 @@ public class DeviceManagementConfigServiceImpl implements DeviceManagementConfig
@Path ( "/configurations" )
@Produces ( MediaType . APPLICATION_JSON )
public Response getConfiguration ( @HeaderParam ( "token" ) String token ,
@QueryParam ( "properties" ) String properties ) {
@QueryParam ( "properties" ) String properties ,
@QueryParam ( "withAccessToken" ) boolean withAccessToken ) {
DeviceManagementProviderService dms = DeviceMgtAPIUtils . getDeviceManagementService ( ) ;
try {
if ( token = = null | | token . isEmpty ( ) ) {
@ -102,7 +107,8 @@ public class DeviceManagementConfigServiceImpl implements DeviceManagementConfig
deviceProps . put ( "token" , token ) ;
DeviceConfiguration devicesConfiguration =
dms . getDeviceConfiguration ( deviceProps ) ;
setAccessTokenToDeviceConfigurations ( devicesConfiguration ) ;
if ( withAccessToken ) setAccessTokenToDeviceConfigurations ( devicesConfiguration ) ;
else setOTPTokenToDeviceConfigurations ( devicesConfiguration ) ;
return Response . status ( Response . Status . OK ) . entity ( devicesConfiguration ) . build ( ) ;
} catch ( DeviceManagementException e ) {
String msg = "Error occurred while retrieving configurations" ;
@ -214,6 +220,33 @@ public class DeviceManagementConfigServiceImpl implements DeviceManagementConfig
}
}
private void setOTPTokenToDeviceConfigurations ( DeviceConfiguration deviceConfiguration )
throws DeviceManagementException {
OneTimePinDTO oneTimePinData = new OneTimePinDTO ( ) ;
oneTimePinData . setEmail ( OTPEmailTypes . DEVICE_ENROLLMENT . toString ( ) ) ;
oneTimePinData . setEmailType ( OTPEmailTypes . DEVICE_ENROLLMENT . toString ( ) ) ;
oneTimePinData . setUsername ( deviceConfiguration . getDeviceOwner ( ) ) ;
PrivilegedCarbonContext . startTenantFlow ( ) ;
PrivilegedCarbonContext . getThreadLocalCarbonContext ( ) . setTenantDomain (
deviceConfiguration . getTenantDomain ( ) , true ) ;
oneTimePinData . setTenantId ( PrivilegedCarbonContext . getThreadLocalCarbonContext ( ) . getTenantId ( ) ) ;
PrivilegedCarbonContext . endTenantFlow ( ) ;
OTPManagementService otpManagementService = DeviceMgtAPIUtils . getOtpManagementService ( ) ;
try {
OneTimePinDTO oneTimePinDTO = otpManagementService . generateOneTimePin ( oneTimePinData , true ) ;
if ( oneTimePinDTO = = null ) {
String msg = "Null value returned when generating OTP token for " + oneTimePinData . getOtpToken ( ) ;
log . error ( msg ) ;
throw new DeviceManagementException ( msg ) ;
}
deviceConfiguration . setAccessToken ( oneTimePinDTO . getOtpToken ( ) ) ;
} catch ( OTPManagementException ex ) {
String msg = "Error occurred while generating one time pin: " + ex . getMessage ( ) ;
log . error ( msg , ex ) ;
throw new DeviceManagementException ( msg , ex ) ;
}
}
@Override
@Path ( "/tenants" )
@GET