Add Stronghold features into main

Author: frankie567

fief/apps/auth/app.py

@@ -48,7 +48,7 @@ def include_routers(router: APIRouter) -> APIRouter:
     allow_headers=["Authorization", "X-Requested-With"],
 )
 
-app.add_middleware(BabelMiddleware, **get_babel_middleware_kwargs())  # type: ignore
+app.add_middleware(BabelMiddleware, **get_babel_middleware_kwargs())
 app.include_router(oauth_router, include_in_schema=False)
 app.include_router(default_tenant_router)
 app.include_router(tenant_router)

fief/dependencies/branding.py

@@ -1,2 +1,5 @@
+from fief.settings import settings
+
+
 async def get_show_branding() -> bool:
-    return True
+    return settings.branding

fief/dependencies/fief.py

@@ -16,7 +16,7 @@ def _is_uvicorn_ssl() -> bool:
         for task in asyncio.all_tasks():
             coroutine = task.get_coro()
             if coroutine is not None:
-                frame = coroutine.cr_frame
+                frame = coroutine.cr_frame  # type: ignore
                 if frame is not None:
                     args = frame.f_locals
                     if self := args.get("self"):

fief/middlewares/cors.py

@@ -5,7 +5,7 @@
 
 
 class CORSMiddlewarePath(CORSMiddleware):
-    def __init__(self, path_regex: str, *args, **kwargs) -> None:
+    def __init__(self, *args, path_regex: str, **kwargs) -> None:
         super().__init__(*args, **kwargs)
         self.path_regex = re.compile(path_regex)
 

fief/settings_class.py

@@ -143,6 +143,9 @@ class Settings(BaseSettings):
     password_min_length: int = 8
     password_min_score: int = Field(ge=0, le=4, default=3)
 
+    branding: bool = True
+    override_templates_directory: DirectoryPath | None = None
+
     model_config = SettingsConfigDict(
         env_file=".env", extra="ignore", secrets_dir=initial_settings.secrets_dir
     )
@@ -205,4 +208,6 @@ def get_email_provider(self) -> EmailProvider:
         return provider_class(**self.email_provider_params)
 
     def get_templates_directory(self) -> Sequence[Path]:
+        if self.override_templates_directory:
+            return [Path(self.override_templates_directory), TEMPLATES_DIRECTORY]
         return [TEMPLATES_DIRECTORY]