services/auth.py

from passlib.context import CryptContext
from sqlmodel import Session, select
from typing import Optional

from models.user import User, UserCreate
from models.db import get_session_context # Using context manager for direct use
from services.logger import app_logger

pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")

def verify_password(plain_password: str, hashed_password: str) -> bool:
    return pwd_context.verify(plain_password, hashed_password)

def get_password_hash(password: str) -> str:
    return pwd_context.hash(password)

def get_user(db: Session, username: str) -> Optional[User]:
    statement = select(User).where(User.username == username)
    user = db.exec(statement).first()
    return user

# In services/auth.py
def create_user_in_db(user_data: UserCreate) -> Optional[User]:
    hashed_password = get_password_hash(user_data.password)
    db_user = User(
        username=user_data.username,
        hashed_password=hashed_password,
        email=user_data.email,
        full_name=user_data.full_name
        # The 'chat_sessions' attribute is an empty list by default here, not yet loaded from DB
    )
    try:
        with get_session_context() as db: # Session starts here
            existing_user = get_user(db, user_data.username)
            if existing_user:
                app_logger.warning(f"User {user_data.username} already exists.")
                return None
            
            db.add(db_user)
            # db.commit() # Commit is handled by context manager
            db.refresh(db_user) # This makes sure all attributes are up-to-date from DB *within this session*
        # Session closes here. db_user is now "detached".
        return db_user # Returning the detached db_user object
    except Exception as e:
        app_logger.error(f"Error creating user {user_data.username}: {e}")
        return None