add hacky fix for KNOWN_TOPICS issue

Author: JJTech0130

pypush/apns/lifecycle.py

@@ -205,6 +205,7 @@ async def _update_filter(self):
     @asynccontextmanager
     async def _filter(self, topics: list[str]):
         for topic in topics:
+            protocol.ADD_KNOWN_TOPIC(topic)
             self._filters[topic] = self._filters.get(topic, 0) + 1
         await self._update_filter()
         yield
@@ -215,6 +216,7 @@ async def _filter(self, topics: list[str]):
         await self._update_filter()
 
     async def mint_scoped_token(self, topic: str) -> bytes:
+        protocol.ADD_KNOWN_TOPIC(topic)
         topic_hash = sha1(topic.encode()).digest()
         await self._send(
             protocol.ScopedTokenCommand(token=await self.base_token, topic=topic_hash)

pypush/apns/protocol.py

@@ -12,6 +12,12 @@
 KNOWN_TOPICS_LOOKUP = {sha1(topic.encode()).digest():topic for topic in KNOWN_TOPICS}
 # fmt: on
 
+def ADD_KNOWN_TOPIC(topic: str):
+    # Add a topic and hash to the known topics and lookup
+    if topic not in KNOWN_TOPICS:
+        KNOWN_TOPICS.add(topic)
+        topic_hash = sha1(topic.encode()).digest()
+        KNOWN_TOPICS_LOOKUP[topic_hash] = topic
 
 @dataclass
 class Command:
@@ -177,6 +183,9 @@ def __post_init__(self):
             and self.outgoing is not None
         ) and not (self._token_topic_1 is not None and self._token_topic_2 is not None):
             raise ValueError("topic, token, and outgoing must be set.")
+        
+        if self.topic is not None and isinstance(self.topic, str):
+            ADD_KNOWN_TOPIC(self.topic)
 
         if self.outgoing is True:
             assert self.topic and self.token