Fixed: Blank page issue after authentication for some users

Improved handling of custom URL parameters by storing them as instance variables
Simplified _setup_routes() method for better maintainability
Applied fix from PR #2 for more robust URL parameter handling
Fixed: Route removal logic now correctly removes all default documentation routes
Properly removes /docs, /redoc, and /openapi.json endpoints
Prevents 500 errors when accessing old endpoints
Improved: Example files and documentation
Fixed custom_styling.py to work with uvicorn by adding default app variable
Standardized credentials across all custom styling examples
Added python-multipart to dev dependencies for form data handling
Added clear run instructions in example files

Author: georgekhananaev

.gitignore

@@ -71,3 +71,6 @@ dmypy.json
 # OS specific files
 .DS_Store
 **/.claude/settings.local.json
+
+# Claude Code memory file
+CLAUDE.md

README.md

@@ -223,6 +223,20 @@ pytest --cov=fastapi_docshield
 
 ## Changelog
 
+### Version 0.2.1 (2025-08-17)
+- **Fixed**: Blank page issue after authentication for some users
+  - Improved handling of custom URL parameters by storing them as instance variables
+  - Simplified `_setup_routes()` method for better maintainability
+  - Applied fix from PR #2 for more robust URL parameter handling
+- **Fixed**: Route removal logic now correctly removes all default documentation routes
+  - Properly removes `/docs`, `/redoc`, and `/openapi.json` endpoints
+  - Prevents 500 errors when accessing old endpoints
+- **Improved**: Example files and documentation
+  - Fixed `custom_styling.py` to work with uvicorn by adding default app variable
+  - Standardized credentials across all custom styling examples
+  - Added `python-multipart` to dev dependencies for form data handling
+  - Added clear run instructions in example files
+
 ### Version 0.2.0 (2025-08-17)
 - **Added**: Custom CSS and JavaScript injection support
   - New `custom_css` parameter to inject custom styles into documentation pages

examples/all_features.py

@@ -1,5 +1,5 @@
 """
-Comprehensive example showcasing all DocShield v0.2.0 features.
+Comprehensive example showcasing all DocShield v0.2.1 features.
 
 This example demonstrates:
 - Multiple user credentials
@@ -22,7 +22,7 @@
 # Create FastAPI app with comprehensive API
 app = FastAPI(
     title="DocShield Feature Showcase",
-    description="Demonstrating all features of FastAPI DocShield v0.2.0",
+    description="Demonstrating all features of FastAPI DocShield v0.2.1",
     version="2.0.0"
 )
 
@@ -349,7 +349,7 @@ def get_stats():
     import uvicorn
 
     print("=" * 80)
-    print("🛡️  DocShield v0.2.0 - All Features Showcase")
+    print("🛡️  DocShield v0.2.1 - All Features Showcase")
     print("=" * 80)
     print()
     print("📍 Server: http://localhost:8000")

examples/custom_styling.py

@@ -4,6 +4,21 @@
 This module demonstrates various ways to customize the appearance
 and behavior of your protected documentation pages.
 
+How to run:
+-----------
+1. With uvicorn (default dark theme):
+   uvicorn custom_styling:app --reload
+
+2. Run specific themes directly:
+   python custom_styling.py dark      # Dark theme with gradient
+   python custom_styling.py minimal   # Clean minimal theme  
+   python custom_styling.py corporate # Corporate theme with analytics
+   python custom_styling.py redoc     # ReDoc customization
+
+Credentials for all examples:
+   Username: demo
+   Password: style123
+
 Author: George Khananaev
 """
 
@@ -85,15 +100,15 @@ def root():
 
     DocShield(
         app=app,
-        credentials={"admin": "darkmode123"},
+        credentials={"demo": "style123"},
         custom_css=custom_css,
         custom_js=custom_js
     )
 
     print("=" * 70)
     print("Dark Theme Example")
     print("Access at: http://localhost:8000/docs")
-    print("Credentials: admin/darkmode123")
+    print("Credentials: demo/style123")
     print("=" * 70)
 
     uvicorn.run(app, host="0.0.0.0", port=8000)
@@ -205,15 +220,15 @@ def get_users():
 
     DocShield(
         app=app,
-        credentials={"viewer": "minimal123"},
+        credentials={"demo": "style123"},
         custom_css=custom_css,
         custom_js=custom_js
     )
 
     print("=" * 70)
     print("Minimal Theme Example")
     print("Access at: http://localhost:8001/docs")
-    print("Credentials: viewer/minimal123")
+    print("Credentials: demo/style123")
     print("=" * 70)
 
     uvicorn.run(app, host="0.0.0.0", port=8001)
@@ -363,19 +378,15 @@ def get_analytics():
 
     DocShield(
         app=app,
-        credentials={
-            "admin": "corp@2025",
-            "developer": "dev@2025",
-            "auditor": "audit@2025"
-        },
+        credentials={"demo": "style123"},
         custom_css=custom_css,
         custom_js=custom_js
     )
 
     print("=" * 70)
     print("Corporate Theme Example")
     print("Access at: http://localhost:8002/docs")
-    print("Credentials: admin/corp@2025, developer/dev@2025, or auditor/audit@2025")
+    print("Credentials: demo/style123")
     print("Features: Usage analytics, corporate branding, compliance notices")
     print("=" * 70)
 
@@ -483,7 +494,7 @@ def get_item(item_id: int):
 
     DocShield(
         app=app,
-        credentials={"user": "redoc123"},
+        credentials={"demo": "style123"},
         custom_css=custom_css,
         custom_js=custom_js,
         prefer_local=True  # Use local files for faster loading
@@ -492,13 +503,80 @@ def get_item(item_id: int):
     print("=" * 70)
     print("ReDoc Custom Example")
     print("Access at: http://localhost:8003/redoc")
-    print("Credentials: user/redoc123")
+    print("Credentials: demo/style123")
     print("Features: Reading progress, copy buttons, custom styling")
     print("=" * 70)
 
     uvicorn.run(app, host="0.0.0.0", port=8003)
 
 
+# Create a default app for uvicorn
+# This will be used when running: uvicorn custom_styling:app
+app = FastAPI(
+    title="Custom Styling Examples",
+    description="Examples of DocShield with custom CSS and JavaScript",
+    version="1.0.0"
+)
+
+@app.get("/")
+def root():
+    return {
+        "message": "DocShield Custom Styling Examples",
+        "examples": [
+            "python custom_styling.py dark",
+            "python custom_styling.py minimal", 
+            "python custom_styling.py corporate",
+            "python custom_styling.py redoc"
+        ]
+    }
+
+@app.get("/examples")
+def list_examples():
+    return {
+        "dark": "Dark theme with gradient backgrounds",
+        "minimal": "Clean minimal theme",
+        "corporate": "Corporate theme with analytics",
+        "redoc": "ReDoc customization example"
+    }
+
+# Apply dark theme by default to the app
+custom_css = """
+/* Dark theme customization */
+.swagger-ui {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+}
+
+.swagger-ui .topbar {
+    background-color: #1a1a2e;
+    border-bottom: 2px solid #764ba2;
+}
+
+.swagger-ui .btn.authorize {
+    background-color: #764ba2;
+    border-color: #764ba2;
+}
+"""
+
+custom_js = """
+console.log('🎨 Custom Styling Examples - Default Dark Theme');
+document.addEventListener('DOMContentLoaded', function() {
+    const info = document.querySelector('.info');
+    if (info) {
+        const note = document.createElement('div');
+        note.style.cssText = 'background:#764ba2;color:white;padding:15px;margin-bottom:20px;border-radius:5px;';
+        note.innerHTML = '📝 Note: Run with different themes using: python custom_styling.py [dark|minimal|corporate|redoc]';
+        info.insertBefore(note, info.firstChild);
+    }
+});
+"""
+
+DocShield(
+    app=app,
+    credentials={"demo": "style123"},
+    custom_css=custom_css,
+    custom_js=custom_js
+)
+
 if __name__ == "__main__":
     import sys
 

fastapi_docshield/__init__.py

@@ -9,9 +9,9 @@
 Copyright (c) 2025 George Khananaev
 """
 
-from .docshield import DocShield
+from .docshield import DocShield, __version__
 
-__version__ = "0.2.0"
+__version__ = __version__
 __author__ = "George Khananaev"
 __license__ = "MIT"
 __copyright__ = "Copyright (c) 2025 George Khananaev"

fastapi_docshield/docshield.py

@@ -6,14 +6,14 @@
 Copyright (c) 2025 George Khananaev
 """
 
-from typing import Dict, Optional, Tuple
-from fastapi import FastAPI, HTTPException, status, Request, Depends
-from fastapi.security.utils import get_authorization_scheme_param
+__version__ = "0.2.1"
+
+from typing import Dict, Optional
+from fastapi import FastAPI, HTTPException, status, Depends
 from fastapi.openapi.docs import get_swagger_ui_html, get_redoc_html
 from fastapi.security import HTTPBasic, HTTPBasicCredentials
 from fastapi.responses import HTMLResponse
 import secrets
-import base64
 from .static_handler import StaticHandler
 
 
@@ -67,6 +67,9 @@ def __init__(
         self.docs_url = docs_url
         self.redoc_url = redoc_url
         self.openapi_url = openapi_url
+        self.swagger_js_url = swagger_js_url
+        self.swagger_css_url = swagger_css_url
+        self.redoc_js_url = redoc_js_url
         self.use_cdn_fallback = use_cdn_fallback
         self.prefer_local = prefer_local
         self.custom_css = custom_css
@@ -89,11 +92,7 @@ def __init__(
         app.openapi_url = None
 
         # Set up protected documentation routes
-        self._setup_routes(
-            swagger_js_url=swagger_js_url,
-            swagger_css_url=swagger_css_url,
-            redoc_js_url=redoc_js_url
-        )
+        self._setup_routes()
 
     def _remove_existing_docs_routes(self) -> None:
         """
@@ -108,12 +107,18 @@ def _remove_existing_docs_routes(self) -> None:
         for route in self.app.routes:
             path = getattr(route, "path", "")
 
-            # Skip documentation routes
-            if (path == self.docs_url or 
+            # Skip both original and new documentation routes
+            if (path == self.original_docs_url or 
+                path == self.original_redoc_url or 
+                path == self.original_openapi_url or
+                path == self.docs_url or 
                 path == self.redoc_url or 
                 path == self.openapi_url or
-                path.startswith(f"{self.docs_url}/") or  # Docs static files
-                path == "/openapi.json"):
+                path == "/docs" or  # Default docs
+                path == "/redoc" or  # Default redoc
+                path == "/openapi.json" or  # Default openapi
+                path.startswith("/docs/") or  # Docs static files
+                path.startswith(f"{self.docs_url}/")):  # New docs static files
                 continue
 
             routes_to_keep.append(route)
@@ -154,19 +159,11 @@ def _verify_credentials(self, credentials: HTTPBasicCredentials) -> str:
             headers={"WWW-Authenticate": "Basic"},
         )
 
-    def _setup_routes(
-        self,
-        swagger_js_url: Optional[str],
-        swagger_css_url: Optional[str],
-        redoc_js_url: Optional[str],
-    ) -> None:
+    def _setup_routes(self) -> None:
         """
         Set up all protected documentation endpoints.
         
-        Args:
-            swagger_js_url: Custom Swagger UI JavaScript URL
-            swagger_css_url: Custom Swagger UI CSS URL
-            redoc_js_url: Custom ReDoc JavaScript URL
+        Uses the stored swagger_js_url, swagger_css_url, and redoc_js_url.
         """
         # Set up OpenAPI JSON endpoint
         @self.app.get(self.openapi_url, include_in_schema=False)
@@ -186,16 +183,16 @@ async def get_docs(credentials: HTTPBasicCredentials = Depends(self.security)):
                 self._verify_credentials(credentials)
 
                 # Determine which URLs to use
-                if swagger_js_url is not None or swagger_css_url is not None:
+                if self.swagger_js_url is not None or self.swagger_css_url is not None:
                     # User provided custom URLs, use them
                     kwargs = {
                         "openapi_url": self.openapi_url,
                         "title": self.app.title + " - Swagger UI",
                     }
-                    if swagger_js_url is not None:
-                        kwargs["swagger_js_url"] = swagger_js_url
-                    if swagger_css_url is not None:
-                        kwargs["swagger_css_url"] = swagger_css_url
+                    if self.swagger_js_url is not None:
+                        kwargs["swagger_js_url"] = self.swagger_js_url
+                    if self.swagger_css_url is not None:
+                        kwargs["swagger_css_url"] = self.swagger_css_url
                 elif self.static_handler and self.prefer_local:
                     # Prefer local files
                     js_url, css_url = self.static_handler.get_swagger_urls(prefer_local=True)
@@ -229,12 +226,12 @@ async def get_redoc(credentials: HTTPBasicCredentials = Depends(self.security)):
                 self._verify_credentials(credentials)
 
                 # Determine which URL to use
-                if redoc_js_url is not None:
+                if self.redoc_js_url is not None:
                     # User provided custom URL, use it
                     kwargs = {
                         "openapi_url": self.openapi_url,
                         "title": self.app.title + " - ReDoc",
-                        "redoc_js_url": redoc_js_url,
+                        "redoc_js_url": self.redoc_js_url,
                     }
                 elif self.static_handler and self.prefer_local:
                     # Prefer local files

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "fastapi-docshield"
-version = "0.2.0"
+version = "0.2.1"
 description = "A simple FastAPI integration to protect documentation endpoints with authentication"
 readme = "README.md"
 requires-python = ">=3.7"
@@ -42,6 +42,7 @@ dev = [
     "httpx>=0.23.0",
     "uvicorn>=0.18.0",
     "requests>=2.28.0",
+    "python-multipart>=0.0.5",
 ]
 
 [project.urls]