Ignore asconfig.yaml

.gitignore

@@ -152,4 +152,5 @@ Temporary Items
 
 # Config
 config.json
+asconfig.yaml
 database.db

README.md

@@ -42,6 +42,8 @@ First, copy config.json.example to config.json. Then fill in the fields:
 * `hosts.bare`: Just the (sub)domain of the server.
 * `user_id_format`: A Python `str.format`-style string to format user IDs as
 * `db_url`: A SQLAlchemy URL for the database. See the [SQLAlchemy docs](http://docs.sqlalchemy.org/en/latest/core/engines.html).
+* `bot_owners`: A list of matrix users the bot won't use a `<username>` prefix for. Can be left blank to print a prefix for all matrix users. Useful if you would like the bridge to replace Telegram for yourself, but also want to allow others to use the Matrix room.
+* `print_url_with_image`: Set to `false` to disable sending the Matrix url of an image to Telegram.
 
 **Synapse configuration**
 

config.json.example

@@ -12,8 +12,16 @@
         "bare": "DOMAIN.TLD"
     },
 
+    "bot_owners": [
+      "@youruser:DOMAIN.TLD",
+      "@youruser:matrix.org",
+      "@youruser:example.com"
+    ],
+
     "user_id_format": "@telegram_{}:DOMAIN.TLD",
     "db_url": "sqlite:///database.db",
 
+    "print_url_with_image": true,
+
     "as_port": 5000
 }

telematrix/__init__.py

@@ -12,7 +12,6 @@ from datetime import datetime
 from time import time
 from urllib.parse import unquote, quote, urlparse, parse_qs
 from io import BytesIO
-import re
 
 from PIL import Image
 from aiohttp import web, ClientSession
@@ -39,6 +38,11 @@ try:
         MATRIX_HOST_EXT = CONFIG['hosts']['external']
         MATRIX_HOST_BARE = CONFIG['hosts']['bare']
 
+        try:
+            MATRIX_BOT_OWNERS = CONFIG['bot_owners']
+        except KeyError:
+            MATRIX_BOT_OWNERS = {}
+
         MATRIX_PREFIX = MATRIX_HOST + '_matrix/client/r0/'
         MATRIX_MEDIA_PREFIX = MATRIX_HOST + '_matrix/media/r0/'
 
@@ -46,6 +50,8 @@ try:
         DATABASE_URL = CONFIG['db_url']
 
         AS_PORT = CONFIG['as_port'] if 'as_port' in CONFIG else 5000
+
+        PRINT_URL_WITH_IMAGE = CONFIG['print_url_with_image'] if 'print_url_with_image' in CONFIG else True
 except (OSError, IOError) as exception:
     print('Error opening config file:')
     print(exception)
@@ -83,26 +89,26 @@ def sanitize_html(string):
     soup = BeautifulSoup(string, 'html.parser')
     for tag in soup.find_all(True):
         if tag.name == 'blockquote':
-            tag.string = ('\n' + tag.text).replace('\n', '\n> ')[3:-3]
+            tag.string = ('\n' + tag.text).replace('\n', '\n> ').rstrip('\n>')
         if tag.name not in VALID_TAGS:
             tag.hidden = True
     return soup.renderContents().decode('utf-8')
 
 
-def format_matrix_msg(form, content):
+def format_matrix_msg(form, username, content):
     """
     Formats a matrix message for sending to Telegram
     :param form: The format string of the message, where the first parameter
                  is the username and the second one the message.
+    :param username: The username of the user.
     :param content: The content to be sent.
     :return: The formatted string.
     """
     if 'format' in content and content['format'] == 'org.matrix.custom.html':
-        sanitized = re.sub("<a href=\\\"https://matrix.to/#/@telegram_([0-9]+):{}\\\">(.+?) \(Telegram\)</a>".format(MATRIX_HOST_BARE), "<a href=\"tg://user?id=\\1\">\\2</a>", content['formatted_body'])
-        sanitized = sanitize_html(sanitized)
-        return html.escape(form).format(sanitized), 'HTML'
+        sanitized = sanitize_html(content['formatted_body'])
+        return html.escape(form).format(username, sanitized), 'HTML'
     else:
-        return form.format(html.escape(content['body'])), None
+        return form.format(username, content['body']), None
 
 
 async def download_matrix_file(url, filename):
@@ -184,7 +190,7 @@ async def matrix_transaction(request):
             for alias in aliases:
                 print(alias)
                 if alias.split('_')[0] != '#telegram' \
-                        or alias.split(':')[-1] != MATRIX_HOST_BARE:
+                        or not alias.endswith(MATRIX_HOST_BARE):
                     continue
 
                 tg_id = alias.split('_')[1].split(':')[0]
@@ -206,38 +212,44 @@ async def matrix_transaction(request):
 
             if event['type'] == 'm.room.message':
                 user_id = event['user_id']
-                if matrix_is_telegram(user_id):
-                    continue
-
-
-                sender = db.session.query(db.MatrixUser)\
-                           .filter_by(matrix_id=user_id).first()
-
-                if not sender:
-                    response = await matrix_get('client', 'profile/{}/displayname'
-                                                          .format(user_id), None)
-                    try:
-                        displayname = response['displayname']
-                    except KeyError:
-                        displayname = get_username(user_id)
-                    sender = db.MatrixUser(user_id, displayname)
-                    db.session.add(sender)
+                if user_id in MATRIX_BOT_OWNERS:
+                    displayname = ''
                 else:
-                    displayname = sender.name or get_username(user_id)
+                    if matrix_is_telegram(user_id):
+                        continue
+
+
+                    sender = db.session.query(db.MatrixUser)\
+                               .filter_by(matrix_id=user_id).first()
+
+                    if not sender:
+                        response = await matrix_get('client', 'profile/{}/displayname'
+                                                              .format(user_id), None)
+                        try:
+                            displayname = response['displayname']
+                        except KeyError:
+                            displayname = get_username(user_id)
+                        sender = db.MatrixUser(user_id, displayname)
+                        db.session.add(sender)
+                    else:
+                        displayname = sender.name or get_username(user_id)
                 content = event['content']
 
                 if 'msgtype' not in content:
                     continue
 
                 if content['msgtype'] == 'm.text':
-                    msg, mode = format_matrix_msg('{}', content)
-                    response = await group.send_text("<b>{}:</b> {}".format(displayname, msg), parse_mode='HTML')
+                    prefix = '' if displayname == '' else '<' + displayname + '> '
+                    msg, mode = format_matrix_msg('{}{}', prefix, content)
+                    response = await group.send_text(msg, parse_mode=mode)
                 elif content['msgtype'] == 'm.notice':
-                    msg, mode = format_matrix_msg('{}', content)
-                    response = await group.send_text("[{}] {}".format(displayname, msg), parse_mode=mode)
+                    prefix = '' if displayname == '' else '[' + displayname + '] '
+                    msg, mode = format_matrix_msg('{}{}', prefix, content)
+                    response = await group.send_text(msg, parse_mode=mode)
                 elif content['msgtype'] == 'm.emote':
-                    msg, mode = format_matrix_msg('{}', content)
-                    response = await group.send_text("* {} {}".format(displayname, msg), parse_mode=mode)
+                    prefix = '' if displayname == '' else '* ' + displayname + ' '
+                    msg, mode = format_matrix_msg('{}{}', prefix, content)
+                    response = await group.send_text(msg, parse_mode=mode)
                 elif content['msgtype'] == 'm.image':
                     try:
                         url = urlparse(content['url'])
@@ -250,13 +262,18 @@ async def matrix_transaction(request):
                         # Download the file
                         await download_matrix_file(url, content['body'])
                         with open('/tmp/{}'.format(content['body']), 'rb') as img_file:
-                            # Create the URL and shorten it
-                            url_str = MATRIX_HOST_EXT + \
-                                      '_matrix/media/r0/download/{}{}' \
-                                      .format(url.netloc, quote(url.path))
-                            url_str = await shorten_url(url_str)
+                            url_str = ''
+                            if PRINT_URL_WITH_IMAGE:
+                                # Create the URL and shorten it
+                                url_str = MATRIX_HOST_EXT + \
+                                          '_matrix/media/r0/download/{}{}' \
+                                          .format(url.netloc, quote(url.path))
+                                url_str = await shorten_url(url_str)
+                                url_str = ' (' + url_str + ')'
 
-                            caption = '{} sent an image'.format(displayname)
+                            prefix = '' if displayname == '' else '<' + displayname + '> '
+                            caption = '{}{}{}'.format(prefix,
+                                                            content['body'], url_str)
                             response = await group.send_photo(img_file, caption=caption)
                     except:
                         pass
@@ -449,7 +466,10 @@ async def register_join_matrix(chat, room_id, user_id):
 
     await matrix_put('client', 'profile/{}/displayname'.format(user_id),
                      user_id, {'displayname': name})
-    await matrix_post('client', 'join/{}'.format(room_id), user_id, {})
+    j = await matrix_post('client', 'join/{}'.format(room_id), user_id, {})
+    if 'errcode' in j and j['errcode'] == 'M_FORBIDDEN':
+        print("Error with <{}> joining room <{}>. This is likely because guests are not allowed to join the room."
+              .format(user_id, room_id))
 
 async def update_matrix_displayname_avatar(tg_user):
     name = tg_user['first_name']