portal.py 96 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353
  1. # mautrix-signal - A Matrix-Signal puppeting bridge
  2. # Copyright (C) 2022 Tulir Asokan
  3. #
  4. # This program is free software: you can redistribute it and/or modify
  5. # it under the terms of the GNU Affero General Public License as published by
  6. # the Free Software Foundation, either version 3 of the License, or
  7. # (at your option) any later version.
  8. #
  9. # This program is distributed in the hope that it will be useful,
  10. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. # GNU Affero General Public License for more details.
  13. #
  14. # You should have received a copy of the GNU Affero General Public License
  15. # along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. from __future__ import annotations
  17. from typing import TYPE_CHECKING, Any, AsyncGenerator, Awaitable, Callable, Union, cast
  18. from collections import deque
  19. from uuid import UUID, uuid4
  20. import asyncio
  21. import hashlib
  22. import mimetypes
  23. import os
  24. import os.path
  25. import pathlib
  26. import time
  27. from mausignald.errors import (
  28. AttachmentTooLargeError,
  29. NotConnected,
  30. ProfileUnavailableError,
  31. RPCError,
  32. )
  33. from mausignald.types import (
  34. AccessControlMode,
  35. Address,
  36. AnnouncementsMode,
  37. Attachment,
  38. Group,
  39. GroupAccessControl,
  40. GroupChange,
  41. GroupID,
  42. GroupMember,
  43. GroupMemberRole,
  44. GroupV2,
  45. GroupV2ID,
  46. LinkPreview,
  47. Mention,
  48. MessageData,
  49. Profile,
  50. Quote,
  51. QuotedAttachment,
  52. Reaction,
  53. SharedContact,
  54. Sticker,
  55. )
  56. from mautrix.appservice import AppService, IntentAPI
  57. from mautrix.bridge import BasePortal, RejectMatrixInvite, async_getter_lock
  58. from mautrix.errors import IntentError, MatrixError, MBadState, MForbidden
  59. from mautrix.types import (
  60. AudioInfo,
  61. BeeperMessageStatusEventContent,
  62. ContentURI,
  63. EncryptedEvent,
  64. EncryptedFile,
  65. EventID,
  66. EventType,
  67. FileInfo,
  68. ImageInfo,
  69. JoinRule,
  70. MediaMessageEventContent,
  71. Membership,
  72. MessageEvent,
  73. MessageEventContent,
  74. MessageStatus,
  75. MessageStatusReason,
  76. MessageType,
  77. PowerLevelStateEventContent,
  78. RelatesTo,
  79. RelationType,
  80. RoomID,
  81. TextMessageEventContent,
  82. UserID,
  83. VideoInfo,
  84. )
  85. from mautrix.util import ffmpeg, variation_selector
  86. from mautrix.util.format_duration import format_duration
  87. from mautrix.util.message_send_checkpoint import MessageSendCheckpointStatus
  88. from . import matrix as m, puppet as p, signal as s, user as u
  89. from .config import Config
  90. from .db import (
  91. DisappearingMessage,
  92. Message as DBMessage,
  93. Portal as DBPortal,
  94. Reaction as DBReaction,
  95. )
  96. from .formatter import matrix_to_signal, signal_to_matrix
  97. from .util import id_to_str
  98. if TYPE_CHECKING:
  99. from .__main__ import SignalBridge
  100. try:
  101. from mautrix.crypto.attachments import decrypt_attachment, encrypt_attachment
  102. except ImportError:
  103. encrypt_attachment = decrypt_attachment = None
  104. try:
  105. from signalstickers_client import StickersClient
  106. from signalstickers_client.models import StickerPack
  107. except ImportError:
  108. StickersClient = StickerPack = None
  109. try:
  110. from mautrix.util import magic
  111. except ImportError:
  112. magic = None
  113. StateBridge = EventType.find("m.bridge", EventType.Class.STATE)
  114. StateHalfShotBridge = EventType.find("uk.half-shot.bridge", EventType.Class.STATE)
  115. ChatInfo = Union[Group, GroupV2, GroupV2ID, Profile, Address]
  116. MAX_MATRIX_MESSAGE_SIZE = 30000
  117. BEEPER_LINK_PREVIEWS_KEY = "com.beeper.linkpreviews"
  118. BEEPER_IMAGE_ENCRYPTION_KEY = "beeper:image:encryption"
  119. class UnknownReactionTarget(Exception):
  120. pass
  121. class Portal(DBPortal, BasePortal):
  122. by_mxid: dict[RoomID, Portal] = {}
  123. by_chat_id: dict[tuple[str, str], Portal] = {}
  124. _sticker_meta_cache: dict[str, StickerPack] = {}
  125. disappearing_msg_class = DisappearingMessage
  126. config: Config
  127. matrix: m.MatrixHandler
  128. signal: s.SignalHandler
  129. az: AppService
  130. private_chat_portal_meta: bool
  131. expiration_time: int | None
  132. _main_intent: IntentAPI | None
  133. _create_room_lock: asyncio.Lock
  134. _msgts_dedup: deque[tuple[Address, int]]
  135. _reaction_dedup: deque[tuple[Address, int, str, Address, bool]]
  136. _reaction_lock: asyncio.Lock
  137. _pending_members: set[UUID] | None
  138. _expiration_lock: asyncio.Lock
  139. def __init__(
  140. self,
  141. chat_id: GroupID | Address,
  142. receiver: str,
  143. mxid: RoomID | None = None,
  144. name: str | None = None,
  145. topic: str | None = None,
  146. avatar_hash: str | None = None,
  147. avatar_url: ContentURI | None = None,
  148. name_set: bool = False,
  149. avatar_set: bool = False,
  150. revision: int = 0,
  151. encrypted: bool = False,
  152. relay_user_id: UserID | None = None,
  153. expiration_time: int | None = None,
  154. ) -> None:
  155. super().__init__(
  156. chat_id=chat_id,
  157. receiver=receiver,
  158. mxid=mxid,
  159. name=name,
  160. topic=topic,
  161. avatar_hash=avatar_hash,
  162. avatar_url=avatar_url,
  163. name_set=name_set,
  164. avatar_set=avatar_set,
  165. revision=revision,
  166. encrypted=encrypted,
  167. relay_user_id=relay_user_id,
  168. expiration_time=expiration_time,
  169. )
  170. BasePortal.__init__(self)
  171. self._create_room_lock = asyncio.Lock()
  172. self.log = self.log.getChild(self.chat_id_str)
  173. self._main_intent = None
  174. self._msgts_dedup = deque(maxlen=100)
  175. self._reaction_dedup = deque(maxlen=100)
  176. self._last_participant_update = set()
  177. self._reaction_lock = asyncio.Lock()
  178. self._pending_members = None
  179. self._relay_user = None
  180. self._expiration_lock = asyncio.Lock()
  181. @property
  182. def main_intent(self) -> IntentAPI:
  183. if not self._main_intent:
  184. raise ValueError("Portal must be postinit()ed before main_intent can be used")
  185. return self._main_intent
  186. @property
  187. def is_direct(self) -> bool:
  188. return isinstance(self.chat_id, Address)
  189. @property
  190. def disappearing_enabled(self) -> bool:
  191. return self.is_direct or self.config["signal.enable_disappearing_messages_in_groups"]
  192. def handle_uuid_receive(self, uuid: UUID) -> None:
  193. if not self.is_direct or self.chat_id.uuid:
  194. raise ValueError(
  195. "handle_uuid_receive can only be used for private chat portals with a phone "
  196. "number chat_id"
  197. )
  198. del self.by_chat_id[(self.chat_id_str, self.receiver)]
  199. self.chat_id = Address(uuid=uuid)
  200. self.by_chat_id[(self.chat_id_str, self.receiver)] = self
  201. @classmethod
  202. def init_cls(cls, bridge: "SignalBridge") -> None:
  203. cls.config = bridge.config
  204. cls.matrix = bridge.matrix
  205. cls.signal = bridge.signal
  206. cls.az = bridge.az
  207. cls.loop = bridge.loop
  208. BasePortal.bridge = bridge
  209. cls.private_chat_portal_meta = cls.config["bridge.private_chat_portal_meta"]
  210. # region Misc
  211. async def _send_delivery_receipt(self, event_id: EventID) -> None:
  212. if event_id and self.config["bridge.delivery_receipts"]:
  213. try:
  214. await self.az.intent.mark_read(self.mxid, event_id)
  215. except Exception:
  216. self.log.exception("Failed to send delivery receipt for %s", event_id)
  217. async def _upsert_reaction(
  218. self,
  219. existing: DBReaction,
  220. intent: IntentAPI,
  221. mxid: EventID,
  222. sender: p.Puppet | u.User,
  223. message: DBMessage,
  224. emoji: str,
  225. ) -> None:
  226. if existing:
  227. self.log.debug(
  228. f"_upsert_reaction redacting {existing.mxid} and inserting {mxid}"
  229. f" (message: {message.mxid})"
  230. )
  231. try:
  232. await intent.redact(existing.mx_room, existing.mxid)
  233. except MForbidden:
  234. self.log.debug("Unexpected MForbidden redacting reaction", exc_info=True)
  235. await existing.edit(emoji=emoji, mxid=mxid, mx_room=message.mx_room)
  236. else:
  237. self.log.debug(f"_upsert_reaction inserting {mxid} (message: {message.mxid})")
  238. await DBReaction(
  239. mxid=mxid,
  240. mx_room=message.mx_room,
  241. emoji=emoji,
  242. signal_chat_id=self.chat_id,
  243. signal_receiver=self.receiver,
  244. msg_author=message.sender,
  245. msg_timestamp=message.timestamp,
  246. author=sender.address,
  247. ).insert()
  248. # endregion
  249. # region Matrix event handling
  250. @staticmethod
  251. async def _make_attachment(message: MediaMessageEventContent, path: str) -> Attachment:
  252. outgoing_filename = path
  253. if message.msgtype == MessageType.AUDIO:
  254. outgoing_filename = await ffmpeg.convert_path(
  255. path, ".m4a", output_args=("-c:a", "aac"), remove_input=True
  256. )
  257. message.info.mimetype = "audio/mp4"
  258. attachment = Attachment(
  259. custom_filename=message.body,
  260. content_type=message.info.mimetype,
  261. outgoing_filename=str(outgoing_filename),
  262. )
  263. info = message.info
  264. attachment.width = info.get("w", info.get("width", 0))
  265. attachment.height = info.get("h", info.get("height", 0))
  266. attachment.voice_note = message.msgtype == MessageType.AUDIO
  267. return attachment
  268. def _write_outgoing_file(self, data: bytes) -> str:
  269. dir = pathlib.Path(self.config["signal.outgoing_attachment_dir"])
  270. path = dir.joinpath(f"mautrix-signal-{str(uuid4())}")
  271. try:
  272. with open(path, "wb") as file:
  273. file.write(data)
  274. except FileNotFoundError:
  275. dir.mkdir(mode=0o755, parents=True, exist_ok=True)
  276. with open(path, "wb") as file:
  277. file.write(data)
  278. return str(path)
  279. async def _download_matrix_media(self, message: MediaMessageEventContent) -> str:
  280. # Signal limits files to 100 MB
  281. if message.info and message.info.size and message.info.size > 100 * 10**6:
  282. raise AttachmentTooLargeError({"filename": message.body})
  283. if message.file:
  284. data = await self.main_intent.download_media(message.file.url)
  285. data = decrypt_attachment(
  286. data, message.file.key.key, message.file.hashes.get("sha256"), message.file.iv
  287. )
  288. else:
  289. data = await self.main_intent.download_media(message.url)
  290. return self._write_outgoing_file(data)
  291. async def handle_matrix_message(
  292. self, sender: u.User, message: MessageEventContent, event_id: EventID
  293. ) -> None:
  294. try:
  295. await self._handle_matrix_message(sender, message, event_id)
  296. except Exception as e:
  297. self.log.exception(f"Failed to handle Matrix message {event_id}")
  298. status = (
  299. MessageSendCheckpointStatus.UNSUPPORTED
  300. if isinstance(e, AttachmentTooLargeError)
  301. else MessageSendCheckpointStatus.PERM_FAILURE
  302. )
  303. sender.send_remote_checkpoint(
  304. status, event_id, self.mxid, EventType.ROOM_MESSAGE, message.msgtype, error=e
  305. )
  306. await sender.handle_auth_failure(e)
  307. await self._send_error_notice("message", e)
  308. asyncio.create_task(self._send_message_status(event_id, e))
  309. async def _send_error_notice(self, type_name: str, err: Exception) -> None:
  310. if not self.config["bridge.delivery_error_reports"]:
  311. return
  312. message = f"{type(err).__name__}: {err}"
  313. if isinstance(err, NotConnected):
  314. message = "There was an error connecting to signald."
  315. elif isinstance(err, UnknownReactionTarget):
  316. message = "Could not find message to react to on Signal."
  317. await self._send_message(
  318. self.main_intent,
  319. TextMessageEventContent(
  320. msgtype=MessageType.NOTICE,
  321. body=f"\u26a0 Your {type_name} was not bridged: {message}",
  322. ),
  323. )
  324. async def _send_message_status(self, event_id: EventID, err: Exception | None) -> None:
  325. if not self.config["bridge.message_status_events"]:
  326. return
  327. intent = self.az.intent if self.encrypted else self.main_intent
  328. status = BeeperMessageStatusEventContent(
  329. network=self.bridge_info_state_key,
  330. relates_to=RelatesTo(
  331. rel_type=RelationType.REFERENCE,
  332. event_id=event_id,
  333. ),
  334. )
  335. if err:
  336. status.reason = MessageStatusReason.GENERIC_ERROR
  337. status.error = str(err)
  338. if isinstance(err, AttachmentTooLargeError):
  339. status.reason = MessageStatusReason.UNSUPPORTED
  340. status.status = MessageStatus.FAIL
  341. status.message = "too large file (maximum is 100MB)"
  342. elif isinstance(err, UnknownReactionTarget):
  343. status.status = MessageStatus.FAIL
  344. else:
  345. status.status = MessageStatus.RETRIABLE
  346. else:
  347. status.status = MessageStatus.SUCCESS
  348. status.fill_legacy_booleans()
  349. await intent.send_message_event(
  350. room_id=self.mxid,
  351. event_type=EventType.BEEPER_MESSAGE_STATUS,
  352. content=status,
  353. )
  354. async def _beeper_link_preview_to_signal(
  355. self, beeper_link_preview: dict[str, Any]
  356. ) -> LinkPreview | None:
  357. link_preview = LinkPreview(
  358. url=beeper_link_preview["matched_url"],
  359. title=beeper_link_preview.get("og:title", ""),
  360. description=beeper_link_preview.get("og:description", ""),
  361. )
  362. if BEEPER_IMAGE_ENCRYPTION_KEY in beeper_link_preview or "og:image" in beeper_link_preview:
  363. if BEEPER_IMAGE_ENCRYPTION_KEY in beeper_link_preview:
  364. file = EncryptedFile.deserialize(beeper_link_preview[BEEPER_IMAGE_ENCRYPTION_KEY])
  365. data = await self.main_intent.download_media(file.url)
  366. data = decrypt_attachment(data, file.key.key, file.hashes.get("sha256"), file.iv)
  367. else:
  368. data = await self.main_intent.download_media(beeper_link_preview["og:image"])
  369. attachment_path = self._write_outgoing_file(data)
  370. link_preview.attachment = Attachment(
  371. content_type=beeper_link_preview.get("og:image:type"),
  372. outgoing_filename=attachment_path,
  373. width=beeper_link_preview.get("og:image:width", 0),
  374. height=beeper_link_preview.get("og:image:height", 0),
  375. size=beeper_link_preview.get("matrix:image:size", 0),
  376. )
  377. return link_preview
  378. async def _handle_matrix_message(
  379. self, sender: u.User, message: MessageEventContent, event_id: EventID
  380. ) -> None:
  381. orig_sender = sender
  382. sender, is_relay = await self.get_relay_sender(sender, f"message {event_id}")
  383. if not sender:
  384. orig_sender.send_remote_checkpoint(
  385. status=MessageSendCheckpointStatus.PERM_FAILURE,
  386. event_id=event_id,
  387. room_id=self.mxid,
  388. event_type=EventType.ROOM_MESSAGE,
  389. message_type=message.msgtype,
  390. error="user is not logged in",
  391. )
  392. return
  393. elif is_relay:
  394. await self.apply_relay_message_format(orig_sender, message)
  395. request_id = int(time.time() * 1000)
  396. self._msgts_dedup.appendleft((sender.address, request_id))
  397. quote = None
  398. if message.get_reply_to():
  399. reply = await DBMessage.get_by_mxid(message.get_reply_to(), self.mxid)
  400. # TODO include actual text? either store in db or fetch event from homeserver
  401. if reply is not None:
  402. quote = Quote(id=reply.timestamp, author=reply.sender, text="")
  403. # TODO only send this when it's actually a reply to an attachment?
  404. # Neither Signal Android nor iOS seem to care though, so this works too
  405. quote.attachments = [QuotedAttachment("", "")]
  406. attachments: list[Attachment] | None = None
  407. attachment_path: str | None = None
  408. mentions: list[Mention] | None = None
  409. link_previews: list[LinkPreview] | None = None
  410. if message.msgtype.is_text:
  411. text, mentions = await matrix_to_signal(message)
  412. message_previews = message.get(BEEPER_LINK_PREVIEWS_KEY, [])
  413. potential_link_previews = await asyncio.gather(
  414. *(self._beeper_link_preview_to_signal(m) for m in message_previews)
  415. )
  416. link_previews = [p for p in potential_link_previews if p is not None]
  417. elif message.msgtype.is_media:
  418. attachment_path = await self._download_matrix_media(message)
  419. attachment = await self._make_attachment(message, attachment_path)
  420. attachments = [attachment]
  421. text = message.body if is_relay else None
  422. self.log.trace("Formed outgoing attachment %s", attachment)
  423. else:
  424. self.log.debug(f"Unknown msgtype {message.msgtype} in Matrix message {event_id}")
  425. return
  426. self.log.debug(f"Sending Matrix message {event_id} to Signal with timestamp {request_id}")
  427. retry_count = await self._signal_send_with_retries(
  428. sender,
  429. event_id,
  430. message_type=message.msgtype,
  431. send_fn=lambda *args, **kwargs: self.signal.send(**kwargs),
  432. event_type=EventType.ROOM_MESSAGE,
  433. username=sender.username,
  434. recipient=self.chat_id,
  435. body=text,
  436. mentions=mentions,
  437. previews=link_previews,
  438. quote=quote,
  439. attachments=attachments,
  440. timestamp=request_id,
  441. )
  442. msg = DBMessage(
  443. mxid=event_id,
  444. mx_room=self.mxid,
  445. sender=sender.address,
  446. timestamp=request_id,
  447. signal_chat_id=self.chat_id,
  448. signal_receiver=self.receiver,
  449. )
  450. await msg.insert()
  451. self.log.debug(f"Handled Matrix message {event_id} -> {request_id}")
  452. if attachment_path and self.config["signal.remove_file_after_handling"]:
  453. try:
  454. os.remove(attachment_path)
  455. except FileNotFoundError:
  456. pass
  457. # Handle disappearing messages
  458. if self.expiration_time and self.disappearing_enabled:
  459. dm = DisappearingMessage(self.mxid, event_id, self.expiration_time)
  460. dm.start_timer()
  461. await dm.insert()
  462. asyncio.create_task(self._disappear_event(dm))
  463. sender.send_remote_checkpoint(
  464. MessageSendCheckpointStatus.SUCCESS,
  465. event_id,
  466. self.mxid,
  467. EventType.ROOM_MESSAGE,
  468. message.msgtype,
  469. retry_num=retry_count,
  470. )
  471. await self._send_delivery_receipt(event_id)
  472. asyncio.create_task(self._send_message_status(event_id, err=None))
  473. async def _signal_send_with_retries(
  474. self,
  475. sender: u.User,
  476. event_id: EventID,
  477. send_fn: Callable,
  478. event_type: EventType,
  479. message_type: MessageType | None = None,
  480. **send_args,
  481. ) -> int:
  482. retry_count = 4
  483. last_error_type = NotConnected
  484. for retry_num in range(retry_count):
  485. try:
  486. req_id = uuid4()
  487. self.log.info(
  488. f"Send attempt {retry_num}. Attempting to send {event_id} with {req_id}"
  489. )
  490. await send_fn(sender, event_id, req_id=req_id, **send_args)
  491. return retry_num
  492. except (NotConnected, UnknownReactionTarget) as e:
  493. if retry_num >= retry_count - 1:
  494. break
  495. last_error_type = type(e)
  496. # Only handle NotConnected and UnknownReactionTarget exceptions so that other
  497. # exceptions actually continue to error.
  498. sleep_seconds = retry_num * 2 + 1
  499. msg = (
  500. f"Not connected to signald. Going to sleep for {sleep_seconds}s. Error: {e}"
  501. if isinstance(e, NotConnected)
  502. else f"UnknownReactionTarget: Going to sleep for {sleep_seconds}s. Error: {e}"
  503. )
  504. self.log.exception(msg)
  505. sender.send_remote_checkpoint(
  506. MessageSendCheckpointStatus.WILL_RETRY,
  507. event_id,
  508. self.mxid,
  509. event_type,
  510. message_type=message_type,
  511. error=msg,
  512. retry_num=retry_num,
  513. )
  514. await asyncio.sleep(sleep_seconds)
  515. except Exception as e:
  516. await sender.handle_auth_failure(e)
  517. raise
  518. event_type_name = {
  519. EventType.ROOM_MESSAGE: "message",
  520. EventType.REACTION: "reaction",
  521. }.get(event_type, str(event_type))
  522. raise last_error_type(f"Failed to send {event_type_name} after {retry_count} retries.")
  523. async def handle_matrix_reaction(
  524. self, sender: u.User, event_id: EventID, reacting_to: EventID, emoji: str
  525. ) -> None:
  526. if not await sender.is_logged_in():
  527. self.log.trace(f"Ignoring reaction by non-logged-in user {sender.mxid}")
  528. return
  529. # Signal doesn't seem to use variation selectors at all
  530. emoji = variation_selector.remove(emoji)
  531. try:
  532. retry_count = await self._signal_send_with_retries(
  533. sender,
  534. event_id,
  535. send_fn=self._handle_matrix_reaction,
  536. event_type=EventType.REACTION,
  537. reacting_to=reacting_to,
  538. emoji=emoji,
  539. )
  540. except Exception as e:
  541. self.log.exception(f"Failed to handle Matrix reaction {event_id} to {reacting_to}")
  542. sender.send_remote_checkpoint(
  543. MessageSendCheckpointStatus.PERM_FAILURE,
  544. event_id,
  545. self.mxid,
  546. EventType.REACTION,
  547. error=e,
  548. )
  549. await self._send_error_notice("reaction", e)
  550. await sender.handle_auth_failure(e)
  551. asyncio.create_task(self._send_message_status(event_id, e))
  552. else:
  553. sender.send_remote_checkpoint(
  554. MessageSendCheckpointStatus.SUCCESS,
  555. event_id,
  556. self.mxid,
  557. EventType.REACTION,
  558. retry_num=retry_count,
  559. )
  560. await self._send_delivery_receipt(event_id)
  561. asyncio.create_task(self._send_message_status(event_id, err=None))
  562. async def _handle_matrix_reaction(
  563. self,
  564. sender: u.User,
  565. event_id: EventID,
  566. reacting_to: EventID,
  567. emoji: str,
  568. req_id: UUID | None = None,
  569. ) -> None:
  570. message = await DBMessage.get_by_mxid(reacting_to, self.mxid)
  571. if not message:
  572. self.log.debug(f"Ignoring reaction to unknown event {reacting_to}")
  573. raise UnknownReactionTarget(f"Ignoring reaction to unknown event {reacting_to}")
  574. async with self._reaction_lock:
  575. existing = await DBReaction.get_by_signal_id(
  576. self.chat_id, self.receiver, message.sender, message.timestamp, sender.address
  577. )
  578. if existing and existing.emoji == emoji:
  579. return
  580. dedup_id = (message.sender, message.timestamp, emoji, sender.address, False)
  581. self._reaction_dedup.appendleft(dedup_id)
  582. reaction = Reaction(
  583. emoji=emoji,
  584. remove=False,
  585. target_author=message.sender,
  586. target_sent_timestamp=message.timestamp,
  587. )
  588. self.log.trace(f"{sender.mxid} reacted to {message.timestamp} with {emoji}")
  589. await self.signal.react(
  590. sender.username, recipient=self.chat_id, reaction=reaction, req_id=req_id
  591. )
  592. await self._upsert_reaction(
  593. existing, self.main_intent, event_id, sender, message, emoji
  594. )
  595. async def handle_matrix_redaction(
  596. self, sender: u.User, event_id: EventID, redaction_event_id: EventID
  597. ) -> None:
  598. if not await sender.is_logged_in():
  599. return
  600. message = await DBMessage.get_by_mxid(event_id, self.mxid)
  601. if message:
  602. try:
  603. await message.delete()
  604. await self.signal.remote_delete(
  605. sender.username, recipient=self.chat_id, timestamp=message.timestamp
  606. )
  607. except Exception as e:
  608. self.log.exception(
  609. f"Failed to handle Matrix redaction {redaction_event_id} of "
  610. f"message {event_id} ({message.timestamp})"
  611. )
  612. sender.send_remote_checkpoint(
  613. MessageSendCheckpointStatus.PERM_FAILURE,
  614. redaction_event_id,
  615. self.mxid,
  616. EventType.ROOM_REDACTION,
  617. error=e,
  618. )
  619. await sender.handle_auth_failure(e)
  620. asyncio.create_task(self._send_error_notice("message deletion", e))
  621. asyncio.create_task(self._send_message_status(event_id, e))
  622. else:
  623. self.log.trace(f"Removed {message} after Matrix redaction")
  624. sender.send_remote_checkpoint(
  625. MessageSendCheckpointStatus.SUCCESS,
  626. redaction_event_id,
  627. self.mxid,
  628. EventType.ROOM_REDACTION,
  629. )
  630. await self._send_delivery_receipt(redaction_event_id)
  631. asyncio.create_task(self._send_message_status(redaction_event_id, err=None))
  632. return
  633. reaction = await DBReaction.get_by_mxid(event_id, self.mxid)
  634. if reaction:
  635. try:
  636. await reaction.delete()
  637. remove_reaction = Reaction(
  638. emoji=reaction.emoji,
  639. remove=True,
  640. target_author=reaction.msg_author,
  641. target_sent_timestamp=reaction.msg_timestamp,
  642. )
  643. await self.signal.react(
  644. username=sender.username, recipient=self.chat_id, reaction=remove_reaction
  645. )
  646. except Exception as e:
  647. self.log.exception(
  648. f"Failed to handle Matrix redaction {redaction_event_id} of "
  649. f"reaction {event_id} to {reaction.msg_timestamp}"
  650. )
  651. sender.send_remote_checkpoint(
  652. MessageSendCheckpointStatus.PERM_FAILURE,
  653. redaction_event_id,
  654. self.mxid,
  655. EventType.ROOM_REDACTION,
  656. error=e,
  657. )
  658. await sender.handle_auth_failure(e)
  659. asyncio.create_task(self._send_error_notice("reaction deletion", e))
  660. asyncio.create_task(self._send_message_status(event_id, e))
  661. else:
  662. self.log.trace(f"Removed {reaction} after Matrix redaction")
  663. sender.send_remote_checkpoint(
  664. MessageSendCheckpointStatus.SUCCESS,
  665. redaction_event_id,
  666. self.mxid,
  667. EventType.ROOM_REDACTION,
  668. )
  669. await self._send_delivery_receipt(redaction_event_id)
  670. asyncio.create_task(self._send_message_status(redaction_event_id, err=None))
  671. return
  672. sender.send_remote_checkpoint(
  673. MessageSendCheckpointStatus.PERM_FAILURE,
  674. redaction_event_id,
  675. self.mxid,
  676. EventType.ROOM_REDACTION,
  677. error="No message or reaction found for redaction",
  678. )
  679. status_err = UnknownReactionTarget("No message or reaction found for redaction")
  680. asyncio.create_task(self._send_message_status(redaction_event_id, err=status_err))
  681. async def handle_matrix_join(self, user: u.User) -> None:
  682. if self.is_direct or not await user.is_logged_in():
  683. return
  684. if self._pending_members is None:
  685. self.log.debug(
  686. f"{user.mxid} ({user.uuid}) joined room, but pending_members is None,"
  687. " updating chat info"
  688. )
  689. await self.update_info(user, GroupV2ID(id=self.chat_id))
  690. if self._pending_members is None:
  691. self.log.warning(
  692. f"Didn't get pending member list after info update, {user.mxid} ({user.uuid}) may"
  693. "not be in the group on Signal."
  694. )
  695. elif user.uuid in self._pending_members:
  696. self.log.debug(f"{user.mxid} ({user.uuid}) joined room, accepting invite on Signal")
  697. try:
  698. resp = await self.signal.accept_invitation(user.username, self.chat_id)
  699. self._pending_members.remove(user.uuid)
  700. except RPCError as e:
  701. await self.main_intent.send_notice(
  702. self.mxid, f"\u26a0 Failed to accept invite on Signal: {e}"
  703. )
  704. await user.handle_auth_failure(e)
  705. else:
  706. await self.update_info(user, resp)
  707. async def handle_matrix_leave(self, user: u.User) -> None:
  708. if not await user.is_logged_in():
  709. return
  710. if self.is_direct:
  711. self.log.info(f"{user.mxid} left private chat portal with {self.chat_id}")
  712. if user.username == self.receiver:
  713. self.log.info(
  714. f"{user.mxid} was the recipient of this portal. Cleaning up and deleting..."
  715. )
  716. await self.cleanup_and_delete()
  717. else:
  718. self.log.debug(f"{user.mxid} left portal to {self.chat_id}")
  719. if self.config["bridge.bridge_matrix_leave"]:
  720. await self.signal.leave_group(user.username, self.chat_id)
  721. # TODO cleanup if empty
  722. async def kick_matrix(self, user: u.User | p.Puppet, source: u.User) -> None:
  723. try:
  724. await self.signal.update_group(
  725. source.username, self.chat_id, remove_members=[user.address]
  726. )
  727. except Exception as e:
  728. self.log.exception(f"Failed to kick Signal user: {e}")
  729. info = await self.signal.get_group(source.username, self.chat_id)
  730. if user.address in info.members:
  731. await self.main_intent.invite_user(
  732. self.mxid,
  733. user.mxid,
  734. check_cache=True,
  735. reason=f"Failed to kick Signal user: {e}",
  736. )
  737. await user.intent_for(self).ensure_joined(self.mxid)
  738. async def ban_matrix(self, user: u.User | p.Puppet, source: u.User) -> None:
  739. try:
  740. await self.signal.ban_user(source.username, self.chat_id, users=[user.address])
  741. except Exception as e:
  742. self.log.exception(f"Failed to ban Signal user: {e}")
  743. info = await self.signal.get_group(source.username, self.chat_id)
  744. is_banned = False
  745. if info.banned_members:
  746. for member in info.banned_members:
  747. is_banned = user.address.uuid == member.uuid or is_banned
  748. if not is_banned:
  749. await self.main_intent.unban_user(
  750. self.mxid, user.mxid, reason=f"Failed to ban Signal user: {e}"
  751. )
  752. if user.address in info.members:
  753. await self.main_intent.invite_user(
  754. self.mxid,
  755. user.mxid,
  756. check_cache=True,
  757. )
  758. await user.intent_for(self).ensure_joined(self.mxid)
  759. async def unban_matrix(self, user: u.User | p.Puppet, source: u.User) -> None:
  760. try:
  761. await self.signal.unban_user(source.username, self.chat_id, users=[user.address])
  762. except Exception as e:
  763. self.log.exception(f"Failed to unban Signal user: {e}")
  764. info = await self.signal.get_group(source.username, self.chat_id)
  765. if info.banned_members:
  766. for member in info.banned_members:
  767. if member.uuid == user.address.uuid:
  768. await self.main_intent.ban_user(
  769. self.mxid, user.mxid, reason=f"Failed to unban Signal user: {e}"
  770. )
  771. return
  772. async def handle_matrix_invite(self, invited_by: u.User, user: u.User | p.Puppet) -> None:
  773. if self.is_direct:
  774. raise RejectMatrixInvite("You can't invite additional users to private chats.")
  775. try:
  776. await self.signal.update_group(
  777. invited_by.username, self.chat_id, add_members=[user.address]
  778. )
  779. except RPCError as e:
  780. raise RejectMatrixInvite(str(e)) from e
  781. power_levels = await self.main_intent.get_power_levels(self.mxid)
  782. invitee_pl = power_levels.get_user_level(user.mxid)
  783. if invitee_pl >= 50:
  784. group_member = GroupMember(uuid=user.uuid, role=GroupMemberRole.ADMINISTRATOR)
  785. try:
  786. update_meta = await self.signal.update_group(
  787. invited_by.username, self.chat_id, update_role=group_member
  788. )
  789. self.revision = update_meta.revision
  790. except Exception as e:
  791. self.log.exception(f"Failed to update Signal member role: {e}")
  792. await self._update_power_levels(
  793. await self.signal.get_group(invited_by.username, self.chat_id)
  794. )
  795. async def handle_matrix_name(self, user: u.User, name: str) -> None:
  796. if self.name == name or self.is_direct or not name:
  797. return
  798. sender, is_relay = await self.get_relay_sender(user, "name change")
  799. if not sender:
  800. return
  801. self.name = name
  802. self.log.debug(
  803. f"{user.mxid} changed the group name, sending to Signal through {sender.username}"
  804. )
  805. try:
  806. await self.signal.update_group(sender.username, self.chat_id, title=name)
  807. except Exception as e:
  808. self.log.exception("Failed to update Signal group name")
  809. await user.handle_auth_failure(e)
  810. self.name = None
  811. async def handle_matrix_topic(self, user: u.User, topic: str) -> None:
  812. if self.topic == topic or self.is_direct or not topic:
  813. return
  814. sender, is_relay = await self.get_relay_sender(user, "topic change")
  815. if not sender:
  816. return
  817. self.topic = topic
  818. self.log.debug(
  819. f"{user.mxid} changed the group topic, sending to Signal through {sender.username}"
  820. )
  821. try:
  822. await self.signal.update_group(sender.username, self.chat_id, description=topic)
  823. except Exception:
  824. self.log.exception("Failed to update Signal group description")
  825. self.name = None
  826. async def handle_matrix_avatar(self, user: u.User, url: ContentURI) -> None:
  827. if self.is_direct or not url:
  828. return
  829. sender, is_relay = await self.get_relay_sender(user, "avatar change")
  830. if not sender:
  831. return
  832. data = await self.main_intent.download_media(url)
  833. new_hash = hashlib.sha256(data).hexdigest()
  834. if new_hash == self.avatar_hash and self.avatar_set:
  835. self.log.debug(f"New avatar from Matrix set by {user.mxid} is same as current one")
  836. return
  837. self.avatar_url = url
  838. self.avatar_hash = new_hash
  839. path = self._write_outgoing_file(data)
  840. self.log.debug(
  841. f"{user.mxid} changed the group avatar, sending to Signal through {sender.username}"
  842. )
  843. try:
  844. await self.signal.update_group(sender.username, self.chat_id, avatar_path=path)
  845. self.avatar_set = True
  846. except Exception as e:
  847. self.log.exception("Failed to update Signal group avatar")
  848. await user.handle_auth_failure(e)
  849. self.avatar_set = False
  850. if self.config["signal.remove_file_after_handling"]:
  851. try:
  852. os.remove(path)
  853. except FileNotFoundError:
  854. pass
  855. async def handle_matrix_power_level(
  856. self,
  857. sender: u.User,
  858. levels: PowerLevelStateEventContent,
  859. prev_content: PowerLevelStateEventContent | None = None,
  860. ) -> None:
  861. old_users = prev_content.users if prev_content else None
  862. new_users = levels.users
  863. changes = {}
  864. sender, is_relay = await self.get_relay_sender(sender, "power level change")
  865. if not sender:
  866. return
  867. if not old_users:
  868. changes = new_users
  869. else:
  870. for user, level in new_users.items():
  871. if (
  872. user
  873. and user != self.main_intent.mxid
  874. and (user not in old_users or level != old_users[user])
  875. ):
  876. changes[user] = level
  877. for user, level in old_users.items():
  878. if user and user != self.main_intent.mxid and user not in new_users:
  879. changes[user] = levels.users_default
  880. if changes:
  881. for user, level in changes.items():
  882. address = p.Puppet.get_id_from_mxid(user)
  883. if not address:
  884. mx_user = await u.User.get_by_mxid(user, create=False)
  885. if not mx_user or not mx_user.is_logged_in:
  886. continue
  887. address = mx_user.address
  888. if not address or not address.uuid:
  889. continue
  890. signal_role = (
  891. GroupMemberRole.DEFAULT if level < 50 else GroupMemberRole.ADMINISTRATOR
  892. )
  893. group_member = GroupMember(uuid=address.uuid, role=signal_role)
  894. try:
  895. update_meta = await self.signal.update_group(
  896. sender.username, self.chat_id, update_role=group_member
  897. )
  898. self.revision = update_meta.revision
  899. except Exception as e:
  900. self.log.exception(f"Failed to update Signal member role: {e}")
  901. await self._update_power_levels(
  902. await self.signal.get_group(sender.username, self.chat_id)
  903. )
  904. return
  905. if not prev_content or levels.invite != prev_content.invite:
  906. try:
  907. update_meta = await self.signal.update_group(
  908. username=sender.username,
  909. group_id=self.chat_id,
  910. update_access_control=GroupAccessControl(
  911. members=(
  912. AccessControlMode.MEMBER
  913. if levels.invite == 0
  914. else AccessControlMode.ADMINISTRATOR
  915. ),
  916. attributes=None,
  917. link=None,
  918. ),
  919. )
  920. self.revision = update_meta.revision
  921. except Exception as e:
  922. self.log.exception(f"Failed to update Signal member add permission: {e}")
  923. await self._update_power_levels(
  924. await self.signal.get_group(sender.username, self.chat_id)
  925. )
  926. return
  927. if not prev_content or levels.state_default != prev_content.state_default:
  928. try:
  929. update_meta = await self.signal.update_group(
  930. username=sender.username,
  931. group_id=self.chat_id,
  932. update_access_control=GroupAccessControl(
  933. attributes=(
  934. AccessControlMode.MEMBER
  935. if levels.state_default == 0
  936. else AccessControlMode.ADMINISTRATOR
  937. ),
  938. members=None,
  939. link=None,
  940. ),
  941. )
  942. self.revision = update_meta.revision
  943. except Exception as e:
  944. self.log.exception(f"Failed to update Signal metadata change permission: {e}")
  945. await self._update_power_levels(
  946. await self.signal.get_group(sender.username, self.chat_id)
  947. )
  948. # endregion
  949. # region Signal event handling
  950. @staticmethod
  951. async def _resolve_address(address: Address) -> Address:
  952. puppet = await p.Puppet.get_by_address(address, create=False)
  953. return puppet.address
  954. async def _find_quote_event_id(self, quote: Quote | None) -> MessageEvent | EventID | None:
  955. if not quote:
  956. return None
  957. author_address = await self._resolve_address(quote.author)
  958. reply_msg = await DBMessage.get_by_signal_id(
  959. author_address, quote.id, self.chat_id, self.receiver
  960. )
  961. if not reply_msg:
  962. return None
  963. try:
  964. evt = await self.main_intent.get_event(self.mxid, reply_msg.mxid)
  965. if isinstance(evt, EncryptedEvent):
  966. return await self.matrix.e2ee.decrypt(evt, wait_session_timeout=0)
  967. return evt
  968. except MatrixError:
  969. return reply_msg.mxid
  970. async def _signal_link_preview_to_beeper(
  971. self, link_preview: LinkPreview, intent: IntentAPI
  972. ) -> dict[str, Any]:
  973. beeper_link_preview: dict[str, Any] = {
  974. "matched_url": link_preview.url,
  975. "og:title": link_preview.title,
  976. "og:url": link_preview.url,
  977. "og:description": link_preview.description,
  978. }
  979. # Upload an image corresponding to the link preview if it exists.
  980. if link_preview.attachment and link_preview.attachment.incoming_filename:
  981. beeper_link_preview["og:image:type"] = link_preview.attachment.content_type
  982. beeper_link_preview["og:image:height"] = link_preview.attachment.height
  983. beeper_link_preview["og:image:width"] = link_preview.attachment.width
  984. beeper_link_preview["matrix:image:size"] = link_preview.attachment.size
  985. with open(link_preview.attachment.incoming_filename, "rb") as file:
  986. data = file.read()
  987. if self.config["signal.remove_file_after_handling"]:
  988. os.remove(link_preview.attachment.incoming_filename)
  989. upload_mime_type = link_preview.attachment.content_type
  990. if self.encrypted and encrypt_attachment:
  991. data, beeper_link_preview[BEEPER_IMAGE_ENCRYPTION_KEY] = encrypt_attachment(data)
  992. upload_mime_type = "application/octet-stream"
  993. upload_uri = await intent.upload_media(
  994. data,
  995. mime_type=upload_mime_type,
  996. filename=link_preview.attachment.id,
  997. async_upload=self.config["homeserver.async_media"],
  998. )
  999. if BEEPER_IMAGE_ENCRYPTION_KEY in beeper_link_preview:
  1000. beeper_link_preview[BEEPER_IMAGE_ENCRYPTION_KEY].url = upload_uri
  1001. beeper_link_preview[BEEPER_IMAGE_ENCRYPTION_KEY] = beeper_link_preview[
  1002. BEEPER_IMAGE_ENCRYPTION_KEY
  1003. ].serialize()
  1004. else:
  1005. beeper_link_preview["og:image"] = upload_uri
  1006. return beeper_link_preview
  1007. async def handle_signal_message(
  1008. self, source: u.User, sender: p.Puppet, message: MessageData
  1009. ) -> None:
  1010. if (sender.address, message.timestamp) in self._msgts_dedup:
  1011. self.log.debug(
  1012. f"Ignoring message {message.timestamp} by {sender.uuid} as it was already handled "
  1013. "(message.timestamp in dedup queue)"
  1014. )
  1015. await self.signal.send_receipt(
  1016. source.username, sender.address, timestamps=[message.timestamp]
  1017. )
  1018. return
  1019. self._msgts_dedup.appendleft((sender.address, message.timestamp))
  1020. old_message = await DBMessage.get_by_signal_id(
  1021. sender.address, message.timestamp, self.chat_id, self.receiver
  1022. )
  1023. if old_message is not None:
  1024. self.log.debug(
  1025. f"Ignoring message {message.timestamp} by {sender.uuid} as it was already handled "
  1026. "(message.id found in database)"
  1027. )
  1028. await self.signal.send_receipt(
  1029. source.username, sender.address, timestamps=[message.timestamp]
  1030. )
  1031. return
  1032. self.log.debug(f"Started handling message {message.timestamp} by {sender.uuid}")
  1033. self.log.trace(f"Message content: {message}")
  1034. intent = sender.intent_for(self)
  1035. await intent.set_typing(self.mxid, False)
  1036. event_id = None
  1037. reply_to = await self._find_quote_event_id(message.quote)
  1038. if message.sticker:
  1039. if message.sticker.attachment.incoming_filename:
  1040. content = await self._handle_signal_attachment(
  1041. intent, message.sticker.attachment, sticker=True
  1042. )
  1043. elif StickersClient:
  1044. content = await self._handle_signal_sticker(intent, message.sticker)
  1045. else:
  1046. self.log.debug(
  1047. f"Not handling sticker in {message.timestamp}: no incoming_filename and "
  1048. "signalstickers-client not installed."
  1049. )
  1050. return
  1051. if content:
  1052. if message.sticker.attachment.blurhash:
  1053. content.info["blurhash"] = message.sticker.attachment.blurhash
  1054. content.info["xyz.amorgan.blurhash"] = message.sticker.attachment.blurhash
  1055. await self._add_sticker_meta(message.sticker, content)
  1056. if reply_to and not message.body:
  1057. content.set_reply(reply_to)
  1058. reply_to = None
  1059. content.msgtype = None
  1060. event_id = await self._send_message(
  1061. intent, content, timestamp=message.timestamp, event_type=EventType.STICKER
  1062. )
  1063. for contact in message.contacts:
  1064. content = await self._handle_signal_contact(contact)
  1065. if reply_to and not message.body:
  1066. content.set_reply(reply_to)
  1067. reply_to = None
  1068. event_id = await self._send_message(intent, content, timestamp=message.timestamp)
  1069. is_first_text = True
  1070. for attachment in message.attachments:
  1071. if not attachment.incoming_filename:
  1072. self.log.warning(
  1073. "Failed to bridge attachment, no incoming filename: %s", attachment
  1074. )
  1075. continue
  1076. as_text = (
  1077. is_first_text
  1078. and attachment.content_type == "text/x-signal-plain"
  1079. and attachment.size < MAX_MATRIX_MESSAGE_SIZE
  1080. )
  1081. file_size = attachment.size or os.path.getsize(attachment.incoming_filename)
  1082. if file_size > self.matrix.media_config.upload_size:
  1083. self.log.warning(
  1084. "Failed to bridge attachment %s in %s: file too large",
  1085. attachment.id,
  1086. message.timestamp,
  1087. )
  1088. continue
  1089. content = await self._handle_signal_attachment(intent, attachment, text=as_text)
  1090. if as_text:
  1091. is_first_text = False
  1092. message.body = ""
  1093. if reply_to and not message.body:
  1094. # If there's no text, set the first image as the reply
  1095. content.set_reply(reply_to)
  1096. reply_to = None
  1097. event_id = await self._send_message(intent, content, timestamp=message.timestamp)
  1098. if message.body:
  1099. content = await signal_to_matrix(message)
  1100. if message.previews:
  1101. content[BEEPER_LINK_PREVIEWS_KEY] = await asyncio.gather(
  1102. *(self._signal_link_preview_to_beeper(p, intent) for p in message.previews)
  1103. )
  1104. if reply_to:
  1105. content.set_reply(reply_to)
  1106. event_id = await self._send_message(intent, content, timestamp=message.timestamp)
  1107. if event_id:
  1108. msg = DBMessage(
  1109. mxid=event_id,
  1110. mx_room=self.mxid,
  1111. sender=sender.address,
  1112. timestamp=message.timestamp,
  1113. signal_chat_id=self.chat_id,
  1114. signal_receiver=self.receiver,
  1115. )
  1116. await msg.insert()
  1117. await self.signal.send_receipt(
  1118. source.username, sender.address, timestamps=[message.timestamp]
  1119. )
  1120. await self._send_delivery_receipt(event_id)
  1121. self.log.debug(f"Handled Signal message {message.timestamp} -> {event_id}")
  1122. if message.expires_in_seconds and self.disappearing_enabled:
  1123. await DisappearingMessage(self.mxid, event_id, message.expires_in_seconds).insert()
  1124. self.log.debug(
  1125. f"{event_id} set to be redacted {message.expires_in_seconds} seconds after "
  1126. "room is read"
  1127. )
  1128. else:
  1129. self.log.debug(f"Didn't get event ID for {message.timestamp}")
  1130. async def handle_signal_kicked(self, user: u.User, sender: p.Puppet) -> None:
  1131. self.log.debug(f"{user.mxid} was kicked by {sender.number} from {self.mxid}")
  1132. await self._kick_with_puppet(user, sender)
  1133. async def handle_signal_group_change(self, group_change: GroupChange, source: u.User) -> None:
  1134. if self.revision < group_change.revision:
  1135. self.revision = group_change.revision
  1136. else:
  1137. return
  1138. editor = await p.Puppet.get_by_address(group_change.editor)
  1139. editor_intent = editor.intent_for(self)
  1140. if (
  1141. group_change.delete_members
  1142. or group_change.delete_pending_members
  1143. or group_change.delete_requesting_members
  1144. ):
  1145. for address in (
  1146. (group_change.delete_members or [])
  1147. + (group_change.delete_pending_members or [])
  1148. + (group_change.delete_requesting_members or [])
  1149. ):
  1150. user = await p.Puppet.get_by_address(address)
  1151. if not user:
  1152. continue
  1153. if user == editor:
  1154. await editor_intent.leave_room(self.mxid)
  1155. else:
  1156. try:
  1157. await editor_intent.kick_user(self.mxid, user.mxid)
  1158. except MForbidden:
  1159. try:
  1160. await self.main_intent.kick_user(
  1161. self.mxid, user.mxid, reason=f"removed by {editor.name}"
  1162. )
  1163. except MForbidden as e:
  1164. self.log.debug(f"Could not remove {user.mxid}: {e}")
  1165. except MBadState as e:
  1166. self.log.debug(f"Could not remove {user.mxid}: {e}")
  1167. if group_change.modify_member_roles:
  1168. levels = await editor.intent_for(self).get_power_levels(self.mxid)
  1169. for group_member in group_change.modify_member_roles:
  1170. user = await p.Puppet.get_by_address(Address(uuid=group_member.uuid))
  1171. if not user:
  1172. continue
  1173. if group_member.role == GroupMemberRole.ADMINISTRATOR:
  1174. new_pl = 50
  1175. else:
  1176. new_pl = 0
  1177. levels.users[user.mxid] = new_pl
  1178. await self._try_with_puppet(
  1179. lambda i: i.set_power_levels(self.mxid, levels), puppet=editor
  1180. )
  1181. if group_change.new_banned_members:
  1182. for banned_member in group_change.new_banned_members:
  1183. user = await p.Puppet.get_by_address(Address(uuid=banned_member.uuid))
  1184. if not user:
  1185. continue
  1186. try:
  1187. await editor_intent.ban_user(self.mxid, user.mxid)
  1188. except MForbidden:
  1189. try:
  1190. await self.main_intent.ban_user(
  1191. self.mxid, user.mxid, reason=f"banned by {editor.name}"
  1192. )
  1193. except MForbidden as e:
  1194. self.log.debug(f"Could not ban {user.mxid}: {e}")
  1195. except MBadState as e:
  1196. self.log.debug(f"Could not ban {user.mxid}: {e}")
  1197. if group_change.new_unbanned_members:
  1198. for banned_member in group_change.new_unbanned_members:
  1199. user = await p.Puppet.get_by_address(Address(uuid=banned_member.uuid))
  1200. if not user:
  1201. continue
  1202. try:
  1203. await editor_intent.unban_user(self.mxid, user.mxid)
  1204. except MForbidden:
  1205. try:
  1206. await self.main_intent.unban_user(
  1207. self.mxid, user.mxid, reason=f"unbanned by {editor.name}"
  1208. )
  1209. except MForbidden as e:
  1210. self.log.debug(f"Could not unban {user.mxid}: {e}")
  1211. except MBadState as e:
  1212. self.log.debug(f"Could not unban {user.mxid}: {e}")
  1213. if (
  1214. group_change.new_members
  1215. or group_change.new_pending_members
  1216. or group_change.promote_requesting_members
  1217. ):
  1218. banned_users = await self.az.intent.get_room_members(self.mxid, (Membership.BAN,))
  1219. for group_member in (
  1220. (group_change.new_members or [])
  1221. + (group_change.new_pending_members or [])
  1222. + (group_change.promote_requesting_members or [])
  1223. ):
  1224. user = await p.Puppet.get_by_address(Address(uuid=group_member.uuid))
  1225. if not user:
  1226. continue
  1227. try:
  1228. await source.sync_contact(Address(uuid=group_member.uuid))
  1229. except ProfileUnavailableError:
  1230. self.log.debug(
  1231. f"Profile of puppet with uuid {group_member.uuid} is unavailable"
  1232. )
  1233. if user.mxid in banned_users:
  1234. await self._try_with_puppet(
  1235. lambda i: i.unban_user(self.mxid, user.mxid), puppet=editor
  1236. )
  1237. try:
  1238. await editor_intent.invite_user(self.mxid, user.mxid, check_cache=True)
  1239. except (MForbidden, IntentError):
  1240. try:
  1241. await self.main_intent.invite_user(
  1242. self.mxid,
  1243. user.mxid,
  1244. reason=f"invited by {editor.name}",
  1245. check_cache=True,
  1246. )
  1247. except (MForbidden, IntentError) as e:
  1248. self.log.debug(f"{editor.name} could not invite {user.mxid}: {e}")
  1249. except MBadState as e:
  1250. self.log.debug(f"{editor.name} could not invite {user.mxid}: {e}")
  1251. if (
  1252. group_member
  1253. in (group_change.new_members or [])
  1254. + (group_change.promote_requesting_members or [])
  1255. and not user.is_real_user()
  1256. ):
  1257. try:
  1258. await user.intent_for(self).ensure_joined(self.mxid)
  1259. except IntentError as e:
  1260. self.log.debug(f"{user.name} could not join group: {e}")
  1261. if group_change.promote_pending_members:
  1262. for member in group_change.promote_pending_members:
  1263. user = await p.Puppet.get_by_address(Address(uuid=member.uuid))
  1264. if not user:
  1265. continue
  1266. try:
  1267. await source.sync_contact(Address(uuid=group_member.uuid))
  1268. except ProfileUnavailableError:
  1269. self.log.debug(
  1270. f"Profile of puppet with uuid {group_member.uuid} is unavailable"
  1271. )
  1272. try:
  1273. user.intent_for(self).ensure_joined(self.mxid)
  1274. except IntentError as e:
  1275. self.log.debug(f"{user.name} could not join group: {e}")
  1276. if group_change.new_requesting_members:
  1277. for member in group_change.new_requesting_members:
  1278. user = await p.Puppet.get_by_address(Address(uuid=member.uuid))
  1279. if not user:
  1280. continue
  1281. try:
  1282. await source.sync_contact(Address(uuid=group_member.uuid))
  1283. except ProfileUnavailableError:
  1284. self.log.debug(
  1285. f"Profile of puppet with uuid {group_member.uuid} is unavailable"
  1286. )
  1287. try:
  1288. await user.intent_for(self).knock(self.mxid, reason="via invite link")
  1289. except (MForbidden, MBadState) as e:
  1290. self.log.debug(f"{user.name} failed knock: {e}")
  1291. if group_change.new_access_control:
  1292. ac = group_change.new_access_control
  1293. if ac.attributes or ac.members:
  1294. levels = await editor.intent_for(self).get_power_levels(self.mxid)
  1295. if ac.attributes:
  1296. meta_edit_level = 50 if ac.attributes == AccessControlMode.ADMINISTRATOR else 0
  1297. levels.events[EventType.ROOM_NAME] = meta_edit_level
  1298. levels.events[EventType.ROOM_AVATAR] = meta_edit_level
  1299. levels.events[EventType.ROOM_TOPIC] = meta_edit_level
  1300. if ac.members:
  1301. levels.invite = 50 if ac.members == AccessControlMode.ADMINISTRATOR else 0
  1302. await self._try_with_puppet(
  1303. lambda i: i.set_power_levels(self.mxid, levels), puppet=editor
  1304. )
  1305. if ac.link:
  1306. join_rule = JoinRule.INVITE
  1307. if ac.link == AccessControlMode.ANY and self.config["bridge.public_portals"]:
  1308. join_rule = JoinRule.PUBLIC
  1309. elif ac.link == AccessControlMode.ADMINISTRATOR:
  1310. join_rule = JoinRule.KNOCK
  1311. await self._try_with_puppet(
  1312. lambda i: i.set_join_rule(self.mxid, join_rule), puppet=editor
  1313. )
  1314. if group_change.new_is_announcement_group:
  1315. levels = await editor.intent_for(self).get_power_levels(self.mxid)
  1316. if group_change.new_is_announcement_group == AnnouncementsMode.ENABLED:
  1317. levels.events_default = 50
  1318. elif group_change.new_is_announcement_group == AnnouncementsMode.DISABLED:
  1319. levels.events_default = 0
  1320. await self._try_with_puppet(
  1321. lambda i: i.set_power_levels(self.mxid, levels), puppet=editor
  1322. )
  1323. changed = False
  1324. if group_change.new_description:
  1325. changed = await self._update_topic(group_change.new_description, editor)
  1326. if group_change.new_title:
  1327. changed = await self._update_name(group_change.new_title, editor) or changed
  1328. if group_change.new_avatar:
  1329. changed = (
  1330. await self._update_avatar(
  1331. await self.signal.get_group(
  1332. source.username, self.chat_id, group_change.revision
  1333. ),
  1334. editor,
  1335. )
  1336. or changed
  1337. )
  1338. if changed:
  1339. await self.update_bridge_info()
  1340. await self.update()
  1341. @staticmethod
  1342. async def _make_media_content(
  1343. attachment: Attachment, data: bytes
  1344. ) -> tuple[MediaMessageEventContent, bytes]:
  1345. if attachment.content_type.startswith("image/"):
  1346. msgtype = MessageType.IMAGE
  1347. info = ImageInfo(
  1348. mimetype=attachment.content_type, width=attachment.width, height=attachment.height
  1349. )
  1350. elif attachment.content_type.startswith("video/"):
  1351. msgtype = MessageType.VIDEO
  1352. info = VideoInfo(
  1353. mimetype=attachment.content_type, width=attachment.width, height=attachment.height
  1354. )
  1355. elif attachment.voice_note or attachment.content_type.startswith("audio/"):
  1356. msgtype = MessageType.AUDIO
  1357. info = AudioInfo(
  1358. mimetype=attachment.content_type if not attachment.voice_note else "audio/ogg"
  1359. )
  1360. else:
  1361. msgtype = MessageType.FILE
  1362. info = FileInfo(mimetype=attachment.content_type)
  1363. info.size = attachment.size or len(data)
  1364. if not attachment.custom_filename:
  1365. ext = mimetypes.guess_extension(info.mimetype) or ""
  1366. attachment.custom_filename = attachment.id + ext
  1367. else:
  1368. for ext in mimetypes.guess_all_extensions(info.mimetype):
  1369. if attachment.custom_filename.endswith(ext):
  1370. break
  1371. else:
  1372. attachment.custom_filename += mimetypes.guess_extension(info.mimetype) or ""
  1373. if attachment.blurhash:
  1374. info["blurhash"] = attachment.blurhash
  1375. info["xyz.amorgan.blurhash"] = attachment.blurhash
  1376. content = MediaMessageEventContent(
  1377. msgtype=msgtype, info=info, body=attachment.custom_filename
  1378. )
  1379. # If this is a voice note, add the additional voice message metadata and convert to OGG.
  1380. if attachment.voice_note:
  1381. content["org.matrix.msc1767.file"] = {
  1382. "url": content.url,
  1383. "name": content.body,
  1384. **(content.file.serialize() if content.file else {}),
  1385. **(content.info.serialize() if content.info else {}),
  1386. }
  1387. content["org.matrix.msc3245.voice"] = {}
  1388. data = await ffmpeg.convert_bytes(
  1389. data, ".ogg", output_args=("-c:a", "libopus"), input_mime=attachment.content_type
  1390. )
  1391. info.size = len(data)
  1392. return content, data
  1393. async def _handle_signal_attachment(
  1394. self, intent: IntentAPI, attachment: Attachment, sticker: bool = False, text: bool = False
  1395. ) -> MediaMessageEventContent | TextMessageEventContent:
  1396. self.log.trace(f"Reuploading attachment {attachment}")
  1397. if not attachment.content_type:
  1398. attachment.content_type = (
  1399. magic.mimetype(attachment.incoming_filename)
  1400. if magic is not None
  1401. else "application/octet-stream"
  1402. )
  1403. with open(attachment.incoming_filename, "rb") as file:
  1404. data = file.read()
  1405. if self.config["signal.remove_file_after_handling"]:
  1406. os.remove(attachment.incoming_filename)
  1407. if text:
  1408. assert attachment.content_type == "text/x-signal-plain"
  1409. assert attachment.size < MAX_MATRIX_MESSAGE_SIZE
  1410. content = TextMessageEventContent(msgtype=MessageType.TEXT, body=data.decode("utf-8"))
  1411. return content
  1412. content, data = await self._make_media_content(attachment, data)
  1413. if sticker:
  1414. self._adjust_sticker_size(content.info)
  1415. await self._upload_attachment(intent, content, data, attachment.id)
  1416. return content
  1417. @staticmethod
  1418. async def _handle_signal_contact(contact: SharedContact) -> TextMessageEventContent:
  1419. msg = f"Shared contact: {contact.name!s}"
  1420. if contact.phone:
  1421. msg += "\n"
  1422. for phone in contact.phone:
  1423. msg += f"\nPhone: {phone.value} ({phone.type_or_label})"
  1424. if contact.email:
  1425. msg += "\n"
  1426. for email in contact.email:
  1427. msg += f"\nEmail: {email.value} ({email.type_or_label})"
  1428. content = TextMessageEventContent(msgtype=MessageType.TEXT, body=msg)
  1429. content["fi.mau.signal.contact"] = contact.serialize()
  1430. return content
  1431. async def _add_sticker_meta(self, sticker: Sticker, content: MediaMessageEventContent) -> None:
  1432. try:
  1433. pack = self._sticker_meta_cache[sticker.pack_id]
  1434. except KeyError:
  1435. self.log.debug(f"Fetching sticker pack metadata for {sticker.pack_id}")
  1436. try:
  1437. async with StickersClient() as client:
  1438. pack = await client.get_pack_metadata(sticker.pack_id, sticker.pack_key)
  1439. self._sticker_meta_cache[sticker.pack_id] = pack
  1440. except Exception:
  1441. self.log.warning(
  1442. f"Failed to fetch pack metadata for {sticker.pack_id}", exc_info=True
  1443. )
  1444. pack = None
  1445. if not pack:
  1446. content.info["fi.mau.signal.sticker"] = {
  1447. "id": sticker.sticker_id,
  1448. "pack": {
  1449. "id": sticker.pack_id,
  1450. "key": sticker.pack_key,
  1451. },
  1452. }
  1453. return
  1454. sticker_meta = pack.stickers[sticker.sticker_id]
  1455. content.body = sticker_meta.emoji
  1456. content.info["fi.mau.signal.sticker"] = {
  1457. "id": sticker.sticker_id,
  1458. "emoji": sticker_meta.emoji,
  1459. "pack": {
  1460. "id": pack.id,
  1461. "key": pack.key,
  1462. "title": pack.title,
  1463. "author": pack.author,
  1464. },
  1465. }
  1466. @staticmethod
  1467. def _adjust_sticker_size(info: ImageInfo) -> None:
  1468. if info.width > 256 or info.height > 256:
  1469. if info.width == info.height:
  1470. info.width = info.height = 256
  1471. elif info.width > info.height:
  1472. info.height = int(info.height / (info.width / 256))
  1473. info.width = 256
  1474. else:
  1475. info.width = int(info.width / (info.height / 256))
  1476. info.height = 256
  1477. async def _handle_signal_sticker(
  1478. self, intent: IntentAPI, sticker: Sticker
  1479. ) -> MediaMessageEventContent | None:
  1480. try:
  1481. self.log.debug(f"Fetching sticker {sticker.pack_id}#{sticker.sticker_id}")
  1482. async with StickersClient() as client:
  1483. data = await client.download_sticker(
  1484. sticker.sticker_id, sticker.pack_id, sticker.pack_key
  1485. )
  1486. except Exception:
  1487. self.log.warning(f"Failed to download sticker {sticker.sticker_id}", exc_info=True)
  1488. return None
  1489. info = ImageInfo(
  1490. mimetype=sticker.attachment.content_type,
  1491. size=len(data),
  1492. width=sticker.attachment.width,
  1493. height=sticker.attachment.height,
  1494. )
  1495. self._adjust_sticker_size(info)
  1496. if magic:
  1497. info.mimetype = magic.mimetype(data)
  1498. ext = mimetypes.guess_extension(info.mimetype)
  1499. if not ext and info.mimetype == "image/webp":
  1500. ext = ".webp"
  1501. content = MediaMessageEventContent(
  1502. msgtype=MessageType.IMAGE, info=info, body=f"sticker{ext}"
  1503. )
  1504. await self._upload_attachment(intent, content, data, sticker.attachment.id)
  1505. return content
  1506. async def _upload_attachment(
  1507. self, intent: IntentAPI, content: MediaMessageEventContent, data: bytes, id: str
  1508. ) -> None:
  1509. upload_mime_type = content.info.mimetype
  1510. if self.encrypted and encrypt_attachment:
  1511. data, content.file = encrypt_attachment(data)
  1512. upload_mime_type = "application/octet-stream"
  1513. content.url = await intent.upload_media(
  1514. data,
  1515. mime_type=upload_mime_type,
  1516. filename=id,
  1517. async_upload=self.config["homeserver.async_media"],
  1518. )
  1519. if content.file:
  1520. content.file.url = content.url
  1521. content.url = None
  1522. # This is a hack for bad clients like Element iOS that require a thumbnail
  1523. if content.info.mimetype.startswith("image/"):
  1524. if content.file:
  1525. content.info.thumbnail_file = content.file
  1526. elif content.url:
  1527. content.info.thumbnail_url = content.url
  1528. async def handle_signal_reaction(
  1529. self, sender: p.Puppet, reaction: Reaction, timestamp: int
  1530. ) -> None:
  1531. author_address = await self._resolve_address(reaction.target_author)
  1532. target_id = reaction.target_sent_timestamp
  1533. async with self._reaction_lock:
  1534. dedup_id = (author_address, target_id, reaction.emoji, sender.address, reaction.remove)
  1535. if dedup_id in self._reaction_dedup:
  1536. return
  1537. self._reaction_dedup.appendleft(dedup_id)
  1538. existing = await DBReaction.get_by_signal_id(
  1539. self.chat_id, self.receiver, author_address, target_id, sender.address
  1540. )
  1541. if reaction.remove:
  1542. if existing:
  1543. try:
  1544. await sender.intent_for(self).redact(existing.mx_room, existing.mxid)
  1545. except IntentError:
  1546. await self.main_intent.redact(existing.mx_room, existing.mxid)
  1547. await existing.delete()
  1548. self.log.trace(f"Removed {existing} after Signal removal")
  1549. return
  1550. elif existing and existing.emoji == reaction.emoji:
  1551. return
  1552. message = await DBMessage.get_by_signal_id(
  1553. author_address, target_id, self.chat_id, self.receiver
  1554. )
  1555. if not message:
  1556. self.log.debug(f"Ignoring reaction to unknown message {target_id}")
  1557. return
  1558. intent = sender.intent_for(self)
  1559. matrix_emoji = variation_selector.add(reaction.emoji)
  1560. mxid = await intent.react(message.mx_room, message.mxid, matrix_emoji, timestamp=timestamp)
  1561. self.log.debug(f"{sender.address} reacted to {message.mxid} -> {mxid}")
  1562. await self._upsert_reaction(existing, intent, mxid, sender, message, reaction.emoji)
  1563. async def handle_signal_delete(self, sender: p.Puppet, message_ts: int) -> None:
  1564. message = await DBMessage.get_by_signal_id(
  1565. sender.address, message_ts, self.chat_id, self.receiver
  1566. )
  1567. if not message:
  1568. return
  1569. await message.delete()
  1570. try:
  1571. await sender.intent_for(self).redact(message.mx_room, message.mxid)
  1572. except MForbidden:
  1573. await self.main_intent.redact(message.mx_room, message.mxid)
  1574. # endregion
  1575. # region Matrix -> Signal metadata
  1576. async def create_signal_group(
  1577. self, source: u.User, levels: PowerLevelStateEventContent
  1578. ) -> None:
  1579. user_mxids = await self.az.intent.get_room_members(
  1580. self.mxid, (Membership.JOIN, Membership.INVITE)
  1581. )
  1582. invitee_addresses = []
  1583. for mxid in user_mxids:
  1584. mx_user = await u.User.get_by_mxid(mxid, create=False)
  1585. if mx_user and mx_user.address and mx_user.username != source.username:
  1586. invitee_addresses.append(mx_user.address)
  1587. puppet = await p.Puppet.get_by_mxid(mxid, create=False)
  1588. if puppet:
  1589. invitee_addresses.append(puppet.address)
  1590. avatar_path: str | None = None
  1591. if self.avatar_url:
  1592. avatar_data = await self.az.intent.download_media(self.avatar_url)
  1593. self.avatar_hash = hashlib.sha256(avatar_data).hexdigest()
  1594. avatar_path = self._write_outgoing_file(avatar_data)
  1595. signal_chat = await self.signal.create_group(
  1596. source.username, title=self.name, members=invitee_addresses, avatar_path=avatar_path
  1597. )
  1598. self.name_set = bool(self.name and signal_chat.title)
  1599. self.avatar_set = bool(self.avatar_url and self.avatar_hash and signal_chat.avatar)
  1600. self.chat_id = signal_chat.id
  1601. await self._postinit()
  1602. await self.insert()
  1603. if avatar_path and self.config["signal.remove_file_after_handling"]:
  1604. try:
  1605. os.remove(avatar_path)
  1606. except FileNotFoundError:
  1607. pass
  1608. if self.topic:
  1609. await self.signal.update_group(source.username, self.chat_id, description=self.topic)
  1610. await self.handle_matrix_power_level(source, levels)
  1611. await self.update()
  1612. await self.update_bridge_info()
  1613. # endregion
  1614. # region Updating portal info
  1615. async def update_info(
  1616. self, source: u.User, info: ChatInfo, sender: p.Puppet | None = None
  1617. ) -> None:
  1618. if self.is_direct:
  1619. if not isinstance(info, (Profile, Address)):
  1620. raise ValueError(f"Unexpected type for direct chat update_info: {type(info)}")
  1621. if not self.name:
  1622. puppet = await self.get_dm_puppet()
  1623. if not puppet.name:
  1624. await puppet.update_info(info, source)
  1625. self.name = puppet.name
  1626. return
  1627. if not info.title:
  1628. try:
  1629. info = await self.signal.get_group(source.username, info.id, info.revision or -1)
  1630. except Exception as e:
  1631. await source.handle_auth_failure(e)
  1632. raise
  1633. if not info:
  1634. self.log.debug(
  1635. f"Failed to get full group v2 info through {source.username}, "
  1636. "cancelling update"
  1637. )
  1638. return
  1639. changed = False
  1640. if isinstance(info, Group):
  1641. changed = await self._update_name(info.name, sender) or changed
  1642. elif isinstance(info, GroupV2):
  1643. if self.revision < info.revision:
  1644. self.revision = info.revision
  1645. changed = True
  1646. elif self.revision > info.revision:
  1647. self.log.warning(
  1648. f"Got outdated info when syncing through {source.username} "
  1649. f"({info.revision} < {self.revision}), ignoring..."
  1650. )
  1651. return
  1652. changed = await self._update_name(info.title, sender) or changed
  1653. changed = await self._update_topic(info.description, sender) or changed
  1654. elif isinstance(info, GroupV2ID):
  1655. return
  1656. else:
  1657. raise ValueError(f"Unexpected type for group update_info: {type(info)}")
  1658. changed = await self._update_avatar(info, sender) or changed
  1659. await self._update_participants(source, info, sender)
  1660. try:
  1661. await self._update_power_levels(info)
  1662. except Exception:
  1663. self.log.warning("Error updating power levels", exc_info=True)
  1664. if changed:
  1665. await self.update_bridge_info()
  1666. await self.update()
  1667. async def update_expires_in_seconds(self, sender: p.Puppet, expires_in_seconds: int) -> None:
  1668. if expires_in_seconds == 0:
  1669. expires_in_seconds = None
  1670. if self.expiration_time == expires_in_seconds:
  1671. return
  1672. assert self.mxid
  1673. self.expiration_time = expires_in_seconds
  1674. await self.update()
  1675. time_str = "Off" if expires_in_seconds is None else format_duration(expires_in_seconds)
  1676. body = f"Set the disappearing message timer to {time_str}"
  1677. content = TextMessageEventContent(msgtype=MessageType.NOTICE, body=body)
  1678. await self._send_message(sender.intent_for(self), content)
  1679. async def get_dm_puppet(self) -> p.Puppet | None:
  1680. if not self.is_direct:
  1681. return None
  1682. return await p.Puppet.get_by_address(self.chat_id)
  1683. async def update_info_from_puppet(self, puppet: p.Puppet | None = None) -> None:
  1684. if not self.is_direct:
  1685. return
  1686. if not puppet:
  1687. puppet = await self.get_dm_puppet()
  1688. await self.update_puppet_name(puppet.name, save=False)
  1689. await self.update_puppet_avatar(puppet.avatar_hash, puppet.avatar_url, save=False)
  1690. async def update_puppet_avatar(
  1691. self, new_hash: str, avatar_url: ContentURI, save: bool = True
  1692. ) -> None:
  1693. if not self.encrypted and not self.private_chat_portal_meta:
  1694. return
  1695. if self.avatar_hash != new_hash or not self.avatar_set:
  1696. self.avatar_hash = new_hash
  1697. self.avatar_url = avatar_url
  1698. if self.mxid:
  1699. try:
  1700. await self.main_intent.set_room_avatar(self.mxid, avatar_url)
  1701. self.avatar_set = True
  1702. except Exception:
  1703. self.log.exception("Error setting avatar")
  1704. self.avatar_set = False
  1705. if save:
  1706. await self.update_bridge_info()
  1707. await self.update()
  1708. async def update_puppet_name(self, name: str, save: bool = True) -> None:
  1709. if not self.encrypted and not self.private_chat_portal_meta:
  1710. return
  1711. changed = await self._update_name(name)
  1712. if changed and save:
  1713. await self.update_bridge_info()
  1714. await self.update()
  1715. async def _update_name(self, name: str, sender: p.Puppet | None = None) -> bool:
  1716. if self.name != name or not self.name_set:
  1717. self.name = name
  1718. if self.mxid:
  1719. try:
  1720. await self._try_with_puppet(
  1721. lambda i: i.set_room_name(self.mxid, self.name), puppet=sender
  1722. )
  1723. self.name_set = True
  1724. except Exception:
  1725. self.log.exception("Error setting name")
  1726. self.name_set = False
  1727. return True
  1728. return False
  1729. async def _update_topic(self, topic: str, sender: p.Puppet | None = None) -> bool:
  1730. if self.topic != topic:
  1731. self.topic = topic
  1732. if self.mxid:
  1733. try:
  1734. await self._try_with_puppet(
  1735. lambda i: i.set_room_topic(self.mxid, self.topic), puppet=sender
  1736. )
  1737. except Exception:
  1738. self.log.exception("Error setting topic")
  1739. self.topic = None
  1740. return True
  1741. return False
  1742. async def _try_with_puppet(
  1743. self, action: Callable[[IntentAPI], Awaitable[Any]], puppet: p.Puppet | None = None
  1744. ) -> None:
  1745. if puppet:
  1746. try:
  1747. await action(puppet.intent_for(self))
  1748. except (MForbidden, IntentError):
  1749. await action(self.main_intent)
  1750. else:
  1751. await action(self.main_intent)
  1752. async def _update_avatar(self, info: ChatInfo, sender: p.Puppet | None = None) -> bool:
  1753. path = None
  1754. if isinstance(info, GroupV2):
  1755. path = info.avatar
  1756. elif isinstance(info, Group):
  1757. path = f"group-{self.chat_id}"
  1758. res = await p.Puppet.upload_avatar(self, path, self.main_intent)
  1759. if res is False:
  1760. return False
  1761. self.avatar_hash, self.avatar_url = res
  1762. if not self.mxid:
  1763. return True
  1764. try:
  1765. await self._try_with_puppet(
  1766. lambda i: i.set_room_avatar(self.mxid, self.avatar_url), puppet=sender
  1767. )
  1768. self.avatar_set = True
  1769. except Exception:
  1770. self.log.exception("Error setting avatar")
  1771. self.avatar_set = False
  1772. return True
  1773. async def _update_participants(
  1774. self, source: u.User, info: ChatInfo, sender: p.Puppet | None = None
  1775. ) -> None:
  1776. if not self.mxid or not isinstance(info, (Group, GroupV2)):
  1777. return
  1778. member_events = await self.main_intent.get_members(self.mxid)
  1779. remove_users: set[UserID] = {
  1780. UserID(evt.state_key)
  1781. for evt in member_events
  1782. if evt.content.membership == Membership.JOIN and evt.state_key != self.az.bot_mxid
  1783. }
  1784. pending_members = info.pending_members if isinstance(info, GroupV2) else []
  1785. self._pending_members = {addr.uuid for addr in pending_members}
  1786. for address in info.members:
  1787. user = await u.User.get_by_address(address)
  1788. if user:
  1789. remove_users.discard(user.mxid)
  1790. await self._try_with_puppet(
  1791. lambda i: i.invite_user(self.mxid, user.mxid, check_cache=True), puppet=sender
  1792. )
  1793. puppet = await p.Puppet.get_by_address(address)
  1794. try:
  1795. await source.sync_contact(address)
  1796. except ProfileUnavailableError:
  1797. self.log.debug(f"Profile of puppet with {address} is unavailable")
  1798. await self._try_with_puppet(
  1799. lambda i: i.invite_user(self.mxid, puppet.intent_for(self).mxid, check_cache=True),
  1800. puppet=sender,
  1801. )
  1802. await puppet.intent_for(self).ensure_joined(self.mxid)
  1803. remove_users.discard(puppet.default_mxid)
  1804. for address in pending_members:
  1805. user = await u.User.get_by_address(address)
  1806. if user:
  1807. remove_users.discard(user.mxid)
  1808. await self._try_with_puppet(
  1809. lambda i: i.invite_user(self.mxid, user.mxid, check_cache=True), puppet=sender
  1810. )
  1811. puppet = await p.Puppet.get_by_address(address)
  1812. try:
  1813. await source.sync_contact(address)
  1814. except ProfileUnavailableError:
  1815. self.log.debug(f"Profile of puppet with {address} is unavailable")
  1816. await self._try_with_puppet(
  1817. lambda i: i.invite_user(self.mxid, puppet.intent_for(self).mxid, check_cache=True),
  1818. puppet=sender,
  1819. )
  1820. remove_users.discard(puppet.default_mxid)
  1821. for mxid in remove_users:
  1822. user = await u.User.get_by_mxid(mxid, create=False)
  1823. if user and await user.is_logged_in():
  1824. await self._kick_with_puppet(user, sender)
  1825. puppet = await p.Puppet.get_by_mxid(mxid, create=False)
  1826. if puppet:
  1827. if puppet == sender:
  1828. await puppet.intent_for(self).leave_room(self.mxid)
  1829. else:
  1830. await self._kick_with_puppet(puppet, sender)
  1831. async def _kick_with_puppet(
  1832. self, user: p.Puppet | u.User, sender: p.Puppet | None = None
  1833. ) -> None:
  1834. if sender:
  1835. try:
  1836. await sender.intent_for(self).kick_user(self.mxid, user.mxid)
  1837. except (MForbidden, IntentError):
  1838. await self.main_intent.kick_user(self.mxid, user.mxid, f"kicked by {sender.name}")
  1839. else:
  1840. await self.main_intent.kick_user(
  1841. self.mxid, user.mxid, "Not a member of this Signal group"
  1842. )
  1843. async def _update_power_levels(self, info: ChatInfo) -> None:
  1844. if not self.mxid:
  1845. return
  1846. power_levels = await self.main_intent.get_power_levels(self.mxid)
  1847. power_levels = await self._get_power_levels(power_levels, info=info, is_initial=False)
  1848. await self.main_intent.set_power_levels(self.mxid, power_levels)
  1849. # endregion
  1850. # region Bridge info state event
  1851. @property
  1852. def bridge_info_state_key(self) -> str:
  1853. return f"net.maunium.signal://signal/{self.chat_id_str}"
  1854. @property
  1855. def bridge_info(self) -> dict[str, Any]:
  1856. return {
  1857. "bridgebot": self.az.bot_mxid,
  1858. "creator": self.main_intent.mxid,
  1859. "protocol": {
  1860. "id": "signal",
  1861. "displayname": "Signal",
  1862. "avatar_url": self.config["appservice.bot_avatar"],
  1863. },
  1864. "channel": {
  1865. "id": self.chat_id_str,
  1866. "displayname": self.name,
  1867. "avatar_url": self.avatar_url,
  1868. },
  1869. }
  1870. async def update_bridge_info(self) -> None:
  1871. if not self.mxid:
  1872. self.log.debug("Not updating bridge info: no Matrix room created")
  1873. return
  1874. try:
  1875. self.log.debug("Updating bridge info...")
  1876. await self.main_intent.send_state_event(
  1877. self.mxid, StateBridge, self.bridge_info, self.bridge_info_state_key
  1878. )
  1879. # TODO remove this once https://github.com/matrix-org/matrix-doc/pull/2346 is in spec
  1880. await self.main_intent.send_state_event(
  1881. self.mxid, StateHalfShotBridge, self.bridge_info, self.bridge_info_state_key
  1882. )
  1883. except Exception:
  1884. self.log.warning("Failed to update bridge info", exc_info=True)
  1885. # endregion
  1886. # region Creating Matrix rooms
  1887. async def update_matrix_room(self, source: u.User, info: ChatInfo) -> None:
  1888. if not self.is_direct and not isinstance(info, (Group, GroupV2, GroupV2ID)):
  1889. raise ValueError(f"Unexpected type for updating group portal: {type(info)}")
  1890. elif self.is_direct and not isinstance(info, (Profile, Address)):
  1891. raise ValueError(f"Unexpected type for updating direct chat portal: {type(info)}")
  1892. try:
  1893. await self._update_matrix_room(source, info)
  1894. except Exception:
  1895. self.log.exception("Failed to update portal")
  1896. async def create_matrix_room(self, source: u.User, info: ChatInfo) -> RoomID | None:
  1897. if not self.is_direct and not isinstance(info, (Group, GroupV2, GroupV2ID)):
  1898. raise ValueError(f"Unexpected type for creating group portal: {type(info)}")
  1899. elif self.is_direct and not isinstance(info, (Profile, Address)):
  1900. raise ValueError(f"Unexpected type for creating direct chat portal: {type(info)}")
  1901. if isinstance(info, Group) and not info.members:
  1902. try:
  1903. groups = await self.signal.list_groups(source.username)
  1904. except Exception as e:
  1905. await source.handle_auth_failure(e)
  1906. raise
  1907. info = next(
  1908. (g for g in groups if isinstance(g, Group) and g.group_id == info.group_id), info
  1909. )
  1910. elif isinstance(info, GroupV2ID) and not isinstance(info, GroupV2):
  1911. self.log.debug(
  1912. f"create_matrix_room() called with {info}, fetching full info from signald"
  1913. )
  1914. try:
  1915. info = await self.signal.get_group(source.username, info.id, info.revision or -1)
  1916. except Exception as e:
  1917. await source.handle_auth_failure(e)
  1918. raise
  1919. if not info:
  1920. self.log.warning(f"Full info not found, canceling room creation")
  1921. return None
  1922. else:
  1923. self.log.trace("get_group() returned full info: %s", info)
  1924. if self.mxid:
  1925. await self.update_matrix_room(source, info)
  1926. return self.mxid
  1927. async with self._create_room_lock:
  1928. return await self._create_matrix_room(source, info)
  1929. def _get_invite_content(self, double_puppet: p.Puppet | None) -> dict[str, Any]:
  1930. invite_content = {}
  1931. if double_puppet:
  1932. invite_content["fi.mau.will_auto_accept"] = True
  1933. if self.is_direct:
  1934. invite_content["is_direct"] = True
  1935. return invite_content
  1936. async def _update_matrix_room(self, source: u.User, info: ChatInfo) -> None:
  1937. puppet = await p.Puppet.get_by_custom_mxid(source.mxid)
  1938. await self.main_intent.invite_user(
  1939. self.mxid,
  1940. source.mxid,
  1941. check_cache=True,
  1942. extra_content=self._get_invite_content(puppet),
  1943. )
  1944. if puppet:
  1945. did_join = await puppet.intent.ensure_joined(self.mxid)
  1946. if did_join and self.is_direct:
  1947. await source.update_direct_chats({self.main_intent.mxid: [self.mxid]})
  1948. await self.update_info(source, info)
  1949. async def _get_power_levels(
  1950. self,
  1951. levels: PowerLevelStateEventContent | None = None,
  1952. info: ChatInfo | None = None,
  1953. is_initial: bool = False,
  1954. ) -> PowerLevelStateEventContent:
  1955. levels = levels or PowerLevelStateEventContent()
  1956. bot_pl = levels.get_user_level(self.az.bot_mxid)
  1957. levels.events_default = 0
  1958. if self.is_direct:
  1959. levels.ban = 99
  1960. levels.kick = 99
  1961. levels.invite = 99
  1962. levels.state_default = 0
  1963. meta_edit_level = 0
  1964. else:
  1965. if isinstance(info, GroupV2):
  1966. ac = info.access_control
  1967. for detail in info.member_detail + info.pending_member_detail:
  1968. puppet = await p.Puppet.get_by_address(Address(uuid=detail.uuid))
  1969. puppet_mxid = puppet.intent_for(self).mxid
  1970. current_level = levels.get_user_level(puppet_mxid)
  1971. if bot_pl > current_level and bot_pl >= 50:
  1972. level = current_level
  1973. if puppet.is_real_user:
  1974. if current_level >= 50 and detail.role == GroupMemberRole.DEFAULT:
  1975. level = 0
  1976. elif (
  1977. current_level < 50 and detail.role == GroupMemberRole.ADMINISTRATOR
  1978. ):
  1979. level = 50
  1980. else:
  1981. level = 50 if detail.role == GroupMemberRole.ADMINISTRATOR else 0
  1982. levels.users[puppet_mxid] = level
  1983. announcements = info.announcements
  1984. else:
  1985. ac = GroupAccessControl()
  1986. announcements = AnnouncementsMode.UNKNOWN
  1987. levels.ban = 50
  1988. levels.kick = 50
  1989. levels.invite = 50 if ac.members == AccessControlMode.ADMINISTRATOR else 0
  1990. levels.state_default = 50
  1991. meta_edit_level = 50 if ac.attributes == AccessControlMode.ADMINISTRATOR else 0
  1992. if announcements == AnnouncementsMode.ENABLED:
  1993. levels.events_default = 50
  1994. levels.events[EventType.REACTION] = 0
  1995. levels.events[EventType.ROOM_NAME] = meta_edit_level
  1996. levels.events[EventType.ROOM_AVATAR] = meta_edit_level
  1997. levels.events[EventType.ROOM_TOPIC] = meta_edit_level
  1998. levels.events[EventType.ROOM_ENCRYPTION] = 50 if self.matrix.e2ee else 99
  1999. levels.events[EventType.ROOM_TOMBSTONE] = 99
  2000. levels.users_default = 0
  2001. # Remote delete is only for your own messages
  2002. levels.redact = 99
  2003. if self.main_intent.mxid not in levels.users:
  2004. levels.users[self.main_intent.mxid] = 9001 if is_initial else 100
  2005. return levels
  2006. async def _create_matrix_room(self, source: u.User, info: ChatInfo) -> RoomID | None:
  2007. if self.mxid:
  2008. await self._update_matrix_room(source, info)
  2009. return self.mxid
  2010. await self.update_info(source, info)
  2011. self.log.debug("Creating Matrix room")
  2012. name: str | None = None
  2013. power_levels = await self._get_power_levels(info=info, is_initial=True)
  2014. initial_state = [
  2015. {
  2016. "type": str(StateBridge),
  2017. "state_key": self.bridge_info_state_key,
  2018. "content": self.bridge_info,
  2019. },
  2020. {
  2021. # TODO remove this once https://github.com/matrix-org/matrix-doc/pull/2346 is in spec
  2022. "type": str(StateHalfShotBridge),
  2023. "state_key": self.bridge_info_state_key,
  2024. "content": self.bridge_info,
  2025. },
  2026. {
  2027. "type": str(EventType.ROOM_POWER_LEVELS),
  2028. "content": power_levels.serialize(),
  2029. },
  2030. ]
  2031. invites = []
  2032. if self.config["bridge.encryption.default"] and self.matrix.e2ee:
  2033. self.encrypted = True
  2034. initial_state.append(
  2035. {
  2036. "type": str(EventType.ROOM_ENCRYPTION),
  2037. "content": self.get_encryption_state_event_json(),
  2038. }
  2039. )
  2040. if self.is_direct:
  2041. invites.append(self.az.bot_mxid)
  2042. if self.is_direct and source.address == self.chat_id:
  2043. name = self.name = "Signal Note to Self"
  2044. elif self.encrypted or self.private_chat_portal_meta or not self.is_direct:
  2045. name = self.name
  2046. if self.avatar_url:
  2047. initial_state.append(
  2048. {
  2049. "type": str(EventType.ROOM_AVATAR),
  2050. "content": {"url": self.avatar_url},
  2051. }
  2052. )
  2053. creation_content = {}
  2054. if not self.config["bridge.federate_rooms"]:
  2055. creation_content["m.federate"] = False
  2056. self.mxid = await self.main_intent.create_room(
  2057. name=name,
  2058. topic=self.topic,
  2059. is_direct=self.is_direct,
  2060. initial_state=initial_state,
  2061. invitees=invites,
  2062. creation_content=creation_content,
  2063. # Make sure the power level event in initial_state is allowed
  2064. # even if the server sends a default power level event before it.
  2065. # TODO remove this if the spec is changed to require servers to
  2066. # use the power level event in initial_state
  2067. power_level_override={"users": {self.main_intent.mxid: 9001}},
  2068. )
  2069. if not self.mxid:
  2070. raise Exception("Failed to create room: no mxid returned")
  2071. self.name_set = bool(name)
  2072. self.avatar_set = bool(self.avatar_url)
  2073. if self.encrypted and self.matrix.e2ee and self.is_direct:
  2074. try:
  2075. await self.az.intent.ensure_joined(self.mxid)
  2076. except Exception:
  2077. self.log.warning("Failed to add bridge bot to new private chat {self.mxid}")
  2078. puppet = await p.Puppet.get_by_custom_mxid(source.mxid)
  2079. await self.main_intent.invite_user(
  2080. self.mxid, source.mxid, extra_content=self._get_invite_content(puppet)
  2081. )
  2082. if puppet:
  2083. try:
  2084. await source.update_direct_chats({self.main_intent.mxid: [self.mxid]})
  2085. await puppet.intent.join_room_by_id(self.mxid)
  2086. except MatrixError:
  2087. self.log.debug(
  2088. "Failed to join custom puppet into newly created portal", exc_info=True
  2089. )
  2090. await self.update()
  2091. self.log.debug(f"Matrix room created: {self.mxid}")
  2092. self.by_mxid[self.mxid] = self
  2093. if not self.is_direct:
  2094. await self._update_participants(source, info)
  2095. return self.mxid
  2096. # endregion
  2097. # region Database getters
  2098. async def _postinit(self) -> None:
  2099. self.by_chat_id[(self.chat_id_str, self.receiver)] = self
  2100. if self.mxid:
  2101. self.by_mxid[self.mxid] = self
  2102. if self.is_direct:
  2103. puppet = await self.get_dm_puppet()
  2104. self._main_intent = puppet.default_mxid_intent
  2105. elif not self.is_direct:
  2106. self._main_intent = self.az.intent
  2107. async def delete(self) -> None:
  2108. await DBMessage.delete_all(self.mxid)
  2109. self.by_mxid.pop(self.mxid, None)
  2110. self.mxid = None
  2111. self.name_set = False
  2112. self.avatar_set = False
  2113. self.relay_user_id = None
  2114. self.topic = None
  2115. self.encrypted = False
  2116. await self.update()
  2117. async def save(self) -> None:
  2118. await self.update()
  2119. @classmethod
  2120. def all_with_room(cls) -> AsyncGenerator[Portal, None]:
  2121. return cls._db_to_portals(super().all_with_room())
  2122. @classmethod
  2123. def find_private_chats_with(cls, other_user: Address) -> AsyncGenerator[Portal, None]:
  2124. return cls._db_to_portals(super().find_private_chats_with(other_user))
  2125. @classmethod
  2126. async def _db_to_portals(cls, query: Awaitable[list[Portal]]) -> AsyncGenerator[Portal, None]:
  2127. portals = await query
  2128. for index, portal in enumerate(portals):
  2129. try:
  2130. yield cls.by_chat_id[(portal.chat_id_str, portal.receiver)]
  2131. except KeyError:
  2132. await portal._postinit()
  2133. yield portal
  2134. @classmethod
  2135. @async_getter_lock
  2136. async def get_by_mxid(cls, mxid: RoomID) -> Portal | None:
  2137. try:
  2138. return cls.by_mxid[mxid]
  2139. except KeyError:
  2140. pass
  2141. portal = cast(cls, await super().get_by_mxid(mxid))
  2142. if portal is not None:
  2143. await portal._postinit()
  2144. return portal
  2145. return None
  2146. @classmethod
  2147. async def get_by_chat_id(
  2148. cls, chat_id: GroupID | Address, *, receiver: str = "", create: bool = False
  2149. ) -> Portal | None:
  2150. if isinstance(chat_id, str):
  2151. receiver = ""
  2152. elif not isinstance(chat_id, Address):
  2153. raise ValueError(f"Invalid chat ID type {type(chat_id)}")
  2154. elif not receiver:
  2155. raise ValueError("Direct chats must have a receiver")
  2156. best_id = id_to_str(chat_id)
  2157. portal = await cls._get_by_chat_id(best_id, receiver, create=create, chat_id=chat_id)
  2158. if portal:
  2159. portal.log.debug(f"get_by_chat_id({chat_id}, {receiver}) -> {hex(id(portal))}")
  2160. return portal
  2161. @classmethod
  2162. @async_getter_lock
  2163. async def _get_by_chat_id(
  2164. cls, best_id: str, receiver: str, *, create: bool, chat_id: GroupID | Address
  2165. ) -> Portal | None:
  2166. try:
  2167. return cls.by_chat_id[(best_id, receiver)]
  2168. except KeyError:
  2169. pass
  2170. portal = cast(cls, await super().get_by_chat_id(chat_id, receiver))
  2171. if portal is not None:
  2172. await portal._postinit()
  2173. return portal
  2174. if create:
  2175. portal = cls(chat_id, receiver)
  2176. await portal.insert()
  2177. await portal._postinit()
  2178. return portal
  2179. return None
  2180. # endregion