package net.minecraft.server.jsonrpc;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import java.util.List;
import net.minecraft.resources.Identifier;
import net.minecraft.util.GsonHelper;
import org.jspecify.annotations.Nullable;

public class JsonRPCUtils {
	public static final String JSON_RPC_VERSION = "2.0";
	public static final String OPEN_RPC_VERSION = "1.3.2";

	public static JsonObject createSuccessResult(final JsonElement id, final JsonElement result) {
		JsonObject response = new JsonObject();
		response.addProperty("jsonrpc", "2.0");
		response.add("id", id);
		response.add("result", result);
		return response;
	}

	public static JsonObject createRequest(@Nullable final Integer id, final Identifier method, final List<JsonElement> params) {
		JsonObject request = new JsonObject();
		request.addProperty("jsonrpc", "2.0");
		if (id != null) {
			request.addProperty("id", id);
		}

		request.addProperty("method", method.toString());
		if (!params.isEmpty()) {
			JsonArray jsonArray = new JsonArray(params.size());

			for (JsonElement param : params) {
				jsonArray.add(param);
			}

			request.add("params", jsonArray);
		}

		return request;
	}

	public static JsonObject createError(final JsonElement id, final String message, final int errorCode, @Nullable final String data) {
		JsonObject errorResponse = new JsonObject();
		errorResponse.addProperty("jsonrpc", "2.0");
		errorResponse.add("id", id);
		JsonObject error = new JsonObject();
		error.addProperty("code", errorCode);
		error.addProperty("message", message);
		if (data != null && !data.isBlank()) {
			error.addProperty("data", data);
		}

		errorResponse.add("error", error);
		return errorResponse;
	}

	@Nullable
	public static JsonElement getRequestId(final JsonObject jsonObject) {
		return jsonObject.get("id");
	}

	@Nullable
	public static String getMethodName(final JsonObject jsonObject) {
		return GsonHelper.getAsString(jsonObject, "method", null);
	}

	@Nullable
	public static JsonElement getParams(final JsonObject jsonObject) {
		return jsonObject.get("params");
	}

	@Nullable
	public static JsonElement getResult(final JsonObject jsonObject) {
		return jsonObject.get("result");
	}

	@Nullable
	public static JsonObject getError(final JsonObject jsonObject) {
		return GsonHelper.getAsJsonObject(jsonObject, "error", null);
	}
}