Initial Release of IGNCore version 2.5

modules/core/ban/ban_controller.py

@@ -0,0 +1,79 @@
+import hjson
+
+from core.aochat.BaseModule import BaseModule
+from core.chat_blob import ChatBlob
+from core.command_param_types import Any, Const, Options, Time, Character
+from core.db import DB
+from core.decorators import instance, command
+from core.lookup.character_service import CharacterService
+from core.translation_service import TranslationService
+from core.tyrbot import Tyrbot
+from modules.core.ban.ban_service import BanService
+
+
+@instance()
+class BanController(BaseModule):
+    # noinspection DuplicatedCode
+    def inject(self, registry):
+        self.bot: Tyrbot = registry.get_instance("bot")
+        self.db: DB = registry.get_instance("db")
+        self.text = registry.get_instance("text")
+        self.util = registry.get_instance("util")
+        self.ban_service: BanService = registry.get_instance("ban_service")
+        self.command_alias_service = registry.get_instance("command_alias_service")
+        self.ts: TranslationService = registry.get_instance("translation_service")
+        self.character_service: CharacterService = registry.get_instance("character_service")
+        self.getresp = self.ts.get_response
+
+    def start(self):
+        self.command_alias_service.add_alias("unban", "ban rem")
+        self.ts.register_translation("module/ban", self.load_ban_msg)
+
+    def load_ban_msg(self):
+        with open("modules/core/ban/ban.msg", mode="r", encoding="UTF-8") as f:
+            return hjson.load(f)
+
+    @command(command="ban", params=[Const("list", is_optional=True)], access_level="moderator",
+             description="Show the ban list")
+    def ban_list_cmd(self, _, _1):
+        query = self.db.query("SELECT a.*, p.*, b.reason, b.finished_at, b.created_at, b.sender_char_id from account a "
+                              "left join player p on a.char_id = p.char_id "
+                              "left join ban_list b on a.char_id = b.char_id "
+                              "where main=a.char_id "
+                              "and disabled=1 "
+                              "and p.org_id NOT IN (SELECT org_id from ban_org_list)")
+        blob = ""
+        for row in query:
+            ends = "never" if (row.finished_at or -1) == -1 else self.util.format_datetime(row.finished_at)
+            blob += f"<highlight>{row.name}</highlight> by " \
+                    f"{self.character_service.resolve_char_to_name(row.sender_char_id or self.bot.get_char_id())}\n"
+            if row.sender_char_id != 0 and row.sender_char_id is not None:
+                blob += f"<tab>Added at: <highlight>{self.util.format_datetime(row.created_at)}</highlight> " \
+                        f"Ends: <highlight>{ends}</highlight>\n"
+                blob += f"<tab>Reason: {row.reason or 'None given'}\n"
+            blob += "\n"
+        return ChatBlob("Banned Accounts", blob)
+
+    @command(command="ban", params=[Options(["rem", "remove"]), Character("character")], access_level="moderator",
+             description="Remove a character from the ban list")
+    def ban_remove_cmd(self, _, _1, char):
+        if not char.char_id:
+            return self.getresp("global", "char_not_found", {"char": char.name})
+        elif not self.ban_service.get_ban(char.char_id):
+            return self.getresp("module/ban", "not_banned", {"char": char.name})
+        else:
+            self.ban_service.remove_ban(char.char_id)
+            return self.getresp("module/ban", "unbanned_self", {"char": char.name})
+
+    @command(command="ban",
+             params=[Const("add", is_optional=True), Character("character"), Time("duration", is_optional=True),
+                     Any("reason", is_optional=True)], access_level="moderator",
+             description="Add a character to the ban list")
+    def ban_add_cmd(self, request, _, char, duration, reason):
+        if not char.char_id:
+            return self.getresp("global", "char_not_found", {"char": char.name})
+        elif self.ban_service.get_ban(char.char_id):
+            return f"<highlight>{char.name}</highlight> is already banned."
+        else:
+            self.ban_service.add_ban(char.char_id, request.sender.char_id, duration, reason)
+            request.reply(f"<highlight>{char.name}</highlight> has been added to the ban list.")