feat: Implement setting own presence (Active/Cloaked aka. offline)

Author: tuxuser

tests/test_presence.py

@@ -1,6 +1,8 @@
 from httpx import Response
 import pytest
 
+from xbox.webapi.api.provider.presence.models import PresenceState
+
 from tests.common import get_response_json
 
 
@@ -44,3 +46,27 @@ async def test_presence_own(respx_mock, xbl_client):
     await xbl_client.presence.get_presence_own()
 
     assert route.called
+
+
+@pytest.mark.asyncio
+async def test_presence_own_set(respx_mock, xbl_client):
+    route = respx_mock.put(
+        "https://userpresence.xboxlive.com/users/xuid(2669321029139235)/state"
+    ).mock(return_value=Response(200))
+
+    ret = await xbl_client.presence.set_presence_own(PresenceState.ACTIVE)
+
+    assert route.called
+    assert ret
+
+
+@pytest.mark.asyncio
+async def test_presence_own_set_fail(respx_mock, xbl_client):
+    route = respx_mock.put(
+        "https://userpresence.xboxlive.com/users/xuid(2669321029139235)/state"
+    ).mock(return_value=Response(500))
+
+    ret = await xbl_client.presence.set_presence_own(PresenceState.CLOAKED)
+
+    assert route.called
+    assert not ret

xbox/webapi/api/provider/presence/__init__.py

@@ -8,6 +8,7 @@
     PresenceBatchResponse,
     PresenceItem,
     PresenceLevel,
+    PresenceState,
 )
 
 
@@ -91,3 +92,20 @@ async def get_presence_own(
         )
         resp.raise_for_status()
         return PresenceItem(**resp.json())
+
+    async def set_presence_own(self, presence_state: PresenceState, **kwargs) -> bool:
+        """
+        Set presence of own profile
+
+        Args:
+            presence_state: State of presence
+
+        Returns:
+            `True` on success, `False` otherwise
+        """
+        url = self.PRESENCE_URL + f"/users/xuid({self.client.xuid})/state"
+        data = {"state": presence_state.value}
+        resp = await self.client.session.put(
+            url, json=data, headers=self.HEADERS_PRESENCE, **kwargs
+        )
+        return resp.status_code == 200

xbox/webapi/api/provider/presence/models.py

@@ -11,6 +11,11 @@ class PresenceLevel(str, Enum):
     ALL = "all"
 
 
+class PresenceState(str, Enum):
+    ACTIVE = "Active"
+    CLOAKED = "Cloaked"
+
+
 class LastSeen(CamelCaseModel):
     device_type: str
     title_id: Optional[str]