RawEventsClient.java

/**
 * This file was auto-generated by Fern from our API Definition.
 */
package com.auth0.client.mgmt;

import com.auth0.client.mgmt.core.ClientOptions;
import com.auth0.client.mgmt.core.ManagementApiException;
import com.auth0.client.mgmt.core.ManagementApiHttpResponse;
import com.auth0.client.mgmt.core.ManagementException;
import com.auth0.client.mgmt.core.ObjectMappers;
import com.auth0.client.mgmt.core.QueryStringMapper;
import com.auth0.client.mgmt.core.RequestOptions;
import com.auth0.client.mgmt.core.ResponseBodyReader;
import com.auth0.client.mgmt.core.Stream;
import com.auth0.client.mgmt.errors.BadRequestError;
import com.auth0.client.mgmt.errors.ForbiddenError;
import com.auth0.client.mgmt.errors.GoneError;
import com.auth0.client.mgmt.errors.NotFoundError;
import com.auth0.client.mgmt.errors.TooManyRequestsError;
import com.auth0.client.mgmt.errors.UnauthorizedError;
import com.auth0.client.mgmt.types.EventStreamSubscribeEventsResponseContent;
import com.auth0.client.mgmt.types.SubscribeEventsRequestParameters;
import com.fasterxml.jackson.core.JsonProcessingException;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
import okhttp3.Headers;
import okhttp3.HttpUrl;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import okhttp3.ResponseBody;

public class RawEventsClient {
    protected final ClientOptions clientOptions;

    public RawEventsClient(ClientOptions clientOptions) {
        this.clientOptions = clientOptions;
    }

    /**
     * Subscribe to events via Server-Sent Events (SSE)
     */
    public ManagementApiHttpResponse<Iterable<EventStreamSubscribeEventsResponseContent>> subscribe() {
        return subscribe(SubscribeEventsRequestParameters.builder().build());
    }

    /**
     * Subscribe to events via Server-Sent Events (SSE)
     */
    public ManagementApiHttpResponse<Iterable<EventStreamSubscribeEventsResponseContent>> subscribe(
            RequestOptions requestOptions) {
        return subscribe(SubscribeEventsRequestParameters.builder().build(), requestOptions);
    }

    /**
     * Subscribe to events via Server-Sent Events (SSE)
     */
    public ManagementApiHttpResponse<Iterable<EventStreamSubscribeEventsResponseContent>> subscribe(
            SubscribeEventsRequestParameters request) {
        return subscribe(request, null);
    }

    /**
     * Subscribe to events via Server-Sent Events (SSE)
     */
    public ManagementApiHttpResponse<Iterable<EventStreamSubscribeEventsResponseContent>> subscribe(
            SubscribeEventsRequestParameters request, RequestOptions requestOptions) {
        HttpUrl.Builder httpUrl = HttpUrl.parse(this.clientOptions.environment().getUrl())
                .newBuilder()
                .addPathSegments("events");
        if (!request.getFrom().isAbsent()) {
            QueryStringMapper.addQueryParameter(
                    httpUrl, "from", request.getFrom().orElse(null), false);
        }
        if (!request.getFromTimestamp().isAbsent()) {
            QueryStringMapper.addQueryParameter(
                    httpUrl, "from_timestamp", request.getFromTimestamp().orElse(null), false);
        }
        if (request.getEventType().isPresent()) {
            QueryStringMapper.addQueryParameter(
                    httpUrl, "event_type", request.getEventType().get(), true);
        }
        if (requestOptions != null) {
            requestOptions.getQueryParameters().forEach((_key, _value) -> {
                httpUrl.addQueryParameter(_key, _value);
            });
        }
        Request.Builder _requestBuilder = new Request.Builder()
                .url(httpUrl.build())
                .method("GET", null)
                .headers(Headers.of(clientOptions.headers(requestOptions)));
        Request okhttpRequest = _requestBuilder.build();
        OkHttpClient client = clientOptions.httpClient();
        if (requestOptions != null && requestOptions.getTimeout().isPresent()) {
            client = clientOptions.httpClientWithTimeout(requestOptions);
        }
        client = client.newBuilder().callTimeout(0, TimeUnit.SECONDS).build();
        try {
            Response response = client.newCall(okhttpRequest).execute();
            ResponseBody responseBody = response.body();
            if (response.isSuccessful()) {
                return new ManagementApiHttpResponse<>(
                        Stream.fromSseWithEventDiscrimination(
                                EventStreamSubscribeEventsResponseContent.class,
                                new ResponseBodyReader(response),
                                "type"),
                        response);
            }
            String responseBodyString = responseBody != null ? responseBody.string() : "{}";
            try {
                switch (response.code()) {
                    case 400:
                        throw new BadRequestError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                    case 401:
                        throw new UnauthorizedError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                    case 403:
                        throw new ForbiddenError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                    case 404:
                        throw new NotFoundError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                    case 410:
                        throw new GoneError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                    case 429:
                        throw new TooManyRequestsError(
                                ObjectMappers.JSON_MAPPER.readValue(responseBodyString, Object.class), response);
                }
            } catch (JsonProcessingException ignored) {
                // unable to map error response, throwing generic error
            }
            Object errorBody = ObjectMappers.parseErrorBody(responseBodyString);
            throw new ManagementApiException(
                    "Error with status code " + response.code(), response.code(), errorBody, response);
        } catch (IOException e) {
            throw new ManagementException("Network error executing HTTP request", e);
        }
    }
}