portal.py 66 KB

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