From 5430ddee54313e0ebc5b7b0a19fbb47cce7a72be Mon Sep 17 00:00:00 2001
From: dolfies <jeyalfie47@gmail.com>
Date: Fri, 26 Jan 2024 14:11:06 -0500
Subject: [PATCH] Implement new age_range experiment filter

---
 discord/experiment.py | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/discord/experiment.py b/discord/experiment.py
index 87fbffde3..8f438f148 100644
--- a/discord/experiment.py
+++ b/discord/experiment.py
@@ -28,7 +28,7 @@ from typing import TYPE_CHECKING, Dict, Final, Iterator, List, Optional, Sequenc
 
 from .enums import HubType, try_enum
 from .metadata import Metadata
-from .utils import SequenceProxy, SnowflakeList, murmurhash32
+from .utils import SequenceProxy, SnowflakeList, murmurhash32, utcnow
 
 if TYPE_CHECKING:
     from .abc import Snowflake
@@ -126,6 +126,7 @@ class ExperimentFilters:
     FILTER_KEYS: Final[Dict[int, str]] = {
         1604612045: 'guild_has_feature',
         2404720969: 'guild_id_range',
+        3730341874: 'guild_age_range_days',
         2918402255: 'guild_member_count_range',
         3013771838: 'guild_ids',
         4148745523: 'guild_hub_types',
@@ -143,7 +144,16 @@ class ExperimentFilters:
         self.options: Metadata = self.array_object(data)
 
     def __repr__(self) -> str:
-        keys = ('features', 'id_range', 'member_count_range', 'ids', 'range_by_hash', 'has_vanity_url')
+        keys = (
+            'features',
+            'id_range',
+            'age_range',
+            'member_count_range',
+            'ids',
+            'hub_types',
+            'range_by_hash',
+            'has_vanity_url',
+        )
         attrs = [f'{attr}={getattr(self, attr)!r}' for attr in keys if getattr(self, attr) is not None]
         if attrs:
             return f'<ExperimentFilters {" ".join(attrs)}>'
@@ -168,7 +178,7 @@ class ExperimentFilters:
         return metadata
 
     @staticmethod
-    def in_range(num: int, start: Optional[int], end: Optional[int], /) -> bool:
+    def in_range(num: float, start: Optional[float], end: Optional[float], /) -> bool:
         if start is not None and num < start:
             return False
         if end is not None and num > end:
@@ -189,6 +199,13 @@ class ExperimentFilters:
         if id_range_filter is not None:
             return id_range_filter.min_id, id_range_filter.max_id
 
+    @property
+    def age_range(self) -> Optional[Tuple[Optional[int], Optional[int]]]:
+        """Optional[Tuple[Optional[:class:`int`], Optional[:class:`int`]]]: The range of guild ages that are eligible for the population."""
+        age_range_filter = self.options.guild_age_range_days
+        if age_range_filter is not None:
+            return age_range_filter.min_id, age_range_filter.max_id
+
     @property
     def member_count_range(self) -> Optional[Tuple[Optional[int], Optional[int]]]:
         """Optional[Tuple[Optional[:class:`int`], Optional[:class:`int`]]]: The range of guild member counts that are eligible for the population."""
@@ -253,6 +270,12 @@ class ExperimentFilters:
             if not self.in_range(guild.id, *id_range):
                 return False
 
+        age_range = self.age_range
+        if age_range is not None:
+            # Guild must be within the range of ages (in days)
+            if not self.in_range((utcnow() - guild.created_at).days, *age_range):
+                return False
+
         member_count_range = self.member_count_range
         if member_count_range is not None and guild.member_count is not None:
             # Guild must be within the range of member counts