From 80239b6b63197519691153839f56e86fe99e18cb Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais6@uci.edu>
Date: Sat, 29 Apr 2023 20:34:18 -0700
Subject: Updated code to use enums for certain constants

---
 .../query/VideosByChannelIDQueryBuilder.java       | 40 ++++++++++++++++++++++
 1 file changed, 40 insertions(+)

(limited to 'src/main/java/com/pinapelz/query/VideosByChannelIDQueryBuilder.java')

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;
-- 
cgit v1.2.3