Updated code to use enums for certain constants

1 files changed, 20 insertions, 0 deletions

diff --git a/src/main/java/com/pinapelz/query/VideoByVideoIdQueryBuilder.java b/src/main/java/com/pinapelz/query/VideoByVideoIdQueryBuilder.java
index 7348683..6729916 100644
--- a/src/main/java/com/pinapelz/query/VideoByVideoIdQueryBuilder.java
+++ b/src/main/java/com/pinapelz/query/VideoByVideoIdQueryBuilder.java
@@ -1,5 +1,10 @@
 package com.pinapelz.query;
 
+import com.pinapelz.datatypes.Video;
+import com.pinapelz.vtuber.Language;
+
+import java.util.List;
+
 /**
  * Query builder for getting a video by video id
  */
@@ -21,11 +26,26 @@ public class VideoByVideoIdQueryBuilder {
         return lang;
     }
 
+
     public VideoByVideoIdQueryBuilder setLang(String lang) {
         this.lang = lang;
         return this;
     }
 
+    public VideoByVideoIdQueryBuilder setLang(Language lang) {
+        this.lang = lang.toString();
+        return this;
+    }
+
+    public VideoByVideoIdQueryBuilder setLang(List<Language> language) {
+        String[] languages = new String[language.size()];
+        for (int i = 0; i < language.size(); i++) {
+            languages[i] = language.get(i).toString();
+        }
+        this.lang = String.join(",", languages);
+        return this;
+    }
+
     public String getC() {
         return c;
     }