ONAP-PDP-REST/src/main/java/org/onap/policy/pdp/rest/api/services/GetConfigService.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP-PDP-REST
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.policy.pdp.rest.api.services;
  21
  22 import java.util.ArrayList;
  23 import java.util.Arrays;
  24 import java.util.Collection;
  25 import java.util.HashMap;
  26 import java.util.HashSet;
  27 import java.util.List;
  28 import java.util.Map;
  29 import java.util.UUID;
  30
  31 import javax.json.Json;
  32 import javax.json.JsonArrayBuilder;
  33 import javax.json.JsonObject;
  34 import javax.json.JsonObjectBuilder;
  35
  36 import org.onap.policy.api.ConfigRequestParameters;
  37 import org.onap.policy.api.PolicyConfigException;
  38 import org.onap.policy.api.PolicyConfigStatus;
  39 import org.onap.policy.api.PolicyConfigType;
  40 import org.onap.policy.common.logging.flexlogger.FlexLogger;
  41 import org.onap.policy.common.logging.flexlogger.Logger;
  42 import org.onap.policy.pdp.rest.api.models.PDPResponse;
  43 import org.onap.policy.pdp.rest.api.models.PolicyConfig;
  44 import org.onap.policy.xacml.api.XACMLErrorConstants;
  45 import org.springframework.http.HttpStatus;
  46
  47 public class GetConfigService {
  48     private static final Logger LOGGER = FlexLogger.getLogger(GetConfigService.class.getName());
  49
  50     private ConfigRequestParameters configRequestParameters = null;
  51     private String message = null;
  52     private HttpStatus responseCode = HttpStatus.BAD_REQUEST;
  53     private Collection<PolicyConfig> policyConfigs = null;
  54     private boolean unique = false;
  55
  56     public GetConfigService(ConfigRequestParameters configRequestParameters,
  57             String requestID){
  58         this.configRequestParameters = configRequestParameters;
  59         if(configRequestParameters.getRequestID()==null){
  60             UUID requestUUID = null;
  61             if (requestID != null && !requestID.isEmpty()) {
  62                 try {
  63                     requestUUID = UUID.fromString(requestID);
  64                 } catch (IllegalArgumentException e) {
  65                     requestUUID = UUID.randomUUID();
  66                     LOGGER.info("Generated Random UUID: " + requestUUID.toString(),e);
  67                 }
  68             }else{
  69                 requestUUID = UUID.randomUUID();
  70                 LOGGER.info("Generated Random UUID: " + requestUUID.toString());
  71             }
  72             this.configRequestParameters.setRequestID(requestUUID);
  73         }
  74         policyConfigs = new ArrayList<>();
  75         try{
  76             run();
  77             specialCheck();
  78         }catch(PolicyConfigException e){
  79                 LOGGER.error(e);
  80             PolicyConfig policyConfig = new PolicyConfig();
  81             policyConfig.setPolicyConfigMessage(e.getMessage());
  82             policyConfig.setPolicyConfigStatus(PolicyConfigStatus.CONFIG_NOT_FOUND);
  83             policyConfigs.add(policyConfig);
  84             responseCode = HttpStatus.BAD_REQUEST;
  85         }
  86     }
  87
  88     private void specialCheck() {
  89         if(policyConfigs==null || policyConfigs.isEmpty()){
  90             responseCode = HttpStatus.BAD_REQUEST;
  91         }else if(policyConfigs.size()==1){
  92             for(PolicyConfig policyConfig: policyConfigs){
  93                 if(policyConfig.getPolicyConfigMessage()!=null && policyConfig.getPolicyConfigMessage().contains("PE300")){
  94                     responseCode = HttpStatus.BAD_REQUEST;
  95                 }
  96             }
  97         }else{
  98             responseCode = HttpStatus.OK;
  99         }
 100     }
 101
 102     private void run() throws PolicyConfigException{
 103         // getValidation.
 104         if(!getValidation()){
 105             LOGGER.error(message);
 106             throw new PolicyConfigException(message);
 107         }
 108         // Generate Request.
 109         String modelString = getModel().toString();
 110         LOGGER.debug("Generated JSON Request is: " + modelString);
 111         if(configRequestParameters.getUnique()){
 112             LOGGER.info("Requested for Unique Result only. ");
 113             unique = true;
 114         }
 115         // Process Result.
 116         try {
 117             PDPServices pdpServices = new PDPServices();
 118             responseCode = HttpStatus.OK;
 119             policyConfigs = configResult(pdpServices.generateRequest(modelString, configRequestParameters.getRequestID(), unique, false));
 120             // Filter addition.
 121             policyConfigs = filterResults(policyConfigs, configRequestParameters);
 122         } catch (Exception e) {
 123             LOGGER.error(XACMLErrorConstants.ERROR_DATA_ISSUE + e);
 124             responseCode = HttpStatus.BAD_REQUEST;
 125             throw new PolicyConfigException(XACMLErrorConstants.ERROR_DATA_ISSUE +e);
 126         }
 127     }
 128
 129     private Collection<PolicyConfig> configResult(
 130             Collection<PDPResponse> generateRequest) {
 131         Collection<PolicyConfig> result = new HashSet<>();
 132         if (generateRequest == null) {
 133             return null;
 134         }
 135         if (!generateRequest.isEmpty()) {
 136             for (PDPResponse stdStatus : generateRequest) {
 137                 PolicyConfig policyConfig = new PolicyConfig();
 138                 policyConfig.setConfig(stdStatus.getConfig());
 139                 policyConfig.setMatchingConditions(stdStatus.getMatchingConditions());
 140                 policyConfig.setPolicyConfigMessage(stdStatus.getPolicyConfigMessage());
 141                 policyConfig.setPolicyConfigStatus(stdStatus.getPolicyConfigStatus());
 142                 policyConfig.setPolicyName(stdStatus.getPolicyName());
 143                 policyConfig.setPolicyVersion(stdStatus.getPolicyVersion());
 144                 policyConfig.setProperty(stdStatus.getProperty());
 145                 policyConfig.setResponseAttributes(stdStatus.getResponseAttributes());
 146                 policyConfig.setType(stdStatus.getType());
 147                 policyConfig.setPolicyType(getPolicyType(stdStatus.getPolicyName()));
 148                 result.add(policyConfig);
 149             }
 150         }
 151         return result;
 152     }
 153
 154     // Returns PolicyConfigType based on policyName.
 155     private PolicyConfigType getPolicyType(String policyName) {
 156         if(policyName != null) {
 157             String name = policyName;
 158             if(name.endsWith(".xml")){
 159                 name = name.substring(0, name.substring(0, name.lastIndexOf('.')).lastIndexOf('.'));
 160             }
 161             name = name.substring(name.lastIndexOf('.')+1);
 162             PolicyConfigType extType = extendedServices(name);
 163             if(extType != null) return extType;
 164             if (name.startsWith("Config_BRMS_Param_")) {
 165                 return PolicyConfigType.BRMS_PARAM;
 166             } else if(name.startsWith("Config_BRMS_Raw_")) {
 167                 return PolicyConfigType.BRMS_RAW;
 168             } else if(name.startsWith("Config_Fault_")) {
 169                 return PolicyConfigType.ClosedLoop_Fault;
 170             } else if(name.startsWith("Config_FW_")) {
 171                 return PolicyConfigType.Firewall;
 172             } else if(name.startsWith("Config_PM_")) {
 173                 return PolicyConfigType.ClosedLoop_PM;
 174             } else if(name.startsWith("Config_MS_")) {
 175                 return PolicyConfigType.MicroService;
 176             } else if(name.startsWith("Config_")) {
 177                 return PolicyConfigType.Base;
 178             }
 179         }
 180         return null;
 181     }
 182
 183     public PolicyConfigType extendedServices(String policyName) {
 184         // For extended services policyName will be required.
 185         return null;
 186     }
 187
 188     // Filter logic required for results comparing with requests.
 189     private Collection<PolicyConfig> filterResults(
 190             Collection<PolicyConfig> policyConfigs,
 191             ConfigRequestParameters configRequestParameters) {
 192         List<PolicyConfig> policyConfig = new ArrayList<>();
 193         for(PolicyConfig config: policyConfigs){
 194             if(config.getPolicyName()!=null && configRequestParameters.getPolicyName()!=null && configRequestParameters.getPolicyName().trim().length()>0){
 195                 if(!config.getPolicyName().matches(configRequestParameters.getPolicyName())){
 196                     continue;
 197                 }
 198             }
 199             if(config.getMatchingConditions()!=null && config.getMatchingConditions().size()>0){
 200                 if(configRequestParameters.getOnapName()!=null && configRequestParameters.getOnapName().trim().length()>0 && config.getMatchingConditions().containsKey("ONAPName")){
 201                     if(!config.getMatchingConditions().get("ONAPName").matches(configRequestParameters.getOnapName())){
 202                         continue;
 203                     }
 204                 }
 205                 if(configRequestParameters.getConfigName()!=null && configRequestParameters.getConfigName().trim().length()>0 && config.getMatchingConditions().containsKey("ConfigName")){
 206                     if(!config.getMatchingConditions().get("ConfigName").matches(configRequestParameters.getConfigName())){
 207                         continue;
 208                     }
 209                 }
 210                 if(configRequestParameters.getConfigAttributes()!=null && configRequestParameters.getConfigAttributes().size()>0){
 211                     boolean flag = false;
 212                     for(String key: configRequestParameters.getConfigAttributes().keySet()){
 213                         if(key.equals("RiskType")||key.equals("RiskLevel")||key.equals("guard")||key.equals("TTLDate")){
 214                                 continue;
 215                         }
 216                         if(config.getMatchingConditions().containsKey(key)){
 217                                 if(config.getMatchingConditions().get(key).contains(",")){
 218                                         List<String> elements = Arrays.asList(config.getMatchingConditions().get(key).split(","));
 219                                         if(!elements.contains(configRequestParameters.getConfigAttributes().get(key))){
 220                                                 flag=true;
 221                                         }
 222                                 }else if(!config.getMatchingConditions().get(key).matches(configRequestParameters.getConfigAttributes().get(key))){
 223                                 flag = true;
 224                                 }
 225                         }else{
 226                                 flag = true;
 227                         }
 228                         if(flag){
 229                                 break;
 230                         }
 231                     }
 232                     if(flag){
 233                         continue;
 234                     }
 235                 }
 236             }
 237             policyConfig.add(config);
 238         }
 239         if(policyConfig.isEmpty()){
 240             PolicyConfig pConfig = new PolicyConfig();
 241             pConfig.setPolicyConfigStatus(PolicyConfigStatus.CONFIG_NOT_FOUND);
 242             pConfig.setPolicyConfigMessage(XACMLErrorConstants.ERROR_DATA_ISSUE+"No Match Found, for the parameters sent.");
 243             policyConfig.add(pConfig);
 244         }
 245         return policyConfig;
 246     }
 247
 248     private JsonObject getModel() throws PolicyConfigException{
 249         String policyName = configRequestParameters.getPolicyName();
 250         if(policyName!= null && !policyName.trim().isEmpty() && !policyName.endsWith("xml")){
 251             policyName = policyName + ".[\\d].*";
 252             configRequestParameters.setPolicyName(policyName);
 253         }
 254         JsonArrayBuilder subjectArray = Json.createArrayBuilder();
 255         JsonArrayBuilder resourceArray = Json.createArrayBuilder();
 256         if(configRequestParameters.getPolicyName()!=null){
 257             JsonObjectBuilder subjectBuilder = Json.createObjectBuilder();
 258             subjectBuilder.add("Value", policyName);
 259             subjectBuilder.add("AttributeId", "PolicyName");
 260             subjectArray.add(subjectBuilder);
 261         }else{
 262             LOGGER.info("PolicyName values are not given. ");
 263         }
 264         if(configRequestParameters.getOnapName()!=null){
 265             JsonObjectBuilder subjectBuilder = Json.createObjectBuilder();
 266             subjectBuilder.add("Value", configRequestParameters.getOnapName());
 267             subjectBuilder.add("AttributeId", "ONAPName");
 268             subjectArray.add(subjectBuilder);
 269             if(configRequestParameters.getConfigName()!=null){
 270                 subjectBuilder = Json.createObjectBuilder();
 271                 subjectBuilder.add("Value", configRequestParameters.getConfigName());
 272                 subjectBuilder.add("AttributeId", "ConfigName");
 273                 subjectArray.add(subjectBuilder);
 274                 resourceArray = getResourceArray(configRequestParameters.getConfigAttributes());
 275             }else{
 276                 LOGGER.info("Config Name is not given. ");
 277             }
 278         }else{
 279             LOGGER.info("Onap Name is not given. ");
 280         }
 281         return Json.createObjectBuilder()
 282                 .add("Request",Json.createObjectBuilder()
 283                                 .add("AccessSubject",Json.createObjectBuilder()
 284                                                 .add("Attribute",subjectArray))
 285                                 .add("Action",Json.createObjectBuilder()
 286                                                 .add("Attribute",Json.createObjectBuilder()
 287                                                                 .add("Value","ACCESS")
 288                                                                 .add("AttributeId","urn:oasis:names:tc:xacml:1.0:action:action-id")))
 289                                 .add("Resource",Json.createObjectBuilder()
 290                                                 .add("Attribute",resourceArray
 291                                                                 .add(Json.createObjectBuilder()
 292                                                                         .add("Value","Config")
 293                                                                         .add("AttributeId","urn:oasis:names:tc:xacml:1.0:resource:resource-id")))))
 294                 .build();
 295     }
 296
 297     private JsonArrayBuilder getResourceArray(Map<String, String> configAttributes) throws PolicyConfigException{
 298         JsonArrayBuilder resourceArray = Json.createArrayBuilder();
 299         configAttributes = configRequestParameters.getConfigAttributes();
 300         if (configAttributes != null && !configAttributes.isEmpty()) {
 301             if(!configAttributes.containsKey("RiskType")){
 302                 configAttributes.put("RiskType", ".*");
 303             }
 304             if(!configAttributes.containsKey("RiskLevel")){
 305                 configAttributes.put("RiskLevel", ".*");
 306             }
 307             if(!configAttributes.containsKey("guard")){
 308                 configAttributes.put("guard", ".*");
 309             }
 310             if(!configAttributes.containsKey("TTLDate")){
 311                 configAttributes.put("TTLDate", ".*");
 312             }
 313         }else{
 314             // ConfigAttributes is Null. So add basic values.
 315             configAttributes = new HashMap<>();
 316             configAttributes.put("RiskType", ".*");
 317             configAttributes.put("RiskLevel", ".*");
 318             configAttributes.put("guard", ".*");
 319             configAttributes.put("TTLDate", ".*");
 320         }
 321         for (String key : configAttributes.keySet()) {
 322             if (key.isEmpty()) {
 323                 String message = XACMLErrorConstants.ERROR_DATA_ISSUE + "Cannot have an Empty Key";
 324                 LOGGER.error(message);
 325                 throw new PolicyConfigException(message);
 326             }
 327             JsonObjectBuilder resourceBuilder = Json.createObjectBuilder();
 328             resourceBuilder.add("Value", configAttributes.get(key));
 329             resourceBuilder.add("AttributeId", key);
 330             resourceArray.add(resourceBuilder);
 331         }
 332         return resourceArray;
 333     }
 334
 335     private Boolean getValidation(){
 336         if(configRequestParameters==null){
 337             message = XACMLErrorConstants.ERROR_DATA_ISSUE + "No config Request Parameters given.";
 338             return false;
 339         }
 340         if(configRequestParameters.getOnapName()==null && configRequestParameters.getPolicyName()==null){
 341             message = XACMLErrorConstants.ERROR_DATA_ISSUE + "Cannot proceed without onapName or PolicyName";
 342             return false;
 343         }
 344         return true;
 345     }
 346
 347     public Collection<PolicyConfig> getResult() {
 348         return policyConfigs;
 349     }
 350
 351     public HttpStatus getResponseCode() {
 352         return responseCode;
 353     }
 354 }