Add search functionality

2 files changed, 23 insertions, 1 deletions

diff --git a/src/main/java/musichub/business/MusicHub.java b/src/main/java/musichub/business/MusicHub.java
index f7cf137..add55e6 100644
--- a/src/main/java/musichub/business/MusicHub.java
+++ b/src/main/java/musichub/business/MusicHub.java
@@ -339,4 +339,20 @@ public class MusicHub {
         }
 
     }
+
+    public void searchAudioElement() throws UnsupportedAudioFileException, NoAlbumFoundException, LineUnavailableException, IOException {
+        Scanner scanner = new Scanner(System.in);
+        String word = scanner.next().toLowerCase(Locale.ROOT);
+        List<AudioElement> searchResult = new ArrayList<>();
+        for(AudioElement el : elements){
+            if(el.getTitle().toLowerCase(Locale.ROOT).contains(word) || el.getArtist().toLowerCase(Locale.ROOT).contains(word)){
+                searchResult.add(el);
+                System.out.println(el);
+            }
+        }
+        if (searchResult.size()==1){
+            this.getAudioElement(searchResult, searchResult.get(0).getTitle());
+        }
+    }
+
 }
\ No newline at end of file
diff --git a/src/main/java/musichub/main/Main.java b/src/main/java/musichub/main/Main.java
index e2e4731..480a2c2 100644
--- a/src/main/java/musichub/main/Main.java
+++ b/src/main/java/musichub/main/Main.java
@@ -12,7 +12,8 @@ import java.util.Scanner;
 import static musichub.business.PathValidation.isPathValid;
 
 public class Main {
-    public static void main(String[] args) throws UnsupportedAudioFileException, IOException, LineUnavailableException {
+    public static void main(String[] args) throws UnsupportedAudioFileException, IOException, LineUnavailableException, NoAlbumFoundException {
+
 
         MusicHub theHub = new MusicHub();
 
@@ -240,6 +241,10 @@ public class Main {
                     printAvailableCommands();
                     choice = scan.nextLine();
                     break;
+                case 'r':
+                    //search a music
+                    theHub.searchAudioElement();
+                    break;
                 default:
 
                     break;
@@ -262,4 +267,5 @@ public class Main {
         System.out.println("s: save elements, albums, playlists");
         System.out.println("q: quit program");
     }
+
 }
\ No newline at end of file