Bat/src/main/java/cc/fascinated/bat/common/WebRequest.java

package cc.fascinated.bat.common;

import cc.fascinated.bat.exception.RateLimitException;
import lombok.experimental.UtilityClass;
import org.springframework.http.HttpStatus;
import org.springframework.http.HttpStatusCode;
import org.springframework.http.ResponseEntity;
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.web.client.RestClient;

@UtilityClass
public class WebRequest {
    /**
     * The web client.
     */
    private static final RestClient CLIENT;

    static {
        HttpComponentsClientHttpRequestFactory requestFactory = new HttpComponentsClientHttpRequestFactory();
        requestFactory.setConnectTimeout(2500); // 2.5 seconds
        CLIENT = RestClient.builder()
                .requestFactory(requestFactory)
                .build();
    }

    /**
     * Gets a response from the given URL.
     *
     * @param url the url
     * @param <T> the type of the response
     * @return the response
     */
    public static <T> T getAsEntity(String url, Class<T> clazz) throws RateLimitException {
        try {
            ResponseEntity<T> responseEntity = CLIENT.get()
                    .uri(url)
                    .retrieve()
                    .onStatus(HttpStatusCode::isError, (request, response) -> {
                    }) // Don't throw exceptions on error
                    .toEntity(clazz);

            if (responseEntity.getStatusCode().isError()) {
                return null;
            }
            if (responseEntity.getStatusCode().isSameCodeAs(HttpStatus.TOO_MANY_REQUESTS)) {
                throw new RateLimitException("Rate limit reached");
            }
            return responseEntity.getBody();
        } catch (RateLimitException e) {
            throw e;
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * Gets a response from the given URL.
     *
     * @param url the url
     * @return the response
     */
    public static ResponseEntity<?> get(String url, Class<?> clazz) {
        return CLIENT.get()
                .uri(url)
                .retrieve()
                .onStatus(HttpStatusCode::isError, (request, response) -> {
                }) // Don't throw exceptions on error
                .toEntity(clazz);
    }

    /**
     * Gets a response from the given URL.
     *
     * @param url the url
     * @return the response
     */
    public static ResponseEntity<?> head(String url, Class<?> clazz) {
        return CLIENT.head()
                .uri(url)
                .retrieve()
                .onStatus(HttpStatusCode::isError, (request, response) -> {
                }) // Don't throw exceptions on error
                .toEntity(clazz);
    }
}