Initial Release of IGNCore version 2.5

modules/standard/helpbot/whompah_controller.py

@@ -0,0 +1,108 @@
+from core.chat_blob import ChatBlob
+from core.command_param_types import Any
+from core.db import DB
+from core.decorators import instance, command
+from core.text import Text
+
+
+@instance()
+class WhompahController:
+
+    def inject(self, registry):
+        self.db: DB = registry.get_instance("db")
+        self.text: Text = registry.get_instance("text")
+
+    def pre_start(self):
+        self.db.load_sql_file(self.module_dir + "/sql/" + "whompah_cities.sql", pre_optimized=True)
+        self.db.create_view("whompah_cities")
+        self.db.create_view("whompah_cities_rel")
+
+    @command(command="whompah", params=[], access_level="all",
+             description="Show list of whompah cities")
+    def whompah_list_cmd(self, request):
+        cities = self.db.query("SELECT id, city_name, zone, faction, short_name FROM whompah_cities ORDER BY city_name")
+
+        blob = ""
+        for city in cities:
+            blob += f"{self.text.get_formatted_faction(city.faction.lower(), city.city_name)} " \
+                    f"({self.text.make_tellcmd(city.short_name, f'whompah {city.short_name}')})\n"
+
+        return ChatBlob("Whompah Cities", blob)
+
+    @command(command="whompah", params=[Any("city1"), Any("city2")], access_level="all",
+             description="Show whompah route between two cities")
+    def whompah_travel_cmd(self, request, city_name1, city_name2):
+        city1 = self.get_whompah_city(city_name1)
+        city2 = self.get_whompah_city(city_name2)
+
+        if not city1:
+            return f"Could not find whompah city <highlight>{city_name1}</highlight>."
+        elif not city2:
+            return f"Could not find whompah city <highlight>{city_name2}</highlight>."
+
+        data = self.db.query("SELECT w1.*, w2.city2_id AS city_rel FROM whompah_cities w1 "
+                             "JOIN whompah_cities_rel w2 ON w1.id = w2.city1_id")
+
+        cities = {}
+        for city in data:
+            rel = cities.get(city.id, {}).get("rel", [])
+            rel.append(city.city_rel)
+            cities[city.id] = city
+            cities[city.id]["rel"] = rel
+
+        path = self.format_path(self.find_path(cities, city1.id, city2.id))
+        return " -> ".join(path)
+
+    @command(command="whompah", params=[Any("city")], access_level="all",
+             description="Show whompah destinations for a city")
+    def whompah_city_cmd(self, request, city_name):
+        city = self.get_whompah_city(city_name)
+
+        if not city:
+            return f"Could not find whompah city <highlight>{city_name}</highlight>."
+
+        cities = self.db.query("SELECT w2.* FROM whompah_cities_rel w1 "
+                               "JOIN whompah_cities w2 ON w1.city2_id = w2.id "
+                               "WHERE w1.city1_id = ?", [city.id])
+        msg = f"From {city.city_name} you can get to: "
+        msg += ", ".join(map(lambda x: f"{self.text.get_formatted_faction(x.faction.lower(), x.city_name)} "
+                                       f"({x.short_name})", cities))
+        return msg
+
+    def get_whompah_city(self, city):
+        return self.db.query_single("SELECT id, city_name, zone, faction, short_name FROM whompah_cities "
+                                    "WHERE city_name LIKE ? "
+                                    "OR short_name LIKE ?", [city, city])
+
+    def find_path(self, cities, start_city_id, end_city_id):
+        def get_and_remove(key):
+            value = cities[key]
+            del cities[key]
+            return value
+
+        end_city = get_and_remove(end_city_id)
+
+        if start_city_id == end_city_id:
+            return end_city
+
+        # create stack and initialize
+        # start with ending city and traverse backwards so we don't have to reverse the result
+        stack = [end_city]
+        while stack:
+            root = stack.pop(0)
+            for rel in root["rel"]:
+                if rel in cities:
+                    c = get_and_remove(rel)
+                    c["parent"] = root
+                    if c["id"] == start_city_id:
+                        return c
+                    stack.append(c)
+        return None
+
+    def format_path(self, path):
+        result = []
+        root = path
+        while root:
+            result.append(self.text.get_formatted_faction(root["faction"].lower(), root["city_name"]))
+            root = root.get("parent")
+        return result