Fix: Added code to Standardized YouTube link format for issue Saptars…

…hiSarkar12#601
Tarunkumarkanakam · Sep 2, 2024 · a3ea4d5 · a3ea4d5
1 parent ce5a41f
commit a3ea4d5
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/CLI/src/main/java/main/Drifty_CLI.java b/CLI/src/main/java/main/Drifty_CLI.java
@@ -274,6 +274,38 @@ public static void main(String[] args) {
                     if (isInstagram(link)) {
                         link = formatInstagramLink(link);
                     }
+                    else if (isYoutube(link)) {
+                        String videoId = null;
+
+                        try {
+                            URI uri = URI.create(link);
+                            String domain = uri.getHost();
+
+                            // checking if the domain is youtu.be
+                            if ("youtu.be".equals(domain)) {
+                                String path = uri.getPath();
+                                if (path != null && path.length() > 1) {
+                                    videoId = path.substring(1); // removing the leading "/"
+                                }
+                            }
+                            // checking if the domain is youtube.com
+                            else if ("www.youtube.com".equals(domain) || "youtube.com".equals(domain)) {
+                                Map<String, String> queryParams = extractQueryParams(link, "v");
+                                videoId = queryParams.get("v");
+                            }
+
+                            if (videoId != null) {
+                                // constructing link in youtube.com/watch?v={videoID}
+                                uri = new URI("https", "www.youtube.com", "/watch", "v=" + videoId, null);
+                                link = uri.toString();
+                            } else {
+                                messageBroker.msgLinkError("YouTube video ID not found in the link.");
+                            }
+
+                        } catch (IllegalArgumentException | URISyntaxException e) {
+                            messageBroker.msgLinkError("Failed to process the YouTube link: " + e.getMessage());
+                        }
+                    }
                     messageBroker.msgFilenameInfo("Retrieving filename from link...");
                     fileName = findFilenameInLink(link);
                     if (!Objects.requireNonNull(fileName).isEmpty()) {