1.0.6 debug

main.py

@@ -14,7 +14,7 @@ from typing import Optional
 load_dotenv()
 
 SERVER_NAME = "Nexus Authentication Service"
-VERSION = "1.0.5 debug"
+VERSION = "1.0.6 debug"
 
 # Supabase Configuration
 SUPABASE_URL = os.getenv("SUPABASE_URL")
@@ -64,6 +64,10 @@ class UserResponse(BaseModel):
     access_level: str
     date_joined: datetime
 
+class UsernameAvailabilityResponse(BaseModel):
+    username: str
+    is_available: bool
+
 class UpdateUserRequest(BaseModel):
     password: Optional[str] = None
     email: Optional[EmailStr] = None
@@ -171,8 +175,6 @@ async def signup(request: SignupRequest):
     created_user = await create_user(request.username, hash_password(request.password), request.email)
     return {"message": "User created successfully", "user": created_user}
 
-from datetime import datetime, timedelta, timezone
-
 @auth_router.post("/login", response_model=LoginResponse)
 async def login(request: LoginRequest, user_agent: str = Header(...)):
     # Query the user based on the username
@@ -235,9 +237,6 @@ async def logout(user_id: str, token: str):
 
     return {"message": "Session forcefully expired"}
 
-
-from datetime import datetime, timezone
-
 @auth_router.get("/validate", response_model=TokenResponse)
 async def validate_token(user_id: str, token: str, user_agent: str = Header(...)):
     # Query to validate session by user_id, token, and device
@@ -274,11 +273,24 @@ async def validate_token(user_id: str, token: str, user_agent: str = Header(...)
 
     return TokenResponse(access_token=token)
 
-
 @auth_router.get("/search-users", response_model=List[str])
 async def search_users(query: str):
     users = supabase.table("users").select("username").ilike("username", f"%{query}%").execute()
-    return [user["username"] for user in users.data]
+    usernames = [user["username"] for user in users.data]
+    
+    # Exclude SYSTEM_USER from the list if it's present
+    if SYSTEM_USER and SYSTEM_USER in usernames:
+        usernames.remove(SYSTEM_USER)
+    
+    return usernames
+
+@auth_router.get("/is-username-available", response_model=UsernameAvailabilityResponse)
+async def is_username_available(query: str):
+    users = supabase.table("users").select("username").eq("username", query).execute()
+    return UsernameAvailabilityResponse(
+        username=query,
+        is_available=len(users.data) == 0  # If no users are found, the username is available
+    )
 
 @auth_router.get("/get-user-id", response_model=str)
 async def get_user_id(username: str):