From 0509738b0af300b1bdda908be10e68f13386a6f8 Mon Sep 17 00:00:00 2001
From: Pinapelz <donaldshan1@outlook.com>
Date: Thu, 13 Apr 2023 14:27:34 -0700
Subject: Added check for if organization schedule is requested for exists.

---
 src/main/java/vtuber/ScheduleHandler.java | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'src/main/java/vtuber/ScheduleHandler.java')

diff --git a/src/main/java/vtuber/ScheduleHandler.java b/src/main/java/vtuber/ScheduleHandler.java
index 094309b..2990bbf 100644
--- a/src/main/java/vtuber/ScheduleHandler.java
+++ b/src/main/java/vtuber/ScheduleHandler.java
@@ -2,7 +2,10 @@ package vtuber;
 
 import com.pina.Holodex;
 import com.pina.HolodexException;
+import com.pina.datatypes.Channel;
+import com.pina.datatypes.SimpleChannel;
 import com.pina.datatypes.SimpleVideo;
+import com.pina.query.ChannelQueryBuilder;
 import com.pina.query.VideoQueryBuilder;
 
 import java.util.ArrayList;
@@ -62,4 +65,31 @@ public class ScheduleHandler {
         return upcomingAndLiveVideos;
 
     }
+
+    public boolean organizationExists(String org) {
+        List<Channel> channels = new ArrayList<>();
+        try {
+            channels = holodex.getChannels(new ChannelQueryBuilder().setOrg(org));
+        } catch (HolodexException e) {
+            System.out.println("Couldn't find organization with name " + org);
+            return false;
+        }
+        return channels.size() > 0;
+    }
+
+    public boolean channelExists(String channelId) {
+        Channel channel = new Channel();
+        try{
+            channel = holodex.getChannel(channelId);
+            if (channel.name == null) {
+                throw new HolodexException("Searching channel successful but no results found");
+            }
+        } catch (HolodexException e) {
+            System.out.println("Couldn't find channel with id " + channelId);
+            return false;
+        }
+        return true;
+    }
+
 }
+
-- 
cgit v1.2.3