d7cf3f2d99d54b900e6fdfc45ea1d6aec20e6668
[policy/xacml-pdp.git] / applications / common / src / main / java / org / onap / policy / pdp / xacml / application / common / std / StdCombinedPolicyRequest.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP
4  * ================================================================================
5  * Copyright (C) 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  *
19  * SPDX-License-Identifier: Apache-2.0
20  * ============LICENSE_END=========================================================
21  */
22
23 package org.onap.policy.pdp.xacml.application.common.std;
24
25 import com.att.research.xacml.std.annotations.XACMLAction;
26 import com.att.research.xacml.std.annotations.XACMLRequest;
27 import com.att.research.xacml.std.annotations.XACMLResource;
28 import com.att.research.xacml.std.annotations.XACMLSubject;
29 import java.util.ArrayList;
30 import java.util.Collection;
31 import java.util.Map;
32 import java.util.Map.Entry;
33
34 import lombok.Getter;
35 import lombok.Setter;
36 import lombok.ToString;
37
38 import org.onap.policy.models.decisions.concepts.DecisionRequest;
39
40 @Getter
41 @Setter
42 @ToString
43 @XACMLRequest(ReturnPolicyIdList = true)
44 public class StdCombinedPolicyRequest {
45
46     public static final String POLICY_TYPE_KEY = "policy-type";
47     public static final String POLICY_ID_KEY = "policy-id";
48
49     @XACMLSubject(includeInResults = true)
50     private String onapName;
51
52     @XACMLSubject(attributeId = "urn:org:onap:onap-component", includeInResults = true)
53     private String onapComponent;
54
55     @XACMLSubject(attributeId = "urn:org:onap:onap-instance",  includeInResults = true)
56     private String onapInstance;
57
58     @XACMLAction()
59     private String action;
60
61     @XACMLResource(includeInResults = true)
62     private Collection<String> resource = new ArrayList<>();
63
64     @XACMLResource(attributeId = "urn:org:onap:policy-type", includeInResults = true)
65     private Collection<String> resourcePolicyType = new ArrayList<>();
66
67     public StdCombinedPolicyRequest() {
68         super();
69     }
70
71     /**
72      * Parses the DecisionRequest into a MonitoringRequest.
73      *
74      * @param decisionRequest Input DecisionRequest
75      * @return MonitoringRequest
76      */
77     @SuppressWarnings({"unchecked", "rawtypes"})
78     public static StdCombinedPolicyRequest createInstance(DecisionRequest decisionRequest) {
79         //
80         // Create our request object
81         //
82         StdCombinedPolicyRequest request = new StdCombinedPolicyRequest();
83         //
84         // Add the subject attributes
85         //
86         request.onapName = decisionRequest.getOnapName();
87         request.onapComponent = decisionRequest.getOnapComponent();
88         request.onapInstance = decisionRequest.getOnapInstance();
89         //
90         // Add the action attribute
91         //
92         request.action = decisionRequest.getAction();
93         //
94         // Add the resource attributes
95         //
96         Map<String, Object> resources = decisionRequest.getResource();
97         for (Entry<String, Object> entrySet : resources.entrySet()) {
98             if (POLICY_ID_KEY.equals(entrySet.getKey())) {
99                 if (entrySet.getValue() instanceof Collection) {
100                     addPolicyIds(request, (Collection) entrySet.getValue());
101                 } else if (entrySet.getValue() instanceof String) {
102                     request.resource.add(entrySet.getValue().toString());
103                 }
104                 continue;
105             }
106             if (POLICY_TYPE_KEY.equals(entrySet.getKey())) {
107                 if (entrySet.getValue() instanceof Collection) {
108                     addPolicyTypes(request, (Collection) entrySet.getValue());
109                 } else if (entrySet.getValue() instanceof String) {
110                     request.resourcePolicyType.add(entrySet.getValue().toString());
111                 }
112             }
113         }
114         return request;
115     }
116
117     private static StdCombinedPolicyRequest addPolicyIds(StdCombinedPolicyRequest request, Collection<Object> ids) {
118         for (Object id : ids) {
119             request.resource.add(id.toString());
120         }
121         return request;
122     }
123
124     private static StdCombinedPolicyRequest addPolicyTypes(StdCombinedPolicyRequest request, Collection<Object> types) {
125         for (Object type : types) {
126             request.resourcePolicyType.add(type.toString());
127         }
128         return request;
129     }
130 }