feat: parse new prediction badges over irc (#569)

twitch4j · May 15, 2022 · 73e16a1 · 73e16a1
1 parent 3ba1e1f
commit 73e16a1
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 6 deletions.
diff --git a/chat/src/main/java/com/github/twitch4j/chat/events/channel/IRCMessageEvent.java b/chat/src/main/java/com/github/twitch4j/chat/events/channel/IRCMessageEvent.java
@@ -313,6 +313,30 @@ public OptionalInt getCheererTier() {
         return OptionalInt.empty();
     }
 
+    /**
+     * @return the index of the choice this user is predicting, in an optional wrapper
+     */
+    public OptionalInt getPredictedChoiceIndex() {
+        String predictions = badges.get("predictions");
+
+        if (predictions != null) {
+            int delim = predictions.indexOf('-');
+            try {
+                return OptionalInt.of(Integer.parseInt(predictions.substring(delim + 1)));
+            } catch (NumberFormatException ignored) {
+            }
+        }
+
+        return OptionalInt.empty();
+    }
+
+    /**
+     * @return the title of the choice this user is predicting, in an optional wrapper
+     */
+    public Optional<String> getPredictedChoiceTitle() {
+        return Optional.ofNullable(badgeInfo.get("predictions")).map(EscapeUtils::unescapeTagValue);
+    }
+
 	/**
 	 * Gets a optional tag from the irc message
      *

diff --git a/common/src/main/java/com/github/twitch4j/common/enums/CommandPermission.java b/common/src/main/java/com/github/twitch4j/common/enums/CommandPermission.java
@@ -47,12 +47,14 @@ public enum CommandPermission {
     CURRENT_HYPE_TRAIN_CONDUCTOR,
 
     /**
-     * Participated in the most recent predictions event for the blue/first option
+     * Participated in the most recent predictions event for a blue option
+     * <p>
+     * Warning: when there are three or more prediction choices, they are all blue
      */
     PREDICTIONS_BLUE,
 
     /**
-     * Participated in the most recent predictions event for the pink/second option
+     * Participated in the most recent predictions event for a pink option
      */
     PREDICTIONS_PINK,
 

diff --git a/common/src/main/java/com/github/twitch4j/common/util/TwitchUtils.java b/common/src/main/java/com/github/twitch4j/common/util/TwitchUtils.java
@@ -58,7 +58,7 @@ public static Set<CommandPermission> getPermissionsFromTags(@NonNull Map<String,
                 permissionSet.add(CommandPermission.MODERATOR);
             }
             // Partner
-            if (badges.containsKey("partner")) {
+            if (badges.containsKey("partner") || badges.containsKey("ambassador")) {
                 permissionSet.add(CommandPermission.PARTNER);
             }
             // VIP
@@ -112,10 +112,9 @@ public static Set<CommandPermission> getPermissionsFromTags(@NonNull Map<String,
             String predictionBadge = badges.get("predictions");
             if (StringUtils.isNotEmpty(predictionBadge)) {
                 char first = predictionBadge.charAt(0);
-                char last = predictionBadge.charAt(predictionBadge.length() - 1);
-                if (first == 'b' || last == '1') {
+                if (first == 'b') {
                     permissionSet.add(CommandPermission.PREDICTIONS_BLUE);
-                } else if (first == 'p' || last == '2') {
+                } else if (first == 'p') {
                     permissionSet.add(CommandPermission.PREDICTIONS_PINK);
                 }
             }

diff --git a/pubsub/src/main/java/com/github/twitch4j/pubsub/domain/ChannelPointsReward.java b/pubsub/src/main/java/com/github/twitch4j/pubsub/domain/ChannelPointsReward.java
@@ -30,6 +30,7 @@ public class ChannelPointsReward {
 	private GlobalCooldown globalCooldown;
 	private Integer redemptionsRedeemedCurrentStream;
 	private Instant cooldownExpiresAt;
+    private String templateId;
 
 	@Data
 	public static class Image {