How to use com.hotels.styx.api

/**
 * Creates a builder with an HTTP method and URI.
 *
 * @param method HTTP method
 * @param uri    URI
 */
public Builder(HttpMethod method, String uri) {
  this.method = method;
  this.url = Url.Builder.url(uri).build();
  this.secure = url.isSecure();
  headers(new HttpHeaders.Builder());
}

public static LiveHttpRequest doOnComplete(LiveHttpRequest request, Runnable action) {
  return request.newBuilder()
      .body(it -> it.doOnEnd(ifSuccessful(action)))
      .build();
}

public static LiveHttpResponse doOnError(LiveHttpResponse response, Consumer<Throwable> action) {
  return response.newBuilder()
      .body(it -> it.doOnEnd(ifError(action)))
      .build();
}

@Override
public Eventual<LiveHttpResponse> intercept(LiveHttpRequest request, Chain chain) {
  LiveHttpRequest newRequest = request.newBuilder()
      .header(VIA, viaHeader(request))
      .build();
  return chain.proceed(newRequest)
      .map(response -> response.newBuilder()
          .header(VIA, viaHeader(response))
          .build());
}

Builder(LiveHttpRequest request) {
  this.id = request.id();
  this.method = request.method();
  this.url = request.url();
  this.version = request.version();
  this.headers = request.headers().newBuilder();
  this.body = request.body();
}

private static HttpResponse responseWithoutBody(HttpResponse response) {
  return response.newBuilder()
      .header(CONTENT_LENGTH, 0)
      .removeHeader(TRANSFER_ENCODING)
      .removeBody()
      .build();
}

@Override
public Eventual<LiveHttpResponse> intercept(LiveHttpRequest request, Chain chain) {
  if (CONNECT.equals(request.method())) {
    return Eventual.of(response(METHOD_NOT_ALLOWED).build());
  } else {
    return chain.proceed(request);
  }
}

public SELF headers(HttpHeader... headers) {
  for (HttpHeader header : headers) {
    checkNotCookie(header.name());
    this.headers.set(header.name(), header.value());
  }
  return (SELF) this;
}

private static Eventual<PluginEnabledState> requestedNewState(LiveHttpRequest request) {
  return request.aggregate(MAX_CONTENT_SIZE)
      .map(fullRequest -> fullRequest.bodyAs(UTF_8))
      .map(PluginToggleHandler::parseToBoolean)
      .map(PluginEnabledState::fromBoolean);
}

private static HttpRequest addUserAgent(HttpRequest request, String userAgent) {
  return request.newBuilder()
      .addHeader(USER_AGENT, userAgent)
      .build();
}

/**
 * Set the response to be chunked.
 *
 * @return {@code this}
 */
public SELF chunked() {
  headers().add(TRANSFER_ENCODING, CHUNKED);
  headers().remove(CONTENT_LENGTH);
  return (SELF) this;
}

/**
 * Returns the value of the header with the specified {@code name}.
 * If there is more than one header value for the specified header name, the first value is returned.
 *
 * @return the value of the header with the specified {@code name} if present
 */
default Optional<String> header(CharSequence name) {
  return headers().get(name);
}

/**
 * Enables Keep-Alive.
 *
 * @return {@code this}
 */
public Builder enableKeepAlive() {
  this.headers().add(CONNECTION, KEEP_ALIVE);
  return this;
}

/**
 * Decodes the "Set-Cookie" headers in this response and returns the cookies.
 *
 * @return a set of cookies
 */
public Set<ResponseCookie> cookies() {
  return decode(headers.getAll(SET_COOKIE));
}

/**
 * Returns the values of the headers with the specified {@code name}.
 *
 * @param name the name of the headers
 * @return A {@link List} of header values which will be empty if no values are found
 */
default List<String> headers(CharSequence name) {
  return headers().getAll(name);
}

/**
 * Returns the values of the headers with the specified {@code name}.
 *
 * @param name the name of the headers
 * @return A {@link List} of header values which will be empty if no values are found
 */
default List<String> headers(CharSequence name) {
  return headers().getAll(name);
}

public static LiveHttpRequest doOnError(LiveHttpRequest request, Consumer<Throwable> action) {
  return request.newBuilder()
      .body(it -> it.doOnEnd(ifError(action)))
      .build();
}

public static LiveHttpResponse doOnComplete(LiveHttpResponse response, Runnable action) {
  return response.newBuilder()
      .body(it -> it.doOnEnd(ifSuccessful(action)))
      .build();
}

public static LiveHttpRequest doFinally(LiveHttpRequest request, Consumer<Optional<Throwable>> action) {
  return request.newBuilder()
      .body(it -> it.doOnEnd(action))
      .build();
}

public static LiveHttpResponse doFinally(LiveHttpResponse response, Consumer<Optional<Throwable>> action) {
  return response.newBuilder()
      .body(it -> it.doOnEnd(action::accept))
      .build();
}