Merge pull request #25 from akshatjainei/test102

Added support for URL Validation
SaptarshiSarkar12 · Oct 1, 2022 · a1f4304 · a1f4304
2 parents 2735ac1 + 6d70df4
commit a1f4304
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/Drifty_CLI.java b/src/Drifty_CLI.java
@@ -1,3 +1,4 @@
+import java.net.URL;
 import java.util.Scanner;
 
 public class Drifty_CLI {
@@ -24,6 +25,16 @@ public static void main(String[] args) {
         while(true) {
             System.out.print("Enter the link to the file : ");
             String link = SC.next();
+            while (true) {
+                if (!validURL(link)) {
+                    System.out.println("Invalid URL. Please enter again");
+                    System.out.print("Enter the link to the file : ");
+                    link = SC.next();
+                }
+                else{
+                    break;
+                }
+            }
             SC.nextLine();
             System.out.print("Enter the filename (with file extension) : ");
             String fName = SC.nextLine();
@@ -69,4 +80,14 @@ private static void enterDownloadsFolder(){
             downloadsFolder = downloadsFolder + System.getProperty("file.separator");
         }
     }
+
+    private static boolean validURL(String link){
+        try{
+            new URL(link).toURI();
+            return true;
+        }
+        catch (Exception e){
+            return false;
+        }
+    }
 }