added token and details

App/Users/Schema.py

@@ -2,6 +2,7 @@ from pydantic import BaseModel, Field, constr, ConfigDict
 from typing import Optional
 from datetime import datetime
 from passlib.context import CryptContext
+from decimal import Decimal
 
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
@@ -12,11 +13,15 @@ MAC_PATTERN = r"^([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})$"
 
 class UserResponse(BaseModel):
     id: str
-    phoneNumber: str
     name: str
+    phoneNumber: str
+    balance: Decimal
     account_locked: bool
     model_config = ConfigDict(from_attributes=True)
 
+    # class Config:
+    #     orm_mode = True
+
 
 # Register User Request
 class RegisterUserRequest(BaseModel):

App/Users/UserRoutes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, HTTPException, status
+from fastapi import APIRouter, HTTPException, status, Depends
 from .Schema import (
     RegisterUserRequest,
     LoginUserRequest,
@@ -17,6 +17,7 @@ from App.Subscriptions.Model import Subscription
 from App.Android.Android import AndroidClient
 from App.Android.Schema import RegisterUserRequest as AndroidRegister
 from App.Templates.Templates import MessageTemplate
+from .dependencies import get_current_active_user
 
 # JWT Configurations
 SECRET_KEY = "your_secret_key_here"
@@ -173,3 +174,13 @@ async def toggle_user_status(user_id: str):
         else "User enabled successfully."
     )
     return BaseResponse(code=200, message=message)
+
+
+@user_router.get(
+    "/user/me", response_model=UserResponse, status_code=status.HTTP_200_OK
+)
+async def get_user_details(current_user: User = Depends(get_current_active_user)):
+    """
+    Get the current user's details and balance.
+    """
+    return UserResponse.from_orm(current_user)

App/Users/dependencies.py

@@ -0,0 +1,28 @@
+from fastapi import Depends, HTTPException, status
+from jose import jwt
+from App.Users.Model import User
+from fastapi.security import OAuth2PasswordBearer
+
+SECRET_KEY = "your_secret_key_here"
+ALGORITHM = "HS256"
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="user/login")
+
+
+async def get_current_user(token: str = Depends(oauth2_scheme)):
+    credentials_exception = HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail="Could not validate credentials",
+        headers={"WWW-Authenticate": "Bearer"},
+    )
+    try:
+        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
+        user = await User.get_or_none(phoneNumber=payload.get("sub"))
+        if user is None:
+            raise credentials_exception
+    except jwt.JWTError:
+        raise credentials_exception
+    return user
+
+
+async def get_current_active_user(current_user: User = Depends(get_current_user)):
+    return current_user