8 files changed, 297 insertions, 37 deletions

@@ -65,6 +65,11 @@ class AssignIds {
         UpdateNextId(room.panels());
         UpdateNextId(room.masteries());
         UpdateNextId(room.keyholders());
+        UpdateNextId(room.ports());
+      }
+
+      if (map.has_rte()) {
+        UpdateNextId(map.rte());
       }
     }
 
@@ -92,10 +97,31 @@ class AssignIds {
   void ProcessMap(std::filesystem::path path) {
     std::string map_name = path.filename().string();
 
+    ProcessMapMetadata(path / "metadata.txtpb", map_name);
     ProcessDoorsFile(path / "doors.txtpb", map_name);
     ProcessRooms(path / "rooms", map_name);
   }
 
+  void ProcessMapMetadata(std::filesystem::path path,
+                          const std::string& current_map_name) {
+    if (!std::filesystem::exists(path)) {
+      return;
+    }
+
+    auto metadata = ReadMessageFromFile<HumanMap>(path.string());
+    auto& maps = *output_.mutable_maps();
+
+    if (metadata.has_rte_room()) {
+      if (!id_mappings_.maps().contains(current_map_name) ||
+          !id_mappings_.maps().at(current_map_name).has_rte()) {
+        maps[current_map_name].set_rte(next_id_++);
+      } else {
+        maps[current_map_name].set_rte(
+            id_mappings_.maps().at(current_map_name).rte());
+      }
+    }
+  }
+
   void ProcessDoorsFile(std::filesystem::path path,
                         const std::string& current_map_name) {
     if (!std::filesystem::exists(path)) {
@@ -111,7 +137,8 @@ class AssignIds {
 
   void ProcessDoor(const HumanDoor& h_door,
                    const std::string& current_map_name) {
-    if (h_door.type() == DoorType::EVENT) {
+    if (h_door.type() == DoorType::EVENT && !h_door.latch() &&
+        !h_door.legacy_location()) {
       return;
     }
 
@@ -245,6 +272,37 @@ class AssignIds {
                                              .at(h_keyholder.name());
       }
     }
+
+    for (const HumanPort& h_port : h_room.ports()) {
+      if (h_port.no_shuffle()) {
+        continue;
+      }
+
+      auto& maps = *output_.mutable_maps();
+      auto& rooms = *maps[current_map_name].mutable_rooms();
+      auto& ports = *rooms[h_room.name()].mutable_ports();
+
+      if (!id_mappings_.maps().contains(current_map_name) ||
+          !id_mappings_.maps()
+               .at(current_map_name)
+               .rooms()
+               .contains(h_room.name()) ||
+          !id_mappings_.maps()
+               .at(current_map_name)
+               .rooms()
+               .at(h_room.name())
+               .ports()
+               .contains(h_port.name())) {
+        ports[h_port.name()] = next_id_++;
+      } else {
+        ports[h_port.name()] = id_mappings_.maps()
+                                   .at(current_map_name)
+                                   .rooms()
+                                   .at(h_room.name())
+                                   .ports()
+                                   .at(h_port.name());
+      }
+    }
   }
 
   void ProcessSpecialIds() {
@@ -309,9 +367,13 @@ class AssignIds {
  private:
   void UpdateNextId(const google::protobuf::Map<std::string, uint64_t>& ids) {
     for (const auto& [_, id] : ids) {
-      if (id > next_id_) {
-        next_id_ = id;
-      }
+      UpdateNextId(id);
+    }
+  }
+
+  void UpdateNextId(uint64_t id) {
+    if (id > next_id_) {
+      next_id_ = id;
     }
   }
 
@@ -88,9 +88,28 @@ class DataPacker {
     uint64_t map_id = container_.FindOrAddMap(map_name);
     Map& map = *container_.all_objects().mutable_maps(map_id);
 
+    map.set_type(metadata.type());
+    map.set_daedalus_only_mode(metadata.daedalus_only_mode());
+
     if (metadata.has_display_name()) {
       map.set_display_name(metadata.display_name());
     }
+
+    if (metadata.has_worldport_entrance()) {
+      map.set_worldport_entrance(container_.FindOrAddPort(
+          map_name, metadata.worldport_entrance().room(),
+          metadata.worldport_entrance().name(), std::nullopt, std::nullopt));
+    }
+
+    if (metadata.has_rte_room()) {
+      map.set_rte_room(container_.FindOrAddRoom(map_name, metadata.rte_room(),
+                                                std::nullopt));
+    }
+
+    if (metadata.has_rte_trigger_pos()) {
+      *map.mutable_rte_trigger_pos() = metadata.rte_trigger_pos();
+      *map.mutable_rte_trigger_scale() = metadata.rte_trigger_scale();
+    }
   }
 
   void ProcessRooms(std::filesystem::path path,
@@ -113,6 +132,10 @@ class DataPacker {
       room.set_panel_display_name(h_room.panel_display_name());
     }
 
+    if (h_room.has_daedalus_only_allow()) {
+      room.set_daedalus_only_allow(h_room.daedalus_only_allow());
+    }
+
     for (const HumanPanel& h_panel : h_room.panels()) {
       room.add_panels(ProcessPanel(h_panel, current_map_name, room.name()));
     }
@@ -186,6 +209,10 @@ class DataPacker {
       panel.set_display_name(h_panel.display_name());
     }
 
+    if (h_panel.has_exclude_from_panelsanity()) {
+      panel.set_exclude_from_panelsanity(h_panel.exclude_from_panelsanity());
+    }
+
     return panel_id;
   }
 
@@ -239,7 +266,14 @@ class DataPacker {
     Port& port = *container_.all_objects().mutable_ports(port_id);
 
     port.set_path(h_port.path());
-    port.set_orientation(h_port.orientation());
+    port.set_display_name(h_port.display_name());
+
+    if (h_port.no_shuffle()) {
+      port.set_no_shuffle(h_port.no_shuffle());
+    } else {
+      *port.mutable_destination() = h_port.destination();
+      port.set_rotation(h_port.rotation());
+    }
 
     // Setting this explicitly because the Godot protobuf doesn't support
     // custom defaults.
@@ -396,8 +430,8 @@ class DataPacker {
           container_.FindOrAddDoor(map_name, di.name(), current_map_name));
     }
 
-    for (const std::string& ending_name : h_door.endings()) {
-      door.add_endings(container_.FindOrAddEnding(ending_name));
+    if (h_door.has_white_ending()) {
+      door.set_white_ending(h_door.white_ending());
     }
 
     if (h_door.has_control_center_color()) {
@@ -417,6 +451,22 @@ class DataPacker {
     if (h_door.has_double_letters()) {
       door.set_double_letters(h_door.double_letters());
     }
+
+    if (h_door.has_latch()) {
+      door.set_latch(h_door.latch());
+    }
+
+    if (h_door.has_legacy_location()) {
+      door.set_legacy_location(h_door.legacy_location());
+    }
+
+    if (h_door.has_daedalus_only_allow()) {
+      door.set_daedalus_only_allow(h_door.daedalus_only_allow());
+    }
+
+    if (h_door.has_daedalus_only_always_item()) {
+      door.set_daedalus_only_always_item(h_door.daedalus_only_always_item());
+    }
   }
 
   void ProcessConnectionsFile(std::filesystem::path path,
@@ -483,6 +533,16 @@ class DataPacker {
       r_connection.set_cyan_ending(human_connection.cyan_ending());
     }
 
+    if (human_connection.has_mint_ending()) {
+      f_connection.set_mint_ending(human_connection.mint_ending());
+      r_connection.set_mint_ending(human_connection.mint_ending());
+    }
+
+    if (human_connection.has_vanilla_only()) {
+      f_connection.set_vanilla_only(human_connection.vanilla_only());
+      r_connection.set_vanilla_only(human_connection.vanilla_only());
+    }
+
     container_.AddConnection(f_connection);
     if (!human_connection.oneway()) {
       container_.AddConnection(r_connection);
@@ -612,6 +672,10 @@ class DataPacker {
           container_.FindOrAddDoor(di.map(), di.name(), std::nullopt);
       group.add_doors(door_id);
     }
+
+    if (h_group.has_daedalus_only_always_item()) {
+      group.set_daedalus_only_always_item(h_group.daedalus_only_always_item());
+    }
   }
 
   void ProcessGlobalMetadataFile(std::filesystem::path path) {
@@ -620,7 +684,7 @@ class DataPacker {
     }
 
     auto h_metadata = ReadMessageFromFile<HumanGlobalMetadata>(path.string());
-    container_.all_objects().set_version(h_metadata.version());
+    *container_.all_objects().mutable_version() = h_metadata.version();
   }
 
   void ProcessIdsFile(std::filesystem::path path) {
@@ -655,6 +719,17 @@ class DataPacker {
               .mutable_keyholders(keyholder_id)
               ->set_ap_id(ap_id);
         }
+
+        for (const auto& [port_name, ap_id] : room.ports()) {
+          uint64_t port_id = container_.FindOrAddPort(
+              map_name, room_name, port_name, std::nullopt, std::nullopt);
+          container_.all_objects().mutable_ports(port_id)->set_ap_id(ap_id);
+        }
+      }
+
+      if (map.has_rte()) {
+        uint64_t map_id = container_.FindOrAddMap(map_name);
+        container_.all_objects().mutable_maps(map_id)->set_rte_ap_id(map.rte());
       }
     }
 
@@ -64,6 +64,13 @@ IdMappings ReadIdsFromYaml(const std::string& filename) {
                   keyholder_it.second.as<uint64_t>();
             }
           }
+
+          if (room_it.second["ports"]) {
+            for (const auto& port_it : room_it.second["ports"]) {
+              (*room_ids.mutable_ports())[port_it.first.as<std::string>()] =
+                  port_it.second.as<uint64_t>();
+            }
+          }
         }
       }
 
@@ -73,6 +80,10 @@ IdMappings ReadIdsFromYaml(const std::string& filename) {
               door_it.second.as<uint64_t>();
         }
       }
+
+      if (map_it.second["rte"]) {
+        map_ids.set_rte(map_it.second["rte"].as<uint64_t>());
+      }
     }
   }
 
@@ -146,6 +157,12 @@ void WriteIdsAsYaml(const IdMappings& ids, const std::string& filename) {
                                    keyholder_id;
                              });
 
+          OperateOnSortedMap(
+              room_ids.ports(),
+              [&room_node](const std::string& port_name, uint64_t port_id) {
+                room_node["ports"][port_name] = port_id;
+              });
+
           map_node["rooms"][room_name] = std::move(room_node);
         });
 
@@ -155,6 +172,10 @@ void WriteIdsAsYaml(const IdMappings& ids, const std::string& filename) {
           map_node["doors"][door_name] = door_id;
         });
 
+    if (map_ids.has_rte()) {
+      map_node["rte"] = map_ids.rte();
+    }
+
     result["maps"][map_name] = std::move(map_node);
   });
 
@@ -74,9 +74,35 @@ class HumanProcessor {
     MapInfo& map_info = info_.maps[current_map_name];
 
     auto metadata = ReadMessageFromFile<HumanMap>(path.string());
+    map_info.definitions.push_back(metadata);
+
     for (const std::string& path : metadata.excluded_nodes()) {
       map_info.game_nodes[path].uses++;
     }
+
+    for (const std::string& path : metadata.custom_nodes()) {
+      map_info.game_nodes[path].defined = true;
+    }
+
+    if (metadata.has_worldport_entrance()) {
+      auto port_identifier = GetCompletePortIdentifier(
+          metadata.worldport_entrance(), current_map_name, std::nullopt);
+      if (port_identifier) {
+        PortInfo& port_info = info_.ports[*port_identifier];
+        port_info.map_worldport_entrances.push_back(current_map_name);
+      } else {
+        map_info.malformed_worldport_entrance = metadata.worldport_entrance();
+      }
+    }
+
+    if (metadata.has_rte_room()) {
+      RoomIdentifier room_identifier;
+      room_identifier.set_map(current_map_name);
+      room_identifier.set_name(metadata.rte_room());
+
+      RoomInfo& room_info = info_.rooms[room_identifier];
+      room_info.map_rtes_referenced_by.push_back(current_map_name);
+    }
   }
 
   void ProcessRooms(std::filesystem::path path,
@@ -358,11 +384,6 @@ class HumanProcessor {
       DoorInfo& other_door_info = info_.doors[complete_door_identifier];
       other_door_info.doors_referenced_by.push_back(door_identifier);
     }
-
-    for (const std::string& ei : h_door.endings()) {
-      EndingInfo& ending_info = info_.endings[ei];
-      ending_info.doors_referenced_by.push_back(door_identifier);
-    }
   }
 
   void ProcessConnectionsFile(std::filesystem::path path,
@@ -560,12 +581,14 @@ class HumanProcessor {
     auto ids = ReadIdsFromYaml(path.string());
 
     DoorIdentifier di;
-    PanelIdentifier pi;
+    PanelIdentifier pai;
+    PortIdentifier poi;
     KeyholderIdentifier ki;
 
     for (const auto& [map_name, map] : ids.maps()) {
       di.set_map(map_name);
-      pi.set_map(map_name);
+      pai.set_map(map_name);
+      poi.set_map(map_name);
       ki.set_map(map_name);
 
       for (const auto& [door_name, ap_id] : map.doors()) {
@@ -576,13 +599,14 @@ class HumanProcessor {
       }
 
       for (const auto& [room_name, room] : map.rooms()) {
-        pi.set_room(room_name);
+        pai.set_room(room_name);
+        poi.set_room(room_name);
         ki.set_room(room_name);
 
         for (const auto& [panel_name, ap_id] : room.panels()) {
-          pi.set_name(panel_name);
+          pai.set_name(panel_name);
 
-          PanelInfo& panel_info = info_.panels[pi];
+          PanelInfo& panel_info = info_.panels[pai];
           panel_info.has_id = true;
         }
 
@@ -596,6 +620,18 @@ class HumanProcessor {
           KeyholderInfo& keyholder_info = info_.keyholders[ki];
           keyholder_info.has_id = true;
         }
+
+        for (const auto& [port_name, ap_id] : room.ports()) {
+          poi.set_name(port_name);
+
+          PortInfo& port_info = info_.ports[poi];
+          port_info.has_id = true;
+        }
+      }
+
+      if (map.has_rte()) {
+        MapInfo& map_info = info_.maps[map_name];
+        map_info.has_rte_id = true;
       }
     }
 
@@ -21,7 +21,9 @@ void Run(const std::string& mapdir, const std::string& repodir) {
 int main(int argc, char** argv) {
   if (argc != 3) {
     std::cout << "Incorrect argument count." << std::endl;
-    std::cout << "Usage: validator [path to map directory] [path to Lingo 2 repository]" << std::endl;
+    std::cout << "Usage: validator [path to map directory] [path to Lingo 2 "
+                 "repository]"
+              << std::endl;
     return 1;
   }
 
@@ -27,6 +27,11 @@ struct GameNodeInfo {
 
 struct MapInfo {
   std::map<std::string, GameNodeInfo> game_nodes;
+
+  std::vector<HumanMap> definitions;
+  bool has_rte_id = false;
+
+  std::optional<PortIdentifier> malformed_worldport_entrance;
 };
 
 struct RoomInfo {
@@ -35,6 +40,7 @@ struct RoomInfo {
   std::vector<DoorIdentifier> doors_referenced_by;
   std::vector<PanelIdentifier> panels_referenced_by;
   std::vector<HumanConnection> connections_referenced_by;
+  std::vector<std::string> map_rtes_referenced_by;
 };
 
 struct DoorInfo {
@@ -54,9 +60,11 @@ struct DoorInfo {
 
 struct PortInfo {
   std::vector<HumanPort> definitions;
+  bool has_id = false;
 
   std::vector<HumanConnection> connections_referenced_by;
   std::vector<HumanConnection> target_connections_referenced_by;
+  std::vector<std::string> map_worldport_entrances;
 };
 
 struct PaintingInfo {
@@ -104,8 +112,6 @@ struct LetterInfo {
 struct EndingInfo {
   std::vector<RoomIdentifier> defined_in;
   bool has_id = false;
-
-  std::vector<DoorIdentifier> doors_referenced_by;
 };
 
 struct PanelNameInfo {
@@ -1,5 +1,6 @@
 #include "validator.h"
 
+#include <algorithm>
 #include <iostream>
 
 #include "proto/human.pb.h"
@@ -69,6 +70,27 @@ class Validator {
                   << " is not defined in the game file." << std::endl;
       }
     }
+
+    if (map_info.malformed_worldport_entrance) {
+      std::cout << "The worldport entrance for map " << map_name
+                << " is malformed." << std::endl;
+    }
+
+    if (map_info.has_rte_id) {
+      if (!std::any_of(
+              map_info.definitions.begin(), map_info.definitions.end(),
+              [](const HumanMap& h_map) { return h_map.has_rte_room(); })) {
+        std::cout << "Map " << map_name << " has an RTE ID but no RTE room."
+                  << std::endl;
+      }
+    } else {
+      if (std::any_of(
+              map_info.definitions.begin(), map_info.definitions.end(),
+              [](const HumanMap& h_map) { return h_map.has_rte_room(); })) {
+        std::cout << "Map " << map_name << " has an RTE room but no RTE ID."
+                  << std::endl;
+      }
+    }
   }
 
   void ValidateRoom(const RoomIdentifier& room_identifier,
@@ -94,6 +116,10 @@ class Validator {
         std::cout << "    CONNECTION " << connection.ShortDebugString()
                   << std::endl;
       }
+
+      for (const std::string& map_name : room_info.map_rtes_referenced_by) {
+        std::cout << "    MAP RTE " << map_name << std::endl;
+      }
     } else if (room_info.definitions.size() > 1) {
       std::cout << "Room " << room_identifier.ShortDebugString()
                 << " was defined multiple times." << std::endl;
@@ -106,7 +132,7 @@ class Validator {
       return false;
     }
 
-    if (h_door.keyholders_size() > 0 || h_door.endings_size() > 0 ||
+    if (h_door.keyholders_size() > 0 || h_door.white_ending() ||
         h_door.complete_at() > 0) {
       return true;
     }
@@ -220,16 +246,23 @@ class Validator {
                   << " needs an explicit location name." << std::endl;
       }
 
-      if (h_door.double_letters() &&
-          (h_door.type() == DoorType::STANDARD ||
-           h_door.type() == DoorType::LOCATION_ONLY ||
-           h_door.type() == DoorType::GRAVESTONE)) {
-        std::cout << "Door " << door_identifier.ShortDebugString()
-                  << " is a location that depends on double_letters."
-                  << std::endl;
+      if (h_door.type() == DoorType::STANDARD ||
+          h_door.type() == DoorType::LOCATION_ONLY ||
+          h_door.type() == DoorType::GRAVESTONE || h_door.legacy_location()) {
+        if (h_door.double_letters()) {
+          std::cout << "Door " << door_identifier.ShortDebugString()
+                    << " is a location that depends on double_letters."
+                    << std::endl;
+        }
+
+        if (!h_door.has_location_room()) {
+          std::cout << "Door " << door_identifier.ShortDebugString()
+                    << " is missing a location_room." << std::endl;
+        }
       }
 
-      bool needs_id = (h_door.type() != DoorType::EVENT);
+      bool needs_id = (h_door.type() != DoorType::EVENT || h_door.latch() ||
+                       h_door.legacy_location());
       if (door_info.has_id != needs_id) {
         if (needs_id) {
           std::cout << "Door " << door_identifier.ShortDebugString()
@@ -253,6 +286,14 @@ class Validator {
         std::cout << "    CONNECTION " << connection.ShortDebugString()
                   << std::endl;
       }
+
+      for (const std::string& map_name : port_info.map_worldport_entrances) {
+        std::cout << "    MAP (worldport_entrance) " << map_name << std::endl;
+      }
+
+      if (port_info.has_id) {
+        std::cout << "    An AP ID is present." << std::endl;
+      }
     } else if (port_info.definitions.size() > 1) {
       std::cout << "Port " << port_identifier.ShortDebugString()
                 << " was defined multiple times." << std::endl;
@@ -273,6 +314,29 @@ class Validator {
         }
       }
     }
+
+    for (const HumanPort& port : port_info.definitions) {
+      if (!port.no_shuffle()) {
+        if (!port.has_destination()) {
+          std::cout << "Port " << port_identifier.ShortDebugString()
+                    << " is shuffleable and missing a destination."
+                    << std::endl;
+        }
+        if (!port.has_rotation()) {
+          std::cout << "Port " << port_identifier.ShortDebugString()
+                    << " is shuffleable and missing a rotation." << std::endl;
+        }
+        if (!port_info.has_id) {
+          std::cout << "Port " << port_identifier.ShortDebugString()
+                    << " is missing an AP ID." << std::endl;
+        }
+      } else {
+        if (port_info.has_id) {
+          std::cout << "Port " << port_identifier.ShortDebugString()
+                    << " should not have an AP ID." << std::endl;
+        }
+      }
+    }
   }
 
   void ValidatePainting(const PaintingIdentifier& painting_identifier,
@@ -459,12 +523,6 @@ class Validator {
       std::cout << "Ending " << ending_name
                 << " has no definition, but was referenced:" << std::endl;
 
-      for (const DoorIdentifier& door_identifier :
-           ending_info.doors_referenced_by) {
-        std::cout << "    DOOR " << door_identifier.ShortDebugString()
-                  << std::endl;
-      }
-
       if (ending_info.has_id) {
         std::cout << "    An AP ID is present." << std::endl;
       }
@@ -1,4 +1,4 @@
-#ifndef TOOLS_VALIDATOR_VALIDATOR_H_
+#ifndef TOOLS_VALIDATOR_VALIDATOR_H
 #define TOOLS_VALIDATOR_VALIDATOR_H
 
 namespace com::fourisland::lingo2_archipelago {