From 49f439e761a5c6bf27310db8d706ce9af8574fc3 Mon Sep 17 00:00:00 2001 From: Kavin <20838718+FireMasterK@users.noreply.github.com> Date: Fri, 24 Dec 2021 14:17:02 +0000 Subject: [PATCH] Add null-check for SponsorBlock endpoint (#154) --- src/main/java/me/kavin/piped/utils/SponsorBlockUtils.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/main/java/me/kavin/piped/utils/SponsorBlockUtils.java b/src/main/java/me/kavin/piped/utils/SponsorBlockUtils.java index 00e1a88..b971266 100644 --- a/src/main/java/me/kavin/piped/utils/SponsorBlockUtils.java +++ b/src/main/java/me/kavin/piped/utils/SponsorBlockUtils.java @@ -8,6 +8,8 @@ import java.nio.charset.StandardCharsets; import java.security.MessageDigest; import java.security.NoSuchAlgorithmException; +import org.apache.commons.lang3.StringUtils; + import com.grack.nanojson.JsonArray; import com.grack.nanojson.JsonObject; import com.grack.nanojson.JsonParser; @@ -15,12 +17,16 @@ import com.grack.nanojson.JsonParserException; import com.grack.nanojson.JsonWriter; import me.kavin.piped.consts.Constants; +import me.kavin.piped.utils.resp.InvalidRequestResponse; public class SponsorBlockUtils { public static final String getSponsors(String id, String categories) throws IOException, InterruptedException, NoSuchAlgorithmException, JsonParserException { + if (StringUtils.isEmpty(categories)) + return Constants.mapper.writeValueAsString(new InvalidRequestResponse()); + String hash = toSha256(id); URI uri = URI.create("https://sponsor.ajay.app/api/skipSegments/" + URLUtils.silentEncode(hash.substring(0, 4))