feat: invalid enum detection

Author: mslaursen

fastapi_forge/dtos.py

@@ -373,6 +373,33 @@ class ProjectSpec(_Base):
     models: list[Model] = []
     custom_enums: list[CustomEnum] = []
 
+    @model_validator(mode="after")
+    def _validate_enums(self) -> Self:
+        valid_enum_names = {custom_enum.name for custom_enum in self.custom_enums}
+
+        invalid_fields = [
+            (model.name, field.name, field.type_enum)
+            for model in self.models
+            for field in model.fields
+            if (
+                field.type == FieldDataTypeEnum.ENUM
+                and (field.type_enum is None or field.type_enum not in valid_enum_names)
+            )
+        ]
+
+        if invalid_fields:
+            error_lines = [
+                f"• {model_name}.{field_name} (ref: '{type_enum}')"
+                for model_name, field_name, type_enum in invalid_fields
+            ]
+            raise ValueError(
+                f"Invalid enum references ({len(invalid_fields)}):\n"
+                + "\n".join(error_lines)
+                + f"\nValid enums: {', '.join(sorted(valid_enum_names)) or 'none'}"
+            )
+
+        return self
+
     @model_validator(mode="after")
     def _validate_models(self) -> Self:
         model_names = [model.name for model in self.models]

fastapi_forge/project_io.py

@@ -247,11 +247,13 @@ def _load_project_to_dict(self) -> dict[str, Any]:
         with self.project_path.open() as stream:
             return yaml.safe_load(stream)["project"]
 
-    def load(self) -> Any:
+    def load(self) -> ProjectSpec:
         return ProjectSpec(**self._load_project_to_dict())
 
     @classmethod
-    def load_from_conn_string(cls, conn_string: str, schema: str = "public") -> Any:
+    def load_from_conn_string(
+        cls, conn_string: str, schema: str = "public"
+    ) -> ProjectSpec:
         db_info = _inspect_postgres_schema(conn_string, schema)
         db_schema = db_info["schema_data"]
         db_name = db_info["database_name"]