Initial OpenECOMP policy/engine commit
[policy/engine.git] / PyPDPServer / src / main / java / org / openecomp / policy / pypdp / ConfigRequest.java
diff --git a/PyPDPServer/src/main/java/org/openecomp/policy/pypdp/ConfigRequest.java b/PyPDPServer/src/main/java/org/openecomp/policy/pypdp/ConfigRequest.java
new file mode 100644 (file)
index 0000000..694d010
--- /dev/null
@@ -0,0 +1,182 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ECOMP Policy Engine
+ * ================================================================================
+ * Copyright (C) 2017 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.
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.policy.pypdp;
+
+import java.io.StringWriter;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Properties;
+import java.util.UUID;
+
+import javax.json.JsonObject;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamResult;
+
+import org.openecomp.policy.api.ConfigRequestParameters;
+import org.openecomp.policy.api.PolicyConfig;
+import org.openecomp.policy.api.PolicyConfigException;
+import org.openecomp.policy.api.PolicyConfigStatus;
+import org.openecomp.policy.api.PolicyType;
+import org.openecomp.policy.pypdp.model_pojo.PepConfigPolicyNameRequest;
+import org.openecomp.policy.pypdp.model_pojo.PyPolicyConfig;
+import org.openecomp.policy.std.StdPolicyConfig;
+import org.openecomp.policy.std.StdPolicyEngine;
+import org.w3c.dom.Document;
+
+import org.openecomp.policy.xacml.api.XACMLErrorConstants;
+
+import org.openecomp.policy.common.logging.eelf.PolicyLogger;
+
+public class ConfigRequest {
+       
+       private StdPolicyEngine pe;
+       public ConfigRequest(StdPolicyEngine pe){
+               this.pe= pe;
+       }
+       
+       public Collection<PyPolicyConfig> run(ConfigRequestParameters pep, String requestID, String userID, String passcode) {
+               PolicyLogger.debug("... Request Params : \n"
+                               + "configName " + pep.getConfigName() + "\n"
+                               + "ecompName"  + pep.getEcompName() + "\n"
+                               + "policyName" + pep.getPolicyName() + "\n");
+               StdPolicyConfig policyConfig = new StdPolicyConfig();
+               Collection<PyPolicyConfig> result = new ArrayList<PyPolicyConfig>();
+               // construct a UUID from the request string
+               if(pep.getRequestID()==null){
+                       UUID requestUUID = null;
+                       if (requestID != null && !requestID.isEmpty()) {
+                               try {
+                                       requestUUID = UUID.fromString(requestID);
+                               }
+                               catch (IllegalArgumentException e) {
+                                       requestUUID = UUID.randomUUID();
+                                       PolicyLogger.info("Generated Random UUID: " + requestUUID.toString());
+                               }
+                       }
+                       pep.setRequestID(requestUUID);
+               }
+               try {
+                       PolicyLogger.debug("\n\n calling PEP.. ");
+                       Collection<PolicyConfig> pConfigs = pe.configRequest(pep, userID, passcode);
+                       for(PolicyConfig pConfig: pConfigs){
+                               PyPolicyConfig pyPolicyConfig = checkResponse(pConfig);
+                               result.add(pyPolicyConfig);
+                       }
+                       return result;
+               } catch(Exception e){
+                       policyConfig.setConfigStatus(e.getMessage(), PolicyConfigStatus.CONFIG_NOT_FOUND);
+                       PyPolicyConfig pyPolicyConfig = checkResponse(policyConfig);
+                       result.add(pyPolicyConfig);
+                       return result;
+               }
+       }
+       
+       public Collection<PyPolicyConfig> run(PepConfigPolicyNameRequest pep, String requestID, String userID, String passcode) {
+               PolicyLogger.debug("... Request Params : \n"
+                               + "policyName" + pep.getPolicyName() + "\n");
+               StdPolicyConfig policyConfig = new StdPolicyConfig();
+               Collection<PyPolicyConfig> result = new ArrayList<PyPolicyConfig>();
+               // construct a UUID from the request string
+               UUID requestUUID = null;
+               if (requestID != null && !requestID.isEmpty()) {
+                       try {
+                               requestUUID = UUID.fromString(requestID);
+                       }
+                       catch (IllegalArgumentException e) {
+                               requestUUID = UUID.randomUUID();
+                               PolicyLogger.info("Generated Random UUID: " + requestUUID.toString());
+                       }
+               }
+               if(pep.getPolicyName()!= null && !pep.getPolicyName().isEmpty()) {
+                       try {
+                               Collection<PolicyConfig> pConfigs = pe.configPolicyName(pep.getPolicyName(), requestUUID, userID, passcode);
+                               for(PolicyConfig pConfig: pConfigs){
+                                       PyPolicyConfig pyPolicyConfig = checkResponse(pConfig);
+                                       result.add(pyPolicyConfig);
+                               }
+                               return result;
+                       } catch (PolicyConfigException e) {
+                               policyConfig.setConfigStatus(e.getMessage(), PolicyConfigStatus.CONFIG_NOT_FOUND);
+                               PyPolicyConfig pyPolicyConfig = checkResponse(policyConfig);
+                               result.add(pyPolicyConfig);
+                               return result;
+                       }
+               }
+               else {
+                       policyConfig.setConfigStatus(XACMLErrorConstants.ERROR_DATA_ISSUE + "PolicyFile Name is empty", PolicyConfigStatus.CONFIG_NOT_FOUND);
+                       PyPolicyConfig pyPolicyConfig = checkResponse(policyConfig);
+                       result.add(pyPolicyConfig);
+                       return result;
+               }
+       }
+
+       public PyPolicyConfig checkResponse(PolicyConfig pConfig) {
+               PyPolicyConfig policyConfig = new PyPolicyConfig();
+               policyConfig.setPolicyConfigMessage(pConfig.getPolicyConfigMessage());
+               policyConfig.setPolicyConfigStatus(pConfig.getPolicyConfigStatus());
+               policyConfig.setType(pConfig.getType());
+               policyConfig.setPolicyName(pConfig.getPolicyName());
+               policyConfig.setMatchingConditions(pConfig.getMatchingConditions());
+               policyConfig.setResponseAttributes(pConfig.getResponseAttributes());
+               policyConfig.setPolicyVersion(pConfig.getPolicyVersion());
+               if (pConfig.getPolicyConfigStatus().equals(PolicyConfigStatus.CONFIG_RETRIEVED)) {
+                       PolicyType policyType = policyConfig.getType();
+                       if(policyType.equals(PolicyType.PROPERTIES)) {
+                               Properties properties = pConfig.toProperties();
+                               Map<String, String> propVal = new HashMap<String, String>();
+                               for(String name: properties.stringPropertyNames()) {
+                                       propVal.put(name, properties.getProperty(name));
+                               }
+                               policyConfig.setProperty(propVal);
+                       } else if(policyType.equals(PolicyType.OTHER)) {
+                               String other = pConfig.toOther();
+                               policyConfig.setConfig(other);
+                       } else if (policyType.equals(PolicyType.JSON)) {
+                               JsonObject json = pConfig.toJSON();
+                               policyConfig.setConfig(json.toString());
+                       } else if (policyType.equals(PolicyType.XML)) {
+                               Document document = pConfig.toXML();
+                               DOMSource domSource = new DOMSource(document);
+                               StringWriter writer = new StringWriter();
+                               StreamResult result = new StreamResult(writer);
+                               TransformerFactory tf = TransformerFactory.newInstance();
+                               Transformer transformer;
+                               try {
+                                       transformer = tf.newTransformer();
+                                       transformer.transform(domSource, result);
+                                       policyConfig.setConfig(writer.toString());
+                               } catch (TransformerException e) {
+                                       policyConfig.setConfig(null);
+                                       policyConfig.setPolicyConfigMessage(XACMLErrorConstants.ERROR_SCHEMA_INVALID + "XML error in the Configuration. " + e.getMessage());
+                                       policyConfig.setPolicyConfigStatus(PolicyConfigStatus.CONFIG_NOT_FOUND);
+                               }
+                       }
+               } else {
+                       policyConfig.setConfig(null);
+               }
+               return policyConfig;
+       }
+}