Consolidate PolicyRestAdapter setup
[policy/engine.git] / ONAP-PAP-REST / src / main / java / org / onap / policy / pap / xacml / rest / handler / SavePolicyHandler.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP-PAP-REST
4  * ================================================================================
5  * Copyright (C) 2017-2019 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
21 package org.onap.policy.pap.xacml.rest.handler;
22
23 import com.att.research.xacml.util.XACMLProperties;
24
25 import java.io.IOException;
26 import java.util.HashMap;
27
28 import javax.servlet.ServletException;
29 import javax.servlet.http.HttpServletRequest;
30 import javax.servlet.http.HttpServletResponse;
31 import javax.xml.parsers.ParserConfigurationException;
32
33 import org.onap.policy.common.logging.eelf.PolicyLogger;
34 import org.onap.policy.common.logging.flexlogger.FlexLogger;
35 import org.onap.policy.common.logging.flexlogger.Logger;
36 import org.onap.policy.pap.xacml.rest.policycontroller.PolicyCreation;
37 import org.onap.policy.rest.adapter.PolicyRestAdapter;
38 import org.onap.policy.utils.PolicyUtils;
39 import org.onap.policy.xacml.std.pap.StdPAPPolicy;
40 import org.xml.sax.SAXException;
41
42 public class SavePolicyHandler {
43     private static final Logger logger = FlexLogger.getLogger(SavePolicyHandler.class);
44     private HashMap<String, String> ErrorHeaders = null;
45
46     public void doPolicyAPIPut(HttpServletRequest request, HttpServletResponse response) throws IOException {
47         String operation = request.getParameter("operation");
48         String policyType = request.getParameter("policyType");
49         String apiflag = request.getParameter("apiflag");
50         PolicyCreation creation = new PolicyCreation();
51         if (policyType != null) {
52             PolicyRestAdapter policyAdapter = new PolicyRestAdapter();
53             if ("update".equalsIgnoreCase(operation)) {
54                 policyAdapter.setEditPolicy(true);
55             } else {
56                 policyAdapter.setEditPolicy(false);
57             }
58
59             // get the request content into a String
60             String json = null;
61             // read the inputStream into a buffer (trick found online scans entire input looking for end-of-file)
62             java.util.Scanner scanner = new java.util.Scanner(request.getInputStream());
63             scanner.useDelimiter("\\A");
64             json = scanner.hasNext() ? scanner.next() : "";
65             scanner.close();
66
67             if (policyAdapter.isEditPolicy()) {
68                 PolicyLogger.info("SavePolicyHandler: JSON request from API to update a policy: " + json);
69             } else {
70                 PolicyLogger.info("SavePolicyHandler: JSON request from API to create a policy: " + json);
71             }
72
73             // convert Object sent as JSON into local object
74             StdPAPPolicy policy = PolicyUtils.jsonStringToObject(json, StdPAPPolicy.class);
75             // Set policyAdapter values including parentPath (Common to all policy types)
76             try {
77                 PolicyLogger.info("SavePolicyHandler: Setting parameter values to PolicyAdapter");
78                 policyAdapter = setDataToPolicyAdapter(policy, policyType, apiflag);
79
80                 if (!extendedPolicyOptions(policyAdapter, response)) {
81                     creation.savePolicy(policyAdapter, response);
82                 }
83                 if ("update".equalsIgnoreCase(operation)) {
84                     response.addHeader("operation", "update");
85                 } else {
86                     response.addHeader("operation", "create");
87                 }
88             } catch (Exception e1) {
89                 logger.error("Could not set data to policy adapter " + e1.getMessage(), e1);
90             }
91         }
92     }
93
94     private PolicyRestAdapter setDataToPolicyAdapter(StdPAPPolicy policy, String policyType, String apiflag)
95             throws ParserConfigurationException, ServletException, SAXException, IOException {
96         PolicyRestAdapter policyAdapter = new PolicyRestAdapter();
97         policyAdapter.setApiflag(apiflag);
98         /*
99          * set policy adapter values for Building JSON object containing policy data
100          */
101         // Common among policy types
102         policyAdapter.setPolicyName(policy.getPolicyName());
103         policyAdapter.setPolicyDescription(policy.getPolicyDescription());
104         policyAdapter.setOnapName(policy.getOnapName()); // Config Base and Decision Policies
105         policyAdapter.setRuleCombiningAlgId("urn:oasis:names:tc:xacml:3.0:rule-combining-algorithm:permit-overrides");
106
107         policyAdapter.setPolicyType(policyType);
108         policyAdapter.setDynamicFieldConfigAttributes(policy.getDynamicFieldConfigAttributes());
109         policyAdapter.setEditPolicy(policy.isEditPolicy());
110         // Config Specific
111         policyAdapter.setConfigName(policy.getConfigName()); // Base and Firewall
112         policyAdapter.setConfigBodyData(policy.getConfigBodyData()); // Base
113         policyAdapter.setConfigType((policy.getConfigType() != null) ? policy.getConfigType().toUpperCase() : null); // Base
114         policyAdapter.setJsonBody(policy.getJsonBody()); // Firewall, ClosedLoop
115         policyAdapter.setConfigPolicyType(policy.getConfigPolicyType());
116         policyAdapter.setDraft(policy.isDraft()); // ClosedLoop_Fault
117         policyAdapter.setServiceType(policy.getServiceType()); // ClosedLoop_PM
118         policyAdapter.setUuid(policy.getUuid()); // Micro Service
119         policyAdapter.setLocation(policy.getMsLocation()); // Micro Service
120         policyAdapter.setPriority(policy.getPriority()); // Micro Service
121         policyAdapter.setPolicyScope(policy.getDomainDir());
122         policyAdapter.setRiskType(policy.getRiskType()); // Safe Policy Attributes
123         policyAdapter.setRiskLevel(policy.getRiskLevel());// Safe Policy Attributes
124         policyAdapter.setGuard(policy.getGuard());// Safe Policy Attributes
125         policyAdapter.setTtlDate(policy.getTTLDate());// Safe Policy Attributes
126         policyAdapter.setBrmsParamBody(policy.getDrlRuleAndUIParams());
127         policyAdapter.setBrmsDependency(policy.getBrmsDependency()); // BRMS Policies.
128         policyAdapter.setBrmsController(policy.getBrmsController()); // BRMS Policies.
129         // Action Policy Specific
130         policyAdapter.setActionAttribute(policy.getActionAttribute()); // comboDictValue
131         policyAdapter.setActionPerformer(policy.getActionPerformer());
132         policyAdapter.setDynamicRuleAlgorithmLabels(policy.getDynamicRuleAlgorithmLabels());
133         policyAdapter.setDynamicRuleAlgorithmCombo(policy.getDynamicRuleAlgorithmCombo());
134         policyAdapter.setDynamicRuleAlgorithmField1(policy.getDynamicRuleAlgorithmField1());
135         policyAdapter.setDynamicRuleAlgorithmField2(policy.getDynamicRuleAlgorithmField2());
136         // Decision Policy Specific
137         policyAdapter.setDynamicSettingsMap(policy.getDynamicSettingsMap());
138         policyAdapter.setRuleProvider(policy.getProviderComboBox());
139         policyAdapter.setDomainDir(policyAdapter.getPolicyScope());
140         policyAdapter.setRainydayMap(policy.getTreatments());
141         policyAdapter.setRawXacmlPolicy(policy.getRawXacmlPolicy());
142
143         return policyAdapter;
144     }
145
146     public boolean extendedPolicyOptions(PolicyRestAdapter policyAdapter, HttpServletResponse response) {
147         return false;
148     }
149
150     public void addErrorHeader(String key, String value) {
151         if (ErrorHeaders == null) {
152             ErrorHeaders = new HashMap<>();
153         }
154         ErrorHeaders.put(key, value);
155     }
156
157     public static SavePolicyHandler getInstance() {
158         try {
159             Class<?> savePolicyHandler = Class.forName(
160                     XACMLProperties.getProperty("savePolicy.impl.className", SavePolicyHandler.class.getName()));
161             SavePolicyHandler instance = (SavePolicyHandler) savePolicyHandler.newInstance();
162             return instance;
163         } catch (Exception e) {
164             logger.error(e.getMessage(), e);
165         }
166         return null;
167     }
168 }