portal.py 61 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380
  1. # mautrix-signal - A Matrix-Signal puppeting bridge
  2. # Copyright (C) 2021 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 typing import (Dict, Tuple, Optional, List, Deque, Any, Union, AsyncGenerator, Awaitable, Set,
  17. Callable, TYPE_CHECKING, cast)
  18. from html import escape as escape_html
  19. from collections import deque
  20. from uuid import UUID, uuid4
  21. from string import Template
  22. from datetime import datetime
  23. import mimetypes
  24. import pathlib
  25. import hashlib
  26. import asyncio
  27. import os.path
  28. import time
  29. import os
  30. from mausignald.types import (Address, MessageData, Reaction, Quote, Group, Contact, Profile,
  31. Attachment, GroupID, GroupV2ID, GroupV2, Mention, Sticker,
  32. GroupAccessControl, AccessControlMode, GroupMemberRole)
  33. from mausignald.errors import RPCError, ResponseError
  34. from mautrix.appservice import AppService, IntentAPI
  35. from mautrix.bridge import BasePortal, async_getter_lock
  36. from mautrix.types import (EventID, MessageEventContent, RoomID, EventType, MessageType, Format,
  37. MessageEvent, EncryptedEvent, ContentURI, MediaMessageEventContent,
  38. TextMessageEventContent, ImageInfo, VideoInfo, FileInfo, AudioInfo,
  39. PowerLevelStateEventContent, UserID, SingleReceiptEventContent)
  40. from mautrix.util.format_duration import format_duration
  41. from mautrix.util.bridge_state import BridgeStateEvent
  42. from mautrix.util.message_send_checkpoint import MessageSendCheckpointStatus
  43. from mautrix.errors import MatrixError, MForbidden, IntentError
  44. from .db import (Portal as DBPortal, Message as DBMessage, Reaction as DBReaction,
  45. DisappearingMessage)
  46. from .config import Config
  47. from .formatter import matrix_to_signal, signal_to_matrix
  48. from .util import id_to_str
  49. from . import user as u, puppet as p, matrix as m, signal as s
  50. if TYPE_CHECKING:
  51. from .__main__ import SignalBridge
  52. try:
  53. from mautrix.crypto.attachments import encrypt_attachment, decrypt_attachment
  54. except ImportError:
  55. encrypt_attachment = decrypt_attachment = None
  56. try:
  57. from signalstickers_client import StickersClient
  58. from signalstickers_client.models import StickerPack
  59. except ImportError:
  60. StickersClient = StickerPack = None
  61. try:
  62. import magic
  63. except ImportError:
  64. magic = None
  65. StateBridge = EventType.find("m.bridge", EventType.Class.STATE)
  66. StateHalfShotBridge = EventType.find("uk.half-shot.bridge", EventType.Class.STATE)
  67. ChatInfo = Union[Group, GroupV2, GroupV2ID, Contact, Profile, Address]
  68. class Portal(DBPortal, BasePortal):
  69. by_mxid: Dict[RoomID, 'Portal'] = {}
  70. by_chat_id: Dict[Tuple[str, str], 'Portal'] = {}
  71. _sticker_meta_cache: Dict[str, StickerPack] = {}
  72. config: Config
  73. matrix: 'm.MatrixHandler'
  74. signal: 's.SignalHandler'
  75. az: AppService
  76. private_chat_portal_meta: bool
  77. expiration_time: Optional[int]
  78. _main_intent: Optional[IntentAPI]
  79. _create_room_lock: asyncio.Lock
  80. _msgts_dedup: Deque[Tuple[Address, int]]
  81. _reaction_dedup: Deque[Tuple[Address, int, str]]
  82. _reaction_lock: asyncio.Lock
  83. _pending_members: Optional[Set[UUID]]
  84. _relay_user: Optional['u.User']
  85. _expiration_lock: asyncio.Lock
  86. def __init__(self, chat_id: Union[GroupID, Address], receiver: str,
  87. mxid: Optional[RoomID] = None, name: Optional[str] = None,
  88. avatar_hash: Optional[str] = None, avatar_url: Optional[ContentURI] = None,
  89. name_set: bool = False, avatar_set: bool = False, revision: int = 0,
  90. encrypted: bool = False, relay_user_id: Optional[UserID] = None,
  91. expiration_time: Optional[int] = None) -> None:
  92. super().__init__(chat_id, receiver, mxid, name, avatar_hash, avatar_url,
  93. name_set, avatar_set, revision, encrypted, relay_user_id, expiration_time)
  94. self._create_room_lock = asyncio.Lock()
  95. self.log = self.log.getChild(self.chat_id_str)
  96. self._main_intent = None
  97. self._msgts_dedup = deque(maxlen=100)
  98. self._reaction_dedup = deque(maxlen=100)
  99. self._last_participant_update = set()
  100. self._reaction_lock = asyncio.Lock()
  101. self._pending_members = None
  102. self._relay_user = None
  103. self._expiration_lock = asyncio.Lock()
  104. @property
  105. def has_relay(self) -> bool:
  106. return self.config["bridge.relay.enabled"] and bool(self.relay_user_id)
  107. async def get_relay_user(self) -> Optional['u.User']:
  108. if not self.has_relay:
  109. return None
  110. if self._relay_user is None:
  111. self._relay_user = await u.User.get_by_mxid(self.relay_user_id)
  112. return self._relay_user if await self._relay_user.is_logged_in() else None
  113. async def set_relay_user(self, user: Optional['u.User']) -> None:
  114. self._relay_user = user
  115. self.relay_user_id = user.mxid if user else None
  116. await self.save()
  117. @property
  118. def main_intent(self) -> IntentAPI:
  119. if not self._main_intent:
  120. raise ValueError("Portal must be postinit()ed before main_intent can be used")
  121. return self._main_intent
  122. @property
  123. def is_direct(self) -> bool:
  124. return isinstance(self.chat_id, Address)
  125. def handle_uuid_receive(self, uuid: UUID) -> None:
  126. if not self.is_direct or self.chat_id.uuid:
  127. raise ValueError("handle_uuid_receive can only be used for private chat portals with "
  128. "a phone number chat_id")
  129. del self.by_chat_id[(self.chat_id_str, self.receiver)]
  130. self.chat_id = Address(uuid=uuid)
  131. self.by_chat_id[(self.chat_id_str, self.receiver)] = self
  132. @classmethod
  133. async def init_cls(cls, bridge: 'SignalBridge') -> None:
  134. cls.config = bridge.config
  135. cls.matrix = bridge.matrix
  136. cls.signal = bridge.signal
  137. cls.az = bridge.az
  138. cls.loop = bridge.loop
  139. BasePortal.bridge = bridge
  140. cls.private_chat_portal_meta = cls.config["bridge.private_chat_portal_meta"]
  141. for dm in await DisappearingMessage.get_all():
  142. if dm.expiration_ts:
  143. asyncio.create_task(cls._expire_event(dm))
  144. # region Misc
  145. async def _send_delivery_receipt(self, event_id: EventID) -> None:
  146. if event_id and self.config["bridge.delivery_receipts"]:
  147. try:
  148. await self.az.intent.mark_read(self.mxid, event_id)
  149. except Exception:
  150. self.log.exception("Failed to send delivery receipt for %s", event_id)
  151. async def _upsert_reaction(self, existing: DBReaction, intent: IntentAPI, mxid: EventID,
  152. sender: Union['p.Puppet', 'u.User'], message: DBMessage, emoji: str
  153. ) -> None:
  154. if existing:
  155. self.log.debug(f"_upsert_reaction redacting {existing.mxid} and inserting {mxid}"
  156. f" (message: {message.mxid})")
  157. try:
  158. await intent.redact(existing.mx_room, existing.mxid)
  159. except MForbidden:
  160. self.log.debug("Unexpected MForbidden redacting reaction", exc_info=True)
  161. await existing.edit(emoji=emoji, mxid=mxid, mx_room=message.mx_room)
  162. else:
  163. self.log.debug(f"_upsert_reaction inserting {mxid} (message: {message.mxid})")
  164. await DBReaction(mxid=mxid, mx_room=message.mx_room, emoji=emoji,
  165. signal_chat_id=self.chat_id, signal_receiver=self.receiver,
  166. msg_author=message.sender, msg_timestamp=message.timestamp,
  167. author=sender.address).insert()
  168. # endregion
  169. # region Matrix event handling
  170. @staticmethod
  171. def _make_attachment(message: MediaMessageEventContent, path: str) -> Attachment:
  172. attachment = Attachment(custom_filename=message.body, content_type=message.info.mimetype,
  173. outgoing_filename=path)
  174. info = message.info
  175. attachment.width = info.get("w", info.get("width", 0))
  176. attachment.height = info.get("h", info.get("height", 0))
  177. attachment.voice_note = message.msgtype == MessageType.AUDIO
  178. return attachment
  179. def _write_outgoing_file(self, data: bytes) -> str:
  180. dir = pathlib.Path(self.config["signal.outgoing_attachment_dir"])
  181. path = dir.joinpath(f"mautrix-signal-{str(uuid4())}")
  182. try:
  183. with open(path, "wb") as file:
  184. file.write(data)
  185. except FileNotFoundError:
  186. dir.mkdir(mode=0o755, parents=True, exist_ok=True)
  187. with open(path, "wb") as file:
  188. file.write(data)
  189. return str(path)
  190. async def _download_matrix_media(self, message: MediaMessageEventContent) -> str:
  191. if message.file:
  192. data = await self.main_intent.download_media(message.file.url)
  193. data = decrypt_attachment(data, message.file.key.key,
  194. message.file.hashes.get("sha256"), message.file.iv)
  195. else:
  196. data = await self.main_intent.download_media(message.url)
  197. return self._write_outgoing_file(data)
  198. async def get_displayname(self, user: 'u.User') -> str:
  199. return await self.main_intent.get_room_displayname(self.mxid, user.mxid) or user.mxid
  200. async def _apply_msg_format(self, sender: 'u.User', content: MessageEventContent) -> None:
  201. if not isinstance(content, TextMessageEventContent) or content.format != Format.HTML:
  202. content.format = Format.HTML
  203. content.formatted_body = escape_html(content.body).replace("\n", "<br/>")
  204. tpl = (self.config[f"relaybot.message_formats.[{content.msgtype.value}]"]
  205. or "$sender_displayname: $message")
  206. displayname = await self.get_displayname(sender)
  207. username, _ = self.az.intent.parse_user_id(sender.mxid)
  208. tpl_args = dict(sender_mxid=sender.mxid,
  209. sender_username=username,
  210. sender_displayname=escape_html(displayname),
  211. message=content.formatted_body,
  212. body=content.body,
  213. formatted_body=content.formatted_body)
  214. content.formatted_body = Template(tpl).safe_substitute(tpl_args)
  215. content.body = Template(tpl).safe_substitute(tpl_args)
  216. if content.msgtype == MessageType.EMOTE:
  217. content.msgtype = MessageType.TEXT
  218. async def _get_relay_sender(self, sender: 'u.User', evt_identifier: str
  219. ) -> Tuple[Optional['u.User'], bool]:
  220. if await sender.is_logged_in():
  221. return sender, False
  222. if not self.has_relay:
  223. self.log.debug(f"Ignoring {evt_identifier} from non-logged-in user {sender.mxid}"
  224. " in chat with no relay user")
  225. return None, True
  226. relay_sender = await self.get_relay_user()
  227. if not relay_sender:
  228. self.log.debug(f"Ignoring {evt_identifier} from non-logged-in user {sender.mxid}: "
  229. f"relay user {self.relay_user_id} is not set up correctly")
  230. return None, True
  231. return relay_sender, True
  232. async def handle_matrix_message(self, sender: 'u.User', message: MessageEventContent,
  233. event_id: EventID) -> None:
  234. orig_sender = sender
  235. sender, is_relay = await self._get_relay_sender(sender, f"message {event_id}")
  236. if not sender:
  237. return
  238. elif is_relay:
  239. await self._apply_msg_format(orig_sender, message)
  240. request_id = int(time.time() * 1000)
  241. self._msgts_dedup.appendleft((sender.address, request_id))
  242. quote = None
  243. if message.get_reply_to():
  244. reply = await DBMessage.get_by_mxid(message.get_reply_to(), self.mxid)
  245. # TODO include actual text? either store in db or fetch event from homeserver
  246. if reply is not None:
  247. quote = Quote(id=reply.timestamp, author=reply.sender, text="")
  248. attachments: Optional[List[Attachment]] = None
  249. attachment_path: Optional[str] = None
  250. mentions: Optional[List[Mention]] = None
  251. if message.msgtype.is_text:
  252. text, mentions = await matrix_to_signal(message)
  253. elif message.msgtype.is_media:
  254. attachment_path = await self._download_matrix_media(message)
  255. attachment = self._make_attachment(message, attachment_path)
  256. attachments = [attachment]
  257. text = message.body if is_relay else None
  258. self.log.trace("Formed outgoing attachment %s", attachment)
  259. else:
  260. self.log.debug(f"Unknown msgtype {message.msgtype} in Matrix message {event_id}")
  261. return
  262. self.log.debug(f"Sending Matrix message {event_id} to Signal with timestamp {request_id}")
  263. try:
  264. await self.signal.send(username=sender.username, recipient=self.chat_id, body=text,
  265. mentions=mentions, quote=quote, attachments=attachments,
  266. timestamp=request_id)
  267. except Exception as e:
  268. sender.send_remote_checkpoint(
  269. MessageSendCheckpointStatus.PERM_FAILURE,
  270. event_id,
  271. self.mxid,
  272. EventType.ROOM_MESSAGE,
  273. message.msgtype,
  274. error=e,
  275. )
  276. auth_failed = (
  277. "org.whispersystems.signalservice.api.push.exceptions.AuthorizationFailedException"
  278. )
  279. if isinstance(e, ResponseError) and auth_failed in e.data.get("exceptions", []):
  280. await sender.push_bridge_state(BridgeStateEvent.BAD_CREDENTIALS, error=str(e))
  281. await self._send_message(
  282. self.main_intent,
  283. TextMessageEventContent(
  284. msgtype=MessageType.NOTICE,
  285. body=f"\u26a0 Your message was not bridged: {e}"))
  286. else:
  287. sender.send_remote_checkpoint(
  288. MessageSendCheckpointStatus.SUCCESS,
  289. event_id,
  290. self.mxid,
  291. EventType.ROOM_MESSAGE,
  292. message.msgtype,
  293. )
  294. await self._send_delivery_receipt(event_id)
  295. msg = DBMessage(mxid=event_id, mx_room=self.mxid, sender=sender.address,
  296. timestamp=request_id,
  297. signal_chat_id=self.chat_id, signal_receiver=self.receiver)
  298. await msg.insert()
  299. self.log.debug(f"Handled Matrix message {event_id} -> {request_id}")
  300. if attachment_path and self.config["signal.remove_file_after_handling"]:
  301. try:
  302. os.remove(attachment_path)
  303. except FileNotFoundError:
  304. pass
  305. # Handle disappearing messages
  306. if self.expiration_time:
  307. disappearing_message = DisappearingMessage(
  308. self.mxid, event_id, self.expiration_time
  309. )
  310. await disappearing_message.insert()
  311. asyncio.create_task(Portal._expire_event(disappearing_message))
  312. async def handle_matrix_reaction(self, sender: 'u.User', event_id: EventID,
  313. reacting_to: EventID, emoji: str) -> None:
  314. if not await sender.is_logged_in():
  315. self.log.trace(f"Ignoring reaction by non-logged-in user {sender.mxid}")
  316. return
  317. # Signal doesn't seem to use variation selectors at all
  318. emoji = emoji.rstrip("\ufe0f")
  319. message = await DBMessage.get_by_mxid(reacting_to, self.mxid)
  320. if not message:
  321. self.log.debug(f"Ignoring reaction to unknown event {reacting_to}")
  322. return
  323. existing = await DBReaction.get_by_signal_id(self.chat_id, self.receiver, message.sender,
  324. message.timestamp, sender.address)
  325. if existing and existing.emoji == emoji:
  326. return
  327. dedup_id = (message.sender, message.timestamp, emoji)
  328. self._reaction_dedup.appendleft(dedup_id)
  329. async with self._reaction_lock:
  330. reaction = Reaction(emoji=emoji, remove=False,
  331. target_author=message.sender,
  332. target_sent_timestamp=message.timestamp)
  333. try:
  334. await self.signal.react(username=sender.username, recipient=self.chat_id,
  335. reaction=reaction)
  336. except Exception as e:
  337. sender.send_remote_checkpoint(
  338. MessageSendCheckpointStatus.PERM_FAILURE,
  339. event_id,
  340. self.mxid,
  341. EventType.REACTION,
  342. error=e,
  343. )
  344. else:
  345. self.log.trace(f"{sender.mxid} reacted to {message.timestamp} with {emoji}")
  346. sender.send_remote_checkpoint(
  347. MessageSendCheckpointStatus.SUCCESS,
  348. event_id,
  349. self.mxid,
  350. EventType.REACTION,
  351. )
  352. await self._upsert_reaction(existing, self.main_intent, event_id, sender, message,
  353. emoji)
  354. await self._send_delivery_receipt(event_id)
  355. async def handle_matrix_redaction(self, sender: 'u.User', event_id: EventID,
  356. redaction_event_id: EventID) -> None:
  357. if not await sender.is_logged_in():
  358. return
  359. message = await DBMessage.get_by_mxid(event_id, self.mxid)
  360. if message:
  361. try:
  362. await message.delete()
  363. await self.signal.remote_delete(sender.username, recipient=self.chat_id,
  364. timestamp=message.timestamp)
  365. except Exception as e:
  366. self.log.exception("Removing message failed")
  367. sender.send_remote_checkpoint(
  368. MessageSendCheckpointStatus.PERM_FAILURE,
  369. redaction_event_id,
  370. self.mxid,
  371. EventType.ROOM_REDACTION,
  372. error=e,
  373. )
  374. else:
  375. self.log.trace(f"Removed {message} after Matrix redaction")
  376. sender.send_remote_checkpoint(
  377. MessageSendCheckpointStatus.SUCCESS,
  378. redaction_event_id,
  379. self.mxid,
  380. EventType.ROOM_REDACTION,
  381. )
  382. await self._send_delivery_receipt(redaction_event_id)
  383. return
  384. reaction = await DBReaction.get_by_mxid(event_id, self.mxid)
  385. if reaction:
  386. try:
  387. await reaction.delete()
  388. remove_reaction = Reaction(emoji=reaction.emoji, remove=True,
  389. target_author=reaction.msg_author,
  390. target_sent_timestamp=reaction.msg_timestamp)
  391. await self.signal.react(username=sender.username, recipient=self.chat_id,
  392. reaction=remove_reaction)
  393. except Exception as e:
  394. self.log.exception("Removing reaction failed")
  395. sender.send_remote_checkpoint(
  396. MessageSendCheckpointStatus.PERM_FAILURE,
  397. redaction_event_id,
  398. self.mxid,
  399. EventType.ROOM_REDACTION,
  400. error=e,
  401. )
  402. else:
  403. self.log.trace(f"Removed {reaction} after Matrix redaction")
  404. sender.send_remote_checkpoint(
  405. MessageSendCheckpointStatus.SUCCESS,
  406. redaction_event_id,
  407. self.mxid,
  408. EventType.ROOM_REDACTION,
  409. )
  410. await self._send_delivery_receipt(redaction_event_id)
  411. return
  412. sender.send_remote_checkpoint(
  413. MessageSendCheckpointStatus.PERM_FAILURE,
  414. redaction_event_id,
  415. self.mxid,
  416. EventType.ROOM_REDACTION,
  417. error=f"No message or reaction found for redaction",
  418. )
  419. async def handle_matrix_join(self, user: 'u.User') -> None:
  420. if self.is_direct or not await user.is_logged_in():
  421. return
  422. if self._pending_members is None:
  423. self.log.debug(f"{user.mxid} ({user.uuid}) joined room, but pending_members is None,"
  424. " updating chat info")
  425. await self.update_info(user, GroupV2ID(id=self.chat_id))
  426. if self._pending_members is None:
  427. self.log.warning(f"Didn't get pending member list after info update, "
  428. f"{user.mxid} ({user.uuid}) may not be in the group on Signal.")
  429. elif user.uuid in self._pending_members:
  430. self.log.debug(f"{user.mxid} ({user.uuid}) joined room, accepting invite on Signal")
  431. try:
  432. resp = await self.signal.accept_invitation(user.username, self.chat_id)
  433. self._pending_members.remove(user.uuid)
  434. except RPCError as e:
  435. await self.main_intent.send_notice(self.mxid, "\u26a0 Failed to accept invite "
  436. f"on Signal: {e}")
  437. else:
  438. await self.update_info(user, resp)
  439. async def handle_matrix_leave(self, user: 'u.User') -> None:
  440. if not await user.is_logged_in():
  441. return
  442. if self.is_direct:
  443. self.log.info(f"{user.mxid} left private chat portal with {self.chat_id}")
  444. if user.username == self.receiver:
  445. self.log.info(f"{user.mxid} was the recipient of this portal. "
  446. "Cleaning up and deleting...")
  447. await self.cleanup_and_delete()
  448. else:
  449. self.log.debug(f"{user.mxid} left portal to {self.chat_id}")
  450. # TODO cleanup if empty
  451. async def handle_matrix_name(self, user: 'u.User', name: str) -> None:
  452. if self.name == name or self.is_direct or not name:
  453. return
  454. sender, is_relay = await self._get_relay_sender(user, "name change")
  455. if not sender:
  456. return
  457. self.name = name
  458. self.log.debug(f"{user.mxid} changed the group name, "
  459. f"sending to Signal through {sender.username}")
  460. try:
  461. await self.signal.update_group(sender.username, self.chat_id, title=name)
  462. except Exception:
  463. self.log.exception("Failed to update Signal group name")
  464. self.name = None
  465. async def handle_matrix_avatar(self, user: 'u.User', url: ContentURI) -> None:
  466. if self.is_direct or not url:
  467. return
  468. sender, is_relay = await self._get_relay_sender(user, "avatar change")
  469. if not sender:
  470. return
  471. data = await self.main_intent.download_media(url)
  472. new_hash = hashlib.sha256(data).hexdigest()
  473. if new_hash == self.avatar_hash and self.avatar_set:
  474. self.log.debug(f"New avatar from Matrix set by {user.mxid} is same as current one")
  475. return
  476. self.avatar_url = url
  477. self.avatar_hash = new_hash
  478. path = self._write_outgoing_file(data)
  479. self.log.debug(f"{user.mxid} changed the group avatar, "
  480. f"sending to Signal through {sender.username}")
  481. try:
  482. await self.signal.update_group(sender.username, self.chat_id, avatar_path=path)
  483. self.avatar_set = True
  484. except Exception:
  485. self.log.exception("Failed to update Signal group avatar")
  486. self.avatar_set = False
  487. if self.config["signal.remove_file_after_handling"]:
  488. try:
  489. os.remove(path)
  490. except FileNotFoundError:
  491. pass
  492. @classmethod
  493. async def _expire_event(cls, disappearing_message: DisappearingMessage):
  494. """
  495. Expire a :class:`DisappearingMessage`. This should only be called once the message has been
  496. read, as the timer for redaction will start immediately, and there is no (supported)
  497. mechanism to stop the countdown, even after bridge restart.
  498. """
  499. room_id = disappearing_message.room_id
  500. event_id = disappearing_message.mxid
  501. wait = disappearing_message.expiration_seconds
  502. # If there is an expiration_ts, then there was probably a bridge restart, so we have to
  503. # resume the countdown. This is fairly likely to occur if the disappearance timeout is
  504. # weeks.
  505. # If there is not an expiration_ts, then set one.
  506. now = datetime.now().timestamp()
  507. if disappearing_message.expiration_ts is not None:
  508. wait = (disappearing_message.expiration_ts / 1000) - now
  509. else:
  510. disappearing_message.expiration_ts = int((now + wait) * 1000)
  511. await disappearing_message.update()
  512. if wait < 0:
  513. wait = 0
  514. portal = await cls.get_by_mxid(room_id)
  515. if not portal:
  516. portal.log.warning(f"No portal for {room_id}")
  517. portal.log.debug(f"Redacting {event_id} in {wait} seconds")
  518. await asyncio.sleep(wait)
  519. async with portal._expiration_lock:
  520. if not await DisappearingMessage.get(room_id, event_id):
  521. portal.log.debug(
  522. f"{event_id} no longer in disappearing messages list, not redacting"
  523. )
  524. return
  525. portal.log.debug(f"Redacting {event_id} because it was expired")
  526. try:
  527. await portal.main_intent.redact(room_id, event_id)
  528. portal.log.debug(f"Redacted {event_id} successfully")
  529. except Exception as e:
  530. portal.log.warning("Redacting expired event didn't work", e)
  531. finally:
  532. await DisappearingMessage.delete(room_id, event_id)
  533. async def handle_read_receipt(self, event_id: EventID, data: SingleReceiptEventContent):
  534. # Start the redaction timers for all of the disappearing messages in the room when the user
  535. # reads the room. This is the behavior of the Signal clients.
  536. for disappearing_message in await DisappearingMessage.get_all_for_room(self.mxid):
  537. asyncio.create_task(Portal._expire_event(disappearing_message))
  538. # endregion
  539. # region Signal event handling
  540. @staticmethod
  541. async def _resolve_address(address: Address) -> Address:
  542. puppet = await p.Puppet.get_by_address(address, create=False)
  543. return puppet.address
  544. async def _find_quote_event_id(self, quote: Optional[Quote]
  545. ) -> Optional[Union[MessageEvent, EventID]]:
  546. if not quote:
  547. return None
  548. author_address = await self._resolve_address(quote.author)
  549. reply_msg = await DBMessage.get_by_signal_id(author_address, quote.id,
  550. self.chat_id, self.receiver)
  551. if not reply_msg:
  552. return None
  553. try:
  554. evt = await self.main_intent.get_event(self.mxid, reply_msg.mxid)
  555. if isinstance(evt, EncryptedEvent):
  556. return await self.matrix.e2ee.decrypt(evt, wait_session_timeout=0)
  557. return evt
  558. except MatrixError:
  559. return reply_msg.mxid
  560. async def handle_signal_message(self, source: 'u.User', sender: 'p.Puppet',
  561. message: MessageData) -> None:
  562. if (sender.address, message.timestamp) in self._msgts_dedup:
  563. self.log.debug(f"Ignoring message {message.timestamp} by {sender.uuid}"
  564. " as it was already handled (message.timestamp in dedup queue)")
  565. await self.signal.send_receipt(source.username, sender.address,
  566. timestamps=[message.timestamp])
  567. return
  568. old_message = await DBMessage.get_by_signal_id(sender.address, message.timestamp,
  569. self.chat_id, self.receiver)
  570. if old_message is not None:
  571. self.log.debug(f"Ignoring message {message.timestamp} by {sender.uuid}"
  572. " as it was already handled (message.id found in database)")
  573. await self.signal.send_receipt(source.username, sender.address,
  574. timestamps=[message.timestamp])
  575. return
  576. self.log.debug(f"Started handling message {message.timestamp} by {sender.uuid}")
  577. self.log.trace(f"Message content: {message}")
  578. self._msgts_dedup.appendleft((sender.address, message.timestamp))
  579. intent = sender.intent_for(self)
  580. await intent.set_typing(self.mxid, False)
  581. event_id = None
  582. reply_to = await self._find_quote_event_id(message.quote)
  583. if message.sticker:
  584. if message.sticker.attachment.incoming_filename:
  585. content = await self._handle_signal_attachment(intent, message.sticker.attachment,
  586. sticker=True)
  587. elif StickersClient:
  588. content = await self._handle_signal_sticker(intent, message.sticker)
  589. else:
  590. self.log.debug(f"Not handling sticker in {message.timestamp}: no incoming_filename"
  591. " and signalstickers-client not installed.")
  592. return
  593. if content:
  594. if message.sticker.attachment.blurhash:
  595. content.info["blurhash"] = message.sticker.attachment.blurhash
  596. content.info["xyz.amorgan.blurhash"] = message.sticker.attachment.blurhash
  597. await self._add_sticker_meta(message.sticker, content)
  598. if reply_to and not message.body:
  599. content.set_reply(reply_to)
  600. reply_to = None
  601. event_id = await self._send_message(intent, content, timestamp=message.timestamp,
  602. event_type=EventType.STICKER)
  603. for attachment in message.attachments:
  604. if not attachment.incoming_filename:
  605. self.log.warning("Failed to bridge attachment, no incoming filename: %s",
  606. attachment)
  607. continue
  608. content = await self._handle_signal_attachment(intent, attachment)
  609. if reply_to and not message.body:
  610. # If there's no text, set the first image as the reply
  611. content.set_reply(reply_to)
  612. reply_to = None
  613. event_id = await self._send_message(intent, content, timestamp=message.timestamp)
  614. if message.body:
  615. content = await signal_to_matrix(message)
  616. if reply_to:
  617. content.set_reply(reply_to)
  618. event_id = await self._send_message(intent, content, timestamp=message.timestamp)
  619. if event_id:
  620. msg = DBMessage(mxid=event_id, mx_room=self.mxid,
  621. sender=sender.address, timestamp=message.timestamp,
  622. signal_chat_id=self.chat_id, signal_receiver=self.receiver)
  623. await msg.insert()
  624. await self.signal.send_receipt(source.username, sender.address,
  625. timestamps=[message.timestamp])
  626. await self._send_delivery_receipt(event_id)
  627. self.log.debug(f"Handled Signal message {message.timestamp} -> {event_id}")
  628. if message.expires_in_seconds:
  629. disappearing_message = DisappearingMessage(
  630. self.mxid, event_id, message.expires_in_seconds
  631. )
  632. await disappearing_message.insert()
  633. self.log.debug(
  634. f"{event_id} set to be redacted {message.expires_in_seconds} seconds after "
  635. "room is read"
  636. )
  637. else:
  638. self.log.debug(f"Didn't get event ID for {message.timestamp}")
  639. @staticmethod
  640. def _make_media_content(attachment: Attachment) -> MediaMessageEventContent:
  641. if attachment.content_type.startswith("image/"):
  642. msgtype = MessageType.IMAGE
  643. info = ImageInfo(mimetype=attachment.content_type,
  644. width=attachment.width, height=attachment.height)
  645. elif attachment.content_type.startswith("video/"):
  646. msgtype = MessageType.VIDEO
  647. info = VideoInfo(mimetype=attachment.content_type,
  648. width=attachment.width, height=attachment.height)
  649. elif attachment.voice_note or attachment.content_type.startswith("audio/"):
  650. msgtype = MessageType.AUDIO
  651. info = AudioInfo(mimetype=attachment.content_type)
  652. else:
  653. msgtype = MessageType.FILE
  654. info = FileInfo(mimetype=attachment.content_type)
  655. if not attachment.custom_filename:
  656. ext = mimetypes.guess_extension(attachment.content_type) or ""
  657. attachment.custom_filename = attachment.id + ext
  658. if attachment.blurhash:
  659. info["blurhash"] = attachment.blurhash
  660. info["xyz.amorgan.blurhash"] = attachment.blurhash
  661. return MediaMessageEventContent(msgtype=msgtype, info=info,
  662. body=attachment.custom_filename)
  663. async def _handle_signal_attachment(self, intent: IntentAPI, attachment: Attachment,
  664. sticker: bool = False) -> MediaMessageEventContent:
  665. self.log.trace(f"Reuploading attachment {attachment}")
  666. if not attachment.content_type:
  667. attachment.content_type = (magic.from_file(attachment.incoming_filename, mime=True)
  668. if magic is not None else "application/octet-stream")
  669. content = self._make_media_content(attachment)
  670. if sticker:
  671. self._adjust_sticker_size(content.info)
  672. with open(attachment.incoming_filename, "rb") as file:
  673. data = file.read()
  674. if self.config["signal.remove_file_after_handling"]:
  675. os.remove(attachment.incoming_filename)
  676. await self._upload_attachment(intent, content, data, attachment.id)
  677. return content
  678. async def _add_sticker_meta(self, sticker: Sticker, content: MediaMessageEventContent) -> None:
  679. try:
  680. pack = self._sticker_meta_cache[sticker.pack_id]
  681. except KeyError:
  682. self.log.debug(f"Fetching sticker pack metadata for {sticker.pack_id}")
  683. try:
  684. async with StickersClient() as client:
  685. pack = await client.get_pack_metadata(sticker.pack_id, sticker.pack_key)
  686. self._sticker_meta_cache[sticker.pack_id] = pack
  687. except Exception:
  688. self.log.warning(f"Failed to fetch pack metadata for {sticker.pack_id}",
  689. exc_info=True)
  690. pack = None
  691. if not pack:
  692. content.info["fi.mau.signal.sticker"] = {
  693. "id": sticker.sticker_id,
  694. "pack": {
  695. "id": sticker.pack_id,
  696. "key": sticker.pack_key,
  697. },
  698. }
  699. return
  700. sticker_meta = pack.stickers[sticker.sticker_id]
  701. content.body = sticker_meta.emoji
  702. content.info["fi.mau.signal.sticker"] = {
  703. "id": sticker.sticker_id,
  704. "emoji": sticker_meta.emoji,
  705. "pack": {
  706. "id": pack.id,
  707. "key": pack.key,
  708. "title": pack.title,
  709. "author": pack.author,
  710. },
  711. }
  712. @staticmethod
  713. def _adjust_sticker_size(info: ImageInfo) -> None:
  714. if info.width > 256 or info.height > 256:
  715. if info.width == info.height:
  716. info.width = info.height = 256
  717. elif info.width > info.height:
  718. info.height = int(info.height / (info.width / 256))
  719. info.width = 256
  720. else:
  721. info.width = int(info.width / (info.height / 256))
  722. info.height = 256
  723. async def _handle_signal_sticker(self, intent: IntentAPI, sticker: Sticker
  724. ) -> Optional[MediaMessageEventContent]:
  725. try:
  726. self.log.debug(f"Fetching sticker {sticker.pack_id}#{sticker.sticker_id}")
  727. async with StickersClient() as client:
  728. data = await client.download_sticker(sticker.sticker_id,
  729. sticker.pack_id, sticker.pack_key)
  730. except Exception:
  731. self.log.warning(f"Failed to download sticker {sticker.sticker_id}", exc_info=True)
  732. return None
  733. info = ImageInfo(mimetype=sticker.attachment.content_type, size=len(data),
  734. width=sticker.attachment.width, height=sticker.attachment.height)
  735. self._adjust_sticker_size(info)
  736. if magic:
  737. info.mimetype = magic.from_buffer(data, mime=True)
  738. ext = mimetypes.guess_extension(info.mimetype)
  739. if not ext and info.mimetype == "image/webp":
  740. ext = ".webp"
  741. content = MediaMessageEventContent(msgtype=MessageType.IMAGE, info=info,
  742. body=f"sticker{ext}")
  743. await self._upload_attachment(intent, content, data, sticker.attachment.id)
  744. return content
  745. async def _upload_attachment(self, intent: IntentAPI, content: MediaMessageEventContent,
  746. data: bytes, id: str) -> None:
  747. upload_mime_type = content.info.mimetype
  748. if self.encrypted and encrypt_attachment:
  749. data, content.file = encrypt_attachment(data)
  750. upload_mime_type = "application/octet-stream"
  751. content.url = await intent.upload_media(data, mime_type=upload_mime_type, filename=id)
  752. if content.file:
  753. content.file.url = content.url
  754. content.url = None
  755. # This is a hack for bad clients like Element iOS that require a thumbnail
  756. if content.info.mimetype.startswith("image/"):
  757. if content.file:
  758. content.info.thumbnail_file = content.file
  759. elif content.url:
  760. content.info.thumbnail_url = content.url
  761. async def handle_signal_reaction(self, sender: 'p.Puppet', reaction: Reaction,
  762. timestamp: int) -> None:
  763. author_address = await self._resolve_address(reaction.target_author)
  764. target_id = reaction.target_sent_timestamp
  765. async with self._reaction_lock:
  766. dedup_id = (author_address, target_id, reaction.emoji)
  767. if dedup_id in self._reaction_dedup:
  768. return
  769. self._reaction_dedup.appendleft(dedup_id)
  770. existing = await DBReaction.get_by_signal_id(self.chat_id, self.receiver,
  771. author_address, target_id, sender.address)
  772. if reaction.remove:
  773. if existing:
  774. try:
  775. await sender.intent_for(self).redact(existing.mx_room, existing.mxid)
  776. except IntentError:
  777. await self.main_intent.redact(existing.mx_room, existing.mxid)
  778. await existing.delete()
  779. self.log.trace(f"Removed {existing} after Signal removal")
  780. return
  781. elif existing and existing.emoji == reaction.emoji:
  782. return
  783. message = await DBMessage.get_by_signal_id(author_address, target_id,
  784. self.chat_id, self.receiver)
  785. if not message:
  786. self.log.debug(f"Ignoring reaction to unknown message {target_id}")
  787. return
  788. intent = sender.intent_for(self)
  789. # TODO add variation selectors to emoji before sending to Matrix
  790. mxid = await intent.react(message.mx_room, message.mxid, reaction.emoji,
  791. timestamp=timestamp)
  792. self.log.debug(f"{sender.address} reacted to {message.mxid} -> {mxid}")
  793. await self._upsert_reaction(existing, intent, mxid, sender, message, reaction.emoji)
  794. async def handle_signal_delete(self, sender: 'p.Puppet', message_ts: int) -> None:
  795. message = await DBMessage.get_by_signal_id(sender.address, message_ts,
  796. self.chat_id, self.receiver)
  797. if not message:
  798. return
  799. await message.delete()
  800. try:
  801. await sender.intent_for(self).redact(message.mx_room, message.mxid)
  802. except MForbidden:
  803. await self.main_intent.redact(message.mx_room, message.mxid)
  804. # endregion
  805. # region Updating portal info
  806. async def update_info(self, source: 'u.User', info: ChatInfo,
  807. sender: Optional['p.Puppet'] = None) -> None:
  808. if self.is_direct:
  809. if not isinstance(info, (Contact, Profile, Address)):
  810. raise ValueError(f"Unexpected type for direct chat update_info: {type(info)}")
  811. if not self.name:
  812. puppet = await p.Puppet.get_by_address(self.chat_id)
  813. if not puppet.name:
  814. await puppet.update_info(info)
  815. self.name = puppet.name
  816. return
  817. if isinstance(info, GroupV2ID):
  818. info = await self.signal.get_group(source.username, info.id, info.revision or -1)
  819. if not info:
  820. self.log.debug(f"Failed to get full group v2 info through {source.username}, "
  821. "cancelling update")
  822. return
  823. changed = False
  824. if isinstance(info, Group):
  825. changed = await self._update_name(info.name, sender) or changed
  826. elif isinstance(info, GroupV2):
  827. if self.revision < info.revision:
  828. self.revision = info.revision
  829. changed = True
  830. elif self.revision > info.revision:
  831. self.log.warning(f"Got outdated info when syncing through {source.username} "
  832. f"({info.revision} < {self.revision}), ignoring...")
  833. return
  834. changed = await self._update_name(info.title, sender) or changed
  835. elif isinstance(info, GroupV2ID):
  836. return
  837. else:
  838. raise ValueError(f"Unexpected type for group update_info: {type(info)}")
  839. changed = await self._update_avatar(info, sender) or changed
  840. await self._update_participants(source, info)
  841. try:
  842. await self._update_power_levels(info)
  843. except Exception:
  844. self.log.warning("Error updating power levels", exc_info=True)
  845. if changed:
  846. await self.update_bridge_info()
  847. await self.update()
  848. async def update_expires_in_seconds(self, sender: 'p.Puppet', expires_in_seconds: int) -> None:
  849. if expires_in_seconds == 0:
  850. expires_in_seconds = None
  851. if self.expiration_time == expires_in_seconds:
  852. return
  853. assert self.mxid
  854. self.expiration_time = expires_in_seconds
  855. await self.update()
  856. time_str = format_duration(expires_in_seconds)
  857. await self.main_intent.send_notice(
  858. self.mxid,
  859. text=f'{sender.name} set the disappearing message timer to {time_str}.',
  860. html=f'<a href="https://matrix.to/#/{sender.mxid}">{sender.name}</a> set the '
  861. f'disappearing message timer to {time_str}.'
  862. )
  863. async def update_puppet_avatar(self, new_hash: str, avatar_url: ContentURI) -> None:
  864. if not self.encrypted and not self.private_chat_portal_meta:
  865. return
  866. if self.avatar_hash != new_hash or not self.avatar_set:
  867. self.avatar_hash = new_hash
  868. self.avatar_url = avatar_url
  869. if self.mxid:
  870. try:
  871. await self.main_intent.set_room_avatar(self.mxid, avatar_url)
  872. self.avatar_set = True
  873. except Exception:
  874. self.log.exception("Error setting avatar")
  875. self.avatar_set = False
  876. await self.update_bridge_info()
  877. await self.update()
  878. async def update_puppet_name(self, name: str) -> None:
  879. if not self.encrypted and not self.private_chat_portal_meta:
  880. return
  881. changed = await self._update_name(name)
  882. if changed:
  883. await self.update_bridge_info()
  884. await self.update()
  885. async def _update_name(self, name: str, sender: Optional['p.Puppet'] = None) -> bool:
  886. if self.name != name or not self.name_set:
  887. self.name = name
  888. if self.mxid:
  889. try:
  890. await self._try_with_puppet(lambda i: i.set_room_name(self.mxid, self.name),
  891. puppet=sender)
  892. self.name_set = True
  893. except Exception:
  894. self.log.exception("Error setting name")
  895. self.name_set = False
  896. return True
  897. return False
  898. async def _try_with_puppet(self, action: Callable[[IntentAPI], Awaitable[Any]],
  899. puppet: Optional['p.Puppet'] = None) -> None:
  900. if puppet:
  901. try:
  902. await action(puppet.intent_for(self))
  903. except (MForbidden, IntentError):
  904. await action(self.main_intent)
  905. else:
  906. await action(self.main_intent)
  907. async def _update_avatar(self, info: ChatInfo, sender: Optional['p.Puppet'] = None) -> bool:
  908. path = None
  909. if isinstance(info, GroupV2):
  910. path = info.avatar
  911. elif isinstance(info, Group):
  912. path = f"group-{self.chat_id}"
  913. res = await p.Puppet.upload_avatar(self, path, self.main_intent)
  914. if res is False:
  915. return False
  916. self.avatar_hash, self.avatar_url = res
  917. if not self.mxid:
  918. return True
  919. try:
  920. await self._try_with_puppet(lambda i: i.set_room_avatar(self.mxid, self.avatar_url),
  921. puppet=sender)
  922. self.avatar_set = True
  923. except Exception:
  924. self.log.exception("Error setting avatar")
  925. self.avatar_set = False
  926. return True
  927. async def _update_participants(self, source: 'u.User', info: ChatInfo) -> None:
  928. if not self.mxid or not isinstance(info, (Group, GroupV2)):
  929. return
  930. pending_members = info.pending_members if isinstance(info, GroupV2) else []
  931. self._pending_members = {addr.uuid for addr in pending_members}
  932. for address in info.members:
  933. user = await u.User.get_by_address(address)
  934. if user:
  935. await self.main_intent.invite_user(self.mxid, user.mxid)
  936. puppet = await p.Puppet.get_by_address(address)
  937. await source.sync_contact(address)
  938. await puppet.intent_for(self).ensure_joined(self.mxid)
  939. for address in pending_members:
  940. user = await u.User.get_by_address(address)
  941. if user:
  942. await self.main_intent.invite_user(self.mxid, user.mxid)
  943. puppet = await p.Puppet.get_by_address(address)
  944. await source.sync_contact(address)
  945. await self.main_intent.invite_user(self.mxid, puppet.intent_for(self).mxid)
  946. async def _update_power_levels(self, info: ChatInfo) -> None:
  947. if not self.mxid:
  948. return
  949. power_levels = await self.main_intent.get_power_levels(self.mxid)
  950. power_levels = await self._get_power_levels(power_levels, info=info, is_initial=False)
  951. await self.main_intent.set_power_levels(self.mxid, power_levels)
  952. # endregion
  953. # region Bridge info state event
  954. @property
  955. def bridge_info_state_key(self) -> str:
  956. return f"net.maunium.signal://signal/{self.chat_id}"
  957. @property
  958. def bridge_info(self) -> Dict[str, Any]:
  959. return {
  960. "bridgebot": self.az.bot_mxid,
  961. "creator": self.main_intent.mxid,
  962. "protocol": {
  963. "id": "signal",
  964. "displayname": "Signal",
  965. "avatar_url": self.config["appservice.bot_avatar"],
  966. },
  967. "channel": {
  968. "id": str(self.chat_id),
  969. "displayname": self.name,
  970. "avatar_url": self.avatar_url,
  971. }
  972. }
  973. async def update_bridge_info(self) -> None:
  974. if not self.mxid:
  975. self.log.debug("Not updating bridge info: no Matrix room created")
  976. return
  977. try:
  978. self.log.debug("Updating bridge info...")
  979. await self.main_intent.send_state_event(self.mxid, StateBridge,
  980. self.bridge_info, self.bridge_info_state_key)
  981. # TODO remove this once https://github.com/matrix-org/matrix-doc/pull/2346 is in spec
  982. await self.main_intent.send_state_event(self.mxid, StateHalfShotBridge,
  983. self.bridge_info, self.bridge_info_state_key)
  984. except Exception:
  985. self.log.warning("Failed to update bridge info", exc_info=True)
  986. # endregion
  987. # region Creating Matrix rooms
  988. async def update_matrix_room(self, source: 'u.User', info: ChatInfo) -> None:
  989. if not self.is_direct and not isinstance(info, (Group, GroupV2, GroupV2ID)):
  990. raise ValueError(f"Unexpected type for updating group portal: {type(info)}")
  991. elif self.is_direct and not isinstance(info, (Contact, Profile, Address)):
  992. raise ValueError(f"Unexpected type for updating direct chat portal: {type(info)}")
  993. try:
  994. await self._update_matrix_room(source, info)
  995. except Exception:
  996. self.log.exception("Failed to update portal")
  997. async def create_matrix_room(self, source: 'u.User', info: ChatInfo) -> Optional[RoomID]:
  998. if not self.is_direct and not isinstance(info, (Group, GroupV2, GroupV2ID)):
  999. raise ValueError(f"Unexpected type for creating group portal: {type(info)}")
  1000. elif self.is_direct and not isinstance(info, (Contact, Profile, Address)):
  1001. raise ValueError(f"Unexpected type for creating direct chat portal: {type(info)}")
  1002. if isinstance(info, Group) and not info.members:
  1003. groups = await self.signal.list_groups(source.username)
  1004. info = next((g for g in groups
  1005. if isinstance(g, Group) and g.group_id == info.group_id), info)
  1006. elif isinstance(info, GroupV2ID) and not isinstance(info, GroupV2):
  1007. self.log.debug(f"create_matrix_room() called with {info}, "
  1008. "fetching full info from signald")
  1009. info = await self.signal.get_group(source.username, info.id,
  1010. info.revision or -1)
  1011. if not info:
  1012. self.log.warning(f"Full info not found, canceling room creation")
  1013. return None
  1014. else:
  1015. self.log.trace("get_group() returned full info: %s", info)
  1016. if self.mxid:
  1017. await self.update_matrix_room(source, info)
  1018. return self.mxid
  1019. async with self._create_room_lock:
  1020. return await self._create_matrix_room(source, info)
  1021. def _get_invite_content(self, double_puppet: Optional['p.Puppet']) -> Dict[str, Any]:
  1022. invite_content = {}
  1023. if double_puppet:
  1024. invite_content["fi.mau.will_auto_accept"] = True
  1025. if self.is_direct:
  1026. invite_content["is_direct"] = True
  1027. return invite_content
  1028. async def _update_matrix_room(self, source: 'u.User', info: ChatInfo) -> None:
  1029. puppet = await p.Puppet.get_by_custom_mxid(source.mxid)
  1030. await self.main_intent.invite_user(self.mxid, source.mxid, check_cache=True,
  1031. extra_content=self._get_invite_content(puppet))
  1032. if puppet:
  1033. did_join = await puppet.intent.ensure_joined(self.mxid)
  1034. if did_join and self.is_direct:
  1035. await source.update_direct_chats({self.main_intent.mxid: [self.mxid]})
  1036. await self.update_info(source, info)
  1037. async def _get_power_levels(self, levels: Optional[PowerLevelStateEventContent] = None,
  1038. info: Optional[ChatInfo] = None, is_initial: bool = False
  1039. ) -> PowerLevelStateEventContent:
  1040. levels = levels or PowerLevelStateEventContent()
  1041. if self.is_direct:
  1042. levels.ban = 99
  1043. levels.kick = 99
  1044. levels.invite = 99
  1045. levels.state_default = 0
  1046. meta_edit_level = 0
  1047. else:
  1048. if isinstance(info, GroupV2):
  1049. ac = info.access_control
  1050. for detail in info.member_detail + info.pending_member_detail:
  1051. puppet = await p.Puppet.get_by_address(Address(uuid=detail.uuid))
  1052. level = 50 if detail.role == GroupMemberRole.ADMINISTRATOR else 0
  1053. levels.users[puppet.intent_for(self).mxid] = level
  1054. else:
  1055. ac = GroupAccessControl()
  1056. levels.ban = 50
  1057. levels.kick = 50
  1058. levels.invite = 50 if ac.members == AccessControlMode.ADMINISTRATOR else 0
  1059. levels.state_default = 50
  1060. meta_edit_level = 50 if ac.attributes == AccessControlMode.ADMINISTRATOR else 0
  1061. levels.events[EventType.ROOM_NAME] = meta_edit_level
  1062. levels.events[EventType.ROOM_AVATAR] = meta_edit_level
  1063. levels.events[EventType.ROOM_TOPIC] = meta_edit_level
  1064. levels.events[EventType.ROOM_ENCRYPTION] = 50 if self.matrix.e2ee else 99
  1065. levels.events[EventType.ROOM_TOMBSTONE] = 99
  1066. levels.users_default = 0
  1067. levels.events_default = 0
  1068. # Remote delete is only for your own messages
  1069. levels.redact = 99
  1070. if self.main_intent.mxid not in levels.users:
  1071. levels.users[self.main_intent.mxid] = 9001 if is_initial else 100
  1072. return levels
  1073. async def _create_matrix_room(self, source: 'u.User', info: ChatInfo) -> Optional[RoomID]:
  1074. if self.mxid:
  1075. await self._update_matrix_room(source, info)
  1076. return self.mxid
  1077. await self.update_info(source, info)
  1078. self.log.debug("Creating Matrix room")
  1079. name: Optional[str] = None
  1080. power_levels = await self._get_power_levels(info=info, is_initial=True)
  1081. initial_state = [{
  1082. "type": str(StateBridge),
  1083. "state_key": self.bridge_info_state_key,
  1084. "content": self.bridge_info,
  1085. }, {
  1086. # TODO remove this once https://github.com/matrix-org/matrix-doc/pull/2346 is in spec
  1087. "type": str(StateHalfShotBridge),
  1088. "state_key": self.bridge_info_state_key,
  1089. "content": self.bridge_info,
  1090. }, {
  1091. "type": str(EventType.ROOM_POWER_LEVELS),
  1092. "content": power_levels.serialize(),
  1093. }]
  1094. invites = []
  1095. if self.config["bridge.encryption.default"] and self.matrix.e2ee:
  1096. self.encrypted = True
  1097. initial_state.append({
  1098. "type": str(EventType.ROOM_ENCRYPTION),
  1099. "content": {"algorithm": "m.megolm.v1.aes-sha2"},
  1100. })
  1101. if self.is_direct:
  1102. invites.append(self.az.bot_mxid)
  1103. if self.is_direct and source.address == self.chat_id:
  1104. name = self.name = "Signal Note to Self"
  1105. elif self.encrypted or self.private_chat_portal_meta or not self.is_direct:
  1106. name = self.name
  1107. if self.avatar_url:
  1108. initial_state.append({
  1109. "type": str(EventType.ROOM_AVATAR),
  1110. "content": {"url": self.avatar_url},
  1111. })
  1112. if self.config["appservice.community_id"]:
  1113. initial_state.append({
  1114. "type": "m.room.related_groups",
  1115. "content": {"groups": [self.config["appservice.community_id"]]},
  1116. })
  1117. creation_content = {}
  1118. if not self.config["bridge.federate_rooms"]:
  1119. creation_content["m.federate"] = False
  1120. self.mxid = await self.main_intent.create_room(
  1121. name=name,
  1122. is_direct=self.is_direct,
  1123. initial_state=initial_state,
  1124. invitees=invites,
  1125. creation_content=creation_content,
  1126. )
  1127. if not self.mxid:
  1128. raise Exception("Failed to create room: no mxid returned")
  1129. self.name_set = bool(name)
  1130. self.avatar_set = bool(self.avatar_url)
  1131. if self.encrypted and self.matrix.e2ee and self.is_direct:
  1132. try:
  1133. await self.az.intent.ensure_joined(self.mxid)
  1134. except Exception:
  1135. self.log.warning("Failed to add bridge bot "
  1136. f"to new private chat {self.mxid}")
  1137. puppet = await p.Puppet.get_by_custom_mxid(source.mxid)
  1138. await self.main_intent.invite_user(self.mxid, source.mxid,
  1139. extra_content=self._get_invite_content(puppet))
  1140. if puppet:
  1141. try:
  1142. await source.update_direct_chats({self.main_intent.mxid: [self.mxid]})
  1143. await puppet.intent.join_room_by_id(self.mxid)
  1144. except MatrixError:
  1145. self.log.debug("Failed to join custom puppet into newly created portal",
  1146. exc_info=True)
  1147. await self.update()
  1148. self.log.debug(f"Matrix room created: {self.mxid}")
  1149. self.by_mxid[self.mxid] = self
  1150. if not self.is_direct:
  1151. await self._update_participants(source, info)
  1152. # TODO
  1153. # in_community = await source._community_helper.add_room(source._community_id, self.mxid)
  1154. # DBUserPortal(user=source.fbid, portal=self.fbid, portal_receiver=self.fb_receiver,
  1155. # in_community=in_community).upsert()
  1156. return self.mxid
  1157. # endregion
  1158. # region Database getters
  1159. async def _postinit(self) -> None:
  1160. self.by_chat_id[(self.chat_id_str, self.receiver)] = self
  1161. if self.mxid:
  1162. self.by_mxid[self.mxid] = self
  1163. if self.is_direct:
  1164. puppet = await p.Puppet.get_by_address(self.chat_id)
  1165. self._main_intent = puppet.default_mxid_intent
  1166. elif not self.is_direct:
  1167. self._main_intent = self.az.intent
  1168. async def delete(self) -> None:
  1169. await DBMessage.delete_all(self.mxid)
  1170. self.by_mxid.pop(self.mxid, None)
  1171. self.mxid = None
  1172. self.encrypted = False
  1173. await self.update()
  1174. async def save(self) -> None:
  1175. await self.update()
  1176. @classmethod
  1177. def all_with_room(cls) -> AsyncGenerator['Portal', None]:
  1178. return cls._db_to_portals(super().all_with_room())
  1179. @classmethod
  1180. def find_private_chats_with(cls, other_user: Address) -> AsyncGenerator['Portal', None]:
  1181. return cls._db_to_portals(super().find_private_chats_with(other_user))
  1182. @classmethod
  1183. async def _db_to_portals(cls, query: Awaitable[List['Portal']]
  1184. ) -> AsyncGenerator['Portal', None]:
  1185. portals = await query
  1186. for index, portal in enumerate(portals):
  1187. try:
  1188. yield cls.by_chat_id[(portal.chat_id_str, portal.receiver)]
  1189. except KeyError:
  1190. await portal._postinit()
  1191. yield portal
  1192. @classmethod
  1193. @async_getter_lock
  1194. async def get_by_mxid(cls, mxid: RoomID) -> Optional['Portal']:
  1195. try:
  1196. return cls.by_mxid[mxid]
  1197. except KeyError:
  1198. pass
  1199. portal = cast(cls, await super().get_by_mxid(mxid))
  1200. if portal is not None:
  1201. await portal._postinit()
  1202. return portal
  1203. return None
  1204. @classmethod
  1205. async def get_by_chat_id(cls, chat_id: Union[GroupID, Address], *, receiver: str = "",
  1206. create: bool = False) -> Optional['Portal']:
  1207. if isinstance(chat_id, str):
  1208. receiver = ""
  1209. elif not isinstance(chat_id, Address):
  1210. raise ValueError(f"Invalid chat ID type {type(chat_id)}")
  1211. elif not receiver:
  1212. raise ValueError("Direct chats must have a receiver")
  1213. best_id = id_to_str(chat_id)
  1214. portal = await cls._get_by_chat_id(best_id, receiver, create=create, chat_id=chat_id)
  1215. if portal:
  1216. portal.log.debug(f"get_by_chat_id({chat_id}, {receiver}) -> {hex(id(portal))}")
  1217. return portal
  1218. @classmethod
  1219. @async_getter_lock
  1220. async def _get_by_chat_id(cls, best_id: str, receiver: str, *, create: bool,
  1221. chat_id: Union[GroupID, Address]) -> Optional['Portal']:
  1222. try:
  1223. return cls.by_chat_id[(best_id, receiver)]
  1224. except KeyError:
  1225. pass
  1226. portal = cast(cls, await super().get_by_chat_id(chat_id, receiver))
  1227. if portal is not None:
  1228. await portal._postinit()
  1229. return portal
  1230. if create:
  1231. portal = cls(chat_id, receiver)
  1232. await portal.insert()
  1233. await portal._postinit()
  1234. return portal
  1235. return None
  1236. # endregion