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

---
 .../pinapelz/query/VideoByVideoIdQueryBuilder.java   | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

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

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