All Downloads are FREE. Search and download functionalities are using the official Maven repository.

software.xdev.mockserver.mappers.MockServerHttpRequestToFullHttpRequest Maven / Gradle / Ivy

Go to download

MockServer NeoLight Core

There is a newer version: 1.0.8
Show newest version
/*
 * Copyright © 2024 XDEV Software (https://xdev.software)
 *
 * 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 software.xdev.mockserver.mappers;

import static io.netty.handler.codec.http.HttpHeaderNames.ACCEPT_ENCODING;
import static io.netty.handler.codec.http.HttpHeaderNames.CONNECTION;
import static io.netty.handler.codec.http.HttpHeaderNames.CONTENT_LENGTH;
import static io.netty.handler.codec.http.HttpHeaderNames.CONTENT_TYPE;
import static io.netty.handler.codec.http.HttpHeaderNames.COOKIE;
import static io.netty.handler.codec.http.HttpHeaderNames.HOST;
import static io.netty.handler.codec.http.HttpHeaderNames.TRANSFER_ENCODING;
import static io.netty.handler.codec.http.HttpHeaderValues.CLOSE;
import static io.netty.handler.codec.http.HttpHeaderValues.DEFLATE;
import static io.netty.handler.codec.http.HttpHeaderValues.GZIP;
import static io.netty.handler.codec.http.HttpHeaderValues.KEEP_ALIVE;
import static io.netty.handler.codec.http.HttpUtil.isKeepAlive;
import static software.xdev.mockserver.util.StringUtils.isNotBlank;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.netty.buffer.ByteBuf;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpScheme;
import io.netty.handler.codec.http.HttpVersion;
import io.netty.handler.codec.http.QueryStringEncoder;
import io.netty.handler.codec.http2.HttpConversionUtil;
import software.xdev.mockserver.codec.BodyDecoderEncoder;
import software.xdev.mockserver.model.Header;
import software.xdev.mockserver.model.HttpRequest;
import software.xdev.mockserver.model.NottableString;
import software.xdev.mockserver.model.Parameter;
import software.xdev.mockserver.model.Protocol;
import software.xdev.mockserver.proxyconfiguration.ProxyConfiguration;


public class MockServerHttpRequestToFullHttpRequest
{
	private static final Logger LOG = LoggerFactory.getLogger(MockServerHttpRequestToFullHttpRequest.class);
	private final Map proxyConfigurations;
	private final BodyDecoderEncoder bodyDecoderEncoder;
	
	public MockServerHttpRequestToFullHttpRequest(
		final Map proxyConfigurations)
	{
		this.proxyConfigurations = proxyConfigurations;
		this.bodyDecoderEncoder = new BodyDecoderEncoder();
	}
	
	public FullHttpRequest mapMockServerRequestToNettyRequest(final HttpRequest httpRequest)
	{
		// method
		final HttpMethod httpMethod = HttpMethod.valueOf(httpRequest.getMethod("GET"));
		try
		{
			// the request
			final FullHttpRequest request = new DefaultFullHttpRequest(
				HttpVersion.HTTP_1_1,
				httpMethod,
				this.getURI(httpRequest, this.proxyConfigurations),
				this.getBody(httpRequest));
			
			// headers
			this.setHeader(httpRequest, request);
			
			// cookies
			this.setCookies(httpRequest, request);
			
			return request;
		}
		catch(final Exception ex)
		{
			LOG.error("Exception encoding request{}", httpRequest, ex);
			return new DefaultFullHttpRequest(
				HttpVersion.HTTP_1_1,
				httpMethod,
				this.getURI(httpRequest, this.proxyConfigurations));
		}
	}
	
	@SuppressWarnings("HttpUrlsUsage")
	public String getURI(
		final HttpRequest httpRequest,
		final Map proxyConfigurations)
	{
		String uri = "";
		if(httpRequest.getPath() != null)
		{
			if(httpRequest.getQueryStringParameters() != null && isNotBlank(httpRequest.getQueryStringParameters()
				.getRawParameterString()))
			{
				uri = httpRequest.getPath().getValue() + "?" + httpRequest.getQueryStringParameters()
					.getRawParameterString();
			}
			else
			{
				final QueryStringEncoder queryStringEncoder = new QueryStringEncoder(httpRequest.getPath().getValue());
				for(final Parameter parameter : httpRequest.getQueryStringParameterList())
				{
					for(final NottableString value : parameter.getValues())
					{
						queryStringEncoder.addParam(parameter.getName().getValue(), value.getValue());
					}
				}
				uri = queryStringEncoder.toString();
			}
		}
		if(proxyConfigurations != null && proxyConfigurations.get(ProxyConfiguration.Type.HTTP) != null)
		{
			if(isNotBlank(httpRequest.getFirstHeader(HOST.toString())))
			{
				uri = "http://" + httpRequest.getFirstHeader(HOST.toString()) + uri;
			}
			else if(httpRequest.getRemoteAddress() != null)
			{
				uri = "http://" + httpRequest.getRemoteAddress() + uri;
			}
		}
		return uri;
	}
	
	private ByteBuf getBody(final HttpRequest httpRequest)
	{
		return this.bodyDecoderEncoder.bodyToByteBuf(
			httpRequest.getBody(),
			httpRequest.getFirstHeader(CONTENT_TYPE.toString()));
	}
	
	private void setCookies(final HttpRequest httpRequest, final FullHttpRequest request)
	{
		if(!httpRequest.getCookieList().isEmpty())
		{
			final List cookies = new ArrayList<>();
			for(final software.xdev.mockserver.model.Cookie cookie : httpRequest.getCookieList())
			{
				cookies.add(new io.netty.handler.codec.http.cookie.DefaultCookie(
					cookie.getName().getValue(),
					cookie.getValue().getValue()));
			}
			request.headers()
				.set(COOKIE.toString(), io.netty.handler.codec.http.cookie.ClientCookieEncoder.LAX.encode(cookies));
		}
	}
	
	private void setHeader(final HttpRequest httpRequest, final FullHttpRequest request)
	{
		for(final Header header : httpRequest.getHeaderList())
		{
			final String headerName = header.getName().getValue();
			// do not set hop-by-hop headers
			if(!headerName.equalsIgnoreCase(CONTENT_LENGTH.toString())
				&& !headerName.equalsIgnoreCase(TRANSFER_ENCODING.toString())
				&& !headerName.equalsIgnoreCase(HOST.toString())
				&& !headerName.equalsIgnoreCase(ACCEPT_ENCODING.toString()))
			{
				if(!header.getValues().isEmpty())
				{
					for(final NottableString headerValue : header.getValues())
					{
						request.headers().add(headerName, headerValue.getValue());
					}
				}
				else
				{
					request.headers().add(headerName, "");
				}
			}
		}
		
		if(isNotBlank(httpRequest.getFirstHeader(HOST.toString())))
		{
			request.headers().add(HOST, httpRequest.getFirstHeader(HOST.toString()));
		}
		request.headers().set(ACCEPT_ENCODING, GZIP + "," + DEFLATE);
		if(Protocol.HTTP_2.equals(httpRequest.getProtocol()))
		{
			request.headers().add(HttpConversionUtil.ExtensionHeaderNames.SCHEME.text(), HttpScheme.HTTP.name());
			final Integer streamId = httpRequest.getStreamId();
			if(streamId != null)
			{
				request.headers().add(HttpConversionUtil.ExtensionHeaderNames.STREAM_ID.text(), streamId);
			}
		}
		request.headers().set(CONTENT_LENGTH, request.content().readableBytes());
		if(isKeepAlive(request))
		{
			request.headers().set(CONNECTION, KEEP_ALIVE);
		}
		else
		{
			request.headers().set(CONNECTION, CLOSE);
		}
		
		if(!request.headers().contains(CONTENT_TYPE))
		{
			if(httpRequest.getBody() != null
				&& httpRequest.getBody().getContentType() != null)
			{
				request.headers().set(CONTENT_TYPE, httpRequest.getBody().getContentType());
			}
		}
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy