Add entity ranges automatically

Author: henry232323

src/spectrum/models/content.py

@@ -62,6 +62,7 @@ class ContentBlock:
     type: str
     blocks: list[Block] = None
     data: Optional[dict] = None
+    reply_id: int = None
 
     def __post_init__(self):
         self.id = int(self.id) if isinstance(self.id, str) and self.id.isnumeric() else self.id

src/spectrum/models/lobby.py

@@ -2,6 +2,7 @@
 
 from . import message, abc
 from .. import httpclient
+from ..util.entity_ranges import get_entity_ranges
 
 
 class Lobby(abc.Identifier, abc.Subscription):
@@ -81,12 +82,18 @@ def leader(self):
     def __repr__(self):
         return f"Lobby(id={repr(self.id)}, name={repr(self.name)})"
 
-    async def send(self, content: str):
+    async def send(self, content: str, add_links=True):
+        entity_ranges = []
+
+        if add_links:
+            entity_ranges = get_entity_ranges(content)
+
+
         payload = await self._client._http.send_message({
             "lobby_id": self.id,
             "content_state": {"blocks": [
                 {"key": "bpeol", "text": content, "type": "unstyled", "depth": 0,
-                 "inlineStyleRanges": [], "entityRanges": [], "data": {}}], "entityMap": {}},
+                 "inlineStyleRanges": [], "entityRanges": entity_ranges, "data": {}}], "entityMap": {}},
             "plaintext": content, "media_id": None, "highlight_role_id": None
         })
 

src/spectrum/util/entity_ranges.py

@@ -0,0 +1,15 @@
+import re
+from typing import List
+
+from spectrum.models.content import EntityRange
+
+url_pattern = re.compile(
+    r"https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_\+.~#?&//=]*)")
+
+
+def get_entity_ranges(url: str) -> List[EntityRange]:
+    ranges = []
+    for i, match in enumerate(url_pattern.finditer(url)):
+        ranges.append(EntityRange(match.start(), match.end() - match.start(), i))
+
+    return ranges