Initial Release of IGNCore version 2.5

modules/standard/wants/wants_controller.py

@@ -0,0 +1,113 @@
+import time
+
+from core.chat_blob import ChatBlob
+from core.command_param_types import Any, Int, Const, Options, Item
+from core.decorators import instance, command
+
+
+@instance()
+class WantsController:
+    def inject(self, registry):
+        self.db = registry.get_instance("db")
+        self.text = registry.get_instance("text")
+        self.account_service = registry.get_instance("account_service")
+
+    def start(self):
+        self.db.exec("CREATE TABLE IF NOT EXISTS wants ("
+                     "id INT PRIMARY KEY AUTO_INCREMENT, "
+                     "char_id INT NOT NULL, "
+                     "want TEXT NOT NULL,"
+                     "created_at INT NOT NULL,"
+                     "INDEX `char_id` (`char_id`))")
+
+    @command(command="wants", params=[], access_level="member",
+             description="Show your wants")
+    def wants_list_cmd(self, request):
+        alts = self.account_service.get_alts(request.sender.char_id)
+
+        cnt = 0
+        blob = ""
+        for alt in alts:
+            data = self.db.query("SELECT * FROM wants WHERE char_id = ? ORDER BY created_at DESC", [alt.char_id])
+            alt_cnt = len(data)
+            cnt += alt_cnt
+
+            if alt_cnt:
+                for row in data:
+                    blob += "%s %s\n\n" % (row.want, self.text.make_tellcmd("Remove", "wants remove %d" % row.id))
+
+        return ChatBlob("Wants for %s (%d)" % (alts[0].name, cnt), blob)
+
+    @command(command="wants", params=[Const("add"), Any("item")], access_level="member",
+             description="Add a want")
+    def wants_add_cmd(self, request, _, want):
+        self.db.exec("INSERT INTO wants (char_id, want, created_at) "
+                     "VALUES (?, ?, ?)",
+                     [request.sender.char_id, want, int(time.time())])
+
+        return "Want added successfully."
+
+    @command(command="wants", params=[Options(["rem", "remove"]), Int("want_id")], access_level="member",
+             description="Remove a want")
+    def wants_remove_cmd(self, request, _, want_id):
+        want = self.db.query_single("SELECT n.*, p.name FROM wants n "
+                                    "LEFT JOIN player p ON n.char_id = p.char_id "
+                                    "WHERE n.id = ?",
+                                    [want_id])
+
+        if not want:
+            return f"Could not find want with ID <highlight>{want_id:d}</highlight>."
+
+        if self.account_service.get_main(request.sender.char_id).char_id \
+                != self.account_service.get_main(want.char_id).char_id:
+            return f"You must be a confirmed alt of <highlight>{want.name}</highlight> to remove this want."
+
+        self.db.exec("DELETE FROM wants WHERE id = ?", [want_id])
+
+        return f"Want with ID <highlight>{want_id:d}</highlight> deleted successfully."
+
+    @command(command="wants", params=[Const("search"), Item("item")], access_level="member",
+             description="Search wants by itemref")
+    def wants_search_itemref_cmd(self, request, _, item):
+        return self.search_wants(item.name)
+
+    @command(command="wants", params=[Const("search"), Any("name")], access_level="member",
+             description="Search wants by name")
+    def wants_search_name_cmd(self, request, _, wants_search):
+        return self.search_wants(wants_search)
+
+    def search_wants(self, wants_search):
+        wants = self.db.query("SELECT w.char_id, w.want, p.name FROM wants w "
+                              "LEFT JOIN player p ON w.char_id = p.char_id "
+                              "WHERE want LIKE ?",
+                              [f"%{wants_search}%"])
+
+        blob = ""
+        for want in wants:
+            alts = self.account_service.get_alts(want.char_id)
+            main_name = alts[0].name
+            blob += f"<header2>{main_name}</header2>\n{want.want}\n\n"
+
+        return ChatBlob("Search Results (%d)" % len(wants), blob)
+
+    @command(command="wants", params=[Const("list")], access_level="member",
+             description="Shows all wants")
+    def wants_all_cmd(self, request, _):
+        sql = "SELECT w.*, p.name FROM wants w \
+               LEFT JOIN alts a ON w.char_id = a.char_id \
+               LEFT JOIN alts a2 ON (a2.group_id = a.group_id AND a2.status = 2) \
+               LEFT JOIN player p ON p.char_id = COALESCE(a2.char_id, w.char_id) \
+               ORDER BY p.name"
+
+        data = self.db.query(sql)
+
+        blob = ""
+        current_main_name = ""
+        for want in data:
+            if want.name != current_main_name:
+                blob += f"\n<header2>{want.name}</header2>\n"
+                current_main_name = want.name
+
+            blob += want.want + "\n"
+
+        return ChatBlob(f"Wants List ({len(data):d})", blob)