aai-traversal/src/main/java/org/onap/aai/interceptors/pre/RequestHeaderManipulation.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 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.aai.interceptors.pre;
  22
  23 import java.io.IOException;
  24 import java.util.Collections;
  25 import java.util.regex.Matcher;
  26 import java.util.regex.Pattern;
  27
  28 import javax.annotation.Priority;
  29 import javax.servlet.http.HttpServletRequest;
  30 import javax.ws.rs.container.ContainerRequestContext;
  31 import javax.ws.rs.container.ContainerRequestFilter;
  32 import javax.ws.rs.container.PreMatching;
  33 import javax.ws.rs.core.MultivaluedMap;
  34
  35 import org.onap.aai.interceptors.AAIContainerFilter;
  36 import org.onap.aai.interceptors.AAIHeaderProperties;
  37 import org.springframework.beans.factory.annotation.Autowired;
  38
  39 @PreMatching
  40 @Priority(AAIRequestFilterPriority.HEADER_MANIPULATION)
  41 public class RequestHeaderManipulation extends AAIContainerFilter implements ContainerRequestFilter {
  42
  43         @Autowired
  44         private HttpServletRequest httpServletRequest;
  45
  46         private static final Pattern versionedEndpoint = Pattern.compile("^/aai/(v\\d+)");
  47
  48         @Override
  49         public void filter(ContainerRequestContext requestContext) throws IOException {
  50
  51                 String uri = httpServletRequest.getRequestURI();
  52                 this.addRequestContext(uri, requestContext.getHeaders());
  53
  54         }
  55
  56         private void addRequestContext(String uri, MultivaluedMap<String, String> requestHeaders) {
  57
  58                 String rc = "";
  59
  60         Matcher match = versionedEndpoint.matcher(uri);
  61         if (match.find()) {
  62             rc = match.group(1);
  63         }
  64
  65                 if (requestHeaders.containsKey(AAIHeaderProperties.REQUEST_CONTEXT)) {
  66                         requestHeaders.remove(AAIHeaderProperties.REQUEST_CONTEXT);
  67                 }
  68                 requestHeaders.put(AAIHeaderProperties.REQUEST_CONTEXT, Collections.singletonList(rc));
  69         }
  70
  71 }