import datetime, logging import json import gettext from urllib.parse import quote_plus from src.configloader import settings from src.misc import link_formatter, create_article_path, DiscordMessage, send_to_discord, escape_formatting from src.i18n import discussion_formatters _ = discussion_formatters.gettext discussion_logger = logging.getLogger("rcgcdw.discussion_formatter") def compact_formatter(post_type, post, article_paths): """Compact formatter for Fandom discussions.""" message = None if post_type == "FORUM": author = post["createdBy"]["name"] author_url = "<{url}f/u/{creatorId}>".format(url=settings["fandom_discussions"]["wiki_url"], creatorId=post["creatorId"]) elif post["creatorIp"]: author = post["creatorIp"][1:] author_url = "<{url}wiki/Special:Contributions{creatorIp}>".format(url=settings["fandom_discussions"]["wiki_url"], creatorIp=post["creatorIp"]) else: author = post["createdBy"]["name"] author_url = "<{url}wiki/User:{author}>".format(url=settings["fandom_discussions"]["wiki_url"], author=author) author_url = link_formatter(create_article_path("User:{user}".format(user=author))) if post_type == "FORUM": if not post["isReply"]: thread_funnel = post.get("funnel") msg_text = _("[{author}]({author_url}) created [{title}](<{url}f/p/{threadId}>) in {forumName}") if thread_funnel == "POLL": msg_text = _("[{author}]({author_url}) created a poll [{title}](<{url}f/p/{threadId}>) in {forumName}") elif thread_funnel == "QUIZ": msg_text = _("[{author}]({author_url}) created a quiz [{title}](<{url}f/p/{threadId}>) in {forumName}") elif thread_funnel != "TEXT": discussion_logger.warning("The type of {} is an unknown discussion post type. Please post an issue on the project page to have it added https://gitlab.com/piotrex43/RcGcDw/-/issues.".format(thread_funnel)) message = "📝 "+msg_text.format(author=author, author_url=author_url, title=post["title"], url=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"], forumName=post["forumName"]) else: message = "📝 "+_("[{author}]({author_url}) created a [reply](<{url}f/p/{threadId}/r/{postId}>) to [{title}](<{url}f/p/{threadId}>) in {forumName}").format(author=author, author_url=author_url, url=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"], postId=post["id"], title=post["_embedded"]["thread"][0]["title"], forumName=post["forumName"]) elif post_type == "WALL": user_wall = _("unknown") # Fail safe if post["forumName"].endswith(' Message Wall'): user_wall = post["forumName"][:-13] if not post["isReply"]: message = "✉️ "+_("[{author}]({author_url}) created [{title}](<{url}wiki/Message_Wall:{user_wall}?threadId={threadId}>) on [{user}'s Message Wall](<{url}wiki/Message_Wall:{user_wall}>)").format(author=author, author_url=author_url, title=post["title"], url=settings["fandom_discussions"]["wiki_url"], user=user_wall, user_wall=quote_plus(user_wall.replace(" ", "_")), threadId=post["threadId"]) else: message = "📩 "+_("[{author}]({author_url}) created a [reply](<{url}wiki/Message_Wall:{user_wall}?threadId={threadId}#{replyId}>) to [{title}](<{url}wiki/Message_Wall:{user_wall}?threadId={threadId}>) on [{user}'s Message Wall](<{url}wiki/Message_Wall:{user_wall}>)").format(author=author, author_url=author_url, url=settings["fandom_discussions"]["wiki_url"], title=post["_embedded"]["thread"][0]["title"], user=user_wall, user_wall=quote_plus(user_wall.replace(" ", "_")), threadId=post["threadId"], replyId=post["id"]) elif post_type == "ARTICLE_COMMENT": if article_paths is None: article_paths = {"title": _("unknown"), "fullUrl": "{wiki}wiki/{article}".format(wiki=settings["fandom_discussions"]["wiki_url"], article=_("unknown").replace(" ", "_"))} # No page known if not post["isReply"]: message = "🗒️ "+_( "[{author}]({author_url}) created a [comment](<{url}?commentId={commentId}>) on [{article}](<{url}>)").format( author=author, author_url=author_url, url=article_paths["fullUrl"], article=article_paths["title"], commentId=post["threadId"]) else: message = "🗒️ "+_( "[{author}]({author_url}) created a [reply](<{url}?commentId={commentId}&replyId={replyId}>) to a [comment](<{url}?commentId={commentId}>) on [{article}](<{url}>)").format( author=author, author_url=author_url, url=article_paths["fullUrl"], article=article_paths["title"], commentId=post["threadId"], replyId=post["id"]) else: discussion_logger.warning("No entry for {event} with params: {params}".format(event=post_type, params=post)) if not settings["support"]: return else: message = "❓ "+_("Unknown event `{event}` by [{author}]({author_url}), report it on the [support server](<{support}>).").format( event=post_type, author=author, author_url=author_url, support=settings["support"]) send_to_discord(DiscordMessage("compact", "discussion", settings["fandom_discussions"]["webhookURL"], content=message), meta={"request_type": "POST"}) def embed_formatter(post_type, post, article_paths): """Embed formatter for Fandom discussions.""" embed = DiscordMessage("embed", "discussion", settings["fandom_discussions"]["webhookURL"]) if post_type == "FORUM": embed.set_author(post["createdBy"]["name"], "{url}f/u/{creatorId}".format(url=settings["fandom_discussions"]["wiki_url"], creatorId=post["creatorId"]), icon_url=post["createdBy"]["avatarUrl"]) elif post["creatorIp"]: embed.set_author(post["creatorIp"][1:], "{url}wiki/Special:Contributions{creatorIp}".format(url=settings["fandom_discussions"]["wiki_url"], creatorIp=post["creatorIp"])) else: embed.set_author(post["createdBy"]["name"], "{url}wiki/User:{creator}".format(url=settings["fandom_discussions"]["wiki_url"], creator=post["createdBy"]["name"]), icon_url=post["createdBy"]["avatarUrl"]) if settings["fandom_discussions"]["appearance"]["embed"]["show_content"]: if post.get("jsonModel") is not None: npost = DiscussionsFromHellParser(post) embed["description"] = npost.parse() if npost.image_last: embed["image"]["url"] = npost.image_last embed["description"] = embed["description"].replace(npost.image_last, "") else: # Fallback when model is not available embed["description"] = post.get("rawContent", "") embed["footer"]["text"] = post["forumName"] embed["timestamp"] = datetime.datetime.fromtimestamp(post["creationDate"]["epochSecond"], tz=datetime.timezone.utc).isoformat() if post_type == "FORUM": if not post["isReply"]: embed["url"] = "{url}f/p/{threadId}".format(url=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"]) embed["title"] = _("Created \"{title}\"").format(title=post["title"]) thread_funnel = post.get("funnel") if thread_funnel == "POLL": embed.event_type = "discussion/forum/poll" embed["title"] = _("Created a poll \"{title}\"").format(title=post["title"]) if settings["fandom_discussions"]["appearance"]["embed"]["show_content"]: poll = post["poll"] image_type = False if poll["answers"][0]["image"] is not None: image_type = True for num, option in enumerate(poll["answers"]): embed.add_field(option["text"] if image_type is True else _("Option {}").format(num+1), option["text"] if image_type is False else _("__[View image]({image_url})__").format(image_url=option["image"]["url"]), inline=True) elif thread_funnel == "QUIZ": embed.event_type = "discussion/forum/quiz" embed["title"] = _("Created a quiz \"{title}\"").format(title=post["title"]) if settings["fandom_discussions"]["appearance"]["embed"]["show_content"]: quiz = post["_embedded"]["quizzes"][0] embed["description"] = quiz["title"] if quiz["image"] is not None: embed["image"]["url"] = quiz["image"] elif thread_funnel == "TEXT": embed.event_type = "discussion/forum/post" else: discussion_logger.warning("The type of {} is an unknown discussion post type. Please post an issue on the project page to have it added https://gitlab.com/piotrex43/RcGcDw/-/issues.".format(thread_funnel)) if post["_embedded"]["thread"][0]["tags"]: tag_displayname = [] for tag in post["_embedded"]["thread"][0]["tags"]: tag_displayname.append("[{title}]({url})".format(title=tag["articleTitle"], url=create_article_path(tag["articleTitle"]))) if len(", ".join(tag_displayname)) > 1000: embed.add_field(_("Tags"), _("{} tags").format(len(post["_embedded"]["thread"][0]["tags"]))) else: embed.add_field(_("Tags"), ", ".join(tag_displayname)) else: embed.event_type = "discussion/forum/reply" embed["title"] = _("Replied to \"{title}\"").format(title=post["_embedded"]["thread"][0]["title"]) embed["url"] = "{url}f/p/{threadId}/r/{postId}".format(url=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"], postId=post["id"]) elif post_type == "WALL": user_wall = _("unknown") # Fail safe if post["forumName"].endswith(' Message Wall'): user_wall = post["forumName"][:-13] if not post["isReply"]: embed.event_type = "discussion/wall/post" embed["url"] = "{url}wiki/Message_Wall:{user_wall}?threadId={threadId}".format(url=settings["fandom_discussions"]["wiki_url"], user_wall=quote_plus(user_wall.replace(" ", "_")), threadId=post["threadId"]) embed["title"] = _("Created \"{title}\" on {user}'s Message Wall").format(title=post["title"], user=user_wall) else: embed.event_type = "discussion/wall/reply" embed["url"] = "{url}wiki/Message_Wall:{user_wall}?threadId={threadId}#{replyId}".format(url=settings["fandom_discussions"]["wiki_url"], user_wall=quote_plus(user_wall.replace(" ", "_")), threadId=post["threadId"], replyId=post["id"]) embed["title"] = _("Replied to \"{title}\" on {user}'s Message Wall").format(title=post["_embedded"]["thread"][0]["title"], user=user_wall) elif post_type == "ARTICLE_COMMENT": if article_paths is None: article_page = {"title": _("unknown"), "fullUrl": "{wiki}wiki/{article}".format(wiki=settings["fandom_discussions"]["wiki_url"], article=_( "unknown").replace(" ", "_"))} # No page known if not post["isReply"]: embed.event_type = "discussion/comment/post" embed["url"] = "{url}?commentId={commentId}".format(url=article_paths["fullUrl"], commentId=post["threadId"]) embed["title"] = _("Commented on {article}").format(article=article_paths["title"]) else: embed.event_type = "discussion/comment/reply" embed["url"] = "{url}?commentId={commentId}&replyId={replyId}".format(url=article_paths["fullUrl"], commentId=post["threadId"], replyId=post["id"]) embed["title"] = _("Replied to a comment on {article}").format(article=article_paths["title"]) embed["footer"]["text"] = article_paths["title"] else: discussion_logger.warning("No entry for {event} with params: {params}".format(event=post_type, params=post)) embed["title"] = _("Unknown event `{event}`").format(event=post_type) embed["color"] = 0 if settings.get("support", None): change_params = "[```json\n{params}\n```]({support})".format(params=json.dumps(post, indent=2), support=settings["support"]) if len(change_params) > 1000: embed.add_field(_("Report this on the support server"), settings["support"]) else: embed.add_field(_("Report this on the support server"), change_params) embed.finish_embed() send_to_discord(embed, meta={"request_type": "POST"}) class DiscussionsFromHellParser: """This class converts fairly convoluted Fandom jsonModal of a discussion post into Markdown formatted usable thing. Takes string, returns string. Kudos to MarkusRost for allowing me to implement this formatter based on his code in Wiki-Bot.""" def __init__(self, post): self.post = post self.jsonModal = json.loads(post.get("jsonModel", "{}")) self.markdown_text = "" self.item_num = 1 self.image_last = None def parse(self) -> str: """Main parsing logic""" self.parse_content(self.jsonModal["content"]) if len(self.markdown_text) > 2000: self.markdown_text = self.markdown_text[0:2000] + "…" return self.markdown_text def parse_content(self, content, ctype=None): self.image_last = None for item in content: if ctype == "bulletList": self.markdown_text += "\t• " if ctype == "orderedList": self.markdown_text += "\t{num}. ".format(num=self.item_num) self.item_num += 1 if item["type"] == "text": if "marks" in item: prefix, suffix = self.convert_marks(item["marks"]) self.markdown_text = "{old}{pre}{text}{suf}".format(old=self.markdown_text, pre=prefix, text=escape_formatting(item["text"]), suf=suffix) else: if ctype == "code_block": self.markdown_text += item["text"] # ignore formatting on preformatted text which cannot have additional formatting anyways else: self.markdown_text += escape_formatting(item["text"]) elif item["type"] == "paragraph": if "content" in item: self.parse_content(item["content"], item["type"]) self.markdown_text += "\n" elif item["type"] == "openGraph": if not item["attrs"]["wasAddedWithInlineLink"]: self.markdown_text = "{old}{link}\n".format(old=self.markdown_text, link=item["attrs"]["url"]) elif item["type"] == "image": try: discussion_logger.debug(item["attrs"]["id"]) if item["attrs"]["id"] is not None: self.markdown_text = "{old}{img_url}\n".format(old=self.markdown_text, img_url=self.post["_embedded"]["contentImages"][int(item["attrs"]["id"])]["url"]) self.image_last = self.post["_embedded"]["contentImages"][int(item["attrs"]["id"])]["url"] except (IndexError, ValueError): discussion_logger.warning("Image {} not found.".format(item["attrs"]["id"])) discussion_logger.debug(self.markdown_text) elif item["type"] == "code_block": self.markdown_text += "```\n" if "content" in item: self.parse_content(item["content"], item["type"]) self.markdown_text += "\n```\n" elif item["type"] == "bulletList": if "content" in item: self.parse_content(item["content"], item["type"]) elif item["type"] == "orderedList": self.item_num = 1 if "content" in item: self.parse_content(item["content"], item["type"]) elif item["type"] == "listItem": self.parse_content(item["content"], item["type"]) def convert_marks(self, marks): prefix = "" suffix = "" for mark in marks: if mark["type"] == "mention": prefix += "[" suffix = "]({wiki}f/u/{userid}){suffix}".format(wiki=settings["fandom_discussions"]["wiki_url"], userid=mark["attrs"]["userId"], suffix=suffix) elif mark["type"] == "strong": prefix += "**" suffix = "**{suffix}".format(suffix=suffix) elif mark["type"] == "link": prefix += "[" suffix = "]({link}){suffix}".format(link=mark["attrs"]["href"], suffix=suffix) elif mark["type"] == "em": prefix += "_" suffix = "_" + suffix return prefix, suffix