Enhance message retrieval: update receive_text and get_message functions to support additional parameters for improved flexibility

Keniis0712 · Keniis0712 · commit bcf97e33e187 · 2025-09-21T20:01:13.000+08:00
diff --git a/StealthIM/apis/message.py b/StealthIM/apis/message.py
@@ -100,12 +100,15 @@ async def get_message(
         session: str,
         groupid: int,
         from_id: int,
-        direction: int
+        old_to_new: bool,
+        sync: bool,
+        limit: int = 128,
 ):
-    # noinspection PyStatementEffect
-    direction  # currently unused
-
-    api_address = f'{url}/api/v1/message/{groupid}?msgid={from_id}'
+    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}'
+    )
     logger.debug(f"Called API get_message with url {api_address}")
     headers = {
         "Authorization": f"Bearer {session}",
diff --git a/StealthIM/group.py b/StealthIM/group.py
@@ -88,13 +88,15 @@ async def send_text(self, message: str) -> SendMessageResult:
             StealthIM.apis.message.MessageType.Text
         )
 
-    async def receive_text(self, from_id: int = 0):
+    async def receive_text(self, from_id: int = 0, old_to_new: bool = True, sync: bool = True, limit: int = 128):
         gen = StealthIM.apis.message.get_message(
             self.user.server.url,
             self.user.session,
             self.group_id,
             from_id,
-            0
+            old_to_new,
+            sync,
+            limit
         )
         async for data in gen:
             yield data
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stealthimpysdk"       # pip 安装名
-version = "0.1.3"             # 版本
+version = "0.1.4"             # 版本
 description = "The SDK for StealthIM"
 readme = "README.md"
 license = {file = "LICENSE"}
