pyfedi/app/chat/util.py

from flask import flash, current_app
from flask_login import current_user
from flask_babel import _
from sqlalchemy import text

from app import db
from app.activitypub.signature import post_request
from app.models import User, ChatMessage, Notification, utcnow, Conversation
from app.utils import allowlist_html, shorten_string, gibberish, markdown_to_html


def send_message(message: str, conversation_id: int) -> ChatMessage:
    conversation = Conversation.query.get(conversation_id)
    reply = ChatMessage(sender_id=current_user.id, conversation_id=conversation.id,
                        body=message, body_html=allowlist_html(markdown_to_html(message)))
    conversation.updated_at = utcnow()
    db.session.add(reply)
    db.session.commit()
    for recipient in conversation.members:
        if recipient.id != current_user.id:
            if recipient.is_local():
                # Notify local recipient
                notify = Notification(title=shorten_string('New message from ' + current_user.display_name()),
                                      url=f'/chat/{conversation_id}#message_{reply.id}',
                                      user_id=recipient.id,
                                      author_id=current_user.id)
                db.session.add(notify)
                recipient.unread_notifications += 1
                db.session.commit()
            else:
                # Federate reply
                reply_json = {
                    "actor": current_user.profile_id(),
                    "id": f"https://{current_app.config['SERVER_NAME']}/activities/create/{gibberish(15)}",
                    "object": {
                        "attributedTo": current_user.profile_id(),
                        "content": reply.body_html,
                        "id": f"https://{current_app.config['SERVER_NAME']}/private_message/{reply.id}",
                        "mediaType": "text/html",
                        "published": utcnow().isoformat() + 'Z',  # Lemmy is inconsistent with the date format they use
                        "source": {
                            "content": reply.body,
                            "mediaType": "text/markdown"
                        },
                        "to": [
                            recipient.profile_id()
                        ],
                        "type": "ChatMessage"
                    },
                    "to": [
                        recipient.profile_id()
                    ],
                    "type": "Create"
                }
                success = post_request(recipient.ap_inbox_url, reply_json, current_user.private_key,
                                       current_user.profile_id() + '#main-key')
                if not success:
                    flash(_('Message failed to send to %(name)s.', name=recipient.link()), 'error')

    flash(_('Message sent.'))
    return reply