Merge pull request #11 from madeyoga/add-language

Add language parameter to YoutubeClient.search
Shiroechi · Nov 17, 2020 · c66d388 · c66d388
2 parents 86fc299 + 8285e9d
commit c66d388
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/main/java/YouTubeSearchApi/YoutubeClient.java b/src/main/java/YouTubeSearchApi/YoutubeClient.java
@@ -25,10 +25,11 @@ public YoutubeClient() {
         this.YOUTUBE_BASE_URL = "https://www.youtube.com/";
     }
 
-    public List<YoutubeVideo> search(String keywords, int maxResults) throws IOException, NoResultFoundException {
+    public List<YoutubeVideo> search(String keywords, int maxResults, String language) throws IOException, NoResultFoundException {
         String startFeature = "ytInitialData";
         String encodedKeywords = URLEncoder.encode(keywords, StandardCharsets.UTF_8.toString());
-        String searchUrl = this.YOUTUBE_BASE_URL + "results?search_query=" + encodedKeywords;
+        String searchUrl = this.YOUTUBE_BASE_URL + "results?gl=AU&hl=" + language
+                + "&persist_gl=1&persist_hl=1&search_query=" + encodedKeywords;
 
         // try get feature 3 times
         String pageContent = "";

diff --git a/src/test/java/YouTubeSearchApi/TestYoutubeClientSearch.java b/src/test/java/YouTubeSearchApi/TestYoutubeClientSearch.java
@@ -17,7 +17,7 @@ public static void main(String[] args) {
         List<YoutubeVideo> videos;
 
         try {
-            videos = client.search("CHiCO Love Letter", 5);
+            videos = client.search("yoasobi", 5, "en");
             System.out.println(videos.size());
             System.out.println(videos.toString());
         } catch (IOException | NoResultFoundException e) {