Added first controls

src/myfasthtml/auth/utils.py

@@ -15,7 +15,7 @@ from fasthtml.common import RedirectResponse, Beforeware
 from jose import jwt, JWTError
 
 # Configuration
-API_BASE_URL = "http://localhost:5001"  # Base URL for FastAPI backend
+API_BASE_URL = "http://localhost:5003"  # Base URL for FastAPI backend
 JWT_SECRET = "jwt-secret-to-change"  # Must match FastAPI secret
 JWT_ALGORITHM = "HS256"
 TOKEN_REFRESH_THRESHOLD_MINUTES = 5  # Refresh token if expires in less than 5 minutes
@@ -63,14 +63,12 @@ def auth_before(request, session):
   Args:
       request: Starlette request object
       session: FastHTML session object
-
   Returns:
       RedirectResponse to login page if authentication fails, None otherwise
   """
   # Get tokens from session
   access_token = session.get('access_token')
   refresh_token = session.get('refresh_token')
-  print(f"path={request.scope['path']}, {session=}, {access_token=}, {refresh_token=}")
   # If no access token, redirect to login
   if not access_token:
     return RedirectResponse('/login', status_code=303)
@@ -163,13 +161,14 @@ def check_token_expiry(token: str) -> Optional[float]:
     return None
 
 
-def login_user(email: str, password: str) -> Optional[Dict[str, Any]]:
+def login_user(email: str, password: str, base_url: str = None) -> Optional[Dict[str, Any]]:
   """
   Authenticate user with email and password.
 
   Args:
       email: User email address
       password: User password
+      base_url:
 
   Returns:
       Dictionary containing access_token, refresh_token, and user_info if successful,
@@ -177,7 +176,7 @@ def login_user(email: str, password: str) -> Optional[Dict[str, Any]]:
   """
   try:
     response = http_client.post(
-      f"{API_BASE_URL}/auth/login",
+      f"{base_url or API_BASE_URL}/auth/login",
       data={"username": email, "password": password},
       headers={"Content-Type": "application/x-www-form-urlencoded"},
       timeout=10.0
@@ -196,7 +195,7 @@ def login_user(email: str, password: str) -> Optional[Dict[str, Any]]:
     return None
 
 
-def register_user(email: str, username: str, password: str) -> Optional[Dict[str, Any]]:
+def register_user(email: str, username: str, password: str, base_url: str = None) -> Optional[Dict[str, Any]]:
   """
   Register a new user.
 
@@ -204,14 +203,14 @@ def register_user(email: str, username: str, password: str) -> Optional[Dict[str
       email: User email address
       username: User name
       password: User password
-
+      base_url:
   Returns:
       Dictionary containing success message if registration succeeds,
       None if registration fails
   """
   try:
     response = http_client.post(
-      f"{API_BASE_URL}/auth/register",
+      f"{base_url or API_BASE_URL}/auth/register",
       json={"email": email, "username": username, "password": password},
       timeout=10.0
     )
@@ -224,20 +223,20 @@ def register_user(email: str, username: str, password: str) -> Optional[Dict[str
     return None
 
 
-def refresh_access_token(refresh_token: str) -> Optional[Dict[str, Any]]:
+def refresh_access_token(refresh_token: str, base_url: str = None) -> Optional[Dict[str, Any]]:
   """
   Refresh the access token using a refresh token.
 
   Args:
       refresh_token: Valid refresh token
-
+      base_url:
   Returns:
       Dictionary containing new access_token and refresh_token if successful,
       None if refresh fails
   """
   try:
     response = http_client.post(
-      f"{API_BASE_URL}/auth/refresh",
+      f"{base_url or API_BASE_URL}/auth/refresh",
       json={"refresh_token": refresh_token},
       timeout=10.0
     )
@@ -254,20 +253,20 @@ def refresh_access_token(refresh_token: str) -> Optional[Dict[str, Any]]:
     return None
 
 
-def get_user_info(access_token: str) -> Optional[Dict[str, Any]]:
+def get_user_info(access_token: str, base_url: str = None) -> Optional[Dict[str, Any]]:
   """
   Get current user information using access token.
 
   Args:
       access_token: Valid access token
-
+      base_url:
   Returns:
       Dictionary containing user information if successful,
       None if request fails
   """
   try:
     response = http_client.get(
-      f"{API_BASE_URL}/auth/me",
+      f"{base_url or API_BASE_URL}/auth/me",
       headers={"Authorization": f"Bearer {access_token}"},
       timeout=10.0
     )
@@ -280,19 +279,36 @@ def get_user_info(access_token: str) -> Optional[Dict[str, Any]]:
     return None
 
 
-def logout_user(refresh_token: str) -> bool:
+def save_user_info(access_token: str, user_profile: dict, base_url: str = None):
+  try:
+    response = http_client.patch(
+      f"{base_url or API_BASE_URL}/auth/me",
+      headers={"Authorization": f"Bearer {access_token}"},
+      timeout=10.0,
+      json=user_profile
+    )
+    
+    if response.status_code == 200:
+      return response.json()
+    
+    return None
+  except httpx.HTTPError:
+    return None
+
+
+def logout_user(refresh_token: str, base_url: str = None) -> bool:
   """
   Logout user by revoking the refresh token.
 
   Args:
       refresh_token: Refresh token to revoke
-
+      base_url:
   Returns:
       True if logout successful, False otherwise
   """
   try:
     response = http_client.post(
-      f"{API_BASE_URL}/auth/logout",
+      f"{base_url or API_BASE_URL}/auth/logout",
       json={"refresh_token": refresh_token},
       timeout=10.0
     )