com.microsoft.graph.content.MSBatchRequestContent Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of com.liferay.document.library.opener.onedrive.web
Liferay Document Library Opener OneDrive Web
There is a newer version: 3.0.61
Show newest version
package com.microsoft.graph.content;

import java.io.IOException;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ThreadLocalRandom;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.google.gson.JsonPrimitive;
import com.google.gson.JsonParseException;

import okhttp3.Headers;
import okhttp3.Request;
import okhttp3.RequestBody;
import okio.Buffer;

public class MSBatchRequestContent {
	private final LinkedHashMap batchRequestStepsHashMap;

	// Maximum number of requests that can be sent in a batch
	public static final int MAX_NUMBER_OF_REQUESTS = 20;

	/*
	 * Creates Batch request content using list provided
	 *
	 * @param batchRequestStepsArray List of batch steps for batching
	 */
	public MSBatchRequestContent(final List batchRequestStepsArray) {
		if (batchRequestStepsArray.size() > MAX_NUMBER_OF_REQUESTS)
			throw new IllegalArgumentException("Number of batch request steps cannot exceed " + MAX_NUMBER_OF_REQUESTS);

		this.batchRequestStepsHashMap = new LinkedHashMap<>();
		for (final MSBatchRequestStep requestStep : batchRequestStepsArray)
			addBatchRequestStep(requestStep);
	}

	/*
	 * Creates empty batch request content
	 */
	public MSBatchRequestContent() {
		this.batchRequestStepsHashMap = new LinkedHashMap<>();
	}

	/*
	 * @param batchRequestStep Batch request step adding to batch content
	 *
	 * @return true or false based on addition or no addition of batch request step
	 * given
	 */
	public boolean addBatchRequestStep(final MSBatchRequestStep batchRequestStep) {
		if (batchRequestStepsHashMap.containsKey(batchRequestStep.getRequestId()) ||
			batchRequestStepsHashMap.size() >= MAX_NUMBER_OF_REQUESTS)
			return false;
		batchRequestStepsHashMap.put(batchRequestStep.getRequestId(), batchRequestStep);
		return true;
	}

	/**
	 * Add steps to batch from OkHttp.Request
	 * @param request the request to add to the batch
	 * @param arrayOfDependsOnIds ids of steps this step depends on
	 * @return the step id
	 */
	public String addBatchRequestStep(final Request request, final String... arrayOfDependsOnIds) {
		String requestId;
		do {
			requestId = Integer.toString(ThreadLocalRandom.current().nextInt(1, Integer.MAX_VALUE));
		} while(batchRequestStepsHashMap.keySet().contains(requestId));
		if(addBatchRequestStep(new MSBatchRequestStep(requestId, request, Arrays.asList(arrayOfDependsOnIds))))
			return requestId;
		else
			throw new IllegalArgumentException("unable to add step to batch. Number of batch request steps cannot exceed " + MAX_NUMBER_OF_REQUESTS);
	}

	/*
	 * @param requestId Id of Batch request step to be removed
	 *
	 * @return true or false based on removal or no removal of batch request step
	 * with given id
	 */
	public boolean removeBatchRequestStepWithId(final String requestId) {
		boolean removed = false;
		if (batchRequestStepsHashMap.containsKey(requestId)) {
			batchRequestStepsHashMap.remove(requestId);
			removed = true;
			for (final Map.Entry steps : batchRequestStepsHashMap.entrySet()) {
				if (steps.getValue() != null && steps.getValue().getArrayOfDependsOnIds() != null) {
					while (steps.getValue().getArrayOfDependsOnIds().remove(requestId))
						;
				}
			}
		}
		return removed;
	}

	/*
	 * @return Batch request content's json as String
	 */
	public String getBatchRequestContent() {
		final JsonObject batchRequestContentMap = new JsonObject();
		final JsonArray batchContentArray = new JsonArray();
		for (final Map.Entry requestStep : batchRequestStepsHashMap.entrySet()) {
			batchContentArray.add(getBatchRequestObjectFromRequestStep(requestStep.getValue()));
		}
		batchRequestContentMap.add("requests", batchContentArray);

		final String content = batchRequestContentMap.toString();
		return content;
	}

	private static final Pattern protocolAndHostReplacementPattern = Pattern.compile("(?i)^http[s]?:\\/\\/graph\\.microsoft\\.com\\/(?>v1\\.0|beta)\\/?"); // (?i) case insensitive
	private JsonObject getBatchRequestObjectFromRequestStep(final MSBatchRequestStep batchRequestStep) {
		final JsonObject contentmap = new JsonObject();
		contentmap.add("id", new JsonPrimitive(batchRequestStep.getRequestId()));

        final Matcher protocolAndHostReplacementMatcher = protocolAndHostReplacementPattern.matcher(batchRequestStep.getRequest().url().toString());

		final String url =  protocolAndHostReplacementMatcher.replaceAll("");
		contentmap.add("url", new JsonPrimitive(url));

		contentmap.add("method", new JsonPrimitive(batchRequestStep.getRequest().method().toString()));

		final Headers headers = batchRequestStep.getRequest().headers();
		if (headers != null && headers.size() != 0) {
			final JsonObject headerMap = new JsonObject();
			for (final Map.Entry> entry : headers.toMultimap().entrySet()) {
				headerMap.add(entry.getKey(), new JsonPrimitive(getHeaderValuesAsString(entry.getValue())));
			}
			contentmap.add("headers", headerMap);
		}

		final List arrayOfDependsOnIds = batchRequestStep.getArrayOfDependsOnIds();
		if (arrayOfDependsOnIds != null) {
			final JsonArray array = new JsonArray();
			for (final String dependsOnId : arrayOfDependsOnIds)
				array.add(dependsOnId);
			contentmap.add("dependsOn", array);
		}

		final RequestBody body = batchRequestStep.getRequest().body();
		if (body != null) {
			try {
				contentmap.add("body", requestBodyToJSONObject(batchRequestStep.getRequest()));
			} catch (IOException | JsonParseException e) {
				e.printStackTrace();
			}
		}
		return contentmap;
	}

	private String getHeaderValuesAsString(final List list) {
		if (list == null || list.size() == 0)
			return "";
		final StringBuilder builder = new StringBuilder(list.get(0));
		for (int i = 1; i < list.size(); i++) {
			builder.append(";");
			builder.append(list.get(i));
		}
		return builder.toString();
	}

	private JsonObject requestBodyToJSONObject(final Request request) throws IOException, JsonParseException {
		if (request == null || request.body() == null)
			return null;
		final Request copy = request.newBuilder().build();
		final Buffer buffer = new Buffer();
		copy.body().writeTo(buffer);
		final String requestBody = buffer.readUtf8();
		if(requestBody == null || requestBody == "")
			return null;
		final JsonElement requestBodyElement = JsonParser.parseString(requestBody);
		if(requestBodyElement == null || !requestBodyElement.isJsonObject())
			return null;
		else
			return requestBodyElement.getAsJsonObject();
	}

}