aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/crud/util/CrudServiceUtil.java
blob: 5eeb38ed5420361346fa85d6db693a4d941d90a1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017-2018 Amdocs
 * ================================================================================
 * 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.onap.crud.util;

import java.util.AbstractMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import javax.ws.rs.core.EntityTag;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response.Status;
import org.onap.aai.db.props.AAIProperties;
import org.onap.crud.exception.CrudException;
import org.onap.crud.parser.EdgePayload;
import org.onap.crud.parser.util.EdgePayloadUtil;
import org.onap.schema.EdgeRulesLoader;
import org.onap.schema.OxmModelLoader;
import org.onap.schema.RelationshipSchema;

import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonPrimitive;

public class CrudServiceUtil {

  private static Gson gson = new Gson();
  
  @SuppressWarnings({"unchecked", "rawtypes"})
  public static Object validateFieldType(String value, Class clazz) throws CrudException {
    try {
      if (clazz.isAssignableFrom(Integer.class)) {
        return Integer.parseInt(value);
      } else if (clazz.isAssignableFrom(Long.class)) {
        return Long.parseLong(value);
      } else if (clazz.isAssignableFrom(Float.class)) {
        return Float.parseFloat(value);
      } else if (clazz.isAssignableFrom(Double.class)) {
        return Double.parseDouble(value);
      } else if (clazz.isAssignableFrom(Boolean.class)) {

		// If the value is an IN/OUT direction, this gets seen as a boolean, so
        // check for that first.
        if (value.equals("OUT") || value.equals("IN")) {
          return value;
        }

        if (!value.equals("true") && !value.equals("false")) {
          throw new CrudException("Invalid propertry value: " + value, Status.BAD_REQUEST);
        }
        return Boolean.parseBoolean(value);
      } else {
        return value;
      }
    } catch (Exception e) {
      throw new CrudException("Invalid property value: " + value, Status.BAD_REQUEST);
    }
  }

  public static void loadModels() throws CrudException {
    // load the schemas
    try {
      OxmModelLoader.loadModels();
      EdgeRulesLoader.loadModels ();
    } catch (Exception e) {
      throw new CrudException(e);
    }
  }

  /**
   * This method will merge header property from app id in request payload if not already populated
   * @param propertiesFromRequest
   * @param headers
   * @param isAdd
   * @return
   */
    @SuppressWarnings("unchecked")
    public static JsonElement mergeHeaderInFoToPayload(JsonElement propertiesFromRequest, HttpHeaders headers,
            boolean isAdd) {
    String sourceOfTruth = headers.getRequestHeaders().getFirst("X-FromAppId");  
    Set<Map.Entry<String, JsonElement>> properties = new HashSet<Map.Entry<String, JsonElement>>();
    properties.addAll(propertiesFromRequest.getAsJsonObject().entrySet());
    
    Set<String> propertyKeys = new HashSet<String>();
    for(Map.Entry<String, JsonElement> property : properties) {
      propertyKeys.add(property.getKey());
    }
    
    if(!propertyKeys.contains(AAIProperties.LAST_MOD_SOURCE_OF_TRUTH)) {
        properties.add(new AbstractMap.SimpleEntry<String, JsonElement>(AAIProperties.LAST_MOD_SOURCE_OF_TRUTH,
            (new JsonPrimitive(sourceOfTruth))));
    }
   
    if(isAdd && !propertyKeys.contains(AAIProperties.SOURCE_OF_TRUTH)) {
        properties.add(new AbstractMap.SimpleEntry<String, JsonElement>(AAIProperties.SOURCE_OF_TRUTH,
            (new JsonPrimitive(sourceOfTruth))));
    }

    Object[] propArray = properties.toArray();
    StringBuilder sb = new StringBuilder();
    sb.append("{");
    boolean first=true;
    for(int i=0; i<propArray.length; i++) {

      Map.Entry<String, JsonElement> entry = (Entry<String, JsonElement>) propArray[i];
      if(!first) {
        sb.append(",");
      }
      sb.append("\"").append(entry.getKey()).append("\"").append(":").append(entry.getValue());
      first=false;
    }
    sb.append("}");

    return gson.fromJson(sb.toString(), JsonElement.class);
  }

  public static EntityTag getETagFromHeader(MultivaluedMap<String, String> headers) {
    EntityTag entityTag = null;
    if (headers != null && headers.containsKey(CrudServiceConstants.CRD_HEADER_ETAG)) {
      String value = headers.getFirst(CrudServiceConstants.CRD_HEADER_ETAG);
      entityTag = new EntityTag(value.replace("\"", ""));
    }
    return entityTag;
  }

  public static String determineEdgeType(EdgePayload payload, String version) throws CrudException {
    RelationshipSchema schema = EdgeRulesLoader.getSchemaForVersion(version);
    
    if (payload.getSource() == null || payload.getTarget() == null) {
      throw new CrudException("Source/Target not specified", Status.BAD_REQUEST);
    }
    
    Set<String> edgeTypes = schema.getValidRelationTypes(EdgePayloadUtil.getVertexNodeType(payload.getSource()), 
        EdgePayloadUtil.getVertexNodeType(payload.getTarget()));
    
    if (edgeTypes.size() == 0) {
      throw new CrudException("No valid relationship types from " + payload.getSource() + " to " + payload.getTarget(), Status.BAD_REQUEST);
    }
    
    if (edgeTypes.size() > 1) {
      throw new CrudException("Multiple possible relationship types from " + payload.getSource() + " to " + payload.getTarget(), Status.BAD_REQUEST);
    }
    
    return edgeTypes.iterator().next();
  }

}