update: include role enum for default auth model (#41)

Author: mslaursen

fastapi_forge/frontend/constants.py

@@ -82,3 +82,4 @@
         type=FieldDataTypeEnum.STRING,
     ),
 ]
+DEFAULT_AUTH_USER_ROLE_ENUM_NAME = "UserRole"

fastapi_forge/frontend/panels/project_config_panel.py

@@ -5,14 +5,19 @@
 from pydantic import ValidationError
 
 from fastapi_forge.dtos import (
+    CustomEnum,
+    CustomEnumValue,
     Model,
     ModelField,
     ModelFieldMetadata,
     ModelMetadata,
 )
 from fastapi_forge.enums import FieldDataTypeEnum
 from fastapi_forge.forge import build_project
-from fastapi_forge.frontend.constants import DEFAULT_AUTH_USER_FIELDS
+from fastapi_forge.frontend.constants import (
+    DEFAULT_AUTH_USER_FIELDS,
+    DEFAULT_AUTH_USER_ROLE_ENUM_NAME,
+)
 from fastapi_forge.frontend.notifications import notify_validation_error
 from fastapi_forge.frontend.state import state
 
@@ -210,6 +215,23 @@ async def _handle_builtin_auth_change(
             return
 
         try:
+            auth_role_enum_name = DEFAULT_AUTH_USER_ROLE_ENUM_NAME
+            auth_enum = state.get_enum_by_name(auth_role_enum_name)
+
+            auth_enum_values = [
+                CustomEnumValue(name="USER", value="auto()"),
+                CustomEnumValue(name="ADMIN", value="auto()"),
+            ]
+            if auth_enum and not auth_enum.values:
+                auth_enum.values = auth_enum_values
+
+            if not auth_enum:
+                auth_enum = CustomEnum(
+                    name=auth_role_enum_name,
+                    values=auth_enum_values,
+                )
+                state.custom_enums.append(auth_enum)
+
             auth_user_model = Model(
                 name="auth_user",
                 metadata=ModelMetadata(is_auth_model=True),
@@ -222,6 +244,12 @@ async def _handle_builtin_auth_change(
                         index=True,
                     ),
                     *DEFAULT_AUTH_USER_FIELDS,
+                    ModelField(
+                        name="role",
+                        type=FieldDataTypeEnum.ENUM,
+                        type_enum=auth_role_enum_name,
+                        default_value=auth_enum.values[0].name,
+                    ),
                     ModelField(
                         name="created_at",
                         type=FieldDataTypeEnum.DATETIME,

fastapi_forge/frontend/state.py

@@ -264,5 +264,11 @@ def get_auth_model(self) -> Model | None:
             None,
         )
 
+    def get_enum_by_name(self, name: str) -> CustomEnum | None:
+        return next(
+            (enum for enum in self.custom_enums if enum.name == name),
+            None,
+        )
+
 
 state: ProjectState = ProjectState()