| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306 |
- """Main async client for the Chatto Connect API.
- Chatto migrated from GraphQL to a protobuf-first Connect API in v0.4.x
- (see ADR-042). The client speaks Connect binary protobuf via the official
- ``connectrpc`` Python package and the generated service stubs under
- ``chattolib._pb`` for all request/response operations. Realtime events
- live in ``chattolib.realtime``.
- """
- # mypy: disable-error-code="no-any-return"
- # Rationale: attribute access on generated protobuf messages is Any-typed
- # from mypy's perspective (the generated modules skip type checking via
- # follow_imports=skip). The runtime types are exactly what the return-type
- # annotations claim.
- from __future__ import annotations
- import hashlib
- from collections.abc import Awaitable
- from datetime import datetime
- from pathlib import Path
- from typing import Any, TypeVar
- import httpx
- # ConnectError isn't re-exported publicly by connectrpc.__init__; import from
- # its submodule so the top-level import path stays clean for callers.
- from connectrpc.errors import ConnectError # noqa: E402
- from chattolib._pb.chatto.admin.v1 import (
- event_log_pb2,
- room_layout_pb2,
- )
- from chattolib._pb.chatto.admin.v1 import (
- members_pb2 as admin_members_pb2,
- )
- from chattolib._pb.chatto.admin.v1 import (
- permissions_pb2 as admin_permissions_pb2,
- )
- from chattolib._pb.chatto.admin.v1 import (
- roles_pb2 as admin_roles_pb2,
- )
- from chattolib._pb.chatto.admin.v1 import (
- server_pb2 as admin_server_pb2,
- )
- from chattolib._pb.chatto.api.v1 import (
- account_pb2,
- asset_uploads_pb2,
- attachments_pb2,
- common_pb2,
- external_identities_pb2,
- link_previews_pb2,
- member_directory_pb2,
- messages_pb2,
- notification_preferences_pb2,
- notifications_pb2,
- pagination_pb2,
- presence_pb2,
- push_notifications_pb2,
- reactions_pb2,
- read_state_pb2,
- roles_pb2,
- room_directory_pb2,
- room_timeline_pb2,
- rooms_pb2,
- server_state_pb2,
- threads_pb2,
- user_status_pb2,
- viewer_pb2,
- voice_calls_pb2,
- )
- from chattolib._pb.chatto.auth.v1 import external_identity_auth_pb2
- from chattolib._pb.chatto.discovery.v1 import server_pb2 as discovery_server_pb2
- from chattolib._transport import (
- ServiceClients,
- build_service_clients,
- pb_to_dict,
- translate_connect_error,
- )
- from chattolib.exceptions import ChattoAuthError, ChattoError
- from chattolib.types import (
- ActiveCall,
- AdminMember,
- AdminRole,
- AdminRoomLayoutGroup,
- AdminRoomLayoutItemKind,
- Asset,
- AssetUpload,
- DirectoryMember,
- ExternalIdentityProvider,
- FollowedThread,
- FollowedThreadsPage,
- ImageTransformOptions,
- LinkedExternalIdentity,
- LinkPreview,
- Message,
- Notification,
- NotificationLevel,
- NotificationPreference,
- NotificationsPage,
- Page,
- PresenceStatus,
- Role,
- Room,
- RoomBan,
- RoomDirectoryScope,
- RoomGroup,
- RoomWithViewerState,
- ServerConfig,
- ServerLogin,
- ServerProfile,
- ServerRuntimeConfig,
- TimeFormat,
- TimelinePage,
- User,
- UserSettings,
- ViewerUser,
- format_datetime,
- parse_datetime,
- )
- RES = TypeVar("RES")
- def _page_pb(
- limit: int | None, offset: int | None
- ) -> pagination_pb2.PageRequest | None:
- if limit is None and offset is None:
- return None
- return pagination_pb2.PageRequest(limit=limit or 0, offset=offset or 0)
- def _thumbnail_pb(
- opts: ImageTransformOptions | None,
- ) -> common_pb2.ImageTransformOptions | None:
- if opts is None:
- return None
- return common_pb2.ImageTransformOptions(
- width=opts.width, height=opts.height, fit=opts.fit.value
- )
- def _timestamp_pb(value: datetime | None) -> Any:
- from google.protobuf import timestamp_pb2
- if value is None:
- return None
- ts = timestamp_pb2.Timestamp()
- ts.FromJsonString(format_datetime(value))
- return ts
- class ChattoClient:
- """Async client for the Chatto Connect API.
- Usage::
- async with await ChattoClient.login("user", "pass") as client:
- viewer = await client.get_viewer()
- rooms = await client.list_rooms()
- # Or with an existing token:
- async with ChattoClient(token="cht_...") as client:
- ...
- """
- DEFAULT_BASE_URL = "https://chat.chatto.run"
- def __init__(
- self,
- token: str | None = None,
- *,
- base_url: str = DEFAULT_BASE_URL,
- session_cookie: str | None = None,
- service_clients: ServiceClients | None = None,
- ) -> None:
- self._base_url = base_url.rstrip("/")
- self._token = token
- self._session_cookie = session_cookie
- self._svc = service_clients or build_service_clients(self._base_url)
- self._owns_clients = service_clients is None
- @classmethod
- async def login(
- cls,
- login: str,
- password: str,
- *,
- base_url: str = DEFAULT_BASE_URL,
- ) -> ChattoClient:
- """Authenticate with username and password, returning a connected client.
- Uses Chatto's ``/auth/login`` HTTP endpoint (which is still exposed
- alongside the Connect API) and captures both the returned bearer token
- and any ``chatto_session`` cookie.
- """
- base = base_url.rstrip("/")
- async with httpx.AsyncClient() as http:
- resp = await http.post(
- f"{base}/auth/login",
- json={"login": login, "password": password},
- )
- if resp.status_code == 401:
- raise ChattoAuthError("Invalid credentials")
- resp.raise_for_status()
- body = resp.json()
- token = body.get("token")
- session_cookie = None
- if "set-cookie" in resp.headers:
- for cookie_header in resp.headers.get_list("set-cookie"):
- if cookie_header.startswith("chatto_session="):
- session_cookie = cookie_header.split(";")[0].split("=", 1)[1]
- break
- return cls(token=token, base_url=base_url, session_cookie=session_cookie)
- async def __aenter__(self) -> ChattoClient:
- return self
- async def __aexit__(self, *exc: Any) -> None:
- await self.close()
- async def close(self) -> None:
- if self._owns_clients:
- await self._svc.close()
- # --- Transport ------------------------------------------------------
- @property
- def base_url(self) -> str:
- return self._base_url
- @property
- def token(self) -> str | None:
- return self._token
- @property
- def session_cookie(self) -> str | None:
- return self._session_cookie
- @property
- def services(self) -> ServiceClients:
- """Direct access to the underlying ConnectRPC service clients.
- Useful when a caller wants to reach an RPC that this class doesn't
- expose yet, or wants full protobuf messages instead of the
- dataclass views returned by the high-level helpers.
- """
- return self._svc
- def _headers(self) -> dict[str, str]:
- headers: dict[str, str] = {}
- if self._token:
- headers["Authorization"] = f"Bearer {self._token}"
- if self._session_cookie:
- headers["Cookie"] = f"chatto_session={self._session_cookie}"
- return headers
- async def _rpc(self, coro: Awaitable[RES]) -> RES:
- """Await a ConnectRPC coroutine, translating errors."""
- try:
- return await coro
- except ConnectError as exc:
- raise translate_connect_error(exc) from exc
- # --- Server discovery ----------------------------------------------
- async def get_server(self) -> tuple[ServerProfile, ServerLogin]:
- """Public server profile and login options. Does not require auth."""
- resp = await self._rpc(
- self._svc.server_discovery.get_server(
- discovery_server_pb2.GetServerRequest(),
- headers=self._headers(),
- )
- )
- return (
- ServerProfile.parse(pb_to_dict(resp.profile)),
- ServerLogin.parse(pb_to_dict(resp.login)),
- )
- async def get_motd(self) -> str | None:
- resp = await self._rpc(
- self._svc.server.get_motd(
- server_state_pb2.GetMotdRequest(), headers=self._headers()
- )
- )
- if not resp.HasField("motd"):
- return None
- return resp.motd
- async def get_runtime_config(self) -> ServerRuntimeConfig:
- resp = await self._rpc(
- self._svc.server.get_runtime_config(
- server_state_pb2.GetRuntimeConfigRequest(),
- headers=self._headers(),
- )
- )
- return ServerRuntimeConfig.parse(pb_to_dict(resp.runtime))
- # --- Viewer ---------------------------------------------------------
- async def get_viewer(self) -> dict[str, Any]:
- """Full authenticated viewer snapshot (camelCase JSON dict).
- The response contains ``user``, ``capabilities``, notification
- preferences, permissions and viewer state. Callers that only need the
- current user's public profile should use ``me()`` for a typed result.
- """
- resp = await self._rpc(
- self._svc.viewer.get_viewer(
- viewer_pb2.GetViewerRequest(), headers=self._headers()
- )
- )
- return pb_to_dict(resp)
- async def viewer_user(self) -> ViewerUser | None:
- data = await self.get_viewer()
- return ViewerUser.parse(data.get("user"))
- async def me(self) -> User:
- """Return the authenticated user's public profile."""
- viewer = await self.viewer_user()
- if viewer is None or viewer.profile is None:
- raise ChattoAuthError("No authenticated viewer")
- return viewer.profile
- # --- MyAccount -----------------------------------------------------
- async def update_profile(
- self,
- *,
- display_name: str | None = None,
- login: str | None = None,
- ) -> User:
- req = account_pb2.UpdateProfileRequest()
- if display_name is not None:
- req.display_name = display_name
- if login is not None:
- req.login = login
- resp = await self._rpc(
- self._svc.account.update_profile(req, headers=self._headers())
- )
- user = User.parse(pb_to_dict(resp.user))
- assert user is not None
- return user
- async def upload_avatar(
- self,
- file_path: str | Path,
- *,
- content_type: str = "image/png",
- ) -> User:
- p = Path(file_path)
- req = account_pb2.UploadAvatarRequest(
- image=common_pb2.ImageUpload(
- image=p.read_bytes(),
- filename=p.name,
- content_type=content_type,
- )
- )
- resp = await self._rpc(
- self._svc.account.upload_avatar(req, headers=self._headers())
- )
- user = User.parse(pb_to_dict(resp.user))
- assert user is not None
- return user
- async def delete_avatar(self) -> User:
- resp = await self._rpc(
- self._svc.account.delete_avatar(
- account_pb2.DeleteAvatarRequest(), headers=self._headers()
- )
- )
- user = User.parse(pb_to_dict(resp.user))
- assert user is not None
- return user
- async def update_password(self, new_password: str, current_password: str = "") -> User:
- resp = await self._rpc(
- self._svc.account.update_password(
- account_pb2.UpdatePasswordRequest(
- password=new_password, current_password=current_password
- ),
- headers=self._headers(),
- )
- )
- user = User.parse(pb_to_dict(resp.user))
- assert user is not None
- return user
- async def update_settings(
- self,
- *,
- timezone: str | None = None,
- time_format: TimeFormat | None = None,
- ) -> UserSettings:
- req = account_pb2.UpdateSettingsRequest()
- if timezone is not None:
- req.timezone = timezone
- if time_format is not None:
- req.time_format = time_format.value
- resp = await self._rpc(
- self._svc.account.update_settings(req, headers=self._headers())
- )
- return UserSettings.parse(pb_to_dict(resp.settings))
- async def update_presence(
- self,
- status: PresenceStatus,
- *,
- user_selected: bool = True,
- ) -> PresenceStatus:
- if status in (PresenceStatus.UNSPECIFIED, PresenceStatus.OFFLINE):
- raise ValueError(
- "UNSPECIFIED and OFFLINE cannot be set as presence status; "
- "stop refreshing to go offline"
- )
- req = presence_pb2.UpdatePresenceRequest(
- status=status.value, user_selected=user_selected
- )
- resp = await self._rpc(
- self._svc.account.update_presence(req, headers=self._headers())
- )
- name = presence_pb2.PresenceStatus.Name(resp.status)
- return PresenceStatus(name)
- async def update_custom_status(
- self,
- emoji: str,
- text: str,
- *,
- expires_at: datetime | None = None,
- ) -> dict[str, Any]:
- req = user_status_pb2.UpdateCustomStatusRequest(emoji=emoji, text=text)
- if expires_at is not None:
- req.expires_at.CopyFrom(_timestamp_pb(expires_at))
- resp = await self._rpc(
- self._svc.account.update_custom_status(req, headers=self._headers())
- )
- return pb_to_dict(resp)
- async def delete_custom_status(self) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.account.delete_custom_status(
- user_status_pb2.DeleteCustomStatusRequest(), headers=self._headers()
- )
- )
- return pb_to_dict(resp)
- async def request_account_deletion(self) -> str:
- resp = await self._rpc(
- self._svc.account.request_account_deletion(
- account_pb2.RequestAccountDeletionRequest(),
- headers=self._headers(),
- )
- )
- return resp.confirmation_token
- async def delete_my_account(self, confirmation_token: str) -> bool:
- resp = await self._rpc(
- self._svc.account.delete_my_account(
- account_pb2.DeleteMyAccountRequest(
- confirmation_token=confirmation_token
- ),
- headers=self._headers(),
- )
- )
- return resp.deleted
- # --- Users ----------------------------------------------------------
- async def list_users(
- self,
- *,
- search: str = "",
- limit: int | None = None,
- offset: int | None = None,
- ) -> tuple[list[DirectoryMember], Page]:
- req = member_directory_pb2.ListUsersRequest(search=search)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.users.list_users(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- users = [
- u
- for u in (DirectoryMember.parse(row) for row in data.get("users") or [])
- if u is not None
- ]
- return users, Page.parse(data.get("page"))
- async def get_user(
- self, *, user_id: str | None = None, login: str | None = None
- ) -> DirectoryMember | None:
- if bool(user_id) == bool(login):
- raise ValueError("get_user requires exactly one of user_id or login")
- req = member_directory_pb2.GetUserRequest()
- if user_id:
- req.user_id = user_id
- else:
- assert login is not None
- req.login = login
- resp = await self._rpc(
- self._svc.users.get_user(req, headers=self._headers())
- )
- return DirectoryMember.parse(pb_to_dict(resp.user))
- async def batch_get_users(self, user_ids: list[str]) -> list[DirectoryMember]:
- resp = await self._rpc(
- self._svc.users.batch_get_users(
- member_directory_pb2.BatchGetUsersRequest(user_ids=user_ids),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- u
- for u in (DirectoryMember.parse(row) for row in data.get("users") or [])
- if u is not None
- ]
- # --- Roles (public) -----------------------------------------------
- async def list_roles(self) -> list[Role]:
- resp = await self._rpc(
- self._svc.roles.list_roles(
- roles_pb2.ListRolesRequest(), headers=self._headers()
- )
- )
- data = pb_to_dict(resp)
- return [
- r for r in (Role.parse(row) for row in data.get("roles") or []) if r is not None
- ]
- async def get_role(self, name: str) -> Role | None:
- resp = await self._rpc(
- self._svc.roles.get_role(
- roles_pb2.GetRoleRequest(name=name), headers=self._headers()
- )
- )
- return Role.parse(pb_to_dict(resp.role))
- async def batch_get_roles(self, names: list[str]) -> list[Role]:
- resp = await self._rpc(
- self._svc.roles.batch_get_roles(
- roles_pb2.BatchGetRolesRequest(names=names), headers=self._headers()
- )
- )
- data = pb_to_dict(resp)
- return [
- r for r in (Role.parse(row) for row in data.get("roles") or []) if r is not None
- ]
- # --- Room directory ------------------------------------------------
- async def list_rooms(
- self, scope: RoomDirectoryScope = RoomDirectoryScope.ALL
- ) -> list[RoomWithViewerState]:
- resp = await self._rpc(
- self._svc.room_directory.list_rooms(
- room_directory_pb2.ListRoomsRequest(scope=scope.value),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- r
- for r in (RoomWithViewerState.parse(row) for row in data.get("rooms") or [])
- if r is not None
- ]
- async def list_room_groups(self) -> list[RoomGroup]:
- resp = await self._rpc(
- self._svc.room_directory.list_room_groups(
- room_directory_pb2.ListRoomGroupsRequest(),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- g
- for g in (RoomGroup.parse(row) for row in data.get("groups") or [])
- if g is not None
- ]
- async def get_room_group(self, group_id: str) -> RoomGroup | None:
- resp = await self._rpc(
- self._svc.room_directory.get_room_group(
- room_directory_pb2.GetRoomGroupRequest(group_id=group_id),
- headers=self._headers(),
- )
- )
- return RoomGroup.parse(pb_to_dict(resp.group))
- async def batch_get_room_groups(self, group_ids: list[str]) -> list[RoomGroup]:
- resp = await self._rpc(
- self._svc.room_directory.batch_get_room_groups(
- room_directory_pb2.BatchGetRoomGroupsRequest(group_ids=group_ids),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- g
- for g in (RoomGroup.parse(row) for row in data.get("groups") or [])
- if g is not None
- ]
- async def get_room(self, room_id: str) -> RoomWithViewerState | None:
- resp = await self._rpc(
- self._svc.room_directory.get_room(
- room_directory_pb2.GetRoomRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- return RoomWithViewerState.parse(pb_to_dict(resp.room))
- async def batch_get_rooms(self, room_ids: list[str]) -> list[RoomWithViewerState]:
- resp = await self._rpc(
- self._svc.room_directory.batch_get_rooms(
- room_directory_pb2.BatchGetRoomsRequest(room_ids=room_ids),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- r
- for r in (RoomWithViewerState.parse(row) for row in data.get("rooms") or [])
- if r is not None
- ]
- # --- Room lifecycle & membership -----------------------------------
- async def create_room(
- self,
- name: str,
- group_id: str,
- *,
- description: str = "",
- universal: bool = False,
- ) -> Room:
- resp = await self._rpc(
- self._svc.rooms.create_room(
- rooms_pb2.CreateRoomRequest(
- name=name,
- group_id=group_id,
- description=description,
- universal=universal,
- ),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def update_room(
- self,
- room_id: str,
- *,
- name: str | None = None,
- description: str | None = None,
- universal: bool | None = None,
- ) -> Room:
- req = rooms_pb2.UpdateRoomRequest(room_id=room_id)
- if name is not None:
- req.name = name
- if description is not None:
- req.description = description
- if universal is not None:
- req.universal = universal
- resp = await self._rpc(
- self._svc.rooms.update_room(req, headers=self._headers())
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def archive_room(self, room_id: str) -> Room:
- resp = await self._rpc(
- self._svc.rooms.archive_room(
- rooms_pb2.ArchiveRoomRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def unarchive_room(self, room_id: str) -> Room:
- resp = await self._rpc(
- self._svc.rooms.unarchive_room(
- rooms_pb2.UnarchiveRoomRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def join_room(self, room_id: str) -> Room:
- resp = await self._rpc(
- self._svc.rooms.join_room(
- rooms_pb2.JoinRoomRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def join_room_group(self, group_id: str) -> list[str]:
- resp = await self._rpc(
- self._svc.rooms.join_room_group(
- rooms_pb2.JoinRoomGroupRequest(group_id=group_id),
- headers=self._headers(),
- )
- )
- return list(resp.joined_room_ids)
- async def start_dm(self, participant_ids: list[str]) -> Room:
- resp = await self._rpc(
- self._svc.rooms.start_dm(
- rooms_pb2.StartDMRequest(participant_ids=participant_ids),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def leave_room(self, room_id: str) -> bool:
- resp = await self._rpc(
- self._svc.rooms.leave_room(
- rooms_pb2.LeaveRoomRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- return resp.left
- async def add_member(self, room_id: str, user_id: str) -> DirectoryMember | None:
- resp = await self._rpc(
- self._svc.rooms.add_member(
- rooms_pb2.AddMemberRequest(room_id=room_id, user_id=user_id),
- headers=self._headers(),
- )
- )
- return DirectoryMember.parse(pb_to_dict(resp.member))
- async def remove_member(self, room_id: str, user_id: str) -> bool:
- resp = await self._rpc(
- self._svc.rooms.remove_member(
- rooms_pb2.RemoveMemberRequest(room_id=room_id, user_id=user_id),
- headers=self._headers(),
- )
- )
- return resp.removed
- async def list_room_members(
- self,
- room_id: str,
- *,
- search: str = "",
- limit: int | None = None,
- offset: int | None = None,
- ) -> tuple[list[DirectoryMember], Page]:
- req = member_directory_pb2.ListRoomMembersRequest(room_id=room_id, search=search)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.rooms.list_members(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- members = [
- m
- for m in (DirectoryMember.parse(row) for row in data.get("members") or [])
- if m is not None
- ]
- return members, Page.parse(data.get("page"))
- async def get_room_member(self, room_id: str, user_id: str) -> DirectoryMember | None:
- resp = await self._rpc(
- self._svc.rooms.get_member(
- member_directory_pb2.GetRoomMemberRequest(
- room_id=room_id, user_id=user_id
- ),
- headers=self._headers(),
- )
- )
- return DirectoryMember.parse(pb_to_dict(resp.member))
- async def batch_get_room_members(
- self, room_id: str, user_ids: list[str]
- ) -> list[DirectoryMember]:
- resp = await self._rpc(
- self._svc.rooms.batch_get_members(
- member_directory_pb2.BatchGetRoomMembersRequest(
- room_id=room_id, user_ids=user_ids
- ),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- m
- for m in (DirectoryMember.parse(row) for row in data.get("members") or [])
- if m is not None
- ]
- async def ban_member(
- self,
- room_id: str,
- user_id: str,
- reason: str,
- *,
- expires_at: datetime | None = None,
- ) -> bool:
- req = rooms_pb2.BanMemberRequest(
- room_id=room_id, user_id=user_id, reason=reason
- )
- if expires_at is not None:
- req.expires_at.CopyFrom(_timestamp_pb(expires_at))
- resp = await self._rpc(
- self._svc.rooms.ban_member(req, headers=self._headers())
- )
- return resp.banned
- async def unban_member(self, room_id: str, user_id: str, reason: str) -> bool:
- resp = await self._rpc(
- self._svc.rooms.unban_member(
- rooms_pb2.UnbanMemberRequest(
- room_id=room_id, user_id=user_id, reason=reason
- ),
- headers=self._headers(),
- )
- )
- return resp.unbanned
- async def list_bans(
- self,
- *,
- room_id: str = "",
- limit: int | None = None,
- offset: int | None = None,
- ) -> tuple[list[RoomBan], Page]:
- req = rooms_pb2.ListBansRequest(room_id=room_id)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.rooms.list_bans(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- bans = [
- b
- for b in (RoomBan.parse(row) for row in data.get("bans") or [])
- if b is not None
- ]
- return bans, Page.parse(data.get("page"))
- async def update_typing_indicator(
- self, room_id: str, *, thread_root_event_id: str = ""
- ) -> bool:
- resp = await self._rpc(
- self._svc.rooms.update_typing_indicator(
- rooms_pb2.UpdateTypingIndicatorRequest(
- room_id=room_id, thread_root_event_id=thread_root_event_id
- ),
- headers=self._headers(),
- )
- )
- return resp.updated
- # --- Room timeline / read state -----------------------------------
- async def get_room_events(
- self,
- room_id: str,
- *,
- limit: int | None = None,
- before: str | None = None,
- after: str | None = None,
- ) -> TimelinePage:
- req = room_timeline_pb2.GetRoomEventsRequest(room_id=room_id)
- if limit is not None:
- req.limit = limit
- if before is not None:
- req.before = before
- elif after is not None:
- req.after = after
- resp = await self._rpc(
- self._svc.rooms.get_room_events(req, headers=self._headers())
- )
- return TimelinePage.parse(pb_to_dict(resp.page))
- async def get_room_events_around(
- self,
- room_id: str,
- event_id: str,
- *,
- limit: int | None = None,
- ) -> tuple[TimelinePage, int]:
- req = room_timeline_pb2.GetRoomEventsAroundRequest(
- room_id=room_id, event_id=event_id
- )
- if limit is not None:
- req.limit = limit
- resp = await self._rpc(
- self._svc.rooms.get_room_events_around(req, headers=self._headers())
- )
- return TimelinePage.parse(pb_to_dict(resp.page)), resp.target_index
- async def mark_room_as_read(
- self, room_id: str, up_to_event_id: str = ""
- ) -> tuple[datetime | None, datetime | None]:
- resp = await self._rpc(
- self._svc.rooms.mark_room_as_read(
- read_state_pb2.MarkRoomAsReadRequest(
- room_id=room_id, up_to_event_id=up_to_event_id
- ),
- headers=self._headers(),
- )
- )
- d = pb_to_dict(resp)
- return (
- parse_datetime(d.get("lastReadAt")),
- parse_datetime(d.get("previousLastReadAt")),
- )
- async def list_room_attachments(
- self,
- room_id: str,
- *,
- thumbnail: ImageTransformOptions | None = None,
- limit: int | None = None,
- offset: int | None = None,
- ) -> tuple[list[dict[str, Any]], Page]:
- req = rooms_pb2.ListRoomAttachmentsRequest(room_id=room_id)
- thumb = _thumbnail_pb(thumbnail)
- if thumb is not None:
- req.thumbnail.CopyFrom(thumb)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.rooms.list_room_attachments(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- return list(data.get("attachments") or []), Page.parse(data.get("page"))
- # --- Messages -------------------------------------------------------
- async def fetch_link_preview(self, url: str) -> tuple[LinkPreview | None, str]:
- resp = await self._rpc(
- self._svc.messages.fetch_link_preview(
- link_previews_pb2.FetchLinkPreviewRequest(url=url),
- headers=self._headers(),
- )
- )
- return LinkPreview.parse(pb_to_dict(resp.preview)), resp.preview_token
- async def post_message(
- self,
- room_id: str,
- body: str = "",
- *,
- attachment_asset_ids: list[str] | None = None,
- thread_root_event_id: str = "",
- in_reply_to: str = "",
- also_send_to_channel: bool = False,
- link_preview_token: str = "",
- ) -> Message:
- req = messages_pb2.CreateMessageRequest(
- room_id=room_id,
- body=body,
- thread_root_event_id=thread_root_event_id,
- in_reply_to=in_reply_to,
- also_send_to_channel=also_send_to_channel,
- link_preview_token=link_preview_token,
- )
- if attachment_asset_ids:
- req.attachment_asset_ids.extend(attachment_asset_ids)
- resp = await self._rpc(
- self._svc.messages.create_message(req, headers=self._headers())
- )
- message = Message.parse(pb_to_dict(resp.message))
- assert message is not None
- return message
- async def update_message(
- self,
- room_id: str,
- event_id: str,
- *,
- body: str | None = None,
- also_send_to_channel: bool | None = None,
- ) -> Message:
- req = messages_pb2.UpdateMessageRequest(room_id=room_id, event_id=event_id)
- if body is not None:
- req.body = body
- if also_send_to_channel is not None:
- req.also_send_to_channel = also_send_to_channel
- resp = await self._rpc(
- self._svc.messages.update_message(req, headers=self._headers())
- )
- message = Message.parse(pb_to_dict(resp.message))
- assert message is not None
- return message
- async def delete_message(self, room_id: str, event_id: str) -> bool:
- resp = await self._rpc(
- self._svc.messages.delete_message(
- messages_pb2.DeleteMessageRequest(room_id=room_id, event_id=event_id),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def delete_attachment(
- self, room_id: str, event_id: str, attachment_id: str
- ) -> bool:
- resp = await self._rpc(
- self._svc.messages.delete_attachment(
- messages_pb2.DeleteAttachmentRequest(
- room_id=room_id, event_id=event_id, attachment_id=attachment_id
- ),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def delete_link_preview(self, room_id: str, event_id: str, url: str) -> bool:
- resp = await self._rpc(
- self._svc.messages.delete_link_preview(
- messages_pb2.DeleteLinkPreviewRequest(
- room_id=room_id, event_id=event_id, url=url
- ),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def get_message(self, room_id: str, event_id: str) -> Message | None:
- resp = await self._rpc(
- self._svc.messages.get_message(
- messages_pb2.GetMessageRequest(room_id=room_id, event_id=event_id),
- headers=self._headers(),
- )
- )
- return Message.parse(pb_to_dict(resp.message))
- async def batch_get_messages(
- self, room_id: str, event_ids: list[str]
- ) -> list[Message]:
- resp = await self._rpc(
- self._svc.messages.batch_get_messages(
- messages_pb2.BatchGetMessagesRequest(
- room_id=room_id, event_ids=event_ids
- ),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- m
- for m in (Message.parse(row) for row in data.get("messages") or [])
- if m is not None
- ]
- async def add_reaction(
- self, room_id: str, message_event_id: str, emoji: str
- ) -> bool:
- resp = await self._rpc(
- self._svc.messages.add_reaction(
- reactions_pb2.AddReactionRequest(
- room_id=room_id, message_event_id=message_event_id, emoji=emoji
- ),
- headers=self._headers(),
- )
- )
- return resp.added
- async def remove_reaction(
- self, room_id: str, message_event_id: str, emoji: str
- ) -> bool:
- resp = await self._rpc(
- self._svc.messages.remove_reaction(
- reactions_pb2.RemoveReactionRequest(
- room_id=room_id, message_event_id=message_event_id, emoji=emoji
- ),
- headers=self._headers(),
- )
- )
- return resp.removed
- # --- Threads --------------------------------------------------------
- async def follow_thread(self, room_id: str, thread_root_event_id: str) -> bool:
- resp = await self._rpc(
- self._svc.threads.follow_thread(
- threads_pb2.FollowThreadRequest(
- room_id=room_id, thread_root_event_id=thread_root_event_id
- ),
- headers=self._headers(),
- )
- )
- return resp.following
- async def unfollow_thread(self, room_id: str, thread_root_event_id: str) -> bool:
- resp = await self._rpc(
- self._svc.threads.unfollow_thread(
- threads_pb2.UnfollowThreadRequest(
- room_id=room_id, thread_root_event_id=thread_root_event_id
- ),
- headers=self._headers(),
- )
- )
- return resp.following
- async def list_followed_threads(
- self, *, limit: int | None = None, offset: int | None = None
- ) -> FollowedThreadsPage:
- req = threads_pb2.ListFollowedThreadsRequest()
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.threads.list_followed_threads(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- threads = [FollowedThread.parse(t) for t in data.get("threads") or []]
- users: dict[str, User] = {}
- includes = data.get("includes") or {}
- for uid, user_data in (includes.get("users") or {}).items():
- parsed = User.parse(user_data)
- if parsed is not None:
- users[uid] = parsed
- return FollowedThreadsPage(
- threads=threads, page=Page.parse(data.get("page")), users_by_id=users
- )
- async def get_thread_events(
- self,
- room_id: str,
- thread_root_event_id: str,
- *,
- limit: int | None = None,
- before: str | None = None,
- after: str | None = None,
- ) -> TimelinePage:
- req = room_timeline_pb2.GetThreadEventsRequest(
- room_id=room_id, thread_root_event_id=thread_root_event_id
- )
- if limit is not None:
- req.limit = limit
- if before is not None:
- req.before = before
- elif after is not None:
- req.after = after
- resp = await self._rpc(
- self._svc.threads.get_thread_events(req, headers=self._headers())
- )
- return TimelinePage.parse(pb_to_dict(resp.page))
- async def get_thread_events_around(
- self,
- room_id: str,
- thread_root_event_id: str,
- event_id: str,
- *,
- limit: int | None = None,
- ) -> tuple[TimelinePage, int]:
- req = room_timeline_pb2.GetThreadEventsAroundRequest(
- room_id=room_id,
- thread_root_event_id=thread_root_event_id,
- event_id=event_id,
- )
- if limit is not None:
- req.limit = limit
- resp = await self._rpc(
- self._svc.threads.get_thread_events_around(req, headers=self._headers())
- )
- return TimelinePage.parse(pb_to_dict(resp.page)), resp.target_index
- async def mark_thread_as_read(
- self,
- room_id: str,
- thread_root_event_id: str,
- up_to_event_id: str = "",
- ) -> datetime | None:
- resp = await self._rpc(
- self._svc.threads.mark_thread_as_read(
- read_state_pb2.MarkThreadAsReadRequest(
- room_id=room_id,
- thread_root_event_id=thread_root_event_id,
- up_to_event_id=up_to_event_id,
- ),
- headers=self._headers(),
- )
- )
- d = pb_to_dict(resp)
- return parse_datetime(d.get("previousReadAt"))
- # --- Notifications --------------------------------------------------
- async def list_notifications(
- self, *, limit: int | None = None, offset: int | None = None
- ) -> NotificationsPage:
- req = notifications_pb2.ListNotificationsRequest()
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.notifications.list_notifications(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- notifications = [Notification.parse(n) for n in data.get("notifications") or []]
- return NotificationsPage(
- notifications=notifications, page=Page.parse(data.get("page"))
- )
- async def has_notifications(self) -> bool:
- resp = await self._rpc(
- self._svc.notifications.has_notifications(
- notifications_pb2.HasNotificationsRequest(),
- headers=self._headers(),
- )
- )
- return resp.has_notifications
- async def get_notification(self, notification_id: str) -> Notification | None:
- resp = await self._rpc(
- self._svc.notifications.get_notification(
- notifications_pb2.GetNotificationRequest(notification_id=notification_id),
- headers=self._headers(),
- )
- )
- raw = pb_to_dict(resp).get("notification")
- return Notification.parse(raw) if raw else None
- async def batch_get_notifications(
- self, notification_ids: list[str]
- ) -> list[Notification]:
- resp = await self._rpc(
- self._svc.notifications.batch_get_notifications(
- notifications_pb2.BatchGetNotificationsRequest(
- notification_ids=notification_ids
- ),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [Notification.parse(n) for n in data.get("notifications") or []]
- async def list_room_notifications(
- self,
- room_id: str,
- *,
- limit: int | None = None,
- offset: int | None = None,
- ) -> NotificationsPage:
- req = notifications_pb2.ListRoomNotificationsRequest(room_id=room_id)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.notifications.list_room_notifications(
- req, headers=self._headers()
- )
- )
- data = pb_to_dict(resp)
- return NotificationsPage(
- notifications=[Notification.parse(n) for n in data.get("notifications") or []],
- page=Page.parse(data.get("page")),
- )
- async def list_room_notification_counts(self) -> dict[str, int]:
- resp = await self._rpc(
- self._svc.notifications.list_room_notification_counts(
- notifications_pb2.ListRoomNotificationCountsRequest(),
- headers=self._headers(),
- )
- )
- return {row.room_id: row.total_count for row in resp.room_counts}
- async def dismiss_notification(self, notification_id: str) -> bool:
- resp = await self._rpc(
- self._svc.notifications.dismiss_notification(
- notifications_pb2.DismissNotificationRequest(
- notification_id=notification_id
- ),
- headers=self._headers(),
- )
- )
- return resp.dismissed
- async def dismiss_all_notifications(self) -> int:
- resp = await self._rpc(
- self._svc.notifications.dismiss_all_notifications(
- notifications_pb2.DismissAllNotificationsRequest(),
- headers=self._headers(),
- )
- )
- return resp.dismissed_count
- # --- Notification preferences --------------------------------------
- async def get_server_notification_preference(self) -> NotificationPreference:
- resp = await self._rpc(
- self._svc.notification_prefs.get_server_notification_preference(
- notification_preferences_pb2.GetServerNotificationPreferenceRequest(),
- headers=self._headers(),
- )
- )
- return NotificationPreference.parse(pb_to_dict(resp.preference))
- async def update_server_notification_preference(
- self, level: NotificationLevel
- ) -> NotificationPreference:
- resp = await self._rpc(
- self._svc.notification_prefs.update_server_notification_preference(
- notification_preferences_pb2.UpdateServerNotificationPreferenceRequest(
- level=level.value
- ),
- headers=self._headers(),
- )
- )
- return NotificationPreference.parse(pb_to_dict(resp.preference))
- async def get_room_notification_preference(
- self, room_id: str
- ) -> NotificationPreference:
- resp = await self._rpc(
- self._svc.notification_prefs.get_room_notification_preference(
- notification_preferences_pb2.GetRoomNotificationPreferenceRequest(
- room_id=room_id
- ),
- headers=self._headers(),
- )
- )
- return NotificationPreference.parse(pb_to_dict(resp.preference))
- async def update_room_notification_preference(
- self, room_id: str, level: NotificationLevel
- ) -> NotificationPreference:
- resp = await self._rpc(
- self._svc.notification_prefs.update_room_notification_preference(
- notification_preferences_pb2.UpdateRoomNotificationPreferenceRequest(
- room_id=room_id, level=level.value
- ),
- headers=self._headers(),
- )
- )
- return NotificationPreference.parse(pb_to_dict(resp.preference))
- # --- Push notifications --------------------------------------------
- async def subscribe_push(
- self,
- endpoint: str,
- p256dh: str,
- auth: str,
- *,
- user_agent: str | None = None,
- ) -> bool:
- req = push_notifications_pb2.SubscribePushRequest(
- endpoint=endpoint, p256dh=p256dh, auth=auth
- )
- if user_agent is not None:
- req.user_agent = user_agent
- resp = await self._rpc(
- self._svc.push.subscribe(req, headers=self._headers())
- )
- return resp.subscribed
- async def unsubscribe_push(self, endpoint: str) -> bool:
- resp = await self._rpc(
- self._svc.push.unsubscribe(
- push_notifications_pb2.UnsubscribePushRequest(endpoint=endpoint),
- headers=self._headers(),
- )
- )
- return resp.unsubscribed
- # --- Assets ---------------------------------------------------------
- async def get_asset(
- self,
- room_id: str,
- asset_id: str,
- *,
- thumbnail: ImageTransformOptions | None = None,
- ) -> Asset | None:
- req = attachments_pb2.GetAssetRequest(room_id=room_id, asset_id=asset_id)
- thumb = _thumbnail_pb(thumbnail)
- if thumb is not None:
- req.thumbnail.CopyFrom(thumb)
- resp = await self._rpc(
- self._svc.assets.get_asset(req, headers=self._headers())
- )
- return Asset.parse(pb_to_dict(resp.asset))
- async def batch_get_assets(
- self,
- room_id: str,
- asset_ids: list[str],
- *,
- thumbnail: ImageTransformOptions | None = None,
- ) -> list[Asset]:
- req = attachments_pb2.BatchGetAssetsRequest(
- room_id=room_id, asset_ids=asset_ids
- )
- thumb = _thumbnail_pb(thumbnail)
- if thumb is not None:
- req.thumbnail.CopyFrom(thumb)
- resp = await self._rpc(
- self._svc.assets.batch_get_assets(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- return [
- a
- for a in (Asset.parse(row) for row in data.get("assets") or [])
- if a is not None
- ]
- # --- Asset uploads ------------------------------------------------
- async def create_upload(
- self,
- room_id: str,
- filename: str,
- size: int,
- sha256: str,
- *,
- content_type: str = "",
- ) -> AssetUpload:
- resp = await self._rpc(
- self._svc.asset_uploads.create_upload(
- asset_uploads_pb2.CreateUploadRequest(
- room_id=room_id,
- filename=filename,
- content_type=content_type,
- size=size,
- sha256=sha256,
- ),
- headers=self._headers(),
- )
- )
- upload = AssetUpload.parse(pb_to_dict(resp.upload))
- assert upload is not None
- return upload
- async def upload_chunk(
- self, upload_id: str, offset: int, content: bytes, chunk_sha256: str
- ) -> AssetUpload:
- resp = await self._rpc(
- self._svc.asset_uploads.upload_chunk(
- asset_uploads_pb2.UploadChunkRequest(
- upload_id=upload_id,
- offset=offset,
- content=content,
- chunk_sha256=chunk_sha256,
- ),
- headers=self._headers(),
- )
- )
- upload = AssetUpload.parse(pb_to_dict(resp.upload))
- assert upload is not None
- return upload
- async def get_upload(self, upload_id: str) -> AssetUpload:
- resp = await self._rpc(
- self._svc.asset_uploads.get_upload(
- asset_uploads_pb2.GetUploadRequest(upload_id=upload_id),
- headers=self._headers(),
- )
- )
- upload = AssetUpload.parse(pb_to_dict(resp.upload))
- assert upload is not None
- return upload
- async def complete_upload(
- self, upload_id: str
- ) -> tuple[AssetUpload, Asset | None]:
- resp = await self._rpc(
- self._svc.asset_uploads.complete_upload(
- asset_uploads_pb2.CompleteUploadRequest(upload_id=upload_id),
- headers=self._headers(),
- )
- )
- upload = AssetUpload.parse(pb_to_dict(resp.upload))
- assert upload is not None
- return upload, Asset.parse(pb_to_dict(resp.asset))
- async def cancel_upload(self, upload_id: str) -> AssetUpload:
- resp = await self._rpc(
- self._svc.asset_uploads.cancel_upload(
- asset_uploads_pb2.CancelUploadRequest(upload_id=upload_id),
- headers=self._headers(),
- )
- )
- upload = AssetUpload.parse(pb_to_dict(resp.upload))
- assert upload is not None
- return upload
- async def upload_attachment(
- self,
- room_id: str,
- file_path: str | Path,
- *,
- content_type: str = "",
- filename: str | None = None,
- ) -> Asset:
- """Upload a file as a room attachment and return the resulting Asset."""
- path = Path(file_path)
- data = path.read_bytes()
- size = len(data)
- sha = hashlib.sha256(data).hexdigest()
- upload = await self.create_upload(
- room_id, filename or path.name, size, sha, content_type=content_type
- )
- chunk_size = upload.max_chunk_size or 512 * 1024
- offset = upload.committed_offset
- while offset < size:
- end = min(offset + chunk_size, size)
- chunk = data[offset:end]
- chunk_sha = hashlib.sha256(chunk).hexdigest()
- upload = await self.upload_chunk(upload.upload_id, offset, chunk, chunk_sha)
- if upload.committed_offset <= offset:
- raise ChattoError(
- f"upload stalled at offset {offset} (server reported "
- f"committed_offset={upload.committed_offset})"
- )
- offset = upload.committed_offset
- upload, asset = await self.complete_upload(upload.upload_id)
- if asset is None:
- raise ChattoError("upload completed but server returned no asset")
- return asset
- # --- MyAccount external identities --------------------------------
- async def list_external_identities(
- self,
- ) -> tuple[list[ExternalIdentityProvider], list[LinkedExternalIdentity]]:
- resp = await self._rpc(
- self._svc.account.list_external_identities(
- external_identities_pb2.ListExternalIdentitiesRequest(),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- providers = [
- ExternalIdentityProvider.parse(p) for p in data.get("providers") or []
- ]
- linked = [
- LinkedExternalIdentity.parse(li)
- for li in data.get("linkedIdentities") or []
- ]
- return providers, linked
- async def start_external_identity_link(
- self,
- provider_id: str,
- *,
- redirect_path: str = "",
- current_password: str = "",
- ) -> str:
- resp = await self._rpc(
- self._svc.account.start_external_identity_link(
- external_identities_pb2.StartExternalIdentityLinkRequest(
- provider_id=provider_id,
- redirect_path=redirect_path,
- current_password=current_password,
- ),
- headers=self._headers(),
- )
- )
- return resp.start_url
- async def disconnect_external_identity(
- self, subject_hash: str, *, current_password: str = ""
- ) -> bool:
- resp = await self._rpc(
- self._svc.account.disconnect_external_identity(
- external_identities_pb2.DisconnectExternalIdentityRequest(
- subject_hash=subject_hash, current_password=current_password
- ),
- headers=self._headers(),
- )
- )
- return resp.disconnected
- # --- ExternalIdentityAuthService (public OAuth handoff) -----------
- async def get_pending_external_identity(self, token: str) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.external_auth.get_pending_external_identity(
- external_identity_auth_pb2.GetPendingExternalIdentityRequest(
- token=token
- ),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def create_external_identity_account(
- self, token: str, login: str
- ) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.external_auth.create_external_identity_account(
- external_identity_auth_pb2.CreateExternalIdentityAccountRequest(
- token=token, login=login
- ),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def confirm_external_identity_link(
- self, token: str
- ) -> LinkedExternalIdentity | None:
- resp = await self._rpc(
- self._svc.external_auth.confirm_external_identity_link(
- external_identity_auth_pb2.ConfirmExternalIdentityLinkRequest(
- token=token
- ),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp).get("linkedIdentity")
- return LinkedExternalIdentity.parse(data) if data else None
- async def cancel_external_identity_flow(self, token: str) -> bool:
- resp = await self._rpc(
- self._svc.external_auth.cancel_external_identity_flow(
- external_identity_auth_pb2.CancelExternalIdentityFlowRequest(
- token=token
- ),
- headers=self._headers(),
- )
- )
- return resp.cancelled
- # --- Voice calls ----------------------------------------------------
- async def list_active_calls(self) -> list[ActiveCall]:
- resp = await self._rpc(
- self._svc.voice_calls.list_active_calls(
- voice_calls_pb2.ListActiveCallsRequest(), headers=self._headers()
- )
- )
- data = pb_to_dict(resp)
- return [ActiveCall.parse(c) for c in data.get("calls") or []]
- async def get_active_call(self, room_id: str) -> ActiveCall | None:
- resp = await self._rpc(
- self._svc.voice_calls.get_active_call(
- voice_calls_pb2.GetActiveCallRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- raw = pb_to_dict(resp).get("call")
- return ActiveCall.parse(raw) if raw else None
- async def batch_get_active_calls(self, room_ids: list[str]) -> list[ActiveCall]:
- resp = await self._rpc(
- self._svc.voice_calls.batch_get_active_calls(
- voice_calls_pb2.BatchGetActiveCallsRequest(room_ids=room_ids),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [ActiveCall.parse(c) for c in data.get("calls") or []]
- async def join_call(self, room_id: str) -> bool:
- resp = await self._rpc(
- self._svc.voice_calls.join_call(
- voice_calls_pb2.JoinCallRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- return resp.joined
- async def leave_call(self, room_id: str) -> bool:
- resp = await self._rpc(
- self._svc.voice_calls.leave_call(
- voice_calls_pb2.LeaveCallRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- return resp.left
- async def get_call_token(self, room_id: str) -> str:
- resp = await self._rpc(
- self._svc.voice_calls.get_call_token(
- voice_calls_pb2.GetCallTokenRequest(room_id=room_id),
- headers=self._headers(),
- )
- )
- return resp.token
- # --- Admin: server --------------------------------------------------
- async def admin_get_server_config(self) -> tuple[ServerConfig, ServerProfile]:
- resp = await self._rpc(
- self._svc.admin_server.get_server_config(
- admin_server_pb2.GetServerConfigRequest(),
- headers=self._headers(),
- )
- )
- return (
- ServerConfig.parse(pb_to_dict(resp.config)),
- ServerProfile.parse(pb_to_dict(resp.public_profile)),
- )
- async def admin_update_server_config(
- self,
- *,
- server_name: str | None = None,
- description: str | None = None,
- motd: str | None = None,
- welcome_message: str | None = None,
- ) -> tuple[ServerConfig, ServerProfile]:
- req = admin_server_pb2.UpdateServerConfigRequest()
- if server_name is not None:
- req.server_name = server_name
- if description is not None:
- req.description = description
- if motd is not None:
- req.motd = motd
- if welcome_message is not None:
- req.welcome_message = welcome_message
- resp = await self._rpc(
- self._svc.admin_server.update_server_config(req, headers=self._headers())
- )
- return (
- ServerConfig.parse(pb_to_dict(resp.config)),
- ServerProfile.parse(pb_to_dict(resp.public_profile)),
- )
- async def admin_upload_server_logo(
- self,
- file_path: str | Path,
- *,
- content_type: str = "image/png",
- ) -> ServerProfile:
- p = Path(file_path)
- req = admin_server_pb2.UploadServerLogoRequest(
- image=common_pb2.ImageUpload(
- image=p.read_bytes(), filename=p.name, content_type=content_type
- )
- )
- resp = await self._rpc(
- self._svc.admin_server.upload_server_logo(req, headers=self._headers())
- )
- return ServerProfile.parse(pb_to_dict(resp.public_profile))
- async def admin_delete_server_logo(self) -> ServerProfile:
- resp = await self._rpc(
- self._svc.admin_server.delete_server_logo(
- admin_server_pb2.DeleteServerLogoRequest(),
- headers=self._headers(),
- )
- )
- return ServerProfile.parse(pb_to_dict(resp.public_profile))
- async def admin_upload_server_banner(
- self,
- file_path: str | Path,
- *,
- content_type: str = "image/png",
- ) -> ServerProfile:
- p = Path(file_path)
- req = admin_server_pb2.UploadServerBannerRequest(
- image=common_pb2.ImageUpload(
- image=p.read_bytes(), filename=p.name, content_type=content_type
- )
- )
- resp = await self._rpc(
- self._svc.admin_server.upload_server_banner(req, headers=self._headers())
- )
- return ServerProfile.parse(pb_to_dict(resp.public_profile))
- async def admin_delete_server_banner(self) -> ServerProfile:
- resp = await self._rpc(
- self._svc.admin_server.delete_server_banner(
- admin_server_pb2.DeleteServerBannerRequest(),
- headers=self._headers(),
- )
- )
- return ServerProfile.parse(pb_to_dict(resp.public_profile))
- async def admin_get_server_security_config(self) -> list[str]:
- resp = await self._rpc(
- self._svc.admin_server.get_server_security_config(
- admin_server_pb2.GetServerSecurityConfigRequest(),
- headers=self._headers(),
- )
- )
- return list(resp.blocked_usernames)
- async def admin_update_blocked_usernames(self, usernames: list[str]) -> list[str]:
- resp = await self._rpc(
- self._svc.admin_server.update_blocked_usernames(
- admin_server_pb2.UpdateBlockedUsernamesRequest(
- blocked_usernames=usernames
- ),
- headers=self._headers(),
- )
- )
- return list(resp.blocked_usernames)
- # --- Admin: room layout & sidebar links ---------------------------
- async def admin_list_room_groups(self) -> list[AdminRoomLayoutGroup]:
- resp = await self._rpc(
- self._svc.admin_room_layout.list_room_groups(
- room_layout_pb2.ListRoomGroupsRequest(),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- g
- for g in (
- AdminRoomLayoutGroup.parse(row) for row in data.get("groups") or []
- )
- if g is not None
- ]
- async def admin_create_room_group(
- self, name: str, description: str = ""
- ) -> AdminRoomLayoutGroup:
- resp = await self._rpc(
- self._svc.admin_room_layout.create_room_group(
- room_layout_pb2.CreateRoomGroupRequest(
- name=name, description=description
- ),
- headers=self._headers(),
- )
- )
- group = AdminRoomLayoutGroup.parse(pb_to_dict(resp.group))
- assert group is not None
- return group
- async def admin_update_room_group(
- self,
- group_id: str,
- *,
- name: str | None = None,
- description: str | None = None,
- ) -> AdminRoomLayoutGroup:
- req = room_layout_pb2.UpdateRoomGroupRequest(group_id=group_id)
- if name is not None:
- req.name = name
- if description is not None:
- req.description = description
- resp = await self._rpc(
- self._svc.admin_room_layout.update_room_group(
- req, headers=self._headers()
- )
- )
- group = AdminRoomLayoutGroup.parse(pb_to_dict(resp.group))
- assert group is not None
- return group
- async def admin_delete_room_group(self, group_id: str) -> bool:
- resp = await self._rpc(
- self._svc.admin_room_layout.delete_room_group(
- room_layout_pb2.DeleteRoomGroupRequest(group_id=group_id),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def admin_reorder_room_groups(
- self, ordered_group_ids: list[str]
- ) -> list[AdminRoomLayoutGroup]:
- resp = await self._rpc(
- self._svc.admin_room_layout.reorder_room_groups(
- room_layout_pb2.ReorderRoomGroupsRequest(
- ordered_group_ids=ordered_group_ids
- ),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- g
- for g in (
- AdminRoomLayoutGroup.parse(row) for row in data.get("groups") or []
- )
- if g is not None
- ]
- async def admin_move_room_to_group(self, room_id: str, group_id: str) -> Room:
- resp = await self._rpc(
- self._svc.admin_room_layout.move_room_to_group(
- room_layout_pb2.MoveRoomToGroupRequest(
- room_id=room_id, group_id=group_id
- ),
- headers=self._headers(),
- )
- )
- room = Room.parse(pb_to_dict(resp.room))
- assert room is not None
- return room
- async def admin_reorder_sidebar_items_in_group(
- self,
- group_id: str,
- items: list[tuple[AdminRoomLayoutItemKind, str]],
- ) -> AdminRoomLayoutGroup:
- req = room_layout_pb2.ReorderSidebarItemsInGroupRequest(group_id=group_id)
- for kind, item_id in items:
- item = req.items.add()
- item.kind = kind.value
- item.id = item_id
- resp = await self._rpc(
- self._svc.admin_room_layout.reorder_sidebar_items_in_group(
- req, headers=self._headers()
- )
- )
- group = AdminRoomLayoutGroup.parse(pb_to_dict(resp.group))
- assert group is not None
- return group
- async def admin_create_sidebar_link(
- self, group_id: str, label: str, url: str
- ) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_room_layout.create_sidebar_link(
- room_layout_pb2.CreateSidebarLinkRequest(
- group_id=group_id, label=label, url=url
- ),
- headers=self._headers(),
- )
- )
- sl = resp.sidebar_link
- return {"id": sl.id, "label": sl.label, "url": sl.url}
- async def admin_update_sidebar_link(
- self,
- link_id: str,
- *,
- label: str | None = None,
- url: str | None = None,
- ) -> dict[str, Any]:
- req = room_layout_pb2.UpdateSidebarLinkRequest(link_id=link_id)
- if label is not None:
- req.label = label
- if url is not None:
- req.url = url
- resp = await self._rpc(
- self._svc.admin_room_layout.update_sidebar_link(
- req, headers=self._headers()
- )
- )
- sl = resp.sidebar_link
- return {"id": sl.id, "label": sl.label, "url": sl.url}
- async def admin_delete_sidebar_link(self, link_id: str) -> bool:
- resp = await self._rpc(
- self._svc.admin_room_layout.delete_sidebar_link(
- room_layout_pb2.DeleteSidebarLinkRequest(link_id=link_id),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def admin_move_sidebar_link_to_group(
- self, link_id: str, group_id: str
- ) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_room_layout.move_sidebar_link_to_group(
- room_layout_pb2.MoveSidebarLinkToGroupRequest(
- link_id=link_id, group_id=group_id
- ),
- headers=self._headers(),
- )
- )
- sl = resp.sidebar_link
- return {"id": sl.id, "label": sl.label, "url": sl.url}
- # --- Admin: users --------------------------------------------------
- async def admin_list_members(
- self,
- *,
- search: str = "",
- limit: int | None = None,
- offset: int | None = None,
- ) -> tuple[list[AdminMember], list[Role], Page]:
- req = admin_members_pb2.ListMembersRequest(search=search)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.admin_users.list_members(req, headers=self._headers())
- )
- data = pb_to_dict(resp)
- members = [
- m
- for m in (AdminMember.parse(row) for row in data.get("members") or [])
- if m is not None
- ]
- roles = [
- r for r in (Role.parse(row) for row in data.get("roles") or []) if r is not None
- ]
- return members, roles, Page.parse(data.get("page"))
- async def admin_get_member(
- self,
- *,
- user_id: str | None = None,
- login: str | None = None,
- ) -> dict[str, Any]:
- if bool(user_id) == bool(login):
- raise ValueError("admin_get_member requires exactly one of user_id or login")
- req = admin_members_pb2.GetMemberRequest()
- if user_id:
- req.user_id = user_id
- else:
- assert login is not None
- req.login = login
- resp = await self._rpc(
- self._svc.admin_users.get_member(req, headers=self._headers())
- )
- return pb_to_dict(resp)
- async def admin_batch_get_members(self, user_ids: list[str]) -> list[AdminMember]:
- resp = await self._rpc(
- self._svc.admin_users.batch_get_members(
- admin_members_pb2.BatchGetMembersRequest(user_ids=user_ids),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- m
- for m in (AdminMember.parse(row) for row in data.get("members") or [])
- if m is not None
- ]
- async def admin_assign_role(
- self, user_id: str, role_name: str
- ) -> AdminMember | None:
- resp = await self._rpc(
- self._svc.admin_users.assign_role(
- admin_members_pb2.AssignRoleRequest(
- user_id=user_id, role_name=role_name
- ),
- headers=self._headers(),
- )
- )
- return AdminMember.parse(pb_to_dict(resp.member))
- async def admin_revoke_role(
- self, user_id: str, role_name: str
- ) -> AdminMember | None:
- resp = await self._rpc(
- self._svc.admin_users.revoke_role(
- admin_members_pb2.RevokeRoleRequest(
- user_id=user_id, role_name=role_name
- ),
- headers=self._headers(),
- )
- )
- return AdminMember.parse(pb_to_dict(resp.member))
- async def admin_update_user(
- self,
- user_id: str,
- *,
- display_name: str | None = None,
- login: str | None = None,
- ) -> tuple[User | None, AdminMember | None]:
- req = admin_members_pb2.UpdateUserRequest(user_id=user_id)
- if display_name is not None:
- req.display_name = display_name
- if login is not None:
- req.login = login
- resp = await self._rpc(
- self._svc.admin_users.update_user(req, headers=self._headers())
- )
- return (
- User.parse(pb_to_dict(resp.user)),
- AdminMember.parse(pb_to_dict(resp.member)),
- )
- async def admin_update_user_password(
- self, user_id: str, password: str
- ) -> AdminMember | None:
- resp = await self._rpc(
- self._svc.admin_users.update_user_password(
- admin_members_pb2.UpdateUserPasswordRequest(
- user_id=user_id, password=password
- ),
- headers=self._headers(),
- )
- )
- return AdminMember.parse(pb_to_dict(resp.member))
- async def admin_clear_username_cooldown(self, user_id: str) -> bool:
- resp = await self._rpc(
- self._svc.admin_users.clear_username_cooldown(
- admin_members_pb2.ClearUsernameCooldownRequest(user_id=user_id),
- headers=self._headers(),
- )
- )
- return resp.cleared
- async def admin_delete_user(
- self, user_id: str, *, current_password: str = ""
- ) -> bool:
- resp = await self._rpc(
- self._svc.admin_users.delete_user(
- admin_members_pb2.DeleteUserRequest(
- user_id=user_id, current_password=current_password
- ),
- headers=self._headers(),
- )
- )
- return resp.deleted
- # --- Admin: roles --------------------------------------------------
- async def admin_list_roles(self) -> list[AdminRole]:
- resp = await self._rpc(
- self._svc.admin_roles.list_roles(
- admin_roles_pb2.ListRolesRequest(), headers=self._headers()
- )
- )
- data = pb_to_dict(resp)
- return [
- r
- for r in (AdminRole.parse(row) for row in data.get("roles") or [])
- if r is not None
- ]
- async def admin_get_role(self, name: str) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_roles.get_role(
- admin_roles_pb2.GetRoleRequest(name=name), headers=self._headers()
- )
- )
- return pb_to_dict(resp)
- async def admin_create_role(
- self,
- name: str,
- *,
- display_name: str = "",
- description: str = "",
- pingable: bool = False,
- ) -> AdminRole | None:
- resp = await self._rpc(
- self._svc.admin_roles.create_role(
- admin_roles_pb2.CreateRoleRequest(
- name=name,
- display_name=display_name,
- description=description,
- pingable=pingable,
- ),
- headers=self._headers(),
- )
- )
- return AdminRole.parse(pb_to_dict(resp.role))
- async def admin_update_role(
- self,
- name: str,
- *,
- display_name: str | None = None,
- description: str | None = None,
- pingable: bool | None = None,
- ) -> AdminRole | None:
- req = admin_roles_pb2.UpdateRoleRequest(name=name)
- if display_name is not None:
- req.display_name = display_name
- if description is not None:
- req.description = description
- if pingable is not None:
- req.pingable = pingable
- resp = await self._rpc(
- self._svc.admin_roles.update_role(req, headers=self._headers())
- )
- return AdminRole.parse(pb_to_dict(resp.role))
- async def admin_delete_role(self, name: str) -> bool:
- resp = await self._rpc(
- self._svc.admin_roles.delete_role(
- admin_roles_pb2.DeleteRoleRequest(name=name),
- headers=self._headers(),
- )
- )
- return resp.deleted
- async def admin_reorder_roles(self, role_names: list[str]) -> list[AdminRole]:
- resp = await self._rpc(
- self._svc.admin_roles.reorder_roles(
- admin_roles_pb2.ReorderRolesRequest(role_names=role_names),
- headers=self._headers(),
- )
- )
- data = pb_to_dict(resp)
- return [
- r
- for r in (AdminRole.parse(row) for row in data.get("roles") or [])
- if r is not None
- ]
- # --- Admin: event log / diagnostics / permissions ----------------
- async def admin_list_events(
- self,
- *,
- event_types: list[str] | None = None,
- limit: int | None = None,
- offset: int | None = None,
- ) -> dict[str, Any]:
- req = event_log_pb2.ListEventsRequest()
- if event_types:
- req.event_types.extend(event_types)
- page = _page_pb(limit, offset)
- if page is not None:
- req.page.CopyFrom(page)
- resp = await self._rpc(
- self._svc.admin_event_log.list_events(req, headers=self._headers())
- )
- return pb_to_dict(resp)
- async def admin_list_event_types(self) -> list[str]:
- resp = await self._rpc(
- self._svc.admin_event_log.list_event_types(
- event_log_pb2.ListEventTypesRequest(),
- headers=self._headers(),
- )
- )
- return list(resp.event_types)
- async def admin_get_event(self, event_id: str) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_event_log.get_event(
- event_log_pb2.GetEventRequest(event_id=event_id),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def admin_get_system_info(self) -> dict[str, Any]:
- from chattolib._pb.chatto.admin.v1 import diagnostics_pb2
- resp = await self._rpc(
- self._svc.admin_diagnostics.get_system_info(
- diagnostics_pb2.GetSystemInfoRequest(),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def admin_get_role_permission_matrix(self) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_permissions.get_role_permission_matrix(
- admin_permissions_pb2.GetRolePermissionMatrixRequest(),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def admin_list_role_permission_decisions(
- self, role_name: str
- ) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_permissions.list_role_permission_decisions(
- admin_permissions_pb2.ListRolePermissionDecisionsRequest(
- role_name=role_name
- ),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def admin_get_user_permission_matrix(self, user_id: str) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_permissions.get_user_permission_matrix(
- admin_permissions_pb2.GetUserPermissionMatrixRequest(user_id=user_id),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
- async def admin_list_user_permission_decisions(
- self, user_id: str
- ) -> dict[str, Any]:
- resp = await self._rpc(
- self._svc.admin_permissions.list_user_permission_decisions(
- admin_permissions_pb2.ListUserPermissionDecisionsRequest(user_id=user_id),
- headers=self._headers(),
- )
- )
- return pb_to_dict(resp)
|