From f1f6d4ed6303c2d4a2c23f447533d40d6fe72cf9 Mon Sep 17 00:00:00 2001
From: cyklon73 <cyklon698@gmail.com>
Date: Wed, 28 Aug 2024 10:35:46 +0200
Subject: [PATCH] Implements after a failed api key to try the next one

---
 .../slimeball/features/alerts/youtube/Youtube.java | 14 ++++++++------
 .../alerts/youtube/YoutubeApiErrorEvent.java       |  3 +++
 .../features/alerts/youtube/YoutubeListener.java   |  9 +++++----
 .../features/alerts/youtube/YoutubeVideoEvent.java |  3 ++-
 4 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/Youtube.java b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/Youtube.java
index 42e2c1ad..be64bc61 100644
--- a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/Youtube.java
+++ b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/Youtube.java
@@ -11,6 +11,7 @@
 import de.slimecloud.slimeball.main.SlimeBot;
 import de.slimecloud.slimeball.util.MathUtil;
 import lombok.Cleanup;
+import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 import okhttp3.OkHttpClient;
 import okhttp3.Request;
@@ -27,6 +28,7 @@ public class Youtube {
 
 	private final OkHttpClient client = new OkHttpClient().newBuilder().build();
 
+	@Getter
 	private final String[] keys;
 	private final SlimeBot bot;
 	private final YoutubeConfig config;
@@ -46,7 +48,7 @@ public void startListener() {
 		channels.forEach((channel, delay) -> {
 			bot.getExecutor().scheduleAtFixedRate(() -> {
 				try {
-					check(channel);
+					check(channel, currentKey);
 				} catch (Exception e) {
 					logger.error("Failed to check for new video for channel " + channel, e);
 				}
@@ -54,15 +56,15 @@ public void startListener() {
 		});
 	}
 
-	private void check(String youtubeChannelId) throws IOException {
-		Set<Video> videos = getLastVideo(youtubeChannelId, 5);
+	public void check(String youtubeChannelId, int startKey) throws IOException {
+		Set<Video> videos = getLastVideo(youtubeChannelId, 5, startKey);
 
 		Collection<String> known = bot.getIdMemory().getMemory("youtube");
 		List<String> newIds = new ArrayList<>();
 
 		for (Video video : videos) {
 			if (!known.contains(video.id())) {
-				new YoutubeVideoEvent(youtubeChannelId, video).callEvent();
+				new YoutubeVideoEvent(this, youtubeChannelId, video).callEvent();
 				newIds.add(video.id());
 			}
 		}
@@ -76,7 +78,7 @@ private String getNextKey() {
 	}
 
 	@NotNull
-	public Set<Video> getLastVideo(@NotNull String youtubeChannelId, int limit) throws IOException {
+	public Set<Video> getLastVideo(@NotNull String youtubeChannelId, int limit, int startKey) throws IOException {
 		Request request = new Request.Builder()
 				.url(String.format("https://www.googleapis.com/youtube/v3/search?key=%s&channelId=%s&part=snippet,id&order=date&maxResults=" + limit, getNextKey(), youtubeChannelId))
 				.get()
@@ -89,7 +91,7 @@ public Set<Video> getLastVideo(@NotNull String youtubeChannelId, int limit) thro
 		JsonObject json = JsonParser.parseString(response.body().string()).getAsJsonObject();
 
 		if (code != HttpStatus.SC_OK) {
-			YoutubeApiErrorEvent event = new YoutubeApiErrorEvent(response, json, new HashSet<>());
+			YoutubeApiErrorEvent event = new YoutubeApiErrorEvent(this, youtubeChannelId, startKey, response, json, new HashSet<>());
 			event.callEvent();
 			return event.getVideos();
 		}
diff --git a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeApiErrorEvent.java b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeApiErrorEvent.java
index be98f06e..8986f66f 100644
--- a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeApiErrorEvent.java
+++ b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeApiErrorEvent.java
@@ -13,6 +13,9 @@
 @RequiredArgsConstructor
 public class YoutubeApiErrorEvent extends Event {
 
+	private final Youtube youtube;
+	private final String channelId;
+	private final int startKey;
 	private final Response response;
 	private final JsonObject jsonResponse;
 	private final Set<Video> videos;
diff --git a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeListener.java b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeListener.java
index 96a4988a..70fa805d 100644
--- a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeListener.java
+++ b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeListener.java
@@ -17,21 +17,22 @@ public class YoutubeListener {
 	public void onUpload(YoutubeVideoEvent event) {
 		logger.info("Video Uploaded: {}", event.getVideo());
 		bot.getJda().getGuilds().forEach(g -> bot.loadGuild(g).getYoutube().ifPresent(config ->
-				config.getChannel(event.getYoutubeChannelId()).ifPresentOrElse(channel -> {
-					String msg = event.isLive() ? config.getLiveMessage().get(event.getYoutubeChannelId()) : config.getVideoMessage().get(event.getYoutubeChannelId());
+				config.getChannel(event.getChannelId()).ifPresentOrElse(channel -> {
+					String msg = event.isLive() ? config.getLiveMessage().get(event.getChannelId()) : config.getVideoMessage().get(event.getChannelId());
 
 					channel.sendMessage(msg
-							.replace("%role%", config.getRole(event.getYoutubeChannelId()).map(IMentionable::getAsMention).orElse(""))
+							.replace("%role%", config.getRole(event.getChannelId()).map(IMentionable::getAsMention).orElse(""))
 							.replace("%uploader%", event.getVideo().getChannel().getTitle())
 							.replace("%url%", event.getVideo().getUrl())
 							.replace("%title%", event.getVideo().snippet().title())
 					).queue();
-				}, () -> logger.warn("Cannot send Youtube Notification because channel {} not found", config.getChannelId(event.getYoutubeChannelId()).orElse(null)))
+				}, () -> logger.warn("Cannot send Youtube Notification because channel {} not found", config.getChannelId(event.getChannelId()).orElse(null)))
 		));
 	}
 
 	@EventHandler
 	public void onApiError(YoutubeApiErrorEvent event) {
+
 		logger.warn("Youtube API error {} '{}'", event.getCode(), event.getJsonResponse().getAsJsonObject("error").get("message").getAsString());
 	}
 }
diff --git a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeVideoEvent.java b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeVideoEvent.java
index e141bcea..e7535177 100644
--- a/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeVideoEvent.java
+++ b/src/main/java/de/slimecloud/slimeball/features/alerts/youtube/YoutubeVideoEvent.java
@@ -9,7 +9,8 @@
 @RequiredArgsConstructor
 public class YoutubeVideoEvent extends Event {
 
-	private final String youtubeChannelId;
+	private final Youtube youtube;
+	private final String channelId;
 	private final Video video;
 
 	public boolean isLive() {