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

com.arangodb.shaded.vertx.ext.web.client.impl.HttpRequestImpl Maven / Gradle / Ivy

There is a newer version: 7.9.0
Show newest version
/*
 * Copyright (c) 2011-2013 The original author or authors
 * ------------------------------------------------------
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * and Apache License v2.0 which accompanies this distribution.
 *
 *     The Eclipse Public License is available at
 *     http://www.eclipse.org/legal/epl-v10.html
 *
 *     The Apache License v2.0 is available at
 *     http://www.opensource.org/licenses/apache2.0.php
 *
 * You may elect to redistribute this code under either of these licenses.
 */
package com.arangodb.shaded.vertx.ext.web.client.impl;

import com.arangodb.shaded.netty.handler.codec.http.QueryStringDecoder;
import com.arangodb.shaded.netty.handler.codec.http.QueryStringEncoder;
import com.arangodb.shaded.vertx.core.AsyncResult;
import com.arangodb.shaded.vertx.core.Handler;
import com.arangodb.shaded.vertx.core.MultiMap;
import com.arangodb.shaded.vertx.core.buffer.Buffer;
import com.arangodb.shaded.vertx.core.http.HttpHeaders;
import com.arangodb.shaded.vertx.core.http.HttpMethod;
import com.arangodb.shaded.vertx.core.http.RequestOptions;
import com.arangodb.shaded.vertx.core.json.JsonObject;
import com.arangodb.shaded.vertx.core.net.ProxyOptions;
import com.arangodb.shaded.vertx.core.net.SocketAddress;
import com.arangodb.shaded.vertx.core.streams.ReadStream;
import com.arangodb.shaded.vertx.ext.auth.authentication.Credentials;
import com.arangodb.shaded.vertx.ext.web.client.HttpRequest;
import com.arangodb.shaded.vertx.ext.web.client.HttpResponse;
import com.arangodb.shaded.vertx.ext.web.client.predicate.ResponsePredicate;
import com.arangodb.shaded.vertx.ext.web.codec.BodyCodec;
import com.arangodb.shaded.vertx.ext.web.multipart.MultipartForm;
import com.arangodb.shaded.vertx.uritemplate.UriTemplate;
import com.arangodb.shaded.vertx.uritemplate.Variables;

import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * @author Julien Viet
 */
public class HttpRequestImpl implements HttpRequest {

  private final WebClientBase client;
  private ProxyOptions proxyOptions;
  private SocketAddress serverAddress;
  private MultiMap queryParams;
  private Variables templateParams;
  private HttpMethod method;
  private final UriTemplate absoluteUri;
  private int port;
  private String host;
  private String virtualHost;
  private Object uri;
  private long timeout = -1;
  private boolean followRedirects;
  private Boolean ssl;
  private boolean multipartMixed = true;
  private String traceOperation;
  private List expectations;
  private BodyCodec codec;
  MultiMap headers;

  HttpRequestImpl(WebClientBase client,
                  HttpMethod method,
                  SocketAddress serverAddress,
                  UriTemplate absoluteUri,
                  BodyCodec codec,
                  boolean followRedirects,
                  ProxyOptions proxyOptions,
                  MultiMap headers) {
    Objects.requireNonNull(absoluteUri, "AbsoluteUri cannot be null");
    this.client = client;
    this.absoluteUri = absoluteUri;
    this.serverAddress = serverAddress;
    this.method = method;
    this.ssl = null;
    this.host = null;
    this.port = -1;
    this.uri = null;
    this.codec = codec;
    this.headers = headers;
    this.followRedirects = followRedirects;
    this.proxyOptions = proxyOptions;
  }

  HttpRequestImpl(WebClientBase client,
                  HttpMethod method,
                  SocketAddress serverAddress,
                  Boolean ssl,
                  int port,
                  String host,
                  Object uri,
                  BodyCodec codec,
                  boolean followRedirects,
                  ProxyOptions proxyOptions,
                  MultiMap headers) {
    Objects.requireNonNull(host, "Host cannot be null");
    this.client = client;
    this.absoluteUri = null;
    this.serverAddress = serverAddress;
    this.method = method;
    this.ssl = ssl;
    this.port = port;
    this.host = host;
    this.uri = uri;
    this.codec = codec;
    this.headers = headers;
    this.followRedirects = followRedirects;
    this.proxyOptions = proxyOptions;
  }

  private HttpRequestImpl(HttpRequestImpl other) {
    this.client = other.client;
    this.absoluteUri = other.absoluteUri;
    this.serverAddress = other.serverAddress;
    this.ssl = other.ssl;
    this.method = other.method;
    this.port = other.port;
    this.host = other.host;
    this.uri = other.uri;
    this.codec = other.codec;
    this.headers = other.headers != null ? HttpHeaders.headers().addAll(other.headers) : HttpHeaders.headers();
    this.followRedirects = other.followRedirects;
    this.proxyOptions = other.proxyOptions != null ? new ProxyOptions(other.proxyOptions) : null;
    this.timeout = other.timeout;
    this.queryParams = other.queryParams != null ? MultiMap.caseInsensitiveMultiMap().addAll(other.queryParams) : null;
    this.multipartMixed = other.multipartMixed;
    this.virtualHost = other.virtualHost;
    this.expectations = other.expectations != null ? new ArrayList<>(other.expectations) : null;
  }

  @Override
  public  HttpRequest as(BodyCodec responseCodec) {
    codec = (BodyCodec) responseCodec;
    return (HttpRequest) this;
  }

  @Override
  public BodyCodec bodyCodec() {
    return codec;
  }

  @Override
  public HttpRequest method(HttpMethod value) {
    method = value;
    return this;
  }

  @Override
  public HttpMethod method() {
    return method;
  }

  @Override
  public HttpRequest ssl(Boolean value) {
    ssl = value;
    return this;
  }

  @Override
  public Boolean ssl() {
    return ssl;
  }

  @Override
  public HttpRequest port(int value) {
    port = value;
    return this;
  }

  @Override
  public int port() {
    return port;
  }

  @Override
  public HttpRequest host(String value) {
    Objects.requireNonNull(host, "Host cannot be null");
    host = value;
    return this;
  }

  @Override
  public String host() {
    return host;
  }

  @Override
  public HttpRequest uri(String value) {
    queryParams = null;
    uri = value;
    return this;
  }

  public String uri() {
    return uri.toString();
  }

  @Override
  public HttpRequest virtualHost(String value) {
    virtualHost = value;
    return this;
  }

  @Override
  public String virtualHost() {
    return virtualHost;
  }

  @Override
  public HttpRequest putHeaders(MultiMap headers) {
    headers().addAll(headers);
    return this;
  }

  @Override
  public HttpRequest putHeader(String name, String value) {
    headers().set(name, value);
    return this;
  }

  @Override
  public HttpRequest putHeader(String name, Iterable value) {
    headers().set(name, value);
    return this;
  }

  @Override
  public MultiMap headers() {
    if (headers == null) {
      headers = HttpHeaders.headers();
    }
    return headers;
  }

  @Override
  public HttpRequest authentication(Credentials credentials) {
    putHeader(
      HttpHeaders.AUTHORIZATION.toString(),
      credentials.toHttpAuthorization());

    return this;
  }

  @Override
  public HttpRequest timeout(long value) {
    timeout = value;
    return this;
  }

  @Override
  public long timeout() {
      return timeout;
  }

  @Override
  public HttpRequest addQueryParam(String paramName, String paramValue) {
    queryParams().add(paramName, paramValue);
    return this;
  }

  @Override
  public HttpRequest setQueryParam(String paramName, String paramValue) {
    queryParams().set(paramName, paramValue);
    return this;
  }

  @Override
  public HttpRequest setTemplateParam(String paramName, String paramValue) {
    templateParams().set(paramName, paramValue);
    return this;
  }

  @Override
  public HttpRequest setTemplateParam(String paramName, List paramValue) {
    templateParams().set(paramName, paramValue);
    return this;
  }

  @Override
  public HttpRequest setTemplateParam(String paramName, Map paramValue) {
    templateParams().set(paramName, paramValue);
    return this;
  }

  @Override
  public HttpRequest followRedirects(boolean value) {
    followRedirects = value;
    return this;
  }

  @Override
  public boolean followRedirects() {
    return followRedirects;
  }

  @Override
  public HttpRequest proxy(ProxyOptions proxyOptions) {
    this.proxyOptions = proxyOptions;
    return this;
  }

  @Override
  public ProxyOptions proxy() {
    return proxyOptions;
  }

  @Override
  public HttpRequest expect(ResponsePredicate expectation) {
    if (expectations == null) {
      expectations = new ArrayList<>();
    }
    expectations.add(expectation);
    return this;
  }

  @Override
  public List expectations() {
    return expectations != null ? expectations : Collections.emptyList();
  }

  @Override
  public MultiMap queryParams() {
    if (queryParams == null) {
      queryParams = MultiMap.caseInsensitiveMultiMap();
      if (uri instanceof String) {
        int idx = ((String)uri).indexOf('?');
        if (idx >= 0) {
          QueryStringDecoder dec = new QueryStringDecoder((String)uri);
          dec.parameters().forEach((name, value) -> queryParams.add(name, value));
          uri = ((String)uri).substring(0, idx);
        }
      }
    }
    return queryParams;
  }

  @Override
  public Variables templateParams() {
    if (!(uri instanceof UriTemplate) && !(absoluteUri instanceof UriTemplate)) {
      throw new IllegalStateException();
    }
    if (templateParams == null) {
      templateParams = Variables.variables();
    }
    return templateParams;
  }

  @Override
  public HttpRequest copy() {
    return new HttpRequestImpl<>(this);
  }

  @Override
  public HttpRequest multipartMixed(boolean allow) {
    multipartMixed = allow;
    return this;
  }

  @Override
  public HttpRequest traceOperation(String traceOperation) {
    this.traceOperation = traceOperation;
    return this;
  }

  @Override
  public String traceOperation() {
    return traceOperation;
  }

  @Override
  public boolean multipartMixed() {
    return multipartMixed;
  }

  @Override
  public void sendStream(ReadStream body, Handler>> handler) {
    send(null, body, handler);
  }

  @Override
  public void send(Handler>> handler) {
    send(null, null, handler);
  }

  @Override
  public void sendBuffer(Buffer body, Handler>> handler) {
    send(null, body, handler);
  }

  @Override
  public void sendJsonObject(JsonObject body, Handler>> handler) {
    send("application/json", body,handler);
  }

  @Override
  public void sendJson(Object body, Handler>> handler) {
    send("application/json", body, handler);
  }

  @Override
  public void sendForm(MultiMap body, Handler>> handler) {
    sendForm(body, "UTF-8", handler);
  }

  @Override
  public void sendForm(MultiMap body, String charset, Handler>> handler) {
    MultipartForm parts = MultipartForm.create();
    for (Map.Entry attribute : body) {
      parts.attribute(attribute.getKey(), attribute.getValue());
    }
    parts.setCharset(charset);
    send("application/x-www-form-urlencoded", parts, handler);
  }

  @Override
  public void sendMultipartForm(MultipartForm body, Handler>> handler) {
    send("multipart/form-data", body, handler);
  }

  RequestOptions buildRequestOptions() throws URISyntaxException, MalformedURLException {

    String protocol = null;
    Boolean ssl = null;
    int port = -1;
    String host = null;
    String uri = null;
    if (absoluteUri != null) {
      uri = absoluteUri.expandToString(templateParams(), client.options.getTemplateExpandOptions());
      ClientUri curi = ClientUri.parse(uri);
      uri = curi.uri;
      host = curi.host;
      port = curi.port;
      protocol = curi.protocol;
      ssl = curi.ssl;
    }
    if (this.ssl != null) {
      ssl = this.ssl;
    }
    if (this.port >= 0) {
      port = this.port;
    }
    if (this.host != null) {
      host = this.host;
    }
    if (this.uri != null) {
      if (this.uri instanceof String) {
        uri = (String) this.uri;
      } else {
        uri = ((UriTemplate) this.uri).expandToString(templateParams(), client.options.getTemplateExpandOptions());
      }
    }
    if (queryParams != null) {
      uri = buildUri(uri, queryParams);
    }

    RequestOptions requestOptions = new RequestOptions();
    if (protocol != null && !protocol.equals("http") && !protocol.equals("https")) {
      // we have to create an abs url again to parse it in HttpClient
      URI tmp = new URI(protocol, null, host, port, uri, null, null);
      requestOptions.setServer(this.serverAddress)
        .setMethod(this.method)
        .setAbsoluteURI(tmp.toString());
    } else {
      requestOptions.setServer(this.serverAddress)
        .setMethod(this.method)
        .setHost(host)
        .setPort(port)
        .setURI(uri);
      // if the user specified SSL we always enforce it
      // even if the client has a default, because the default
      // may have been used previously to compute the request options
      if (ssl != null) {
        requestOptions
          .setSsl(ssl);
      }
    }
    if (this.virtualHost != null) {
      if (requestOptions.getServer() == null) {
        requestOptions.setServer(SocketAddress.inetSocketAddress(requestOptions.getPort(), requestOptions.getHost()));
      }
      requestOptions.setHost(this.virtualHost);
    }
    this.mergeHeaders(requestOptions);
    requestOptions.setTimeout(this.timeout);
    requestOptions.setProxyOptions(this.proxyOptions);
    requestOptions.setTraceOperation(this.traceOperation);
    return requestOptions;
  }

  void send(String contentType, Object body, Handler>> handler) {
    HttpContext ctx = client.createContext(handler);
    ctx.prepareRequest(this, contentType, body);
  }

  void mergeHeaders(RequestOptions options) {
    if (headers != null) {
      MultiMap tmp = options.getHeaders();
      if (tmp == null) {
        tmp = MultiMap.caseInsensitiveMultiMap();
        options.setHeaders(tmp);
      }
      tmp.addAll(headers);
    }
  }

  private static String buildUri(String uri, MultiMap queryParams) {
    QueryStringDecoder decoder = new QueryStringDecoder(uri);
    QueryStringEncoder encoder = new QueryStringEncoder(decoder.rawPath());
    decoder.parameters().forEach((name, values) -> {
      for (String value : values) {
        encoder.addParam(name, value);
      }
    });
    queryParams.forEach(param -> {
      encoder.addParam(param.getKey(), param.getValue());
    });
    uri = encoder.toString();
    return uri;
  }

}