RcGcDw/extensions/hooks/matrix.py

#  This file is part of Recent changes Goat compatible Discord webhook (RcGcDw).
#
#  RcGcDw is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  RcGcDw is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with RcGcDw.  If not, see <http://www.gnu.org/licenses/>.
import time
from typing import Optional

import requests
import markdown
import json

from src.configloader import settings
from src.api.context import Context
from src.discord.message import DiscordMessage, DiscordMessageMetadata
from src.api.hook import post_hook

# {
#     "hooks": {
#         "matrix": {
#             "homeserver": "https://example.com/endpoint",
#             "rooms": [],
#             "access_token": "authtoken",
#             "matrix_only": true
#         }
#     }
# }

matrix_hook: Optional[dict] = None
start_time = int(time.time())


def check_if_exists(settings: dict):
    global matrix_hook
    matrix_hook = settings.get("hooks", {}).get("matrix", {})
    return bool(matrix_hook)


@post_hook(priority=50, register=check_if_exists)
def matrix_posthook(message: DiscordMessage, metadata: DiscordMessageMetadata, context: Context, change: dict):
    if not matrix_hook.get("access_token"):
        raise KeyError("Matrix hook requires an access token to run!")

    if settings["appearance"]["mode"] == "embed":
        matrix_json_formatted_body = discord_embed_converter(message).replace("\\:", ":").replace("\\|", "|")
    else:
        matrix_json_formatted_body = discord_compact_converter(message)
    if matrix_hook.get("matrix_only"):
        context.event = None

    data = json.dumps({"msgtype": "m.text", "body": "test", "format": "org.matrix.custom.html", "formatted_body": matrix_json_formatted_body})
    for room in matrix_hook.get("rooms", []):
        response = requests.put("{homeserver}/_matrix/client/v3/rooms/{roomId}/send/{eventType}/{txnId}".format(homeserver=matrix_hook.get("homeserver"),
                                                                                                     roomId=room,
                                                                                                     eventType="m.room.message",
                                                                                                     txnId=str(int(change["rcid"])+start_time)),
                                data=data, headers={"Authorization": "Bearer "+matrix_hook.get("access_token"), "Content-Type": "application/json"})
        print(response.status_code)
        print(response.text)


def discord_embed_converter(embed: DiscordMessage) -> str:
    discord_embed = embed.embed
    matrix_soup = []

    matrix_soup.append("<a href=\"{url}\">{author}</a><br>".format(author=discord_embed.get("author", {}).get("name", "unknown"), url=discord_embed.get("author", {}).get("url", "")))
    matrix_soup.append("<b><a href=\"{url}\">{title}</a></b><br>".format(title=discord_embed.get("title", "Unknown title"), url=discord_embed.get("url", "")))
    if discord_embed.get("description"):
        matrix_soup.append(markdown.markdown(discord_embed.get("description", ""), extensions=['pymdownx.tilde']))
    for field in discord_embed.get("fields", []):
        if field.get("inline"):
            matrix_soup.append((markdown.markdown(field.get("value", ""), extensions=['pymdownx.tilde']), True))
        else:
            matrix_soup.append(("<b>{name}</b><br>{data}".format(name=field.get("name", ""), data=removep(markdown.markdown(field.get("value", ""), extensions=['pymdownx.tilde']))), False))
    final_output = ""
    in_table_mode = False
    while matrix_soup:
        element = matrix_soup.pop(0)
        if isinstance(element, str):
            element = removep(element)
            if in_table_mode:
                final_output += "</tbody></table>"
                in_table_mode = False
            final_output += element
        else:
            if in_table_mode is False:
                final_output+="<table><tbody>"
                in_table_mode = True
            if element[1] is False:
                final_output+="<tr><td colspan=\"2\">"+removep(element[0])+"</td></tr>"
            else:
                if matrix_soup and matrix_soup[0][1] is True:
                    another_element = matrix_soup.pop(0)
                    final_output+="<tr><td>"+removep(element[0])+"</td><td>"+removep(another_element[0])+"</td></tr>"
                else:
                    final_output += "<tr><td colspan=\"2\">" + removep(element[0]) + "</td></tr>"
    if in_table_mode:
        final_output += "</tbody></table>"
    return final_output.replace("\n", "")


def removep(element: str) -> str:
    if element.startswith("<p>") and element.endswith("</p>"):
        return element[3:-4]
    return element


def discord_compact_converter(embed: DiscordMessage) -> str:
    return markdown.markdown(embed.webhook_object["content"], extensions=['pymdownx.tilde'])