From cbc8482b53d8fe0a0d33f2cc75d3a531cf11eca3 Mon Sep 17 00:00:00 2001
From: DevilXD <4180725+DevilXD@users.noreply.github.com>
Date: Fri, 28 Feb 2025 18:29:12 +0100
Subject: [PATCH] Add support for "badge or emote" Twitch campaigns

---
 inventory.py | 32 +++++++++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 3 deletions(-)

diff --git a/inventory.py b/inventory.py
index 30ba9be..9e8adcc 100644
--- a/inventory.py
+++ b/inventory.py
@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import re
 import math
+from enum import Enum
 from itertools import chain
 from typing import TYPE_CHECKING
 from functools import cached_property
@@ -27,13 +28,28 @@ def remove_dimensions(url: URLType) -> URLType:
     return URLType(DIMS_PATTERN.sub('', url))
 
 
+class BenefitType(Enum):
+    UNKNOWN = "UNKNOWN"
+    BADGE = "BADGE"
+    EMOTE = "EMOTE"
+    DIRECT_ENTITLEMENT = "DIRECT_ENTITLEMENT"
+
+    def is_badge_or_emote(self) -> bool:
+        return self in (BenefitType.BADGE, BenefitType.EMOTE)
+
+
 class Benefit:
-    __slots__ = ("id", "name", "image_url")
+    __slots__ = ("id", "name", "type", "image_url")
 
     def __init__(self, data: JsonType):
         benefit_data: JsonType = data["benefit"]
         self.id: str = benefit_data["id"]
         self.name: str = benefit_data["name"]
+        self.type: BenefitType = (
+            BenefitType(benefit_data["distributionType"])
+            if benefit_data["distributionType"] in BenefitType.__members__.values()
+            else BenefitType.UNKNOWN
+        )
         self.image_url: URLType = benefit_data["imageAssetURL"]
 
 
@@ -342,6 +358,16 @@ class DropsCampaign:
     def total_drops(self) -> int:
         return len(self.timed_drops)
 
+    @property
+    def eligible(self) -> bool:
+        return self.linked or self.has_badge_or_emote
+
+    @cached_property
+    def has_badge_or_emote(self) -> bool:
+        return any(
+            benefit.type.is_badge_or_emote() for drop in self.drops for benefit in drop.benefits
+        )
+
     @cached_property
     def finished(self) -> bool:
         return all(d.is_claimed for d in self.drops)
@@ -385,7 +411,7 @@ class DropsCampaign:
 
     def _base_can_earn(self, channel: Channel | None = None) -> bool:
         return (
-            self.linked  # account is connected
+            self.eligible  # account is eligible
             and self.active  # campaign is active
             # channel isn't specified, or there's no ACL, or the channel is in the ACL
             and (channel is None or not self.allowed_channels or channel in self.allowed_channels)
@@ -399,7 +425,7 @@ class DropsCampaign:
         # Same as can_earn, but doesn't check the channel
         # and uses a future timestamp to see if we can earn this campaign later
         return (
-            self.linked
+            self.eligible
             and self.ends_at > datetime.now(timezone.utc)
             and self.starts_at < stamp
             and any(drop.can_earn_within(stamp) for drop in self.drops)