import jwt import datetime from passlib.hash import bcrypt from storage import get_user, add_user SECRET_KEY = "your-secret-key-keep-it-safe" # Change this in production! ALGORITHM = "HS256" ACCESS_TOKEN_EXPIRE_MINUTES = 30 def verify_password(plain_password, hashed_password): return bcrypt.verify(plain_password, hashed_password) def get_password_hash(password): return bcrypt.hash(password) def create_access_token(data: dict, expires_delta: datetime.timedelta = None): to_encode = data.copy() if expires_delta: expire = datetime.datetime.utcnow() + expires_delta else: expire = datetime.datetime.utcnow() + datetime.timedelta(minutes=15) to_encode.update({"exp": expire}) encoded_jwt = jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM) return encoded_jwt def authenticate_user(username: str, password: str): user = get_user(username) if not user: return False if not verify_password(password, user["password"]): return False return user def register_user(username: str, password: str): if get_user(username): raise ValueError("Username already registered") hashed_password = get_password_hash(password) add_user(username, hashed_password) return {"username": username, "message": "User created successfully"}