xyz.gianlu.librespot.api.Utils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of librespot-api Show documentation
There is a newer version: 1.6.4
package xyz.gianlu.librespot.api;

import io.undertow.server.HttpServerExchange;
import io.undertow.util.QueryParameterUtils;
import io.undertow.util.StatusCodes;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Deque;
import java.util.Map;

public final class Utils {
    private static final String INVALID_PARAM_BODY = "{\"name\":\"%s\"}";
    private static final String INTERNAL_ERROR_BODY = "{\"msg\":\"%s\"}";
    private static final String INVALID_PARAM_WITH_REASON_BODY = "{\"name\":\"%s\",\"reason\":\"%s\"}";

    private Utils() {
    }

    @NotNull
    public static Map> readParameters(@NotNull HttpServerExchange exchange) throws IOException {
        Map> map = exchange.getQueryParameters();
        if (!exchange.isInIoThread()) {
            String body;
            try (InputStream in = exchange.getInputStream()) {
                ByteArrayOutputStream out = new ByteArrayOutputStream();
                byte[] buffer = new byte[1024];
                int count;
                while ((count = in.read(buffer)) > 0) out.write(buffer, 0, count);
                body = new String(out.toByteArray());
            }

            return QueryParameterUtils.mergeQueryParametersWithNewQueryString(map, body, "UTF-8");
        } else {
            return map;
        }
    }

    @Nullable
    public static String getFirstString(@NotNull Map> params, @NotNull String key) {
        Deque q = params.get(key);
        return q == null ? null : q.getFirst();
    }

    public static boolean getFirstBoolean(@NotNull Map> params, @NotNull String key) {
        return Boolean.parseBoolean(getFirstString(params, key));
    }

    public static void invalidParameter(@NotNull HttpServerExchange exchange, @NotNull String name) {
        exchange.setStatusCode(StatusCodes.BAD_REQUEST);
        exchange.getResponseSender().send(String.format(INVALID_PARAM_BODY, name));
    }

    public static void invalidParameter(@NotNull HttpServerExchange exchange, @NotNull String name, @NotNull String reason) {
        exchange.setStatusCode(StatusCodes.BAD_REQUEST);
        exchange.getResponseSender().send(String.format(INVALID_PARAM_WITH_REASON_BODY, name, reason));
    }

    public static void internalError(@NotNull HttpServerExchange exchange, @NotNull Exception ex) {
        internalError(exchange, ex.getMessage());
    }

    public static void internalError(@NotNull HttpServerExchange exchange, @NotNull String reason) {
        exchange.setStatusCode(StatusCodes.INTERNAL_SERVER_ERROR);
        exchange.getResponseSender().send(String.format(INTERNAL_ERROR_BODY, reason));
    }
}