Add support for inviting users and initiating chats from Matrix
This commit is contained in:
Tulir Asokan
+69
-15
@@ -13,10 +13,11 @@
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
import re
|
||||
from matrix_client.errors import MatrixRequestError
|
||||
|
||||
from .user import User
|
||||
from .portal import Portal
|
||||
from .puppet import Puppet
|
||||
from .commands import CommandHandler
|
||||
|
||||
|
||||
@@ -26,32 +27,83 @@ class MatrixHandler:
|
||||
self.log = log.getChild("mx")
|
||||
self.commands = CommandHandler(context)
|
||||
|
||||
alias_format = self.config.get("bridge.alias_template", "telegram_{groupname}").format(groupname="(.+)")
|
||||
hs = self.config["homeserver"]["domain"]
|
||||
self.localpart_regex = re.compile(f"@{alias_format}:{hs}")
|
||||
|
||||
self.az.matrix_event_handler(self.handle_event)
|
||||
self.az.intent.set_display_name(
|
||||
self.config.get("appservice.bot_displayname", "Mautrix-Telegram Bridge"))
|
||||
self.config.get("appservice.bot_displayname", "Telegram bridge bot"))
|
||||
|
||||
def is_puppet(self, mxid):
|
||||
match = self.localpart_regex.match(mxid)
|
||||
match = Puppet.mxid_regex.match(mxid)
|
||||
return True if match else False
|
||||
|
||||
def get_puppet(self, mxid):
|
||||
match = Puppet.mxid_regex.match(mxid)
|
||||
if not match:
|
||||
return None
|
||||
return Puppet.get(int(match.group(1)))
|
||||
|
||||
def handle_puppet_invite(self, room, puppet, inviter):
|
||||
self.log.debug(f"{inviter} invited puppet for {puppet.tgid} to {room}")
|
||||
portal = Portal.get_by_mxid(room)
|
||||
if portal:
|
||||
if portal.peer_type == "user":
|
||||
puppet.intent.error_and_leave(
|
||||
room, text="You can not invite additional users to private chats.")
|
||||
return
|
||||
portal.invite_telegram(inviter, puppet)
|
||||
puppet.intent.join_room(room)
|
||||
return
|
||||
try:
|
||||
members = self.az.intent.get_room_members(room)
|
||||
except MatrixRequestError:
|
||||
members = []
|
||||
if self.az.intent.mxid not in members:
|
||||
if len(members) > 1:
|
||||
puppet.intent.error_and_leave(room, text=None, html=(
|
||||
f"Please invite "
|
||||
f"<a href='https://matrix.to/#/{self.az.intent.mxid}'>the bridge bot</a> "
|
||||
f"first if you want to create a Telegram chat."))
|
||||
return
|
||||
|
||||
puppet.intent.join_room(room)
|
||||
existing_portal = Portal.get_by_tgid(puppet.tgid, "user")
|
||||
if existing_portal:
|
||||
try:
|
||||
puppet.intent.invite(existing_portal.mxid, inviter.mxid)
|
||||
puppet.intent.send_notice(room, text=None, html=(
|
||||
"You already have a private chat with me: "
|
||||
f"<a href='https://matrix.to/#/{existing_portal.mxid}'>"
|
||||
"Link to room"
|
||||
"</a>"))
|
||||
puppet.intent.leave_room(room)
|
||||
return
|
||||
except MatrixRequestError:
|
||||
existing_portal.delete()
|
||||
|
||||
portal = Portal(tgid=puppet.tgid, peer_type="user", mxid=room)
|
||||
portal.save()
|
||||
puppet.intent.send_notice(room, "Portal to private chat created.")
|
||||
else:
|
||||
puppet.intent.join_room(room)
|
||||
puppet.intent.send_notice(room, "This puppet will remain inactive until a Telegram "
|
||||
"chat is created for this room.")
|
||||
|
||||
def handle_invite(self, room, user, inviter):
|
||||
inviter = User.get_by_mxid(inviter)
|
||||
if user == self.az.bot_mxid:
|
||||
self.az.intent.join_room(room)
|
||||
return
|
||||
tgid = self.get_puppet(user)
|
||||
if tgid:
|
||||
# TODO handle puppet invite
|
||||
self.log.debug(f"{inviter} invited puppet for {tgid} to {room}")
|
||||
elif not inviter.tgid:
|
||||
return
|
||||
puppet = self.get_puppet(user)
|
||||
if puppet:
|
||||
self.handle_puppet_invite(room, puppet, inviter)
|
||||
return
|
||||
# These can probably be ignored
|
||||
self.log.debug(f"{inviter} invited {user} to {room}")
|
||||
|
||||
def handle_part(self, room, user):
|
||||
self.log.debug(f"{user} left {room}")
|
||||
user = User.get_by_mxid(user, create=False)
|
||||
|
||||
def is_command(self, message):
|
||||
text = message.get("body", "")
|
||||
@@ -68,14 +120,14 @@ class MatrixHandler:
|
||||
sender = User.get_by_mxid(sender)
|
||||
|
||||
portal = Portal.get_by_mxid(room)
|
||||
if portal and not is_command:
|
||||
if sender.tgid and portal and not is_command:
|
||||
portal.handle_matrix_message(sender, message, event_id)
|
||||
return
|
||||
|
||||
if message["msgtype"] != "m.text":
|
||||
return
|
||||
|
||||
is_management = len(self.az.intent.get_joined_users(room)) == 2
|
||||
is_management = len(self.az.intent.get_room_members(room)) == 2
|
||||
if is_command or is_management:
|
||||
try:
|
||||
command, arguments = text.split(" ", 1)
|
||||
@@ -90,12 +142,13 @@ class MatrixHandler:
|
||||
def handle_redaction(self, room, sender, event_id):
|
||||
portal = Portal.get_by_mxid(room)
|
||||
sender = User.get_by_mxid(sender)
|
||||
if portal:
|
||||
if sender.tgid and portal:
|
||||
portal.handle_matrix_deletion(sender, event_id)
|
||||
|
||||
def handle_power_levels(self, room, sender, new, old):
|
||||
portal = Portal.get_by_mxid(room)
|
||||
if portal:
|
||||
sender = User.get_by_mxid(sender)
|
||||
if sender.tgid and portal:
|
||||
sender = User.get_by_mxid(sender)
|
||||
portal.handle_matrix_power_levels(sender, new["users"], old["users"])
|
||||
|
||||
@@ -115,6 +168,7 @@ class MatrixHandler:
|
||||
elif membership == "leave":
|
||||
self.handle_part(evt["room_id"], evt["state_key"])
|
||||
elif membership == "join":
|
||||
# TODO handle when needed
|
||||
pass
|
||||
elif type == "m.room.message":
|
||||
self.handle_message(evt["room_id"], evt["sender"], content, evt["event_id"])
|
||||
|
||||
Reference in New Issue
Block a user