WIP

Author: GordonSmith

tools/wit-codegen/package_registry.cpp

@@ -112,8 +112,26 @@ const InterfaceInfo *WitPackage::get_interface(const std::string &name) const
 
 void WitPackage::add_interface(const InterfaceInfo &interface)
 {
+    // Reserve space to minimize reallocations
+    if (interfaces.capacity() == interfaces.size())
+    {
+        interfaces.reserve(interfaces.size() * 2 + 8);
+    }
+
     interfaces.push_back(interface);
-    interface_map[interface.name] = &interfaces.back();
+
+    // Rebuild the entire map to ensure all pointers are valid
+    // This is necessary because vector may have reallocated
+    rebuild_interface_map();
+}
+
+void WitPackage::rebuild_interface_map()
+{
+    interface_map.clear();
+    for (auto &interface : interfaces)
+    {
+        interface_map[interface.name] = &interface;
+    }
 }
 
 // PackageRegistry implementation

tools/wit-codegen/package_registry.hpp

@@ -69,6 +69,12 @@ struct WitPackage
      * Add an interface to this package
      */
     void add_interface(const InterfaceInfo &interface);
+
+    /**
+     * Rebuild the interface map to point to current vector elements
+     * Call this after adding multiple interfaces or when vector may have reallocated
+     */
+    void rebuild_interface_map();
 };
 
 /**