from functools import wraps
from flask import request
from flask_jwt_extended import get_jwt_identity
from werkzeug.exceptions import Unauthorized

def logged_user_id(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        logged_user_id = get_jwt_identity()
        if not logged_user_id:
            raise Unauthorized('Token inválido ou expirado.')

        # Passe o logged_user_id como argumento para a função
        return f(logged_user_id, *args, **kwargs)
    
    return decorated_function

def format_date(date):
    return date.strftime('%Y-%m-%d') if date else None