Add update contacts method, fixes #79

signalbot/api.py

@@ -2,7 +2,7 @@
 
 import aiohttp
 import websockets
-from typing import Any
+from typing import Any, Optional
 
 
 class SignalAPI:
@@ -159,6 +159,34 @@ async def get_attachment(self, attachment_id: str) -> str:
 
         return base64_string
 
+    async def update_contact(
+        self,
+        receiver: str,
+        expiration_in_seconds: Optional[int] = None,
+        name: Optional[str] = None,
+    ):
+        uri = self._contacts_uri() + "/" + receiver
+        payload = {}
+        if expiration_in_seconds is not None:
+            payload["expiration_in_seconds"] = expiration_in_seconds
+
+        if name is not None:
+            payload["name"] = name
+
+        if payload == {}:
+            raise ContactUpdateError("Empty contact update")
+
+        try:
+            async with aiohttp.ClientSession() as session:
+                resp = await session.put(uri, json=payload)
+                resp.raise_for_status()
+                return resp
+        except (
+            aiohttp.ClientError,
+            aiohttp.http_exceptions.HttpProcessingError,
+        ):
+            raise ContactUpdateError
+
     def _attachment_rest_uri(self):
         return f"http://{self.signal_service}/v1/attachments"
 
@@ -177,6 +205,9 @@ def _typing_indicator_uri(self):
     def _groups_uri(self):
         return f"http://{self.signal_service}/v1/groups/{self.phone_number}"
 
+    def _contacts_uri(self):
+        return f"http://{self.signal_service}/v1/contacts"
+
 
 class ReceiveMessagesError(Exception):
     pass
@@ -208,3 +239,7 @@ class GroupsError(Exception):
 
 class GetAttachmentError(Exception):
     pass
+
+
+class ContactUpdateError(Exception):
+    pass

signalbot/bot.py

@@ -225,6 +225,17 @@ async def stop_typing(self, receiver: str):
         receiver = self._resolve_receiver(receiver)
         await self._signal.stop_typing(receiver)
 
+    async def update_contact(
+        self,
+        receiver: str,
+        expiration_in_seconds: Optional[int] = None,
+        name: Optional[str] = None,
+    ) -> None:
+        receiver = self._resolve_receiver(receiver)
+        await self._signal.update_contact(
+            receiver, expiration_in_seconds=expiration_in_seconds, name=name
+        )
+
     async def _detect_groups(self):
         # reset group lookups to avoid stale data
         self.groups = await self._signal.get_groups()