Initial Release of IGNCore version 2.5

modules/standard/helpbot/playfield_controller.py

@@ -0,0 +1,91 @@
+from core.chat_blob import ChatBlob
+from core.command_param_types import Regex, Int, Any, Const
+from core.db import DB
+from core.decorators import instance, command
+from core.text import Text
+
+
+@instance()
+class PlayfieldController:
+    def inject(self, registry):
+        self.db: DB = registry.get_instance("db")
+        self.text: Text = registry.get_instance("text")
+        self.command_alias_service = registry.get_instance("command_alias_service")
+
+    def pre_start(self):
+        self.db.load_sql_file(self.module_dir + "/sql/" + "playfields.sql", pre_optimized=True)
+        self.db.create_view("playfields")
+
+    def start(self):
+        self.command_alias_service.add_alias("playfields", "playfield")
+
+    @command(command="playfield", params=[Const("all", is_optional=True)], access_level="member",
+             description="Show a list of playfields")
+    def playfield_list_command(self, _, const_all):
+        if const_all:
+            data = self.db.query("SELECT * FROM playfields ORDER BY long_name")
+        else:
+            data = self.db.query("SELECT * FROM playfields WHERE short_name != '' ORDER BY long_name")
+
+        blob = ""
+        for row in data:
+            blob += "[<highlight>%d</highlight>] %s (%s)\n" % (row.id, row.long_name, row.short_name)
+
+        return ChatBlob("Playfields", blob)
+
+    @command(command="waypoint",
+             params=[Regex("waypoint_data", r"\s+.*?Pos: ([0-9.]+), ([0-9.]+), ([0-9.]+), Area: ([a-zA-Z ]+).*",
+                           num_groups=4)],
+             access_level="member",
+             description="Create a waypoint link from F9 output",
+             extended_description="Example: <symbol>waypoint Pos: 123.1, 456.1, 789.1, Area: Perpetual Wastelands")
+    def waypoint1_command(self, _, regex):
+        x_coords, y_coords, _, playfield_arg = regex
+
+        return self.create_waypoint_blob(x_coords, y_coords, playfield_arg)
+
+    @command(command="waypoint",
+             params=[Regex("waypoint_data", r"\s+.*?([0-9.]+) ([0-9.]+) y ([0-9.]+) ([0-9]+).*",
+                           num_groups=4)],
+             access_level="member",
+             description="Create a waypoint link from Shift + F9 output",
+             extended_description="Example: <symbol>waypoint 123.1 456.1 y 789.1 570")
+    def waypoint2_command(self, _, regex):
+        x_coords, y_coords, _, playfield_arg = regex
+
+        return self.create_waypoint_blob(x_coords, y_coords, playfield_arg)
+
+    @command(command="waypoint",
+             params=[Int("x_coords"), Int("y_coords"), Any("playfield")],
+             access_level="member",
+             description="Manually create a waypoint link",
+             extended_description="Example: !waypoint 123 456 PW")
+    def waypoint3_command(self, _, x_coords, y_coords, playfield_arg):
+        return self.create_waypoint_blob(x_coords, y_coords, playfield_arg)
+
+    def create_waypoint_blob(self, x_coords, y_coords, playfield_arg):
+        x_coords = int(float(x_coords))
+        y_coords = int(float(y_coords))
+        playfield = self.get_playfield_by_name(playfield_arg) or self.get_playfield_by_id(playfield_arg)
+
+        if not playfield:
+            return f"Could not find playfield <highlight>{playfield_arg}</highlight>."
+        else:
+            title = f"waypoint: {x_coords}x{y_coords} {playfield.long_name}"
+            blob = f"Zone: {playfield.long_name} ({playfield.id})\n"
+            blob += f"Coords: {x_coords} x {y_coords}\n\n"
+            waypoint = f'/waypoint {x_coords} {y_coords} {playfield.id:d}'
+            blob += f"<center>{self.text.make_chatcmd(self.text.make_image(11336), waypoint)}\n"
+            blob += self.text.make_chatcmd("Click for waypoint", f"/waypoint {x_coords} {y_coords} {playfield.id:d}")
+
+            return ChatBlob(title, blob)
+
+    def get_playfield_by_name(self, name):
+        return self.db.query_single("SELECT * FROM playfields "
+                                    "WHERE long_name LIKE ? "
+                                    "OR short_name LIKE ? "
+                                    "LIMIT 1", [name, name])
+
+    def get_playfield_by_id(self, playfield_id):
+        return self.db.query_single("SELECT * FROM playfields "
+                                    "WHERE id = ?", [playfield_id])