From 6d7f2848d6d6a48dc1de8e926c1bc0164d144b0b Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais6@uci.edu>
Date: Sat, 29 Apr 2023 20:48:06 -0700
Subject: Fixed bug with not being able to pass in singular Orgnization enum

---
 .../pinapelz/query/CommentSearchQueryBuilder.java    | 20 ++++++++++++++++++--
 .../java/com/pinapelz/query/VideoQueryBuilder.java   |  1 +
 .../com/pinapelz/query/VideoSearchQueryBuilder.java  | 19 +++++++++++++++++--
 3 files changed, 36 insertions(+), 4 deletions(-)

(limited to 'src/main/java/com')

diff --git a/src/main/java/com/pinapelz/query/CommentSearchQueryBuilder.java b/src/main/java/com/pinapelz/query/CommentSearchQueryBuilder.java
index 4e0020b..6c1fa90 100644
--- a/src/main/java/com/pinapelz/query/CommentSearchQueryBuilder.java
+++ b/src/main/java/com/pinapelz/query/CommentSearchQueryBuilder.java
@@ -1,6 +1,7 @@
 package com.pinapelz.query;
 
 import com.pinapelz.vtuber.Language;
+import com.pinapelz.vtuber.Organization;
 import com.pinapelz.vtuber.Sort;
 import com.pinapelz.vtuber.Type;
 
@@ -130,11 +131,26 @@ public class CommentSearchQueryBuilder {
         return org;
     }
 
-    public CommentSearchQueryBuilder setOrg(List<String> org) {
-        this.org = org;
+    public CommentSearchQueryBuilder setOrg(String org) {
+        this.org = List.of(org);
         return this;
     }
 
+    public CommentSearchQueryBuilder setOrg(Organization org) {
+        this.org = List.of(org.toString());
+        return this;
+    }
+
+    public CommentSearchQueryBuilder setOrg(List<Organization> org) {
+        List<String> orgs = new ArrayList<String>();
+        for (Organization o : org) {
+            orgs.add(o.toString());
+        }
+        this.org = orgs;
+        return this;
+    }
+
+
     public List<String> getComment() {
         return comment;
     }
diff --git a/src/main/java/com/pinapelz/query/VideoQueryBuilder.java b/src/main/java/com/pinapelz/query/VideoQueryBuilder.java
index c8db237..6a74217 100644
--- a/src/main/java/com/pinapelz/query/VideoQueryBuilder.java
+++ b/src/main/java/com/pinapelz/query/VideoQueryBuilder.java
@@ -3,6 +3,7 @@ package com.pinapelz.query;
 import com.pinapelz.datatypes.Video;
 import com.pinapelz.vtuber.*;
 
+import java.util.ArrayList;
 import java.util.List;
 
 /**
diff --git a/src/main/java/com/pinapelz/query/VideoSearchQueryBuilder.java b/src/main/java/com/pinapelz/query/VideoSearchQueryBuilder.java
index e982ac7..0a42716 100644
--- a/src/main/java/com/pinapelz/query/VideoSearchQueryBuilder.java
+++ b/src/main/java/com/pinapelz/query/VideoSearchQueryBuilder.java
@@ -1,6 +1,7 @@
 package com.pinapelz.query;
 
 import com.pinapelz.vtuber.Language;
+import com.pinapelz.vtuber.Organization;
 import com.pinapelz.vtuber.Sort;
 import com.pinapelz.vtuber.Type;
 
@@ -125,8 +126,22 @@ public class VideoSearchQueryBuilder {
         return org;
     }
 
-    public VideoSearchQueryBuilder setOrg(List<String> org) {
-        this.org = org;
+    public VideoSearchQueryBuilder setOrg(String org) {
+        this.org = List.of(org);
+        return this;
+    }
+
+    public VideoSearchQueryBuilder setOrg(Organization org) {
+        this.org = List.of(org.toString());
+        return this;
+    }
+
+    public VideoSearchQueryBuilder setOrg(List<Organization> org) {
+        List<String> orgs = new ArrayList<String>();
+        for (Organization o : org) {
+            orgs.add(o.toString());
+        }
+        this.org = orgs;
         return this;
     }
 
-- 
cgit v1.2.3