fix(Frontend): Enum Registry not reflecting frontend state (#66)

Author: mslaursen

fastapi_forge/frontend/state.py

@@ -19,6 +19,7 @@
     ModelRelationship,
     ProjectSpec,
 )
+from fastapi_forge.type_info_registry import enum_registry
 
 
 class ProjectState(BaseModel):
@@ -170,6 +171,7 @@ def select_enum(self, enum: CustomEnum) -> None:
     def delete_enum(self, enum: CustomEnum) -> None:
         """Remove an enum from the project."""
         self.custom_enums.remove(enum)
+        enum_registry.remove(enum.name)
         self._deselect_content()
         self._trigger_ui_refresh()
 
@@ -182,6 +184,7 @@ def update_enum_name(self, enum: CustomEnum, new_name: str) -> None:
             notify_enum_exists(new_name)
             return
 
+        enum_registry.update_key(enum.name, new_name)
         enum.name = new_name
 
         if enum == self.selected_enum and self.select_enum_fn:

fastapi_forge/type_info_registry.py

@@ -27,7 +27,7 @@ class TypeInfo:
         faker_field_value: The factory field value for this type (can be a Faker method)
         test_value: Value to insert into models for post/patch tests.
         test_func: A function to call with the `test_value`.
-        encapsulate_assert: Wraps the `test_value` value (e.g, UUID(test_value))
+        encapsulate_assert: Wraps the `test_value` value (e.g, "UUID" => UUID(test_value))')
 
     """
 
@@ -58,6 +58,18 @@ def get(self, key: T) -> TypeInfo:
             raise KeyError(f"Key '{key}' not found.")
         return self._registry[key]
 
+    def remove(self, key: T) -> None:
+        if key not in self:
+            raise KeyError(f"Key '{key}' not found.")
+        del self._registry[key]
+
+    def update_key(self, old_key: T, new_key: T) -> None:
+        if old_key not in self:
+            raise KeyError(
+                f"Key '{old_key}' not found. All names: {self._registry.keys()}"
+            )
+        self._registry[new_key] = self._registry.pop(old_key)
+
     def all(self) -> list[TypeInfo]:
         return list(self._registry.values())