forked from community/device-mgt-core
parent
a45820e845
commit
438b7b64af
@ -0,0 +1,116 @@
|
||||
/*
|
||||
* Copyright (c) 2018 - 2024, 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 io.entgra.device.mgt.core.apimgt.webapp.publisher.util;
|
||||
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.PublisherRESTAPIServices;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.dto.APIApplicationKey;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.dto.APIInfo.Scope;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.dto.AccessTokenInfo;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.exceptions.APIServicesException;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.exceptions.BadRequestException;
|
||||
import io.entgra.device.mgt.core.apimgt.extension.rest.api.exceptions.UnexpectedResponseException;
|
||||
import io.entgra.device.mgt.core.apimgt.webapp.publisher.exception.APIManagerPublisherException;
|
||||
import io.entgra.device.mgt.core.apimgt.webapp.publisher.internal.APIPublisherDataHolder;
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
public class SelfSyncingScopeTree {
|
||||
private static final Log log = LogFactory.getLog(SelfSyncingScopeTree.class);
|
||||
private static final PublisherRESTAPIServices publisherRESTAPIServices =
|
||||
APIPublisherDataHolder.getInstance().getPublisherRESTAPIServices();
|
||||
private final APIApplicationKey apiApplicationKey;
|
||||
private final AccessTokenInfo accessTokenInfo;
|
||||
private final Map<String, SelfSyncingScopeTree> childTrees = new ConcurrentHashMap<>();
|
||||
private String pathKey;
|
||||
private Scope scope;
|
||||
|
||||
public SelfSyncingScopeTree(Scope scope, String permission, APIApplicationKey apiApplicationKey,
|
||||
AccessTokenInfo accessTokenInfo) throws APIManagerPublisherException {
|
||||
this.apiApplicationKey = apiApplicationKey;
|
||||
this.accessTokenInfo = accessTokenInfo;
|
||||
add(scope, permission);
|
||||
}
|
||||
|
||||
public void add(Scope scope, String permission) throws APIManagerPublisherException {
|
||||
List<String> pathKeys = Arrays.stream(StringUtils.split(permission, "/")).collect(Collectors.toList());
|
||||
|
||||
if (pathKeys.size() == 1) {
|
||||
this.pathKey = pathKeys.get(0);
|
||||
|
||||
if (this.scope == null) {
|
||||
this.scope = scope;
|
||||
} else {
|
||||
this.update(scope);
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if (this.pathKey == null) {
|
||||
this.pathKey = pathKeys.get(0);
|
||||
}
|
||||
|
||||
pathKeys.remove(0);
|
||||
|
||||
if (childTrees.containsKey(pathKeys.get(0))) {
|
||||
childTrees.get(pathKeys.get(0)).add(scope, String.join("/", pathKeys));
|
||||
} else {
|
||||
childTrees.put(pathKeys.get(0), new SelfSyncingScopeTree(scope, String.join("/", pathKeys),
|
||||
apiApplicationKey, accessTokenInfo));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public Scope findScope(String permission) {
|
||||
if (this.scope != null) {
|
||||
return this.scope;
|
||||
}
|
||||
|
||||
List<String> pathKeys = Arrays.stream(StringUtils.split(permission, "/")).collect(Collectors.toList());
|
||||
pathKeys.remove(0);
|
||||
return childTrees.get(pathKeys.get(0)).findScope(String.join("/", pathKeys));
|
||||
}
|
||||
|
||||
private void update(Scope scope) throws APIManagerPublisherException {
|
||||
try {
|
||||
if (publisherRESTAPIServices.isSharedScopeNameExists(apiApplicationKey, accessTokenInfo,
|
||||
this.scope.getName())) {
|
||||
if (publisherRESTAPIServices.updateSharedScope(apiApplicationKey, accessTokenInfo, scope)) {
|
||||
this.scope = scope;
|
||||
}
|
||||
} else {
|
||||
log.warn("Found a scope which is not reflect in the APIM end. Scope [ " + scope.getName() + " ]");
|
||||
}
|
||||
} catch (APIServicesException | BadRequestException | UnexpectedResponseException e) {
|
||||
String msg = "Error encountered while updating the scope [ " + scope.getName() + "]";
|
||||
log.error(msg, e);
|
||||
throw new APIManagerPublisherException(msg, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue