Initial Commit

12 files changed, 630 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..c9f47d5
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,128 @@
+onionBot/config.py
+settings/discordToken.txt
+target/classes/audio/Music$1.class
+target/classes/audio/Music$2.class
+target/classes/audio/Music.class
+target/classes/BotTool.class
+target/classes/hololive/HololiveTools$1.class
+target/classes/hololive/HololiveTools.class
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/nijisanji/ScreenShotTool.class
+target/classes/utilities/ScreenShotTool.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-archiver/pom.properties
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+onionBot/config.py
+settings/discordToken.txt
+target/classes/audio/Music$1.class
+target/classes/audio/Music$2.class
+target/classes/audio/Music.class
+target/classes/BotTool.class
+target/classes/hololive/HololiveTools$1.class
+target/classes/hololive/HololiveTools.class
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/utilities/ScreenShotTool.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-archiver/pom.properties
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+target/classes/Main$1.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/Main.class
+.idea/workspace.xml
+.idea/workspace.xml
+holoCli/hololive.txt
+data/nijisanji.txt
+target/classes/nijisanji/NijisanjiTools$1.class
+settings/config.json
+.idea/libraries/Maven__com_googlecode_json_simple_json_simple_1_1_1.xml
+.idea/libraries/Maven__junit_junit_4_10.xml
+.idea/libraries/Maven__org_hamcrest_hamcrest_core_1_1.xml
+.idea/hololiveDiscordBot.iml
+.idea/libraries/Maven__com_googlecode_json_simple_json_simple_1_1_1.xml
+.idea/libraries/Maven__junit_junit_4_10.xml
+.idea/libraries/Maven__org_hamcrest_hamcrest_core_1_1.xml
+.idea/hololiveDiscordBot.iml
+.idea/hololiveDiscordBot.iml
+.idea/workspace.xml
+automaticRefresh/config.py
+holoCli/hololive.txt
+settings/config.json
+target/classes/audio/Music$1.class
+target/classes/audio/Music.class
+target/classes/BotTool.class
+target/classes/hololive/HololiveTools.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-archiver/pom.properties
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+target/classes/audio/Music$1.class
+target/classes/audio/Music$2.class
+target/classes/audio/Music.class
+target/classes/BotTool.class
+target/classes/hololive/HololiveTools$1.class
+target/classes/hololive/HololiveTools.class
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/utilities/AutoRefreshLive.class
+target/classes/utilities/ScreenShotTool.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-archiver/pom.properties
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+.idea/hololiveDiscordBot.iml
+.idea/workspace.xml
+settings/config.json
+target/classes/hololive/HololiveTools.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/utilities/ScreenShotTool.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/hololive/HololiveTools$1.class
+target/classes/BotTool.class
+target/classes/audio/Music.class
+target/classes/audio/Music$2.class
+target/classes/audio/Music$1.class
+settings/config.json
+holoCli/text/timezone
+holoCli/text/hololive_members_en.txt
+holoCli/text/hololive_members.txt
+holoCli/text/help
+holoCli/src/util.py
+holoCli/src/scraping.py
+holoCli/src/fetch_html.py
+holoCli/src/__pycache__/util.cpython-39.pyc
+holoCli/src/__pycache__/util.cpython-38.pyc
+holoCli/src/__pycache__/util.cpython-37.pyc
+holoCli/src/__pycache__/scraping.cpython-39.pyc
+holoCli/src/__pycache__/scraping.cpython-38.pyc
+holoCli/src/__pycache__/fetch_html.cpython-37.pyc
+holoCli/src/__pycache__/fetch_html.cpython-39.pyc
+holoCli/src/__pycache__/scraping.cpython-37.pyc
+holoCli/src/__pycache__/fetch_html.cpython-38.pyc
+holoCli/main.py
+holoCli/hololive.txt
+data/nijisanjiChannel.txt
+data/nijiMemberID.txt
+data/memberList.txt
+settings/config.json
+target/classes/hololive/HololiveTools.class
+target/classes/hololive/KusoNihongoConverter.class
+target/classes/Main.class
+target/classes/nijisanji/NijisanjiTools.class
+target/classes/utilities/ScreenShotTool.class
+target/DiscordBot-1.0-SNAPSHOT-shaded.jar
+target/DiscordBot-1.0-SNAPSHOT.jar
+target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
diff --git a/.idea/.gitignore b/.idea/.gitignore
new file mode 100644
index 0000000..13566b8
--- /dev/null
+++ b/.idea/.gitignore
@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..3b1e12e
--- /dev/null
+++ b/README.md
@@ -0,0 +1,2 @@
+# VStreamTrack
+A Java JDA Bot for Tracking VTuber Livestreams and Schedules
\ No newline at end of file
diff --git a/pom.xml b/pom.xml
new file mode 100644
index 0000000..f18ce4c
--- /dev/null
+++ b/pom.xml
@@ -0,0 +1,93 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>com.pina</groupId>
+    <artifactId>vTuberDiscordBot</artifactId>
+    <version>0.01</version>
+
+    <properties>
+        <maven.compiler.source>1.8</maven.compiler.source>
+        <maven.compiler.target>1.8</maven.compiler.target>
+    </properties>
+
+    <repositories>
+        <repository>
+            <id>jcenter</id>
+            <name>jcenter-bintray</name>
+            <url>https://jcenter.bintray.com</url>
+        </repository>
+        <repository>
+            <id>jitpack.io</id>
+            <url>https://jitpack.io</url>
+        </repository>
+        <repository>
+            <id>dv8tion</id>
+            <name>m2-dv8tion</name>
+            <url>https://m2.dv8tion.net/releases</url>
+        </repository>
+    </repositories>
+    <dependencies>
+        <dependency>
+            <groupId>com.googlecode.json-simple</groupId>
+            <artifactId>json-simple</artifactId>
+            <version>1.1.1</version>
+        </dependency>
+        <dependency>
+            <groupId>net.dv8tion</groupId>
+            <artifactId>JDA</artifactId>
+            <version>4.4.0_350</version>
+        </dependency>
+        <dependency>
+            <groupId>org.jsoup</groupId>
+            <artifactId>jsoup</artifactId>
+            <version>1.15.3</version>
+        </dependency>
+        <dependency>
+            <groupId>com.github.pinapelz</groupId>
+            <artifactId>JHolodex</artifactId>
+            <version>0.05</version>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <defaultGoal>clean package</defaultGoal>
+        <plugins>
+            <plugin>
+                <artifactId>maven-dependency-plugin</artifactId>
+                <version>3.1.2</version>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-shade-plugin</artifactId>
+                <version>3.1.0</version>
+                <executions>
+                    <execution>
+                        <phase>package</phase>
+                        <goals>
+                            <goal>shade</goal>
+                        </goals>
+                        <configuration>
+                            <transformers>
+                                <transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
+                                    <mainClass>Main</mainClass>
+                                </transformer>
+                            </transformers>
+                            <createDependencyReducedPom>false</createDependencyReducedPom>
+                            <shadedArtifactAttached>true</shadedArtifactAttached>
+                        </configuration>
+                    </execution>
+                </executions>
+            </plugin>
+        </plugins>
+        <resources>
+            <resource>
+                <directory>src/main/resources</directory>
+                <filtering>true</filtering>
+            </resource>
+        </resources>
+    </build>
+
+</project>
\ No newline at end of file
diff --git a/settings/upcomingChannels.txt b/settings/upcomingChannels.txt
new file mode 100644
index 0000000..307c790
--- /dev/null
+++ b/settings/upcomingChannels.txt
@@ -0,0 +1,3 @@
+org:Hololive:1055683371193348206
+org:Nijisanji:1055710959919976479
+channel:UC4WvIIAo89_AzGUh1AZ6Dkg:1094112349163638804
diff --git a/src/main/java/Main.java b/src/main/java/Main.java
new file mode 100644
index 0000000..78a2f4d
--- /dev/null
+++ b/src/main/java/Main.java
@@ -0,0 +1,93 @@
+import commands.CommandManager;
+import commands.StatusHandler;
+import common.OrgChannelTuple;
+import fileutils.FileDataProcessor;
+import net.dv8tion.jda.api.JDA;
+import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import net.dv8tion.jda.api.hooks.ListenerAdapter;
+
+import javax.security.auth.login.LoginException;
+import java.util.List;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+
+
+
+public class Main extends ListenerAdapter{
+    private int REFRESH_INTERVAL = 15;
+    private JDA jda;
+    private StatusHandler statusHandler;
+    private JDABuilder jdaBuilder;
+    private FileDataProcessor fileDataProcessor;
+    private CommandManager commandManager;
+
+    public void initializeBot(){
+        fileDataProcessor = new FileDataProcessor();
+        commandManager = new CommandManager(fileDataProcessor.readCredential("holodexAPIKey"));
+        jdaBuilder = JDABuilder.createDefault(fileDataProcessor.readCredential("discordToken"));
+        jdaBuilder.addEventListeners(commandManager);
+        try {
+            jda = jdaBuilder.build();
+            statusHandler = new StatusHandler(jda);
+            statusHandler.updateSlashCommands();
+            System.out.println("Bot is ready!");
+            initializeAutoRefresh();
+        }
+        catch (LoginException e) {
+            System.out.println("Unable to login with the provided token. Please check your token and try again.");
+            throw new RuntimeException(e);
+        }
+
+    }
+
+    public void initializeAutoRefresh(){
+        ScheduledExecutorService ses = Executors.newScheduledThreadPool(1);
+        ses.scheduleAtFixedRate(() -> {
+            try {
+                System.out.println("Refreshing upcoming channels");
+                List<OrgChannelTuple> refreshChannels = fileDataProcessor.getRefreshChannels();
+                if (refreshChannels.size() == 0) {
+                    System.out.println("No channels to refresh");
+                    return;
+                }
+                for (OrgChannelTuple orgChannelTuple : refreshChannels) {
+                    System.out.println("Refreshing " + orgChannelTuple.getType() + " " + orgChannelTuple.getName());
+                    List<MessageEmbed> messageEmbeds = commandManager.updateUpcomingChannel(orgChannelTuple.getName(), orgChannelTuple.getType());
+                    if (messageEmbeds.size() == 0) {
+                        continue;
+                    }
+                    jda.getTextChannelById(orgChannelTuple.getDiscordChannelId()).purgeMessages(
+                            jda.getTextChannelById(orgChannelTuple.getDiscordChannelId()).getIterableHistory().complete());
+                    for (MessageEmbed messageEmbed : messageEmbeds) {
+                        jda.getTextChannelById(orgChannelTuple.getDiscordChannelId()).sendMessageEmbeds(messageEmbed).queue();
+                    }
+                }
+            }
+            catch(NullPointerException ex){
+                System.out.println("Channel is empty. Skipping refresh there");
+            }
+            catch (Exception e) {
+                System.out.println("Error occurred while refreshing upcoming channels");
+                e.printStackTrace();
+            }
+        }, 0, REFRESH_INTERVAL, TimeUnit.MINUTES);
+    }
+
+
+    @Override
+    public void onMessageReceived(MessageReceivedEvent e) {
+        JDA jda = e.getJDA();
+        Message message = e.getMessage();
+        String msg = message.getContentDisplay();
+
+    }
+    public static void main(String args[]) {
+        Main main = new Main();
+        main.initializeBot();
+    }
+}
+
diff --git a/src/main/java/builders/ScheduleMessageBuilder.java b/src/main/java/builders/ScheduleMessageBuilder.java
new file mode 100644
index 0000000..9dd77bf
--- /dev/null
+++ b/src/main/java/builders/ScheduleMessageBuilder.java
@@ -0,0 +1,69 @@
+package builders;
+
+import com.pina.datatypes.SimpleVideo;
+import net.dv8tion.jda.api.EmbedBuilder;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+
+import java.awt.*;
+import java.time.OffsetDateTime;
+import java.util.ArrayList;
+import java.util.List;
+
+public class ScheduleMessageBuilder {
+    private int COLOR = 8805674;
+    public MessageEmbed buildLiveAndUpcomingMessage(List<SimpleVideo> videos){
+        if (videos.size() == 0){
+            return new EmbedBuilder()
+                    .setTitle("No streams upcoming or live. Check back later!")
+                    .setDescription("If you think this is a mistake, please check that you've spelled the organization as listed on Holodex")
+                    .setColor(new Color(COLOR))
+                    .setTimestamp(OffsetDateTime.now())
+                    .build();
+        }
+        String pfp = videos.get(0).channel.photo;
+        EmbedBuilder messageBuilder = new EmbedBuilder()
+                .setThumbnail(pfp)
+                .setTitle("Upcoming and Live Streams for " + videos.get(0).channel.org)
+                .setDescription("The schedule you asked for")
+                .setColor(new Color(COLOR))
+                .setTimestamp(OffsetDateTime.now());
+
+        for (SimpleVideo video : videos){
+            String gmtStartTime = video.start_scheduled;
+            long unixTime = OffsetDateTime.parse(gmtStartTime).toEpochSecond();
+            String unixTimeStr = "<t:" + Long.toString(unixTime) + ":R> ⏰";
+            if (video.status.equals("live")){
+                unixTimeStr = "LIVE \uD83D\uDD34";
+            }
+            String titleText = video.channel.english_name + " - " + unixTimeStr;
+            String videoURL = "https://www.youtube.com/watch?v=" + video.id;
+            messageBuilder.addField(titleText, "["+video.title+"]"+"("+videoURL+")", false);
+        }
+        return messageBuilder.build();
+
+    }
+
+    public ArrayList<MessageEmbed> getUpcomingLiveListMessages(List<SimpleVideo> simpleVideos){
+        ArrayList<MessageEmbed> messageEmbeds = new ArrayList<>();
+        for (SimpleVideo video : simpleVideos){
+            String title = video.channel.english_name + " is streaming soon!   ⏰";
+            String fieldTitle = "Scheduled Start Time";
+            if (video.status.equals("live")){
+                title = video.channel.english_name + " is live!   \uD83D\uDD34";
+                fieldTitle = "Live Since";
+            }
+            String gmtStartTime = video.start_scheduled;
+            long unixTime = OffsetDateTime.parse(gmtStartTime).toEpochSecond();
+            EmbedBuilder embedBuilder = new EmbedBuilder()
+                    .setTitle(title)
+                    .setDescription("["+video.title+"]"+"(https://www.youtube.com/watch?v="+video.id+")")
+                    .addField(fieldTitle, "<t:" + unixTime + ":R>", false)
+                    .setThumbnail(video.channel.photo)
+                    .setImage("https://img.youtube.com/vi/"+video.id+"/maxresdefault.jpg")
+                    .setTimestamp(OffsetDateTime.now());
+            messageEmbeds.add(embedBuilder.build());
+        }
+        return messageEmbeds;
+
+    }
+}
diff --git a/src/main/java/commands/CommandManager.java b/src/main/java/commands/CommandManager.java
new file mode 100644
index 0000000..e374b21
--- /dev/null
+++ b/src/main/java/commands/CommandManager.java
@@ -0,0 +1,52 @@
+package commands;
+
+import builders.ScheduleMessageBuilder;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.interaction.SlashCommandEvent;
+import net.dv8tion.jda.api.hooks.ListenerAdapter;
+import vtuber.ScheduleHandler;
+
+import java.util.ArrayList;
+
+public class CommandManager extends ListenerAdapter {
+    ScheduleHandler scheduleHandler;
+    ScheduleMessageBuilder scb;
+    public CommandManager(String holodexAPIKey) {
+        scheduleHandler = new ScheduleHandler(holodexAPIKey);
+        scb = new ScheduleMessageBuilder();
+        System.out.println("CommandManager initialized");
+    }
+    @Override
+    public void onSlashCommand(SlashCommandEvent e) {
+        String command = e.getName();
+        switch (command) {
+            case "schedule":
+                String organization = e.getOption("organization").getAsString();
+                organization = organization.replaceAll(" ", "%20");
+                MessageEmbed scheduleMessage = scb.buildLiveAndUpcomingMessage(scheduleHandler.getSchedule(organization, 10));
+                e.deferReply().queue();
+                e.getHook().sendMessageEmbeds(scheduleMessage).queue();
+                break;
+            default:
+                e.reply("Unknown command received").queue();
+                break;
+        }
+    }
+
+    public ArrayList<MessageEmbed> updateUpcomingChannel(String name, String type){
+        ArrayList<MessageEmbed> messageEmbeds = new ArrayList<>();
+        switch (type) {
+            case "org":
+                messageEmbeds = scb.getUpcomingLiveListMessages(scheduleHandler.getSchedule(name));
+                break;
+            case "channel":
+                messageEmbeds = scb.getUpcomingLiveListMessages(scheduleHandler.getScheduleChannelId(name));
+                break;
+            default:
+                System.out.println("Unknown type");
+                break;
+        }
+        return messageEmbeds;
+    }
+
+}
diff --git a/src/main/java/commands/StatusHandler.java b/src/main/java/commands/StatusHandler.java
new file mode 100644
index 0000000..ff48632
--- /dev/null
+++ b/src/main/java/commands/StatusHandler.java
@@ -0,0 +1,22 @@
+package commands;
+import net.dv8tion.jda.api.JDA;
+import net.dv8tion.jda.api.interactions.commands.OptionType;
+import net.dv8tion.jda.api.interactions.commands.build.CommandData;
+
+public class StatusHandler {
+    JDA jda;
+    public StatusHandler(JDA jda) {
+        this.jda = jda;
+
+    }
+    public void updateSlashCommands(){
+        jda.upsertCommand(new CommandData("schedule", "Shows upcoming streams and events for a given organization")
+                .addOption(OptionType.STRING, "organization",
+                        "Holodex Organization Name (e.g Hololive, Nijisanji, Phase Connect, PRISM, Production Kawaii)",
+                        true))
+                .queue();
+
+    }
+
+
+}
diff --git a/src/main/java/common/OrgChannelTuple.java b/src/main/java/common/OrgChannelTuple.java
new file mode 100644
index 0000000..8a78dcc
--- /dev/null
+++ b/src/main/java/common/OrgChannelTuple.java
@@ -0,0 +1,37 @@
+package common;
+
+public class OrgChannelTuple {
+    private String name;
+    private long discordChannelId;
+    private String type;
+
+    public OrgChannelTuple(String type, String name, long discordChannelId) {
+        this.name = name;
+        this.type = type;
+        this.discordChannelId = discordChannelId;
+    }
+
+    public String getType() {
+        return type;
+    }
+
+    public void setType(String type) {
+        this.type = type;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public long getDiscordChannelId() {
+        return discordChannelId;
+    }
+
+    public void setDiscordChannelId(long discordChannelId) {
+        this.discordChannelId = discordChannelId;
+    }
+}
diff --git a/src/main/java/fileutils/FileDataProcessor.java b/src/main/java/fileutils/FileDataProcessor.java
new file mode 100644
index 0000000..0315fbb
--- /dev/null
+++ b/src/main/java/fileutils/FileDataProcessor.java
@@ -0,0 +1,58 @@
+package fileutils;
+
+import common.OrgChannelTuple;
+import org.json.simple.JSONObject;
+import org.json.simple.parser.JSONParser;
+import org.json.simple.parser.ParseException;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileReader;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+
+public class FileDataProcessor {
+
+    public static String readCredential(String parameter){
+        try {
+            Object obj = new JSONParser().parse(new FileReader("settings//config.json"));
+            JSONObject jo = (JSONObject) obj;
+            return (String) jo.get(parameter);
+        }
+        catch(FileNotFoundException e){
+            System.out.println("Credential file could not be found. Please create it at settings//config.json");
+        }
+        catch(ParseException ex){
+            System.out.println("Ensure that your credential file is valid JSON");
+        }
+        catch(IOException ex){
+            System.out.println("An error occurred while reading the credential file");
+        }
+        return "";
+
+    }
+
+    public List<OrgChannelTuple> getRefreshChannels(){
+        List<OrgChannelTuple> orgChannelTuples = new ArrayList<>();
+        try{
+            File channelFile = new File("settings//upcomingChannels.txt");
+            if(channelFile.createNewFile()){
+                System.out.println("upcomingChannels.txt created. Please fill it out with the organizations you want to track (refer to README)");
+            }
+            for (String line : Files.readAllLines(Paths.get("settings//upcomingChannels.txt"))) {
+                String type = line.split(":")[0];
+                String name = line.split(":")[1];
+                String channelIdStr = line.split(":")[2];
+                long channelId = Long.parseLong(channelIdStr);
+                orgChannelTuples.add(new OrgChannelTuple(type, name, channelId));
+            }
+        } catch (IOException e) {
+            System.out.println("Unable to create upcomingChannels.txt file for updating Discord Channels");
+        }
+        return orgChannelTuples;
+    }
+
+}
diff --git a/src/main/java/vtuber/ScheduleHandler.java b/src/main/java/vtuber/ScheduleHandler.java
new file mode 100644
index 0000000..094309b
--- /dev/null
+++ b/src/main/java/vtuber/ScheduleHandler.java
@@ -0,0 +1,65 @@
+package vtuber;
+
+import com.pina.Holodex;
+import com.pina.HolodexException;
+import com.pina.datatypes.SimpleVideo;
+import com.pina.query.VideoQueryBuilder;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class ScheduleHandler {
+    Holodex holodex;
+
+    public ScheduleHandler(String apikey) {
+        System.out.println("ScheduleHandler initialized");
+        holodex = new Holodex(apikey);
+
+    }
+
+    public List<SimpleVideo> getSchedule(String org, int limit) {
+        System.out.println("Getting schedule for " + org);
+        List<SimpleVideo> upcomingAndLiveVideos = new ArrayList<>();
+        try {
+            List<SimpleVideo> upcomingVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setOrg(org).setLimit(limit).setStatus("upcoming"));
+            List<SimpleVideo> liveVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setStatus("live").setOrg(org).setLimit(limit));
+            upcomingAndLiveVideos.addAll(liveVideos);
+            upcomingAndLiveVideos.addAll(upcomingVideos);
+        } catch (HolodexException e) {
+            System.out.println("Error getting schedule for " + org);
+            System.out.println(e.getMessage());
+        }
+        return upcomingAndLiveVideos;
+    }
+
+    public List<SimpleVideo> getSchedule(String org) {
+        System.out.println("Getting schedule for " + org);
+        List<SimpleVideo> upcomingAndLiveVideos = new ArrayList<>();
+        try {
+            List<SimpleVideo> upcomingVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setOrg(org).setStatus("upcoming"));
+            List<SimpleVideo> liveVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setStatus("live").setOrg(org));
+            upcomingAndLiveVideos.addAll(liveVideos);
+            upcomingAndLiveVideos.addAll(upcomingVideos);
+        } catch (HolodexException e) {
+            System.out.println("Error getting schedule for " + org);
+            System.out.println(e.getMessage());
+        }
+        return upcomingAndLiveVideos;
+    }
+
+    public List<SimpleVideo> getScheduleChannelId(String channelId) {
+        System.out.println("Getting schedule for " + channelId);
+        List<SimpleVideo> upcomingAndLiveVideos = new ArrayList<>();
+        try {
+            List<SimpleVideo> upcomingVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setChannelId(channelId).setStatus("upcoming"));
+            List<SimpleVideo> liveVideos = holodex.getLiveAndUpcomingVideos(new VideoQueryBuilder().setStatus("live").setChannelId(channelId));
+            upcomingAndLiveVideos.addAll(liveVideos);
+            upcomingAndLiveVideos.addAll(upcomingVideos);
+        } catch (HolodexException e) {
+            System.out.println("Error getting schedule for " + channelId);
+            System.out.println(e.getMessage());
+        }
+        return upcomingAndLiveVideos;
+
+    }
+}