100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
"""User management routes"""
|
|
from fastapi import APIRouter, HTTPException, Header
|
|
from pymongo.errors import DuplicateKeyError
|
|
from db import get_database
|
|
from models import UserCreate, UserUpdate, User
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/register", response_model=dict)
|
|
async def register_user(user_data: UserCreate):
|
|
"""
|
|
Register a new user (called after Firebase Google Auth)
|
|
Stores user profile in MongoDB
|
|
"""
|
|
db = get_database()
|
|
|
|
try:
|
|
user_doc = {
|
|
"email": user_data.email,
|
|
"displayName": user_data.displayName or user_data.email.split("@")[0],
|
|
"photoURL": user_data.photoURL,
|
|
"createdAt": datetime.utcnow(),
|
|
"updatedAt": datetime.utcnow(),
|
|
"theme": "light"
|
|
}
|
|
|
|
result = db.users.insert_one(user_doc)
|
|
user_doc["id"] = str(result.inserted_id)
|
|
|
|
return {
|
|
"id": user_doc["id"],
|
|
"email": user_doc["email"],
|
|
"displayName": user_doc["displayName"],
|
|
"message": "User registered successfully"
|
|
}
|
|
except DuplicateKeyError:
|
|
raise HTTPException(status_code=400, detail="User already exists")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/by-email/{email}", response_model=dict)
|
|
async def get_user_by_email(email: str):
|
|
"""Get user profile by email (called after Firebase Auth)"""
|
|
db = get_database()
|
|
|
|
user = db.users.find_one({"email": email})
|
|
if not user:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
|
|
user["id"] = str(user["_id"])
|
|
return user
|
|
|
|
|
|
@router.put("/update/{user_id}", response_model=dict)
|
|
async def update_user(user_id: str, user_data: UserUpdate):
|
|
"""Update user profile"""
|
|
db = get_database()
|
|
from bson import ObjectId
|
|
|
|
try:
|
|
update_data = user_data.model_dump(exclude_unset=True)
|
|
update_data["updatedAt"] = datetime.utcnow()
|
|
|
|
result = db.users.update_one(
|
|
{"_id": ObjectId(user_id)},
|
|
{"$set": update_data}
|
|
)
|
|
|
|
if result.matched_count == 0:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
|
|
return {"message": "User updated successfully"}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.delete("/{user_id}")
|
|
async def delete_user(user_id: str):
|
|
"""Delete user account and all associated data"""
|
|
db = get_database()
|
|
from bson import ObjectId
|
|
|
|
try:
|
|
# Delete user
|
|
db.users.delete_one({"_id": ObjectId(user_id)})
|
|
|
|
# Delete all entries by user
|
|
db.entries.delete_many({"userId": user_id})
|
|
|
|
# Delete user settings
|
|
db.settings.delete_one({"userId": user_id})
|
|
|
|
return {"message": "User and associated data deleted"}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|