portal.py 69 KB

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