from datetime import timedelta, datetime

import jwt
from app.core.config import settings

def encode_jwt(payload: dict,
               private_key: str = settings.auth_jwt.private_key_path.read_text(),
               algorithm=settings.auth_jwt.algorithm,
               expire_timedelta: timedelta | None = None,
               expire_minutes: int = settings.auth_jwt.access_token_expire_minutes):
    to_encode = payload.copy()
    now = datetime.utcnow()
    if expire_timedelta:
        expire = now + timedelta
    else:
        expire = now + timedelta(minutes=expire_minutes)
    to_encode.update(
        exp=expire,
        iat=now
    )
    encoded = jwt.encode(to_encode,
                         private_key,
                         algorithm=algorithm)
    return encoded

def decode_jwt(token: str | bytes,
               public_key: str = settings.auth_jwt.public_key_path.read_text(),
               algorithm: str = settings.auth_jwt.algorithm):
    decoded = jwt.decode(token, public_key, algorithms=[algorithm])
    return decoded