Add command to get invite link. Fixes #32

README.md

@@ -128,5 +128,5 @@ The bridge does not do this automatically.
   * [x] Joining chats with invite links (`join`)
   * [x] Creating a Telegram chat for an existing Matrix room (`create`)
   * [ ] Upgrading the chat of a portal room into a supergroup (`upgrade`)
-  * [ ] Getting the Telegram invite link to a Matrix room (`invitelink`)
+  * [x] Getting the Telegram invite link to a Matrix room (`invitelink`)
   * [ ] Clean up and forget a portal room (`deleteportal`)

mautrix_telegram/commands.py

@@ -27,7 +27,8 @@
 
 
 def command_handler(func):
-    command_handlers[func.__name__] = func
+    command_handlers[func.__name__.replace("_", "")] = func
+    return func
 
 
 class CommandHandler:
@@ -61,7 +62,7 @@ def handler(self, sender, room, command, args, is_management, is_portal):
                 args.insert(0, command)
                 command = sender.command_status["next"]
             else:
-                command = command_handlers["unknown_command"]
+                command = self.unknown_command
         self._is_management = is_management
         self._is_portal = is_portal
         yield command
@@ -195,7 +196,7 @@ def logout(self, sender, args):
     def search(self, sender, args):
         if len(args) == 0:
             return self.reply("**Usage:** `$cmdprefix+sp search [-r|--remote] <query>")
-        elif not sender.tgid:
+        elif not sender.logged_in:
             return self.reply("This command requires you to be logged in.")
         force_remote = False
         if args[0] in {"-r", "--remote"}:
@@ -222,7 +223,7 @@ def search(self, sender, args):
     def pm(self, sender, args):
         if len(args) == 0:
             return self.reply("**Usage:** `$cmdprefix+sp pm <user identifier>`")
-        elif not sender.tgid:
+        elif not sender.logged_in:
             return self.reply("This command requires you to be logged in.")
 
         user = sender.client.get_entity(args[0])
@@ -234,6 +235,26 @@ def pm(self, sender, args):
         portal.create_matrix_room(sender, user, [sender.mxid])
         self.reply(f"Created private chat room with {pu.Puppet.get_displayname(user, False)}")
 
+    @command_handler
+    def invite_link(self, sender, args):
+        if not sender.logged_in:
+            return self.reply("This command requires you to be logged in.")
+
+        portal = po.Portal.get_by_mxid(self._room_id)
+        if not portal:
+            return self.reply("This is not a portal room.")
+
+        if portal.peer_type == "user":
+            return self.reply("You can't invite users to private chats.")
+
+        try:
+            link = portal.get_invite_link(sender)
+            return self.reply(f"Invite link to {portal.title}: {link}")
+        except ValueError as e:
+            return self.reply(e.args[0])
+        except ChatAdminRequiredError:
+            return self.reply("You don't have the permission to create an invite link.")
+
     def _strip_prefix(self, value, prefixes):
         for prefix in prefixes:
             if value.startswith(prefix):
@@ -244,7 +265,7 @@ def _strip_prefix(self, value, prefixes):
     def join(self, sender, args):
         if len(args) == 0:
             return self.reply("**Usage:** `$cmdprefix+sp join <invite link>")
-        elif not sender.tgid:
+        elif not sender.logged_in:
             return self.reply("This command requires you to be logged in.")
 
         regex = re.compile(r"(?:https?://)?t(?:elegram)?\.(?:dog|me)(?:joinchat/)?/(.+)")
@@ -280,7 +301,7 @@ def create(self, sender, args):
         type = args[0] if len(args) > 0 else "group"
         if type not in {"chat", "group", "supergroup", "channel"}:
             return self.reply("**Usage:** `$cmdprefix+sp create [`group`/`supergroup`/`channel`]")
-        elif not sender.tgid:
+        elif not sender.logged_in:
             return self.reply("This command requires you to be logged in.")
 
         if po.Portal.get_by_mxid(self._room_id):

mautrix_telegram/portal.py

@@ -15,9 +15,10 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 from telethon.tl.functions.messages import (GetFullChatRequest, EditChatAdminRequest,
-                                            CreateChatRequest, AddChatUserRequest)
+                                            CreateChatRequest, AddChatUserRequest,
+                                            ExportChatInviteRequest)
 from telethon.tl.functions.channels import (GetParticipantsRequest, CreateChannelRequest,
-                                            InviteToChannelRequest)
+                                            InviteToChannelRequest, ExportInviteRequest)
 from telethon.errors.rpc_error_list import ChatAdminRequiredError, LocationInvalidError
 from telethon.tl.types import *
 from PIL import Image
@@ -236,6 +237,21 @@ def get_users(self, user, entity):
         elif self.peer_type == "user":
             return [entity], []
 
+    def get_invite_link(self, user):
+        if self.peer_type == "user":
+            raise ValueError("You can't invite users to private chats.")
+        elif self.peer_type == "chat":
+            link = user.client(ExportChatInviteRequest(chat_id=self.tgid))
+        elif self.peer_type == "channel":
+            link = user.client(ExportInviteRequest(channel=user.client.get_input_entity(self.peer)))
+        else:
+            raise ValueError(f"Invalid peer type '{self.peer_type}' for invite link.")
+
+        if isinstance(link, ChatInviteEmpty):
+            raise ValueError("Failed to get invite link.")
+
+        return link.link
+
     # endregion
     # region Matrix event handling