com.browserup.bup.mitmproxy.management.AddonsManagerClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of browserup-proxy-core Show documentation
BrowserUp Proxy - Core
There is a newer version: 3.0.0
package com.browserup.bup.mitmproxy.management;

import com.fasterxml.jackson.databind.ObjectMapper;
import okhttp3.*;
import org.apache.commons.lang3.tuple.Pair;

import java.io.IOException;
import java.util.List;
import java.util.Objects;

public class AddonsManagerClient {
    private final int port;
    private final String host = "localhost";

    public AddonsManagerClient(int port) {
        this.port = port;
    }

    public  T putRequestToAddonsManager(String addOnPath,
                                        String operation,
                                        List> queryParams,
                                        RequestBody requestBody,
                                        Class responseClass) {
        return requestToAddonsManager(addOnPath, operation, queryParams, "PUT", requestBody, responseClass);
    }

    public  T getRequestToAddonsManager(String addOnPath,
                                        String operation,
                                        List> queryParams,
                                        Class responseClass) {
        return requestToAddonsManager(addOnPath, operation, queryParams, "GET", null, responseClass);
    }

    public  T requestToAddonsManager(String addOnPath,
                                        String operation,
                                        List> queryParams,
                                        String method,
                                        RequestBody requestBody,
                                        Class responseClass) {
        OkHttpClient client = new OkHttpClient();
        Request request = new Request.Builder()
                .url(buildRequestUrl(addOnPath, operation, queryParams))
                .method(method, requestBody)
                .build();

        Response response;
        try {
            response = client.newCall(request).execute();
        } catch (IOException ex) {
            throw new RuntimeException("Failed to request manager API", ex);
        }

        try {
            if (responseClass.equals(Void.class)) return null;

            if (responseClass.equals(String.class)) return (T) response.body().string();

            return new ObjectMapper().readerFor(responseClass).readValue(Objects.requireNonNull(response.body()).byteStream());
        } catch (IOException e) {
            throw new RuntimeException("Failed to parse response from manager API", e);
        } finally {
            ResponseBody body = response.body();
            if (body != null) {
                body.close();
            }
        }
    }

    private HttpUrl buildRequestUrl(String addOnPath, String operation, List> queryParams) {
        HttpUrl.Builder builder = new HttpUrl.Builder()
                .host(host)
                .port(port)
                .scheme("http")
                .addPathSegment(addOnPath)
                .addPathSegment(operation);

        queryParams.forEach(p -> builder.addQueryParameter(p.getKey(), p.getValue()));

        return builder.build();
    }
}