Merge "FFix the Bug of Missing fields on View Screen"
[policy/engine.git] / ONAP-REST / src / main / java / org / onap / policy / rest / util / PolicyValidationRequestWrapper.java
index 0724622..f197739 100644 (file)
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP Policy Engine
  * ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -37,6 +37,7 @@ import org.onap.policy.api.AttributeType;
 import org.onap.policy.api.PolicyParameters;
 import org.onap.policy.common.logging.flexlogger.FlexLogger;
 import org.onap.policy.common.logging.flexlogger.Logger;
+import org.onap.policy.rest.adapter.ClosedLoopFaultTrapDatas;
 import org.onap.policy.rest.adapter.PolicyRestAdapter;
 import org.onap.policy.rest.adapter.RainyDayParams;
 import org.onap.policy.rest.adapter.YAMLParams;
@@ -55,24 +56,31 @@ public class PolicyValidationRequestWrapper {
        public PolicyRestAdapter populateRequestParameters(HttpServletRequest request) {
                
                PolicyRestAdapter policyData = null;
-               
+               ClosedLoopFaultTrapDatas trapDatas = null;
+               ClosedLoopFaultTrapDatas faultDatas = null;
                try {
                        ObjectMapper mapper = new ObjectMapper();
                        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
                        JsonNode root = mapper.readTree(request.getReader());
                        policyData = mapper.readValue(root.get("policyData").toString(), PolicyRestAdapter.class);
+                       if(root.get("trapData") != null){
+                               trapDatas = mapper.readValue(root.get("trapData").toString(), ClosedLoopFaultTrapDatas.class);
+                               policyData.setTrapDatas(trapDatas);
+                       }
+                       if(root.get("faultData") != null){
+                               faultDatas = mapper.readValue(root.get("faultData").toString(), ClosedLoopFaultTrapDatas.class);
+                               policyData.setFaultDatas(faultDatas);
+                       }
                        
-                       JsonObject json = null;
-                       if(root != null){
-                               json = stringToJsonObject(root.toString());
-                               
-                               if(json != null){
-                                       if(json.containsKey("policyJSON")){
-                                               policyData.setPolicyJSON(root.get("policyJSON"));
-                                       }else{
-                                               String jsonBodyData = json.getJsonObject("policyData").get("jsonBodyData").toString();
-                                               policyData.setJsonBody(jsonBodyData);
-                                       }
+                       JsonObject json;
+                       json = stringToJsonObject(root.toString());
+                       
+                       if(json != null){
+                               if(json.containsKey("policyJSON")){
+                                       policyData.setPolicyJSON(root.get("policyJSON"));
+                               }else{
+                                       String jsonBodyData = json.getJsonObject("policyData").get("jsonBodyData").toString();
+                                       policyData.setJsonBody(jsonBodyData);
                                }
                        }                       
                                                
@@ -90,9 +98,15 @@ public class PolicyValidationRequestWrapper {
                /*
                 * set policy adapter values for Building JSON object containing policy data
                 */
-               //Common among policy types
+               //Common Policy Fields
                policyData.setPolicyName(parameters.getPolicyName());
                policyData.setOnapName(parameters.getOnapName()); 
+               policyData.setPriority(parameters.getPriority()); //Micro Service
+               policyData.setConfigName(parameters.getConfigName());  //Base and Firewall
+               policyData.setRiskType(parameters.getRiskType()); //Safe parameters Attributes
+               policyData.setRiskLevel(parameters.getRiskLevel());//Safe parameters Attributes
+               policyData.setGuard(String.valueOf(parameters.getGuard()));//Safe parameters Attributes
+               policyData.setTtlDate(convertDate(parameters.getTtlDate()));//Safe parameters Attributes
                
                //Some policies require jsonObject conversion from String for configBody (i.e. MicroService and Firewall)
                JsonObject json = null;
@@ -127,13 +141,13 @@ public class PolicyValidationRequestWrapper {
                                        // Set Matching attributes in RainyDayParams in adapter
                                        RainyDayParams rainyday = new RainyDayParams();
                                        
-                                       rainyday.setServiceType(matching.get("ServiceType"));
-                                       rainyday.setVnfType(matching.get("VNFType"));
-                                       rainyday.setBbid(matching.get("BB_ID"));
-                                       rainyday.setWorkstep(matching.get("WorkStep"));
-                                       
-                                       
-                                       
+                                       if(matching != null) {
+                                               rainyday.setServiceType(matching.get("ServiceType"));
+                                               rainyday.setVnfType(matching.get("VNFType"));
+                                               rainyday.setBbid(matching.get("BB_ID"));
+                                               rainyday.setWorkstep(matching.get("WorkStep"));
+                                       }
+
                                        Map<String, String> treatments = parameters.getTreatments();
                                        ArrayList<Object> treatmentsTableChoices = new ArrayList<>();
                                        
@@ -153,44 +167,44 @@ public class PolicyValidationRequestWrapper {
                                        // Set Matching attributes in YAMLParams in adapter
                                        YAMLParams yamlparams = new YAMLParams();
                                        
-                                       yamlparams.setActor(matching.get("actor"));
-                                       yamlparams.setRecipe(matching.get("recipe"));
-                                       yamlparams.setGuardActiveStart(matching.get("guardActiveStart"));
-                                       yamlparams.setGuardActiveEnd(matching.get("guardActiveEnd"));
-                                       
-                                       if("GUARD_YAML".equals(ruleProvider)){
-                                               yamlparams.setLimit(matching.get("limit"));
-                                               yamlparams.setTimeWindow(matching.get("timeWindow"));
-                                               yamlparams.setTimeUnits(matching.get("timeUnits"));     
-                                       }else{
+                                       if (matching != null) {
+                                               yamlparams.setActor(matching.get("actor"));
+                                               yamlparams.setRecipe(matching.get("recipe"));
+                                               yamlparams.setGuardActiveStart(matching.get("guardActiveStart"));
+                                               yamlparams.setGuardActiveEnd(matching.get("guardActiveEnd"));
                                                
-                                               List<String> blackList = new ArrayList<>();
+                                               if("GUARD_YAML".equals(ruleProvider)){
+                                                       yamlparams.setLimit(matching.get("limit"));
+                                                       yamlparams.setTimeWindow(matching.get("timeWindow"));
+                                                       yamlparams.setTimeUnits(matching.get("timeUnits"));     
+                                               }else{
+                                                       
+                                                       List<String> blackList = new ArrayList<>();
 
-                                               if(!Strings.isNullOrEmpty(matching.get("blackList"))){
-                                                       String[] blackListArray = matching.get("blackList").split(",");
-                                                       for(String element : blackListArray){
-                                                               blackList.add(element);
-                                                       }                                       
-                                               }       
-                                               
-                                               yamlparams.setBlackList(blackList);
+                                                       if(!Strings.isNullOrEmpty(matching.get("blackList"))){
+                                                               String[] blackListArray = matching.get("blackList").split(",");
+                                                               for(String element : blackListArray){
+                                                                       blackList.add(element);
+                                                               }                                       
+                                                       }       
+                                                       
+                                                       yamlparams.setBlackList(blackList);
 
-                                       }                                       
-                                       
+                                               }                                       
+                                       }
                                        policyData.setYamlparams(yamlparams);
-;                                      
                                }
                                
                        } else if("Action".equals(parameters.getPolicyClass().toString())){
                                
-                               ArrayList<Object> ruleAlgorithmChoices = new ArrayList<Object>();
+                               ArrayList<Object> ruleAlgorithmChoices = new ArrayList<>();
                                                                
                                List<String> dynamicLabelRuleAlgorithms = parameters.getDynamicRuleAlgorithmLabels();
                                List<String> dynamicFieldFunctionRuleAlgorithms = parameters.getDynamicRuleAlgorithmFunctions();
                                List<String> dynamicFieldOneRuleAlgorithms = parameters.getDynamicRuleAlgorithmField1();
                                List<String> dyrnamicFieldTwoRuleAlgorithms = parameters.getDynamicRuleAlgorithmField2();
                    
-                               if (dynamicLabelRuleAlgorithms != null && dynamicLabelRuleAlgorithms.size() > 0) {
+                               if (dynamicLabelRuleAlgorithms != null && !dynamicLabelRuleAlgorithms.isEmpty()) {
                        int i = dynamicLabelRuleAlgorithms.size() - 1;
 
                        for (String labelAttr : dynamicLabelRuleAlgorithms) {
@@ -216,15 +230,16 @@ public class PolicyValidationRequestWrapper {
                    policyData.setRuleAlgorithmschoices(ruleAlgorithmChoices);
                    
                    ArrayList<Object> attributeList = new ArrayList<>();
-                   
-                   for (String keyField : matching.keySet()) {
-                                       LinkedHashMap<String, String> attributeMap = new LinkedHashMap<>();
-                                       String key = keyField;
-                                       String value = matching.get(keyField);
-                                       attributeMap.put("key", key);
-                                       attributeMap.put("value", value);
-                                       attributeList.add(attributeMap);
-                               }                   
+                   if (matching != null) {
+                           for (Map.Entry<String, String> entry : matching.entrySet()) {
+                                               LinkedHashMap<String, String> attributeMap = new LinkedHashMap<>();
+                                               String key = entry.getKey();
+                                               String value = entry.getValue();
+                                               attributeMap.put("key", key);
+                                               attributeMap.put("value", value);
+                                               attributeList.add(attributeMap);
+                                       }       
+                   }
                    
                    policyData.setAttributes(attributeList);        
                    policyData.setActionAttributeValue(parameters.getActionAttribute());
@@ -299,6 +314,26 @@ public class PolicyValidationRequestWrapper {
                                        String version = json.get("version").toString().replace("\"", "");
                                        policyData.setVersion(version);
                                }
+                               if(json.containsKey("policyScope")){
+                                       String policyScope = json.get("policyScope").toString().replace("\"", "");
+                                       policyData.setPolicyScope(policyScope);
+                               }
+                               if(json.containsKey("riskType")){
+                                       String riskType = json.get("riskType").toString().replace("\"", "");
+                                       policyData.setRiskType(riskType);
+                               }
+                               if(json.containsKey("riskLevel")){
+                                       String riskLevel = json.get("riskLevel").toString().replace("\"", "");
+                                       policyData.setRiskLevel(riskLevel);
+                               }
+                               if(json.containsKey("guard")){
+                                       String guard = json.get("guard").toString().replace("\"", "");
+                                       policyData.setGuard(guard);
+                               }
+                               } else {
+                           String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ " improper JSON object : " + parameters.getConfigBody();
+                           LOGGER.error(message);
+                           return null;                                
                                }
                                
                        } else if("Fault".equals(parameters.getPolicyConfigType().toString())){
@@ -338,26 +373,15 @@ public class PolicyValidationRequestWrapper {
                                
                        }
                }
-               
-               policyData.setPriority(parameters.getPriority()); //Micro Service
-               policyData.setConfigName(parameters.getConfigName());  //Base and Firewall
-               policyData.setRiskType(parameters.getRiskType()); //Safe parameters Attributes
-               policyData.setRiskLevel(parameters.getRiskLevel());//Safe parameters Attributes
-               policyData.setGuard(String.valueOf(parameters.getGuard()));//Safe parameters Attributes
-               policyData.setTtlDate(convertDate(parameters.getTtlDate()));//Safe parameters Attributes
 
                return policyData;
                                
        }
        
-    private JsonObject stringToJsonObject(String value)
-            throws JsonException, IllegalStateException {
-       
-       try{
-            JsonReader jsonReader = Json.createReader(new StringReader(value));
-            JsonObject object = jsonReader.readObject();
-            jsonReader.close();
-            return object;
+    private JsonObject stringToJsonObject(String value) {
+
+       try(JsonReader jsonReader = Json.createReader(new StringReader(value))){
+                       return jsonReader.readObject();
         } catch(JsonException| IllegalStateException e){
             LOGGER.info(XACMLErrorConstants.ERROR_DATA_ISSUE+ "Improper JSON format... may or may not cause issues in validating the policy: " + value, e);
             return null;