Handle youtube /live/ URLs.

lavalink-devs · Dec 25, 2023 · f55fb79 · f55fb79
1 parent 53eb716
commit f55fb79
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...c/main/java/com/sedmelluq/discord/lavaplayer/source/youtube/DefaultYoutubeLinkRouter.java b/...c/main/java/com/sedmelluq/discord/lavaplayer/source/youtube/DefaultYoutubeLinkRouter.java
@@ -29,7 +29,8 @@ public class DefaultYoutubeLinkRouter implements YoutubeLinkRouter {
         new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + DOMAIN_REGEX + "/.*"), this::routeFromMainDomain),
         new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + SHORT_DOMAIN_REGEX + "/.*"), this::routeFromShortDomain),
         new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + DOMAIN_REGEX + "/embed/.*"), this::routeFromEmbed),
-        new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + DOMAIN_REGEX + "/shorts/.*"), this::routeFromShorts)
+        new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + DOMAIN_REGEX + "/shorts/.*"), this::routeFromShorts),
+        new Extractor(Pattern.compile("^" + PROTOCOL_REGEX + DOMAIN_REGEX + "/live/.*"), this::routeFromShortPath)
     };
 
     @Override
@@ -118,6 +119,12 @@ protected <T> T routeFromShorts(Routes<T> routes, String url) {
         return routeFromUrlWithVideoId(routes, urlInfo.path.substring(8), urlInfo);
     }
 
+    protected <T> T routeFromShortPath(Routes<T> routes, String url) {
+        UrlInfo urlInfo = getUrlInfo(url, true);
+        String[] pathParts = urlInfo.path.split("/");
+        return routeFromUrlWithVideoId(routes, pathParts[pathParts.length - 1], urlInfo);
+    }
+
     private static UrlInfo getUrlInfo(String url, boolean retryValidPart) {
         try {
             if (!url.startsWith("http://") && !url.startsWith("https://")) {