Merge pull request #14 from ColyTeam/feature/search

Search Feature

Author: klegul

main/pom.xml

@@ -86,6 +86,12 @@
             <artifactId>jackson-databind</artifactId>
             <version>2.17.0</version>
         </dependency>
+        <dependency>
+            <groupId>org.junit.jupiter</groupId>
+            <artifactId>junit-jupiter-api</artifactId>
+            <version>5.10.2</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
 </project>

main/src/main/java/com/shirkanesi/magmaplayer/search/SearchResult.java

@@ -0,0 +1,20 @@
+package com.shirkanesi.magmaplayer.search;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Getter;
+
+@Getter
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class SearchResult {
+
+    @JsonProperty("url")
+    private String url;
+
+    @JsonProperty("title")
+    private String title;
+
+    @JsonProperty("channel")
+    private String creator;
+
+}

main/src/main/java/com/shirkanesi/magmaplayer/search/SearchUtil.java

@@ -0,0 +1,32 @@
+package com.shirkanesi.magmaplayer.search;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.List;
+
+public class SearchUtil {
+
+    private static final String SEARCH_COMMANDS = "yt-dlp \"%s:%s\" --print-json --flat-playlist --skip-download --no-warnings";
+
+    private static final ObjectMapper objectMapper = new ObjectMapper();
+
+    public static List<SearchResult> search(String prefix, String query) throws IOException {
+        final String findInformation = String.format(SEARCH_COMMANDS, prefix, query);
+        Process process = Runtime.getRuntime().exec(findInformation);
+
+        List<SearchResult> results = new ArrayList<>();
+        try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(process.getInputStream()))) {
+            String line;
+            while ((line = bufferedReader.readLine()) != null) {
+                    results.add(objectMapper.readValue(line, SearchResult.class));
+            }
+        }
+
+        return results;
+    }
+
+}

main/src/test/java/com/shirkanesi/magmaplayer/search/SearchUtilTest.java

@@ -0,0 +1,54 @@
+package com.shirkanesi.magmaplayer.search;
+
+import org.junit.jupiter.api.Test;
+
+import java.io.IOException;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class SearchUtilTest {
+
+    @Test
+    void testSearchValid() {
+        try {
+            List<SearchResult> results = SearchUtil.search("ytsearch10", "music");
+
+            assertEquals(10, results.size());
+
+            for (SearchResult result : results) {
+                assertNotNull(result);
+                assertNotNull(result.getTitle());
+                assertNotNull(result.getUrl());
+            }
+        } catch (IOException e) {
+            fail(e);
+        }
+    }
+
+    @Test
+    void testSearchSingle() {
+        try {
+            List<SearchResult> results = SearchUtil.search("ytsearch", "music");
+
+            assertEquals(1, results.size());
+
+            assertNotNull(results.get(0));
+            assertNotNull(results.get(0).getTitle());
+            assertNotNull(results.get(0).getUrl());
+        } catch (IOException e) {
+            fail(e);
+        }
+    }
+
+    @Test
+    void testSearchInvalid() {
+        try {
+            List<SearchResult> results = SearchUtil.search("ytsearch10", "hikodfgsoihsdghgoujuoihöfdgs");
+
+            assertEquals(0, results.size());
+        } catch (IOException e) {
+            fail(e);
+        }
+    }
+}