Initial Release of IGNCore version 2.5

2021-08-09 13:18:56 +02:00
commit a83d98c47e
910 changed files with 224171 additions and 0 deletions
@@ -0,0 +1,179 @@
+import html
+import re
+from typing import List
+
+import requests
+from bs4 import BeautifulSoup
+
+from core.chat_blob import ChatBlob
+from core.command_param_types import Any, Int, Item
+from core.decorators import instance, command
+from core.text import Text
+from modules.standard.items.auno_comment import AunoComment
+from modules.standard.items.items_controller import ItemsController
+
+
+@instance()
+class AunoController:
+    def __init__(self):
+        pass
+
+    def inject(self, registry):
+        self.text: Text = registry.get_instance("text")
+        self.items_controller: ItemsController = registry.get_instance("items_controller")
+
+    @command(command="auno", params=[Int("item_id")], access_level="member",
+             description="Fetch comments for item from Auno by item id")
+    def auno_comments_item_id_cmd(self, _, item_id):
+        item = self.items_controller.get_by_item_id(item_id)
+        if item:
+            low_id = item.lowid
+            high_id = item.highid
+            name = item.name
+        else:
+            low_id = item_id
+            high_id = item_id
+            name = item_id
+
+        return self.get_combined_response(low_id, high_id, name)
+
+    @command(command="auno", params=[Item("item_link")], access_level="member",
+             description="Fetch comments for item from Auno by item link")
+    def auno_comments_item_link_cmd(self, _, item):
+        return self.get_combined_response(item.low_id, item.high_id, item.name)
+
+    @command(command="auno", params=[Any("search")], access_level="member",
+             description="Fetch comments for item from Auno by search")
+    def auno_comments_cmd(self, _, search):
+        items = self.items_controller.find_items(search)
+        count = len(items)
+
+        if count > 0:
+            if count > 1:
+                link_txt = f"Multiple search results for \"{search}\" ({count})"
+                return ChatBlob(link_txt, self.multiple_results_blob(items, search))
+            else:
+                return self.get_combined_response(items[0].lowid, items[0].highid, items[0].name)
+        else:
+            return "No items found matching <highlight>%s</highlight>." % search
+
+    def get_combined_response(self, low_id, high_id, name):
+        combined_response = self.get_auno_response(low_id, high_id)
+
+        if len(combined_response) > 0:
+            # high id comments
+            soup = BeautifulSoup(combined_response[0].text, features="html.parser")
+            comments: List[AunoComment] = self.find_comments(soup)
+
+            if len(combined_response) > 1:
+                # low id comments
+                soup = BeautifulSoup(combined_response[1].text, features="html.parser")
+                comments += self.find_comments(soup)
+
+            # sort the comments by date
+            comments.sort(key=lambda comment: comment.date)
+
+            if len(comments) > 0:
+                return ChatBlob(f"Comments for {name} ({len(comments)})",
+                                self.build_comments_blob(comments, name, low_id, high_id))
+            else:
+                return f"No comments found for <highlight>{name}</highlight>."
+        else:
+            return "Error fetching comments from Auno.org."
+
+    def build_comments_blob(self, comments, name, low_id, high_id):
+        link_auno = self.text.make_chatcmd("Auno", f"/start {self.get_auno_request_url(high_id)}")
+        link_aoitems = self.text.make_chatcmd("AOItems", f"/start {self.get_aoitems_request_url(high_id)}")
+
+        item = self.items_controller.get_by_item_id(high_id)
+        blob = ""
+        if item:
+            ql = item.highql
+            blob += f"Item: {self.text.make_item(int(low_id), int(high_id), int(ql), name)}\n"
+        blob += f"Item links: [{link_auno}] [{link_aoitems}]\n\n"
+        blob += "<header2>Comments</header2>\n"
+
+        for comment in comments:
+            blob += html.unescape(comment.content) + "\n"
+            blob += f"<highlight>{comment.author}</highlight> [<grey>{comment.date}</grey>]\n"
+            blob += "\n<pagebreak>"
+
+        return blob
+
+    def multiple_results_blob(self, items, search):
+        max_multiple_results = 40
+
+        blob = f"Found <highlight>{len(items)}</highlight> items matching <highlight>\"{search}\"</highlight>\n"
+        if len(items) > max_multiple_results:
+            blob += "Results have been truncated to only show the first %s results...\n\n" % max_multiple_results
+            items = items[:max_multiple_results]
+
+        for i, item in enumerate(items):
+            itemref = self.text.make_item(item.lowid, item.highid, item.highql, item.name)
+            comments_link = self.text.make_tellcmd("Comments", f"auno {item.highid}")
+            auno_link_h = self.text.make_chatcmd("Auno", f"/start {self.get_auno_request_url(item.highid)}")
+            blob += f"{i + 1}. {itemref}\n | [{comments_link}] [{auno_link_h}]"
+            blob += "\n\n<pagebreak>"
+
+        return blob
+
+    def find_comments(self, soup):
+        comments: List[AunoComment] = []
+
+        brs = soup.find_all("br")
+        for br in brs:
+            br.replace_with("\n")
+
+        trs = soup.find_all(self.tr_contains_comment)
+        for tr in trs:
+            author, date = tr.find("span").string.split("@")
+
+            comment_content = tr.find("div")
+            comment = ""
+
+            for content in comment_content:
+                if type(content) is str:
+                    comment += html.escape(content)
+                else:
+                    comment += html.escape(content.string)
+
+            # when those halfwits think a billion linebreaks are necessary
+            # and because auno's comment output is ever so lovely...
+            # noinspection RegExpUnnecessaryNonCapturingGroup
+            comment = re.sub(r"(\n(?:\n)*(?:\s)*)", "\n", comment)
+
+            comments.append(AunoComment(author.strip(), date.strip(), comment.strip()))
+        return comments
+
+    def tr_contains_comment(self, tag):
+        if tag:
+            if 'id' in tag.attrs:
+                p = re.compile(r"aoc\d+")
+                return p.match(tag['id'])
+
+    def get_auno_response(self, low_id, high_id):
+        auno_request_low = self.get_auno_request_url(low_id)
+        auno_request_high = self.get_auno_request_url(high_id)
+
+        auno_response_h = requests.get(auno_request_high, timeout=5)
+        auno_response_l = None
+
+        if low_id != high_id:
+            auno_response_l = requests.get(auno_request_low, timeout=5)
+
+        combined_response = []
+
+        if auno_response_h:
+            if auno_response_h.status_code == 200:
+                combined_response.append(auno_response_h)
+        if auno_response_l:
+            if auno_response_l.status_code == 200:
+                combined_response.append(auno_response_l)
+
+        return combined_response
+
+    def get_auno_request_url(self, item_id):
+        return "https://auno.org/ao/db.php?id=%s" % item_id
+
+    def get_aoitems_request_url(self, item_id):
+        return "https://aoitems.com/item/%s/" % item_id