com.wavemaker.runtime.rest.RequestDataBuilder Maven / Gradle / Ivy
/*******************************************************************************
* Copyright (C) 2022-2023 WaveMaker, Inc.
*
* 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.
******************************************************************************/
package com.wavemaker.runtime.rest;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Enumeration;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import org.springframework.http.HttpHeaders;
import org.springframework.util.MultiValueMap;
import org.springframework.web.servlet.HandlerMapping;
import org.springframework.web.util.UriComponentsBuilder;
import com.wavemaker.commons.io.NoCloseInputStream;
import com.wavemaker.runtime.rest.model.HttpRequestData;
/**
* @author Uday Shankar
*/
public class RequestDataBuilder {
public HttpRequestData getRequestData(HttpServletRequest httpServletRequest) throws URISyntaxException, IOException {
HttpRequestData httpRequestData = new HttpRequestData();
httpRequestData.setHttpHeaders(getHttpHeaders(httpServletRequest));
httpRequestData.setQueryParametersMap(getQueryParameters(httpServletRequest));
httpRequestData.setPathVariablesMap(getPathVariablesMap(httpServletRequest));
httpRequestData.setRequestBody(getRequestBody(httpServletRequest));
return httpRequestData;
}
private HttpHeaders getHttpHeaders(HttpServletRequest httpServletRequest) {
HttpHeaders httpHeaders = new HttpHeaders();
Enumeration headerNames = httpServletRequest.getHeaderNames();
while (headerNames.hasMoreElements()) {
String headerName = (String) headerNames.nextElement();
Enumeration headersEnumeration = httpServletRequest.getHeaders(headerName);
while (headersEnumeration.hasMoreElements()) {
httpHeaders.add(headerName, headersEnumeration.nextElement());
}
}
return httpHeaders;
}
private MultiValueMap getQueryParameters(HttpServletRequest httpServletRequest) throws URISyntaxException {
String fullUrl = getFullURL(httpServletRequest);
return UriComponentsBuilder.fromUri(new URI(fullUrl)).build().getQueryParams();
}
private Map getPathVariablesMap(HttpServletRequest httpServletRequest) {
return (Map) httpServletRequest.getAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE);
}
private String getFullURL(HttpServletRequest request) {
StringBuffer requestURL = request.getRequestURL();
String queryString = request.getQueryString();
if (queryString == null) {
return requestURL.toString();
} else {
return requestURL.append('?').append(queryString).toString();
}
}
private InputStream getRequestBody(HttpServletRequest httpServletRequest) throws IOException {
return new NoCloseInputStream(httpServletRequest.getInputStream());
}
}