Updated code to use enums for certain constants

1 files changed, 40 insertions, 0 deletions

diff --git a/src/main/java/com/pinapelz/query/VideosByChannelIDQueryBuilder.java b/src/main/java/com/pinapelz/query/VideosByChannelIDQueryBuilder.java
index 821df24..9047f1a 100644
--- a/src/main/java/com/pinapelz/query/VideosByChannelIDQueryBuilder.java
+++ b/src/main/java/com/pinapelz/query/VideosByChannelIDQueryBuilder.java
@@ -1,5 +1,10 @@
 package com.pinapelz.query;
 
+import com.pinapelz.vtuber.ExtraData;
+import com.pinapelz.vtuber.Language;
+
+import java.util.List;
+
 /**
  * Query builder for getting videos by a given channel id
  */
@@ -27,11 +32,46 @@ public class VideosByChannelIDQueryBuilder {
         return this;
     }
 
+    public VideosByChannelIDQueryBuilder setInclude(String[] include) {
+        this.include = String.join(",", include);
+        return this;
+    }
+
+    public VideosByChannelIDQueryBuilder setInclude(ExtraData[] include) {
+        String[] includeStrings = new String[include.length];
+        for (int i = 0; i < include.length; i++) {
+            includeStrings[i] = include[i].toString();
+        }
+        this.include = String.join(",", includeStrings);
+        return this;
+    }
+
+    public VideosByChannelIDQueryBuilder setInclude(ExtraData include) {
+        this.include = include.toString();
+        return this;
+    }
+
+
+
     public VideosByChannelIDQueryBuilder setLang(String lang) {
         this.lang = lang;
         return this;
     }
 
+    public VideosByChannelIDQueryBuilder setLang(Language lang) {
+        this.lang = lang.toString();
+        return this;
+    }
+
+    public VideosByChannelIDQueryBuilder setLang(List<Language> lang){
+        String[] langStrings = new String[lang.size()];
+        for (int i = 0; i < lang.size(); i++) {
+            langStrings[i] = lang.get(i).toString();
+        }
+        this.lang = String.join(",", langStrings);
+        return this;
+    }
+
     public VideosByChannelIDQueryBuilder setLimit(Integer limit) {
         this.limit = limit;
         return this;