forked from community/device-mgt-core
Fix registry loading issue when SSO has enabled. Because of that issue device type EULA displaying page is broken4.x.x
tcdlpds@gmail.com
4 years ago
parent
a61bd6b1db
commit
7b41e2cd0d
@ -0,0 +1,121 @@
|
||||
/*
|
||||
* Copyright (c) 2021, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
*
|
||||
* Entgra (Pvt) Ltd. licenses this file to you under the Apache License,
|
||||
* Version 2.0 (the "License"); you may not use this file except
|
||||
* in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing,
|
||||
* software distributed under the License is distributed on an
|
||||
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
* KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
|
||||
package org.wso2.carbon.device.mgt.extensions.license.mgt.meta.data;
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
||||
import org.wso2.carbon.device.mgt.common.exceptions.DeviceManagementException;
|
||||
import org.wso2.carbon.device.mgt.common.exceptions.MetadataKeyAlreadyExistsException;
|
||||
import org.wso2.carbon.device.mgt.common.exceptions.MetadataManagementException;
|
||||
import org.wso2.carbon.device.mgt.common.license.mgt.License;
|
||||
import org.wso2.carbon.device.mgt.common.license.mgt.LicenseManagementException;
|
||||
import org.wso2.carbon.device.mgt.common.license.mgt.LicenseManager;
|
||||
import org.wso2.carbon.device.mgt.common.metadata.mgt.Metadata;
|
||||
import org.wso2.carbon.device.mgt.common.metadata.mgt.MetadataManagementService;
|
||||
import org.wso2.carbon.device.mgt.core.dto.DeviceType;
|
||||
import org.wso2.carbon.device.mgt.core.service.DeviceManagementProviderService;
|
||||
import org.wso2.carbon.device.mgt.core.util.DeviceManagerUtil;
|
||||
import org.wso2.carbon.device.mgt.extensions.device.type.template.util.DeviceTypePluginConstants;
|
||||
import org.wso2.carbon.device.mgt.extensions.internal.DeviceTypeExtensionDataHolder;
|
||||
|
||||
public class MetaRepositoryBasedLicenseManager implements LicenseManager {
|
||||
|
||||
private static final Log log = LogFactory.getLog(MetaRepositoryBasedLicenseManager.class);
|
||||
|
||||
@Override
|
||||
public License getLicense(String deviceType, String languageCode) throws LicenseManagementException {
|
||||
MetadataManagementService metadataManagementService = DeviceTypeExtensionDataHolder.getInstance()
|
||||
.getMetadataManagementService();
|
||||
String licenceKey = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantDomain()
|
||||
+ DeviceTypePluginConstants.UNDERSCORE + deviceType + DeviceTypePluginConstants.LICENCE_META_KEY_SUFFIX
|
||||
+ languageCode;
|
||||
|
||||
try {
|
||||
Metadata metadata = metadataManagementService.retrieveMetadata(licenceKey);
|
||||
if (metadata == null) {
|
||||
DeviceType deviceTypeData = DeviceManagerUtil
|
||||
.getDeviceType(deviceType, PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId());
|
||||
|
||||
DeviceManagementProviderService deviceManagementProviderService = DeviceTypeExtensionDataHolder
|
||||
.getInstance().getDeviceManagementProviderService();
|
||||
|
||||
License license = deviceManagementProviderService.getLicenseConfig(deviceTypeData.getName());
|
||||
|
||||
if (!StringUtils.isBlank(license.getLanguage()) || !StringUtils.isBlank(license.getName())
|
||||
|| !StringUtils.isBlank(license.getText()) || !StringUtils.isBlank(license.getVersion())) {
|
||||
addLicense(deviceTypeData.getName(), license);
|
||||
return license;
|
||||
} else {
|
||||
license = new License();
|
||||
license.setName(deviceType);
|
||||
license.setVersion("1.0.0");
|
||||
license.setLanguage("en_US");
|
||||
license.setText("This is license text");
|
||||
addLicense(deviceTypeData.getName(), license);
|
||||
return license;
|
||||
}
|
||||
}
|
||||
Gson g = new Gson();
|
||||
return g.fromJson(metadata.getMetaValue(), License.class);
|
||||
} catch (MetadataManagementException e) {
|
||||
String msg = "Error occurred while accessing meta data service to store licence data";
|
||||
log.error(msg, e);
|
||||
throw new LicenseManagementException(msg, e);
|
||||
} catch (DeviceManagementException e) {
|
||||
String msg = "Error occurred while getting device details.";
|
||||
log.error(msg, e);
|
||||
throw new LicenseManagementException(msg, e); }
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addLicense(String deviceType, License license) throws LicenseManagementException {
|
||||
|
||||
String languageCode = license.getLanguage();
|
||||
if (StringUtils.isBlank(languageCode)) {
|
||||
languageCode = DeviceTypePluginConstants.LANGUAGE_CODE_ENGLISH_US;
|
||||
}
|
||||
|
||||
String licenceKey = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantDomain()
|
||||
+ DeviceTypePluginConstants.UNDERSCORE + deviceType + DeviceTypePluginConstants.LICENCE_META_KEY_SUFFIX
|
||||
+ languageCode;
|
||||
|
||||
Metadata metadata = new Metadata();
|
||||
metadata.setMetaKey(licenceKey);
|
||||
metadata.setMetaValue(new Gson().toJson(license));
|
||||
|
||||
MetadataManagementService metadataManagementService = DeviceTypeExtensionDataHolder.getInstance()
|
||||
.getMetadataManagementService();
|
||||
try {
|
||||
metadataManagementService.createMetadata(metadata);
|
||||
} catch (MetadataManagementException e) {
|
||||
String msg = "Error occurred while saving the licence value in meta data repository";
|
||||
log.error(msg, e);
|
||||
throw new LicenseManagementException(msg, e);
|
||||
} catch (MetadataKeyAlreadyExistsException e) {
|
||||
String msg =
|
||||
"Error occurred while saving the licence key and licence key exist. Licence Key: " + licenceKey;
|
||||
log.error(msg, e);
|
||||
throw new LicenseManagementException(msg, e);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue