Update core/security.py

core/security.py

@@ -6,10 +6,9 @@ from fastapi.security import OAuth2PasswordBearer
 from core.config import SECRET_KEY, ALGORITHM, ACCESS_TOKEN_EXPIRE_MINUTES
 from db.mongo import users_collection
 
-# OAuth2 setup — adjust tokenUrl if your API has a prefix like /api
+# OAuth2 setup
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login")
 
-
 # Password hashing context
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
@@ -28,24 +27,24 @@ def create_access_token(data: dict, expires_delta: timedelta = None):
     to_encode.update({"exp": expire})
     return jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM)
 
+# Get the current user from the JWT token
 async def get_current_user(token: str = Depends(oauth2_scheme)):
-    logger.info(f"🔐 Authentication attempt with token: {token[:15]}...")  # Log first part of token
-    
+    print("🔐 Raw token received:", token)
     try:
         payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
-        logger.info(f"📄 Token payload: {payload}")
         email: str = payload.get("sub")
         if email is None:
-            logger.error("❌ Invalid token: subject missing")
             raise HTTPException(status_code=401, detail="Invalid token: subject missing")
     except JWTError as e:
-        logger.error(f"❌ JWTError while decoding token: {str(e)}")
+        print("❌ JWTError while decoding token:", str(e))
         raise HTTPException(status_code=401, detail="Could not validate token")
-
-    user = await users_collection.find_one({"email": email})
-    if not user:
-        logger.error(f"❌ User not found for email: {email}")
-        raise HTTPException(status_code=404, detail="User not found")
-
-    logger.info(f"✅ Authenticated user: {user['email']}")
-    return user
+    
+    try:
+        user = await users_collection.find_one({"email": email})
+        if not user:
+            raise HTTPException(status_code=404, detail="User not found")
+        print("✅ Authenticated user:", user["email"])
+        return user
+    except Exception as e:
+        print("❌ MongoDB error:", str(e))
+        raise HTTPException(status_code=500, detail="Database error")