Refactor message retrieval logic: simplify receive_text parameters, add methods for latest and new text messages

Author: Keniis0712

StealthIM/apis/message.py

@@ -100,14 +100,13 @@ async def get_message(
         session: str,
         groupid: int,
         from_id: int,
-        old_to_new: bool,
         sync: bool,
         limit: int = 128,
 ):
     assert 0 <= limit <= 256, "Limit must be between 0 and 256"
     api_address = (
         f'{url}/api/v1/message/{groupid}?'
-        f'msgid={from_id}&prev={"false" if old_to_new else "true"}&sync={"true" if sync else "false"}&limit={limit}'
+        f'msgid={from_id}&sync={"true" if sync else "false"}&limit={limit}'
     )
     logger.debug(f"Called API get_message with url {api_address}")
     headers = {

StealthIM/group.py

@@ -129,11 +129,10 @@ async def send_text(self, message: str) -> SendMessageResult:
             StealthIM.apis.message.MessageType.Text
         )
 
-    async def receive_text(self, from_id: int = 0, old_to_new: bool = True, sync: bool = True, limit: int = 128):
+    async def receive_text(self, from_id: int = 0, sync: bool = True, limit: int = 128):
         """
         Receive text messages from the group.
         :param from_id: Start message ID
-        :param old_to_new: Sort from old to new
         :param sync: Whether to sync
         :param limit: Maximum number of messages
         :yield: Message data
@@ -143,13 +142,20 @@ async def receive_text(self, from_id: int = 0, old_to_new: bool = True, sync: bo
             self.user.session,
             self.group_id,
             from_id,
-            old_to_new,
             sync,
             limit
         )
         async for data in gen:
             yield data
 
+    async def receive_latest_text(self, limit: int = 128):
+        async for data in self.receive_text(from_id=0, sync=False, limit=limit):
+            yield data
+
+    async def receive_new_text(self, limit: int = 128):
+        async for data in self.receive_text(from_id=-1, limit=limit):
+            yield data
+
     async def recall_message(self, message_id: int):
         return await StealthIM.apis.message.recall_message(self.user.server.url, self.user.session, self.group_id, message_id)