portal.py 83 KB

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