summaryrefslogtreecommitdiffstats
path: root/aai-core/src/main/java/org/onap/aai/restcore/util/URITools.java
blob: 9889a5676b34cd3e3e1b4a85d8f7f2506b0b0501 (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
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 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=========================================================
 *
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 */
package org.onap.aai.restcore.util;

import java.io.UnsupportedEncodingException;
import java.net.URI;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import javax.ws.rs.core.MultivaluedHashMap;
import javax.ws.rs.core.MultivaluedMap;

import org.springframework.web.util.UriUtils;

import org.onap.aai.introspection.Introspector;
import org.onap.aai.introspection.sideeffect.exceptions.AAIMissingRequiredPropertyException;
import org.onap.aai.schema.enums.PropertyMetadata;

public class URITools {

	protected static final Pattern template = Pattern.compile("\\{(.*?)\\}");

	public static MultivaluedMap<String, String> getQueryMap(URI uri) {
		MultivaluedMap<String, String> result = new MultivaluedHashMap<>();
		String queryParams = uri.getRawQuery();
		if (queryParams != null) {
			try {
				String[] sections = queryParams.split("&");
				String[] query = null;
				String key, value = "";
				for (String section : sections) {
					query = section.split("=");
					key = UriUtils.decode(query[0], "UTF-8");
					value = UriUtils.decode(query[1], "UTF-8");
					if (result.containsKey(key)) {
						result.add(key, value);
					} else {
						result.putSingle(key, value);
					}
				}
			} catch (UnsupportedEncodingException e ) {
				
			}
		}
		return result;
		
	}
	
	public static Optional<String> replaceTemplates(Introspector obj, String uriString, PropertyMetadata metadata, boolean replaceWithWildcard) throws AAIMissingRequiredPropertyException {
		String result = uriString;
		final Map<String, String> propMap = URITools.findProperties(obj, uriString, metadata, replaceWithWildcard);
		if (propMap.isEmpty()) {
			return Optional.empty();
		}
		for (Entry<String, String> entry : propMap.entrySet()) {
			result = result.replaceAll("\\{" + entry.getKey() + "\\}", entry.getValue());
		}
		//drop out wildcards if they exist
		result = result.replaceFirst("/[^/]+?(?:/\\*)+", "");
		return Optional.of(result);
	}
	
	private static Map<String, String> findProperties(Introspector obj, String uriString, PropertyMetadata metadata, boolean replaceWithWildcard) throws AAIMissingRequiredPropertyException {
		
		final Map<String, String> result = new HashMap<>();
		final Set<String> missing = new LinkedHashSet<>();
		Matcher m = template.matcher(uriString);
		int properties = 0;
		while (m.find()) {
			String propName = m.group(1);
			String value = obj.getValue(propName);
			properties++;
			if (value != null) {
				result.put(propName, value);
			} else {
				if (replaceWithWildcard) {
					result.put(propName, "*");
				}
				missing.add(propName);
			}
		}
		
		if (!missing.isEmpty() && (properties != missing.size())) {
			throw new AAIMissingRequiredPropertyException("Cannot complete " + metadata.toString() + " uri. Missing properties " + missing);
		}
		return result;
	}
	
}