From fc65d5f0f55f8c0f4dd05213c724e8af1a90b5b0 Mon Sep 17 00:00:00 2001
From: Max Sandholm <max@sandholm.org>
Date: Mon, 3 Apr 2017 20:56:56 +0300
Subject: [PATCH] Telegram profile pictures and display names are updated

The bridge updates the displayname and/or the avatar of the bridged Matrix
user when the Telegram user changes them.

This is done by checking the name and profile pic of the sender when a message
is received from Telegram, and comparing to the database to see if they have
changed. If they have, it will update the database and send the updates to the
Matrix homeserver.
---
 telematrix/__init__.py | 39 +++++++++++++++++++++++++++++++++++++++
 telematrix/database.py |  4 ++--
 2 files changed, 41 insertions(+), 2 deletions(-)

diff --git a/telematrix/__init__.py b/telematrix/__init__.py
index 39524f2..f3663b6 100644
--- a/telematrix/__init__.py
+++ b/telematrix/__init__.py
@@ -200,6 +200,8 @@ async def matrix_transaction(request):
         group = TG_BOT.group(link.tg_room)
 
         try:
+            response = None
+
             if event['type'] == 'm.room.message':
                 user_id = event['user_id']
                 if matrix_is_telegram(user_id):
@@ -448,6 +450,39 @@ async def register_join_matrix(chat, room_id, user_id):
                      user_id, {'displayname': name})
     await matrix_post('client', 'join/{}'.format(room_id), user_id, {})
 
+async def update_matrix_displayname_avatar(tg_user):
+    name = tg_user['first_name']
+    if 'last_name' in tg_user:
+        name += ' ' + tg_user['last_name']
+    name += ' (Telegram)'
+    user_id = USER_ID_FORMAT.format(tg_user['id'])
+    
+    db_user = db.session.query(db.TgUser).filter_by(tg_id=tg_user['id']).first()
+
+    profile_photos = await TG_BOT.get_user_profile_photos(tg_user['id'])
+    pp_file_id = None
+    try:
+        pp_file_id = profile_photos['result']['photos'][0][-1]['file_id']
+    except:
+        pp_file_id = None
+
+    if db_user:
+        if db_user.name != name:
+            await matrix_put('client', 'profile/{}/displayname'.format(user_id), user_id, {'displayname': name})
+            db_user.name = name
+        if db_user.profile_pic_id != pp_file_id:
+            pp_uri, _ = await upload_tgfile_to_matrix(pp_file_id, user_id)
+            await matrix_put('client', 'profile/{}/avatar_url'.format(user_id), user_id, {'avatar_url':pp_uri})
+            db_user.profile_pic_id = pp_file_id
+    else:
+        db_user = db.TgUser(tg_user['id'], name, pp_file_id)
+        await matrix_put('client', 'profile/{}/displayname'.format(user_id), user_id, {'displayname': name})
+        pp_uri, _ = await upload_tgfile_to_matrix(pp_file_id, user_id)
+        await matrix_put('client', 'profile/{}/avatar_url'.format(user_id), user_id, {'avatar_url':pp_uri})
+        db.session.add(db_user)
+    db.session.commit()
+        
+
 @TG_BOT.handle('sticker')
 async def aiotg_sticker(chat, sticker):
     link = db.session.query(db.ChatLink).filter_by(tg_room=chat.id).first()
@@ -455,6 +490,8 @@ async def aiotg_sticker(chat, sticker):
         print('Unknown telegram chat {}: {}'.format(chat, chat.id))
         return
 
+    await update_matrix_displayname_avatar(chat.sender);
+
     room_id = link.matrix_room
     user_id = USER_ID_FORMAT.format(chat.sender['id'])
     txn_id = quote('{}{}'.format(chat.message['message_id'], chat.id))
@@ -501,6 +538,7 @@ async def aiotg_photo(chat, photo):
         print('Unknown telegram chat {}: {}'.format(chat, chat.id))
         return
 
+    await update_matrix_displayname_avatar(chat.sender);
     room_id = link.matrix_room
     user_id = USER_ID_FORMAT.format(chat.sender['id'])
     txn_id = quote('{}{}'.format(chat.message['message_id'], chat.id))
@@ -555,6 +593,7 @@ async def aiotg_message(chat, match):
         print('Unknown telegram chat {}: {}'.format(chat, chat.id))
         return
 
+    await update_matrix_displayname_avatar(chat.sender);
     user_id = USER_ID_FORMAT.format(chat.sender['id'])
     txn_id = quote('{}:{}'.format(chat.message['message_id'], chat.id))
 
diff --git a/telematrix/database.py b/telematrix/database.py
index 9d65099..b4bab21 100644
--- a/telematrix/database.py
+++ b/telematrix/database.py
@@ -32,9 +32,9 @@ class TgUser(Base):
     id = sa.Column(sa.Integer, primary_key=True)
     tg_id = sa.Column(sa.BigInteger)
     name = sa.Column(sa.String)
-    profile_pic_id = sa.Column(sa.String)
+    profile_pic_id = sa.Column(sa.String, nullable=True)
 
-    def __init__(self, tg_id, name, profile_pic_id):
+    def __init__(self, tg_id, name, profile_pic_id=None):
         self.tg_id = tg_id
         self.name = name
         self.profile_pic_id = profile_pic_id