portal.py 63 KB

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